1 /*
2  * Copyright (c) 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 <cstdint>
17 
18 #include "errorcode_convertor.h"
19 #include "module_template.h"
20 #include "napi_utils.h"
21 #include "net_manager_constants.h"
22 #include "netmanager_ext_log.h"
23 #include "networkvpn_client.h"
24 #include "vpn_connection.h"
25 #include "vpn_monitor.h"
26 #ifdef SUPPORT_SYSVPN
27 #include "vpn_async_work.h"
28 #include "add_context.h"
29 #include "delete_context.h"
30 #include "get_list_context.h"
31 #include "get_context.h"
32 #include "get_connected_context.h"
33 #endif // SUPPORT_SYSVPN
34 
35 namespace OHOS {
36 namespace NetManagerStandard {
37 constexpr int32_t ARG_NUM_0 = 0;
38 constexpr int32_t PARAM_ONE = 1;
39 constexpr const char *ON = "on";
40 constexpr const char *OFF = "off";
41 #ifdef SUPPORT_SYSVPN
42 constexpr const char *ADD_SYS_VPN_CONFIG = "addSysVpnConfig";
43 constexpr const char *DELETE_SYS_VPN_CONFIG = "deleteSysVpnConfig";
44 constexpr const char *GET_SYS_VPN_CONFIG_LIST = "getSysVpnConfigList";
45 constexpr const char *GET_SYS_VPN_CONFIG = "getSysVpnConfig";
46 constexpr const char *GET_CONNECTED_SYS_VPN_CONFIG = "getConnectedSysVpnConfig";
47 #endif // SUPPORT_SYSVPN
48 
MakeData(napi_env env,size_t argc,napi_value * argv,EventManager * manager)49 static void *MakeData(napi_env env, size_t argc, napi_value *argv, EventManager *manager)
50 {
51     if ((argc != PARAM_ONE) || (NapiUtils::GetValueType(env, argv[ARG_NUM_0]) != napi_object)) {
52         NETMANAGER_EXT_LOGE("funciton prameter error");
53         napi_throw_error(env, std::to_string(NETMANAGER_EXT_ERR_PARAMETER_ERROR).c_str(), "Parameter error");
54         return nullptr;
55     }
56 
57     int32_t ret = NetworkVpnClient::GetInstance().CreateVpnConnection();
58     if (ret != NETMANAGER_EXT_SUCCESS) {
59         NETMANAGER_EXT_LOGE("execute CreateVpnConnection failed: %{public}d", ret);
60         std::string errorMsg = NetBaseErrorCodeConvertor().ConvertErrorCode(ret);
61         napi_throw_error(env, std::to_string(ret).c_str(), errorMsg.c_str());
62         return nullptr;
63     }
64     return reinterpret_cast<void *>(&NetworkVpnClient::GetInstance());
65 }
66 
CreateVpnConnection(napi_env env,napi_callback_info info)67 static napi_value CreateVpnConnection(napi_env env, napi_callback_info info)
68 {
69     return ModuleTemplate::NewInstance(env, info, VPN_CONNECTION, MakeData, [](napi_env, void *data, void *) {
70         NETMANAGER_EXT_LOGI("finalize VpnConnection");
71     });
72 }
73 
On(napi_env env,napi_callback_info info)74 static napi_value On(napi_env env, napi_callback_info info)
75 {
76     return VpnMonitor::GetInstance().On(env, info);
77 }
78 
Off(napi_env env,napi_callback_info info)79 static napi_value Off(napi_env env, napi_callback_info info)
80 {
81     return VpnMonitor::GetInstance().Off(env, info);
82 }
83 
84 #ifdef SUPPORT_SYSVPN
AddSysVpnConfig(napi_env env,napi_callback_info info)85 static napi_value AddSysVpnConfig(napi_env env, napi_callback_info info)
86 {
87     return ModuleTemplate::Interface<AddContext>(env, info, ADD_SYS_VPN_CONFIG, nullptr,
88         VpnAsyncWork::ExecAddSysVpnConfig, VpnAsyncWork::AddSysVpnConfigCallback);
89 }
90 
DeleteSysVpnConfig(napi_env env,napi_callback_info info)91 static napi_value DeleteSysVpnConfig(napi_env env, napi_callback_info info)
92 {
93     return ModuleTemplate::Interface<DeleteContext>(env, info, DELETE_SYS_VPN_CONFIG, nullptr,
94         VpnAsyncWork::ExecDeleteSysVpnConfig, VpnAsyncWork::DeleteSysVpnConfigCallback);
95 }
96 
GetSysVpnConfigList(napi_env env,napi_callback_info info)97 static napi_value GetSysVpnConfigList(napi_env env, napi_callback_info info)
98 {
99     return ModuleTemplate::Interface<GetListContext>(env, info, GET_SYS_VPN_CONFIG_LIST, nullptr,
100         VpnAsyncWork::ExecGetSysVpnConfigList, VpnAsyncWork::GetSysVpnConfigListCallback);
101 }
102 
GetSysVpnConfig(napi_env env,napi_callback_info info)103 static napi_value GetSysVpnConfig(napi_env env, napi_callback_info info)
104 {
105     return ModuleTemplate::Interface<GetContext>(env, info, GET_SYS_VPN_CONFIG, nullptr,
106         VpnAsyncWork::ExecGetSysVpnConfig, VpnAsyncWork::GetSysVpnConfigCallback);
107 }
108 
GetConnectedSysVpnConfig(napi_env env,napi_callback_info info)109 static napi_value GetConnectedSysVpnConfig(napi_env env, napi_callback_info info)
110 {
111     return ModuleTemplate::Interface<GetConnectedContext>(env, info, GET_CONNECTED_SYS_VPN_CONFIG, nullptr,
112         VpnAsyncWork::ExecGetConnectedSysVpnConfig, VpnAsyncWork::GetConnectedSysVpnConfigCallback);
113 }
114 #endif // SUPPORT_SYSVPN
115 
RegisterVpnModule(napi_env env,napi_value exports)116 napi_value RegisterVpnModule(napi_env env, napi_value exports)
117 {
118     NapiUtils::DefineProperties(env, exports,
119                                 {
120                                     DECLARE_NAPI_FUNCTION(CREATE_VPN_CONNECTION, CreateVpnConnection),
121                                     DECLARE_NAPI_FUNCTION(ON, On),
122                                     DECLARE_NAPI_FUNCTION(OFF, Off),
123                                     #ifdef SUPPORT_SYSVPN
124                                     DECLARE_NAPI_FUNCTION(ADD_SYS_VPN_CONFIG, AddSysVpnConfig),
125                                     DECLARE_NAPI_FUNCTION(DELETE_SYS_VPN_CONFIG, DeleteSysVpnConfig),
126                                     DECLARE_NAPI_FUNCTION(GET_SYS_VPN_CONFIG_LIST, GetSysVpnConfigList),
127                                     DECLARE_NAPI_FUNCTION(GET_SYS_VPN_CONFIG, GetSysVpnConfig),
128                                     DECLARE_NAPI_FUNCTION(GET_CONNECTED_SYS_VPN_CONFIG, GetConnectedSysVpnConfig),
129                                     #endif // SUPPORT_SYSVPN
130                                 });
131     ModuleTemplate::DefineClass(env, exports,
132                                 {
133                                     DECLARE_NAPI_FUNCTION(SET_UP, VpnConnection::SetUp),
134                                     DECLARE_NAPI_FUNCTION(PROTECT, VpnConnection::Protect),
135                                     DECLARE_NAPI_FUNCTION(DESTROY, VpnConnection::Destroy),
136                                 },
137                                 VPN_CONNECTION);
138     return exports;
139 }
140 
141 static napi_module g_vpnModule = {
142     .nm_version = 1,
143     .nm_flags = 0,
144     .nm_filename = nullptr,
145     .nm_register_func = RegisterVpnModule,
146     .nm_modname = VPN_MODULE_NAME,
147     .nm_priv = nullptr,
148     .reserved = {nullptr},
149 };
150 
VpnNapiRegister()151 extern "C" __attribute__((constructor)) void VpnNapiRegister()
152 {
153     napi_module_register(&g_vpnModule);
154 }
155 } // namespace NetManagerStandard
156 } // namespace OHOS