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 "vpn_exec.h"
17
18 #include <cstdint>
19 #include <securec.h>
20
21 #include "napi_utils.h"
22 #include "net_manager_constants.h"
23 #include "netmanager_ext_log.h"
24 #include "networkvpn_client.h"
25 #ifdef SUPPORT_SYSVPN
26 #include "vpn_config_utils.h"
27 #endif // SUPPORT_SYSVPN
28
29 namespace OHOS {
30 namespace NetManagerStandard {
31 namespace VpnExec {
GetVpnConnectionInstance(ContextT * context)32 template <typename ContextT> static inline NetworkVpnClient *GetVpnConnectionInstance(ContextT *context)
33 {
34 auto manager = context->GetManager();
35 return (manager == nullptr) ? nullptr : reinterpret_cast<NetworkVpnClient *>(manager->GetData());
36 }
37
ExecPrepare(PrepareContext * context)38 bool ExecPrepare(PrepareContext *context)
39 {
40 auto vpnClient = GetVpnConnectionInstance(context);
41 if (vpnClient == nullptr) {
42 NETMANAGER_EXT_LOGE("vpnClient is nullptr");
43 return false;
44 }
45 int32_t result = vpnClient->Prepare(context->isExistVpn_, context->isRun_, context->package_);
46 if (result != NETMANAGER_EXT_SUCCESS) {
47 context->SetErrorCode(result);
48 return false;
49 }
50 return true;
51 }
52
ExecSetUp(SetUpContext * context)53 bool ExecSetUp(SetUpContext *context)
54 {
55 auto vpnClient = GetVpnConnectionInstance(context);
56 if (vpnClient == nullptr) {
57 NETMANAGER_EXT_LOGE("vpnClient is nullptr");
58 return false;
59 }
60 int32_t result = vpnClient->SetUpVpn(context->vpnConfig_, context->fd_);
61 if (result != NETMANAGER_EXT_SUCCESS) {
62 context->SetErrorCode(result);
63 return false;
64 }
65 return true;
66 }
67
ExecProtect(ProtectContext * context)68 bool ExecProtect(ProtectContext *context)
69 {
70 auto vpnClient = GetVpnConnectionInstance(context);
71 if (vpnClient == nullptr) {
72 NETMANAGER_EXT_LOGE("vpnClient is nullptr");
73 return false;
74 }
75 int32_t result = vpnClient->Protect(context->socketFd_);
76 if (result != NETMANAGER_EXT_SUCCESS) {
77 context->SetErrorCode(result);
78 return false;
79 }
80 return true;
81 }
82
ExecDestroy(DestroyContext * context)83 bool ExecDestroy(DestroyContext *context)
84 {
85 auto vpnClient = GetVpnConnectionInstance(context);
86 if (vpnClient == nullptr) {
87 NETMANAGER_EXT_LOGE("vpnClient is nullptr");
88 return false;
89 }
90 int32_t result = vpnClient->DestroyVpn();
91 if (result != NETMANAGER_EXT_SUCCESS) {
92 context->SetErrorCode(result);
93 return false;
94 }
95 return true;
96 }
97
98 #ifdef SUPPORT_SYSVPN
ExecAddSysVpnConfig(AddContext * context)99 bool ExecAddSysVpnConfig(AddContext *context)
100 {
101 int32_t result = NetworkVpnClient::GetInstance().AddSysVpnConfig(context->vpnConfig_);
102 if (result != NETMANAGER_EXT_SUCCESS) {
103 context->SetErrorCode(result);
104 return false;
105 }
106 return true;
107 }
108
ExecDeleteSysVpnConfig(DeleteContext * context)109 bool ExecDeleteSysVpnConfig(DeleteContext *context)
110 {
111 int32_t result = NetworkVpnClient::GetInstance().DeleteSysVpnConfig(context->vpnId_);
112 if (result != NETMANAGER_EXT_SUCCESS) {
113 context->SetErrorCode(result);
114 return false;
115 }
116 return true;
117 }
118
ExecGetSysVpnConfigList(GetListContext * context)119 bool ExecGetSysVpnConfigList(GetListContext *context)
120 {
121 int32_t result = NetworkVpnClient::GetInstance().GetSysVpnConfigList(context->vpnList_);
122 if (result != NETMANAGER_EXT_SUCCESS) {
123 context->SetErrorCode(result);
124 return false;
125 }
126 return true;
127 }
128
ExecGetSysVpnConfig(GetContext * context)129 bool ExecGetSysVpnConfig(GetContext *context)
130 {
131 int32_t result = NetworkVpnClient::GetInstance().GetSysVpnConfig(context->vpnConfig_, context->vpnId_);
132 if (result != NETMANAGER_EXT_SUCCESS) {
133 context->SetErrorCode(result);
134 return false;
135 }
136 return true;
137 }
138
ExecGetConnectedSysVpnConfig(GetConnectedContext * context)139 bool ExecGetConnectedSysVpnConfig(GetConnectedContext *context)
140 {
141 int32_t result = NetworkVpnClient::GetInstance().GetConnectedSysVpnConfig(context->vpnConfig_);
142 if (result != NETMANAGER_EXT_SUCCESS) {
143 context->SetErrorCode(result);
144 return false;
145 }
146 return true;
147 }
148 #endif // SUPPORT_SYSVPN
149
PrepareCallback(PrepareContext * context)150 napi_value PrepareCallback(PrepareContext *context)
151 {
152 napi_value obj = NapiUtils::CreateObject(context->GetEnv());
153 NapiUtils::SetBooleanProperty(context->GetEnv(), obj, "isExistVpn", context->isExistVpn_);
154 NapiUtils::SetBooleanProperty(context->GetEnv(), obj, "isRun", context->isRun_);
155 NapiUtils::SetStringPropertyUtf8(context->GetEnv(), obj, "package", context->package_);
156 return obj;
157 }
158
SetUpCallback(SetUpContext * context)159 napi_value SetUpCallback(SetUpContext *context)
160 {
161 return NapiUtils::CreateInt32(context->GetEnv(), context->fd_);
162 }
163
ProtectCallback(ProtectContext * context)164 napi_value ProtectCallback(ProtectContext *context)
165 {
166 return NapiUtils::GetUndefined(context->GetEnv());
167 }
168
DestroyCallback(DestroyContext * context)169 napi_value DestroyCallback(DestroyContext *context)
170 {
171 return NapiUtils::GetUndefined(context->GetEnv());
172 }
173
174 #ifdef SUPPORT_SYSVPN
AddSysVpnConfigCallback(AddContext * context)175 napi_value AddSysVpnConfigCallback(AddContext *context)
176 {
177 return NapiUtils::GetUndefined(context->GetEnv());
178 }
179
DeleteSysVpnConfigCallback(DeleteContext * context)180 napi_value DeleteSysVpnConfigCallback(DeleteContext *context)
181 {
182 return NapiUtils::GetUndefined(context->GetEnv());
183 }
184
GetSysVpnConfigCallback(GetContext * context)185 napi_value GetSysVpnConfigCallback(GetContext *context)
186 {
187 return VpnConfigUtils::CreateNapiVpnConfig(context->GetEnv(), context->vpnConfig_);
188 }
189
GetSysVpnConfigListCallback(GetListContext * context)190 napi_value GetSysVpnConfigListCallback(GetListContext *context)
191 {
192 int32_t index = 0;
193 auto len = context->vpnList_.size();
194 napi_value array = NapiUtils::CreateArray(context->GetEnv(), len);
195 for (const auto &info : context->vpnList_) {
196 napi_value config = NapiUtils::CreateObject(context->GetEnv());
197 NapiUtils::SetStringPropertyUtf8(context->GetEnv(), config, VpnConfigUtils::CONFIG_VPN_ID, info.vpnId_);
198 NapiUtils::SetStringPropertyUtf8(context->GetEnv(), config, VpnConfigUtils::CONFIG_VPN_NAME, info.vpnName_);
199 NapiUtils::SetArrayElement(context->GetEnv(), array, index, config);
200 ++index;
201 }
202 return array;
203 }
204
GetConnectedSysVpnConfigCallback(GetConnectedContext * context)205 napi_value GetConnectedSysVpnConfigCallback(GetConnectedContext *context)
206 {
207 return VpnConfigUtils::CreateNapiVpnConfig(context->GetEnv(), context->vpnConfig_);
208 }
209 #endif // SUPPORT_SYSVPN
210 } // namespace VpnExec
211 } // namespace NetManagerStandard
212 } // namespace OHOS