1 /* 2 * Copyright (c) 2024 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 17 #ifndef PROXY_DEATH_HANDLER_H 18 #define PROXY_DEATH_HANDLER_H 19 20 #include <map> 21 #include <mutex> 22 #include <utility> 23 #include "iremote_object.h" 24 #include "proxy_death_param.h" 25 26 namespace OHOS { 27 namespace Security { 28 namespace AccessToken { 29 class ProxyDeathHandler { 30 public: ProxyDeathHandler()31 ProxyDeathHandler() {} 32 ~ProxyDeathHandler() = default; 33 34 void AddProxyStub(const sptr<IRemoteObject>& proxyStub, std::shared_ptr<ProxyDeathParam>& param); 35 void HandleRemoteDied(const sptr<IRemoteObject>& object); 36 void ReleaseProxyByParam(const std::shared_ptr<ProxyDeathParam>& param); 37 38 protected: 39 void ReleaseProxies(); 40 void ProcessProxyData(const std::shared_ptr<ProxyDeathParam>& param); 41 42 private: 43 std::mutex proxyLock_; 44 using RecipientAndParam = std::pair<sptr<IRemoteObject::DeathRecipient>, std::shared_ptr<ProxyDeathParam>>; 45 std::map<sptr<IRemoteObject>, RecipientAndParam> proxyStubAndRecipientMap_; 46 }; 47 } // namespace AccessToken 48 } // namespace Security 49 } // namespace OHOS 50 #endif // PROXY_DEATH_HANDLER_H 51 52