1 /*
2  * Copyright (c) 2021-2022 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 #include "hisysevent_manager.h"
17 
18 #include "hilog/log.h"
19 #include "hisysevent_base_manager.h"
20 #include "ret_code.h"
21 
22 #undef LOG_DOMAIN
23 #define LOG_DOMAIN 0xD002D08
24 
25 #undef LOG_TAG
26 #define LOG_TAG "HISYSEVENT_MANAGER"
27 
28 namespace OHOS {
29 namespace HiviewDFX {
30 std::unordered_map<std::shared_ptr<HiSysEventListener>,
31         std::shared_ptr<HiSysEventBaseListener>> HiSysEventManager::listenerToBaseMap_;
32 std::mutex HiSysEventManager::listenersMutex_;
33 
AddListener(std::shared_ptr<HiSysEventListener> listener,std::vector<ListenerRule> & rules)34 int32_t HiSysEventManager::AddListener(std::shared_ptr<HiSysEventListener> listener,
35     std::vector<ListenerRule>& rules)
36 {
37     if (listener == nullptr) {
38         HILOG_WARN(LOG_CORE, "add a null listener is not allowed.");
39         return ERR_LISTENER_NOT_EXIST;
40     }
41     std::lock_guard<std::mutex> lock(listenersMutex_);
42     auto baseListener = listenerToBaseMap_[listener];
43     if (baseListener == nullptr) {
44         baseListener = std::make_shared<HiSysEventBaseListener>(listener);
45         listenerToBaseMap_[listener] = baseListener;
46     }
47     return HiSysEventBaseManager::AddListener(baseListener, rules);
48 }
49 
RemoveListener(std::shared_ptr<HiSysEventListener> listener)50 int32_t HiSysEventManager::RemoveListener(std::shared_ptr<HiSysEventListener> listener)
51 {
52     if (listener == nullptr) {
53         HILOG_WARN(LOG_CORE, "remov a null listener is not allowed.");
54         return ERR_LISTENER_NOT_EXIST;
55     }
56     std::lock_guard<std::mutex> lock(listenersMutex_);
57     auto baseListener = listenerToBaseMap_[listener];
58     if (baseListener == nullptr) {
59         HILOG_WARN(LOG_CORE, "no need to remove a listener which has not been added.");
60         return ERR_LISTENER_NOT_EXIST;
61     }
62     auto ret = HiSysEventBaseManager::RemoveListener(baseListener);
63     if (ret == IPC_CALL_SUCCEED) {
64         HILOG_DEBUG(LOG_CORE, "remove listener from local cache.");
65         listenerToBaseMap_.erase(listener);
66     }
67     return ret;
68 }
69 
Query(struct QueryArg & arg,std::vector<QueryRule> & rules,std::shared_ptr<HiSysEventQueryCallback> callback)70 int32_t HiSysEventManager::Query(struct QueryArg& arg, std::vector<QueryRule>& rules,
71     std::shared_ptr<HiSysEventQueryCallback> callback)
72 {
73     auto baseQueryCallback = std::make_shared<HiSysEventBaseQueryCallback>(callback);
74     return HiSysEventBaseManager::Query(arg, rules, baseQueryCallback);
75 }
76 
SetDebugMode(std::shared_ptr<HiSysEventListener> listener,bool mode)77 int32_t HiSysEventManager::SetDebugMode(std::shared_ptr<HiSysEventListener> listener, bool mode)
78 {
79     if (listener == nullptr) {
80         HILOG_WARN(LOG_CORE, "set debug mode on a null listener is not allowed.");
81         return ERR_LISTENER_NOT_EXIST;
82     }
83     std::lock_guard<std::mutex> lock(listenersMutex_);
84     auto baseListener = listenerToBaseMap_[listener];
85     if (baseListener == nullptr) {
86         HILOG_WARN(LOG_CORE, "no need to set debug mode on a listener which has not been added.");
87         return ERR_LISTENER_NOT_EXIST;
88     }
89     return HiSysEventBaseManager::SetDebugMode(baseListener, mode);
90 }
91 } // namespace HiviewDFX
92 } // namespace OHOS
93