1 /*
2 * Copyright (c) 2022-2023 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 "policy_observer_wrapper.h"
17
18 #include "constant.h"
19 #include "module_template.h"
20 #include "napi_constant.h"
21 #include "net_manager_constants.h"
22 #include "net_policy_client.h"
23 #include "net_policy_constants.h"
24
25 namespace OHOS {
26 namespace NetManagerStandard {
27 constexpr const int DELETE_SIZE = 1;
PolicyObserverWrapper()28 PolicyObserverWrapper::PolicyObserverWrapper()
29 : observer_(new NetPolicyCallbackObserver()), manager_(std::make_shared<EventManager>()), registed_(false)
30 {
31 }
32
33 PolicyObserverWrapper::~PolicyObserverWrapper() = default;
34
On(napi_env env,napi_callback_info info,const std::initializer_list<std::string> & events,bool asyncCallback)35 napi_value PolicyObserverWrapper::On(napi_env env, napi_callback_info info,
36 const std::initializer_list<std::string> &events, bool asyncCallback)
37 {
38 size_t paramsCount = MAX_PARAM_NUM;
39 napi_value params[MAX_PARAM_NUM] = {nullptr};
40 NAPI_CALL(env, napi_get_cb_info(env, info, ¶msCount, params, nullptr, nullptr));
41 if (paramsCount != PARAM_OPTIONS_AND_CALLBACK || NapiUtils::GetValueType(env, params[ARG_INDEX_0]) != napi_string ||
42 NapiUtils::GetValueType(env, params[ARG_INDEX_1]) != napi_function) {
43 NETMANAGER_BASE_LOGE("on off once interface para: [string, function]");
44 napi_throw_error(env, std::to_string(NETMANAGER_ERR_PARAMETER_ERROR).c_str(), "Parameter error");
45 return NapiUtils::GetUndefined(env);
46 }
47
48 const auto event = NapiUtils::GetStringFromValueUtf8(env, params[ARG_INDEX_0]);
49 if (std::find(events.begin(), events.end(), event) == events.end()) {
50 NETMANAGER_BASE_LOGE("no find event");
51 return NapiUtils::GetUndefined(env);
52 }
53 if (!registed_) {
54 int32_t ret = NetPolicyClient::GetInstance().RegisterNetPolicyCallback(observer_);
55 NETMANAGER_BASE_LOGI("ret = [%{public}d]", ret);
56 registed_ = (ret == NETMANAGER_SUCCESS);
57 if (!registed_) {
58 NETMANAGER_BASE_LOGE("register callback error");
59 NetBaseErrorCodeConvertor convertor;
60 std::string errorMsg = convertor.ConvertErrorCode(ret);
61 napi_throw_error(env, std::to_string(ret).c_str(), errorMsg.c_str());
62 } else {
63 manager_->AddListener(env, event, params[ARG_INDEX_1], false, asyncCallback);
64 }
65 } else {
66 int32_t ret = NetPolicyClient::GetInstance().CheckPermission();
67 if (ret != NETMANAGER_SUCCESS) {
68 NETMANAGER_BASE_LOGE("register callback error");
69 NetBaseErrorCodeConvertor convertor;
70 std::string errorMsg = convertor.ConvertErrorCode(ret);
71 napi_throw_error(env, std::to_string(ret).c_str(), errorMsg.c_str());
72 } else {
73 manager_->AddListener(env, event, params[ARG_INDEX_1], false, asyncCallback);
74 }
75 }
76 return NapiUtils::GetUndefined(env);
77 }
78
Off(napi_env env,napi_callback_info info,const std::initializer_list<std::string> & events,bool asyncCallback)79 napi_value PolicyObserverWrapper::Off(napi_env env, napi_callback_info info,
80 const std::initializer_list<std::string> &events, bool asyncCallback)
81 {
82 napi_value thisVal = nullptr;
83 size_t paramsCount = MAX_PARAM_NUM;
84 napi_value params[MAX_PARAM_NUM] = {nullptr};
85 NAPI_CALL(env, napi_get_cb_info(env, info, ¶msCount, params, &thisVal, nullptr));
86
87 if (!(paramsCount != PARAM_JUST_OPTIONS || paramsCount != PARAM_OPTIONS_AND_CALLBACK) ||
88 NapiUtils::GetValueType(env, params[ARG_INDEX_0]) != napi_string) {
89 NETMANAGER_BASE_LOGE("on off once interface para: [string, function?]");
90 napi_throw_error(env, std::to_string(NETMANAGER_ERR_PARAMETER_ERROR).c_str(), "Parameter error");
91 return NapiUtils::GetUndefined(env);
92 }
93
94 if (paramsCount == PARAM_OPTIONS_AND_CALLBACK &&
95 NapiUtils::GetValueType(env, params[ARG_INDEX_1]) != napi_function) {
96 NETMANAGER_BASE_LOGE("on off once interface para: [string, function]");
97 return NapiUtils::GetUndefined(env);
98 }
99
100 std::string event = NapiUtils::GetStringFromValueUtf8(env, params[ARG_INDEX_0]);
101 if (std::find(events.begin(), events.end(), event) == events.end()) {
102 return NapiUtils::GetUndefined(env);
103 }
104
105 if (manager_->GetListenerListNum() > DELETE_SIZE) {
106 int32_t ret = NetPolicyClient::GetInstance().CheckPermission();
107 if (ret != NETMANAGER_SUCCESS) {
108 NETMANAGER_BASE_LOGE("unregister ret = %{public}d", ret);
109 NetBaseErrorCodeConvertor convertor;
110 std::string errorMsg = convertor.ConvertErrorCode(ret);
111 napi_throw_error(env, std::to_string(ret).c_str(), errorMsg.c_str());
112 return NapiUtils::GetUndefined(env);
113 }
114 DeleteListener(paramsCount, params, event);
115 return NapiUtils::GetUndefined(env);
116 }
117 if (manager_->GetListenerListNum() <= DELETE_SIZE) {
118 auto ret = NetPolicyClient::GetInstance().UnregisterNetPolicyCallback(observer_);
119 if (ret != NETMANAGER_SUCCESS) {
120 NETMANAGER_BASE_LOGE("unregister ret = %{public}d", ret);
121 NetBaseErrorCodeConvertor convertor;
122 std::string errorMsg = convertor.ConvertErrorCode(ret);
123 napi_throw_error(env, std::to_string(ret).c_str(), errorMsg.c_str());
124 } else {
125 DeleteListener(paramsCount, params, event);
126 registed_ = false;
127 }
128 }
129 return NapiUtils::GetUndefined(env);
130 }
131
DeleteListener(size_t paramsCount,napi_value * params,std::string event)132 void PolicyObserverWrapper::DeleteListener(size_t paramsCount, napi_value *params, std::string event)
133 {
134 if (paramsCount == PARAM_OPTIONS_AND_CALLBACK) {
135 manager_->DeleteListener(event, params[ARG_INDEX_1]);
136 } else {
137 manager_->DeleteListener(event);
138 }
139 }
140
141 } // namespace NetManagerStandard
142 } // namespace OHOS
143