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 #include "net_observer.h"
17
18 #include <pthread.h>
19 #include <unistd.h>
20
21 #include "account_task_manager.h"
22 #include "common_defs.h"
23 #include "device_auth.h"
24 #include "device_auth_defines.h"
25 #include "hc_log.h"
26 #include "net_conn_client.h"
27 #include "net_conn_constants.h"
28
29 using namespace OHOS;
30 using namespace OHOS::NetManagerStandard;
31
32 static sptr<INetConnCallback> g_netCallback = nullptr;
33
StartObserver()34 void NetObserver::StartObserver()
35 {
36 LOGI("[NetObserver]: Start to register net connection callback.");
37 if (isObserverStarted_) {
38 LOGI("[NetObserver]: Observer already started.");
39 return;
40 }
41 isObserverStarted_ = true;
42 NetObserver *observer = static_cast<NetObserver *>(this);
43 regThread_ = std::thread([observer]() {
44 NetSpecifier netSpecifier;
45 NetAllCapabilities netAllCapabilities;
46 netAllCapabilities.netCaps_.insert(NetManagerStandard::NetCap::NET_CAPABILITY_INTERNET);
47 netSpecifier.ident_ = "";
48 netSpecifier.netCapabilities_ = netAllCapabilities;
49 sptr<NetSpecifier> specifier = new NetSpecifier(netSpecifier);
50 constexpr uint32_t RETRY_MAX_TIMES = 10;
51 uint32_t retryCount = 0;
52 do {
53 if (!observer->IsObserverStarted()) {
54 LOGW("[NetObserver]: observer stopped, can not register!");
55 return;
56 }
57 int32_t ret = NetConnClient::GetInstance().RegisterNetConnCallback(specifier, observer, 0);
58 if (ret == NetConnResultCode::NET_CONN_SUCCESS) {
59 LOGI("[NetMgrSubsysterm][RegisterNetConnCallback]: Register net connection callback succeeded.");
60 g_netCallback = observer;
61 return;
62 }
63 retryCount++;
64 LOGW("[NetMgrSubsysterm][RegisterNetConnCallback]: Register failed, errCode = %" LOG_PUB "d, retry.", ret);
65 sleep(1);
66 } while (retryCount < RETRY_MAX_TIMES);
67 LOGE("[NetMgrSubsysterm][RegisterNetConnCallback]: Register net connection callback failed!");
68 });
69 }
70
NetCapabilitiesChange(sptr<NetHandle> & netHandle,const sptr<NetAllCapabilities> & netAllCap)71 int32_t NetObserver::NetCapabilitiesChange(sptr<NetHandle> &netHandle, const sptr<NetAllCapabilities> &netAllCap)
72 {
73 LOGI("[NetObserver]: Net capabilities change.");
74 if (netAllCap == nullptr) {
75 return 0;
76 }
77 return HandleNetAllCap(*netAllCap);
78 }
79
NetLost(sptr<NetHandle> & netHandle)80 int32_t NetObserver::NetLost(sptr<NetHandle> &netHandle)
81 {
82 LOGI("[NetObserver]: Net lost!");
83 return 0;
84 }
85
NetAvailable(sptr<NetHandle> & netHandle)86 int32_t NetObserver::NetAvailable(sptr<NetHandle> &netHandle)
87 {
88 LOGI("[NetObserver]: Net available.");
89 return 0;
90 }
91
HandleNetAllCap(const NetAllCapabilities & netAllCap)92 int32_t NetObserver::HandleNetAllCap(const NetAllCapabilities &netAllCap)
93 {
94 if (netAllCap.netCaps_.count(NET_CAPABILITY_INTERNET) <= 0 ||
95 netAllCap.netCaps_.count(NET_CAPABILITY_VALIDATED) <= 0) {
96 for (auto netCap : netAllCap.netCaps_) {
97 LOGI("[NetObserver]: No net, netCap is: %" LOG_PUB "d", static_cast<int32_t>(netCap));
98 }
99 return 0;
100 }
101 LOGI("[NetObserver]: Net available.");
102 int32_t res = ExecuteAccountAuthCmd(DEFAULT_OS_ACCOUNT, RELOAD_CRED_MGR, nullptr, nullptr);
103 LOGI("[NetObserver]: Reload credential manager res: %" LOG_PUB "d", res);
104 return 0;
105 }
106
IsObserverStarted()107 bool NetObserver::IsObserverStarted()
108 {
109 return isObserverStarted_;
110 }
111
StopObserver()112 void NetObserver::StopObserver()
113 {
114 LOGI("[NetObserver]: Start to unregister net connection callback.");
115 if (!isObserverStarted_) {
116 LOGI("[NetObserver]: Observer already stopped.");
117 return;
118 }
119 isObserverStarted_ = false;
120 if (regThread_.joinable()) {
121 regThread_.join();
122 }
123 if (g_netCallback == nullptr) {
124 LOGI("[NetObserver]: Net connection callback is null.");
125 return;
126 }
127 int32_t ret = NetConnClient::GetInstance().UnregisterNetConnCallback(g_netCallback);
128 if (ret == 0) {
129 LOGI("[NetMgrSubsysterm][UnregisterNetConnCallback]: unregister net connection callback succeeded.");
130 g_netCallback = nullptr;
131 }
132 }