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 <string>
17
18 #include <napi/native_api.h>
19
20 #include "constant.h"
21 #include "get_cellular_rxbytes_context.h"
22 #include "get_iface_rxbytes_context.h"
23 #include "get_iface_stats_context.h"
24 #include "get_iface_uid_stats_context.h"
25 #include "get_uid_rxbytes_context.h"
26 #include "module_template.h"
27 #include "napi_utils.h"
28 #include "statistics_async_work.h"
29 #include "statistics_callback_observer.h"
30 #include "statistics_exec.h"
31 #include "statistics_observer_wrapper.h"
32 #include "update_iface_stats_context.h"
33
34 namespace OHOS {
35 namespace NetManagerStandard {
36 namespace {
37 constexpr const char *STATISTICS_MODULE_NAME = "net.statistics";
38
39 constexpr const char *FUNCTION_GET_CELLULAR_RXBYTES = "getCellularRxBytes";
40 constexpr const char *FUNCTION_GET_CELLULAR_TXBYTES = "getCellularTxBytes";
41 constexpr const char *FUNCTION_GET_ALL_RXBYTES = "getAllRxBytes";
42 constexpr const char *FUNCTION_GET_ALL_TXBYTES = "getAllTxBytes";
43 constexpr const char *FUNCTION_GET_UID_RXBYTES = "getUidRxBytes";
44 constexpr const char *FUNCTION_GET_UID_TXBYTES = "getUidTxBytes";
45 constexpr const char *FUNCTION_GET_IFACE_RXBYTES = "getIfaceRxBytes";
46 constexpr const char *FUNCTION_GET_IFACE_TXBYTES = "getIfaceTxBytes";
47 constexpr const char *FUNCTION_GET_IFACE_STATS = "getTrafficStatsByIface";
48 constexpr const char *FUNCTION_GET_IFACE_UID_STATS = "getTrafficStatsByUid";
49 constexpr const char *FUNCTION_UPDATE_IFACE_STATS = "updateIfacesStats";
50 constexpr const char *FUNCTION_UPDATE_STATS_DATA = "updateStatsData";
51 constexpr const char *FUNCTION_ON = "on";
52 constexpr const char *FUNCTION_OFF = "off";
53 constexpr const char *FUNCTION_GET_SOCKFD_RXBYTES = "getSockfdRxBytes";
54 constexpr const char *FUNCTION_GET_SOCKFD_TXBYTES = "getSockfdTxBytes";
55 } // namespace
56
GetCellularRxBytes(napi_env env,napi_callback_info info)57 napi_value GetCellularRxBytes(napi_env env, napi_callback_info info)
58 {
59 return ModuleTemplate::Interface<GetCellularRxBytesContext>(env, info, FUNCTION_GET_CELLULAR_RXBYTES, nullptr,
60 StatisticsAsyncWork::ExecGetCellularRxBytes,
61 StatisticsAsyncWork::GetCellularRxBytesCallback);
62 }
63
GetCellularTxBytes(napi_env env,napi_callback_info info)64 napi_value GetCellularTxBytes(napi_env env, napi_callback_info info)
65 {
66 return ModuleTemplate::Interface<GetCellularTxBytesContext>(env, info, FUNCTION_GET_CELLULAR_TXBYTES, nullptr,
67 StatisticsAsyncWork::ExecGetCellularTxBytes,
68 StatisticsAsyncWork::GetCellularTxBytesCallback);
69 }
70
GetAllRxBytes(napi_env env,napi_callback_info info)71 napi_value GetAllRxBytes(napi_env env, napi_callback_info info)
72 {
73 return ModuleTemplate::Interface<GetAllRxBytesContext>(env, info, FUNCTION_GET_ALL_RXBYTES, nullptr,
74 StatisticsAsyncWork::ExecGetAllRxBytes,
75 StatisticsAsyncWork::GetAllRxBytesCallback);
76 }
77
GetAllTxBytes(napi_env env,napi_callback_info info)78 napi_value GetAllTxBytes(napi_env env, napi_callback_info info)
79 {
80 return ModuleTemplate::Interface<GetAllTxBytesContext>(env, info, FUNCTION_GET_ALL_TXBYTES, nullptr,
81 StatisticsAsyncWork::ExecGetAllTxBytes,
82 StatisticsAsyncWork::GetAllTxBytesCallback);
83 }
84
GetUidRxBytes(napi_env env,napi_callback_info info)85 napi_value GetUidRxBytes(napi_env env, napi_callback_info info)
86 {
87 return ModuleTemplate::Interface<GetUidRxBytesContext>(env, info, FUNCTION_GET_UID_RXBYTES, nullptr,
88 StatisticsAsyncWork::ExecGetUidRxBytes,
89 StatisticsAsyncWork::GetUidRxBytesCallback);
90 }
91
GetUidTxBytes(napi_env env,napi_callback_info info)92 napi_value GetUidTxBytes(napi_env env, napi_callback_info info)
93 {
94 return ModuleTemplate::Interface<GetUidTxBytesContext>(env, info, FUNCTION_GET_UID_TXBYTES, nullptr,
95 StatisticsAsyncWork::ExecGetUidTxBytes,
96 StatisticsAsyncWork::GetUidTxBytesCallback);
97 }
98
GetSockfdRxBytes(napi_env env,napi_callback_info info)99 napi_value GetSockfdRxBytes(napi_env env, napi_callback_info info)
100 {
101 return ModuleTemplate::Interface<GetSockfdRxBytesContext>(env, info, FUNCTION_GET_SOCKFD_RXBYTES, nullptr,
102 StatisticsAsyncWork::ExecGetSockfdRxBytes,
103 StatisticsAsyncWork::GetSockfdRxBytesCallback);
104 }
105
GetSockfdTxBytes(napi_env env,napi_callback_info info)106 napi_value GetSockfdTxBytes(napi_env env, napi_callback_info info)
107 {
108 return ModuleTemplate::Interface<GetSockfdTxBytesContext>(env, info, FUNCTION_GET_SOCKFD_TXBYTES, nullptr,
109 StatisticsAsyncWork::ExecGetSockfdTxBytes,
110 StatisticsAsyncWork::GetSockfdTxBytesCallback);
111 }
112
GetIfaceRxBytes(napi_env env,napi_callback_info info)113 napi_value GetIfaceRxBytes(napi_env env, napi_callback_info info)
114 {
115 return ModuleTemplate::Interface<GetIfaceRxBytesContext>(env, info, FUNCTION_GET_IFACE_RXBYTES, nullptr,
116 StatisticsAsyncWork::ExecGetIfaceRxBytes,
117 StatisticsAsyncWork::GetIfaceRxBytesCallback);
118 }
119
GetIfaceTxBytes(napi_env env,napi_callback_info info)120 napi_value GetIfaceTxBytes(napi_env env, napi_callback_info info)
121 {
122 return ModuleTemplate::Interface<GetIfaceTxBytesContext>(env, info, FUNCTION_GET_IFACE_TXBYTES, nullptr,
123 StatisticsAsyncWork::ExecGetIfaceTxBytes,
124 StatisticsAsyncWork::GetIfaceTxBytesCallback);
125 }
126
GetIfaceStats(napi_env env,napi_callback_info info)127 napi_value GetIfaceStats(napi_env env, napi_callback_info info)
128 {
129 return ModuleTemplate::Interface<GetIfaceStatsContext>(env, info, FUNCTION_GET_IFACE_STATS, nullptr,
130 StatisticsAsyncWork::ExecGetIfaceStats,
131 StatisticsAsyncWork::GetIfaceStatsCallback);
132 }
133
GetIfaceUidStats(napi_env env,napi_callback_info info)134 napi_value GetIfaceUidStats(napi_env env, napi_callback_info info)
135 {
136 return ModuleTemplate::Interface<GetIfaceUidStatsContext>(env, info, FUNCTION_GET_IFACE_UID_STATS, nullptr,
137 StatisticsAsyncWork::ExecGetIfaceUidStats,
138 StatisticsAsyncWork::GetIfaceUidStatsCallback);
139 }
140
UpdateIfacesStats(napi_env env,napi_callback_info info)141 napi_value UpdateIfacesStats(napi_env env, napi_callback_info info)
142 {
143 return ModuleTemplate::Interface<UpdateIfacesStatsContext>(env, info, FUNCTION_UPDATE_IFACE_STATS, nullptr,
144 StatisticsAsyncWork::ExecUpdateIfacesStats,
145 StatisticsAsyncWork::UpdateIfacesStatsCallback);
146 }
147
UpdateStatsData(napi_env env,napi_callback_info info)148 napi_value UpdateStatsData(napi_env env, napi_callback_info info)
149 {
150 return ModuleTemplate::Interface<UpdateStatsDataContext>(env, info, FUNCTION_UPDATE_STATS_DATA, nullptr,
151 StatisticsAsyncWork::ExecUpdateStatsData,
152 StatisticsAsyncWork::UpdateStatsDataCallback);
153 }
154
On(napi_env env,napi_callback_info info)155 napi_value On(napi_env env, napi_callback_info info)
156 {
157 return StatisticsObserverWrapper::GetInstance().On(env, info, {EVENT_STATS_CHANGE}, false);
158 }
159
Off(napi_env env,napi_callback_info info)160 napi_value Off(napi_env env, napi_callback_info info)
161 {
162 return StatisticsObserverWrapper::GetInstance().Off(env, info, {EVENT_STATS_CHANGE}, false);
163 }
164
InitStatisticsModule(napi_env env,napi_value exports)165 napi_value InitStatisticsModule(napi_env env, napi_value exports)
166 {
167 NapiUtils::DefineProperties(env, exports,
168 {
169 DECLARE_NAPI_FUNCTION(FUNCTION_GET_CELLULAR_RXBYTES, GetCellularRxBytes),
170 DECLARE_NAPI_FUNCTION(FUNCTION_GET_CELLULAR_TXBYTES, GetCellularTxBytes),
171 DECLARE_NAPI_FUNCTION(FUNCTION_GET_ALL_RXBYTES, GetAllRxBytes),
172 DECLARE_NAPI_FUNCTION(FUNCTION_GET_ALL_TXBYTES, GetAllTxBytes),
173 DECLARE_NAPI_FUNCTION(FUNCTION_GET_UID_RXBYTES, GetUidRxBytes),
174 DECLARE_NAPI_FUNCTION(FUNCTION_GET_UID_TXBYTES, GetUidTxBytes),
175 DECLARE_NAPI_FUNCTION(FUNCTION_GET_SOCKFD_RXBYTES, GetSockfdRxBytes),
176 DECLARE_NAPI_FUNCTION(FUNCTION_GET_SOCKFD_TXBYTES, GetSockfdTxBytes),
177 DECLARE_NAPI_FUNCTION(FUNCTION_GET_IFACE_RXBYTES, GetIfaceRxBytes),
178 DECLARE_NAPI_FUNCTION(FUNCTION_GET_IFACE_TXBYTES, GetIfaceTxBytes),
179 DECLARE_NAPI_FUNCTION(FUNCTION_GET_IFACE_STATS, GetIfaceStats),
180 DECLARE_NAPI_FUNCTION(FUNCTION_GET_IFACE_UID_STATS, GetIfaceUidStats),
181 DECLARE_NAPI_FUNCTION(FUNCTION_UPDATE_IFACE_STATS, UpdateIfacesStats),
182 DECLARE_NAPI_FUNCTION(FUNCTION_UPDATE_STATS_DATA, UpdateStatsData),
183 DECLARE_NAPI_FUNCTION(FUNCTION_ON, On),
184 DECLARE_NAPI_FUNCTION(FUNCTION_OFF, Off),
185 });
186 return exports;
187 }
188
189 static napi_module g_statisticsModule = {
190 .nm_version = 1,
191 .nm_flags = 0,
192 .nm_filename = nullptr,
193 .nm_register_func = InitStatisticsModule,
194 .nm_modname = STATISTICS_MODULE_NAME,
195 .nm_priv = nullptr,
196 .reserved = {nullptr},
197 };
198
RegisterNetStatsModule(void)199 extern "C" __attribute__((constructor)) void RegisterNetStatsModule(void)
200 {
201 napi_module_register(&g_statisticsModule);
202 }
203 } // namespace NetManagerStandard
204 } // namespace OHOS
205