1 /*
2 * Copyright (c) 2021-2022 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 "tlssocket_module.h"
17
18 #include <initializer_list>
19 #include <napi/native_common.h>
20
21 #include "common_context.h"
22 #include "event_manager.h"
23 #include "module_template.h"
24 #include "monitor.h"
25 #include "napi_utils.h"
26 #include "netstack_log.h"
27 #include "tls.h"
28 #include "tls_bind_context.h"
29 #include "tls_connect_context.h"
30 #include "tls_extra_context.h"
31 #include "tls_napi_context.h"
32 #include "tls_send_context.h"
33 #include "tlssocket_async_work.h"
34
35 namespace OHOS {
36 namespace NetStack {
37 namespace TlsSocket {
38 namespace {
39 static constexpr const char *PROTOCOL_TLSV13 = "TLSv13";
40 static constexpr const char *PROTOCOL_TLSV12 = "TLSv12";
41
Finalize(napi_env,void * data,void *)42 void Finalize(napi_env, void *data, void *)
43 {
44 auto manager = reinterpret_cast<EventManager *>(data);
45 if (manager != nullptr) {
46 EventManager::SetInvalid(manager);
47 }
48 }
49 } // namespace
50
GetCertificate(napi_env env,napi_callback_info info)51 napi_value TLSSocketModuleExports::TLSSocket::GetCertificate(napi_env env, napi_callback_info info)
52 {
53 return ModuleTemplate::Interface<GetCertificateContext>(env, info, FUNCTION_GET_CERTIFICATE, nullptr,
54 TLSSocketAsyncWork::ExecGetCertificate,
55 TLSSocketAsyncWork::GetCertificateCallback);
56 }
57
GetProtocol(napi_env env,napi_callback_info info)58 napi_value TLSSocketModuleExports::TLSSocket::GetProtocol(napi_env env, napi_callback_info info)
59 {
60 return ModuleTemplate::Interface<GetCipherSuitesContext>(env, info, FUNCTION_GET_PROTOCOL, nullptr,
61 TLSSocketAsyncWork::ExecGetProtocol,
62 TLSSocketAsyncWork::GetProtocolCallback);
63 }
64
Connect(napi_env env,napi_callback_info info)65 napi_value TLSSocketModuleExports::TLSSocket::Connect(napi_env env, napi_callback_info info)
66 {
67 return ModuleTemplate::Interface<TLSConnectContext>(
68 env, info, FUNCTION_CONNECT, nullptr, TLSSocketAsyncWork::ExecConnect, TLSSocketAsyncWork::ConnectCallback);
69 }
70
GetCipherSuites(napi_env env,napi_callback_info info)71 napi_value TLSSocketModuleExports::TLSSocket::GetCipherSuites(napi_env env, napi_callback_info info)
72 {
73 return ModuleTemplate::Interface<GetCipherSuitesContext>(env, info, FUNCTION_GET_CIPHER_SUITE, nullptr,
74 TLSSocketAsyncWork::ExecGetCipherSuites,
75 TLSSocketAsyncWork::GetCipherSuitesCallback);
76 }
77
GetRemoteCertificate(napi_env env,napi_callback_info info)78 napi_value TLSSocketModuleExports::TLSSocket::GetRemoteCertificate(napi_env env, napi_callback_info info)
79 {
80 return ModuleTemplate::Interface<GetRemoteCertificateContext>(env, info, FUNCTION_GET_REMOTE_CERTIFICATE, nullptr,
81 TLSSocketAsyncWork::ExecGetRemoteCertificate,
82 TLSSocketAsyncWork::GetRemoteCertificateCallback);
83 }
84
GetSignatureAlgorithms(napi_env env,napi_callback_info info)85 napi_value TLSSocketModuleExports::TLSSocket::GetSignatureAlgorithms(napi_env env, napi_callback_info info)
86 {
87 return ModuleTemplate::Interface<GetSignatureAlgorithmsContext>(
88 env, info, FUNCTION_GET_SIGNATURE_ALGORITHMS, nullptr, TLSSocketAsyncWork::ExecGetSignatureAlgorithms,
89 TLSSocketAsyncWork::GetSignatureAlgorithmsCallback);
90 }
91
Send(napi_env env,napi_callback_info info)92 napi_value TLSSocketModuleExports::TLSSocket::Send(napi_env env, napi_callback_info info)
93 {
94 return ModuleTemplate::Interface<TLSSendContext>(env, info, FUNCTION_SEND, nullptr, TLSSocketAsyncWork::ExecSend,
95 TLSSocketAsyncWork::SendCallback);
96 }
97
Close(napi_env env,napi_callback_info info)98 napi_value TLSSocketModuleExports::TLSSocket::Close(napi_env env, napi_callback_info info)
99 {
100 return ModuleTemplate::Interface<TLSNapiContext>(env, info, FUNCTION_CLOSE, nullptr, TLSSocketAsyncWork::ExecClose,
101 TLSSocketAsyncWork::CloseCallback);
102 }
103
Bind(napi_env env,napi_callback_info info)104 napi_value TLSSocketModuleExports::TLSSocket::Bind(napi_env env, napi_callback_info info)
105 {
106 return ModuleTemplate::Interface<TLSBindContext>(env, info, FUNCTION_BIND, nullptr, TLSSocketAsyncWork::ExecBind,
107 TLSSocketAsyncWork::BindCallback);
108 }
109
GetState(napi_env env,napi_callback_info info)110 napi_value TLSSocketModuleExports::TLSSocket::GetState(napi_env env, napi_callback_info info)
111 {
112 return ModuleTemplate::Interface<TLSGetStateContext>(
113 env, info, FUNCTION_GET_STATE, nullptr, TLSSocketAsyncWork::ExecGetState, TLSSocketAsyncWork::GetStateCallback);
114 }
115
GetRemoteAddress(napi_env env,napi_callback_info info)116 napi_value TLSSocketModuleExports::TLSSocket::GetRemoteAddress(napi_env env, napi_callback_info info)
117 {
118 return ModuleTemplate::Interface<TLSGetRemoteAddressContext>(env, info, FUNCTION_GET_REMOTE_ADDRESS, nullptr,
119 TLSSocketAsyncWork::ExecGetRemoteAddress,
120 TLSSocketAsyncWork::GetRemoteAddressCallback);
121 }
122
SetExtraOptions(napi_env env,napi_callback_info info)123 napi_value TLSSocketModuleExports::TLSSocket::SetExtraOptions(napi_env env, napi_callback_info info)
124 {
125 return ModuleTemplate::Interface<TLSSetExtraOptionsContext>(env, info, FUNCTION_BIND, nullptr,
126 TLSSocketAsyncWork::ExecSetExtraOptions,
127 TLSSocketAsyncWork::SetExtraOptionsCallback);
128 }
129
On(napi_env env,napi_callback_info info)130 napi_value TLSSocketModuleExports::TLSSocket::On(napi_env env, napi_callback_info info)
131 {
132 return DelayedSingleton<Monitor>::GetInstance()->On(env, info);
133 }
134
Off(napi_env env,napi_callback_info info)135 napi_value TLSSocketModuleExports::TLSSocket::Off(napi_env env, napi_callback_info info)
136 {
137 return DelayedSingleton<Monitor>::GetInstance()->Off(env, info);
138 }
139
DefineTLSSocketClass(napi_env env,napi_value exports)140 void TLSSocketModuleExports::DefineTLSSocketClass(napi_env env, napi_value exports)
141 {
142 std::initializer_list<napi_property_descriptor> functions = {
143 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_GET_CERTIFICATE, TLSSocket::GetCertificate),
144 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_GET_REMOTE_CERTIFICATE, TLSSocket::GetRemoteCertificate),
145 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_GET_SIGNATURE_ALGORITHMS, TLSSocket::GetSignatureAlgorithms),
146 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_GET_PROTOCOL, TLSSocket::GetProtocol),
147 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_CONNECT, TLSSocket::Connect),
148 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_GET_CIPHER_SUITE, TLSSocket::GetCipherSuites),
149 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_SEND, TLSSocket::Send),
150 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_CLOSE, TLSSocket::Close),
151 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_BIND, TLSSocket::Bind),
152 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_GET_STATE, TLSSocket::GetState),
153 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_GET_REMOTE_ADDRESS, TLSSocket::GetRemoteAddress),
154 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_SET_EXTRA_OPTIONS, TLSSocket::SetExtraOptions),
155 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_ON, TLSSocket::On),
156 DECLARE_NAPI_FUNCTION(TLSSocket::FUNCTION_OFF, TLSSocket::Off),
157 };
158 ModuleTemplate::DefineClass(env, exports, functions, INTERFACE_TLS_SOCKET);
159 }
160
InitProtocol(napi_env env,napi_value exports)161 void TLSSocketModuleExports::InitProtocol(napi_env env, napi_value exports)
162 {
163 std::initializer_list<napi_property_descriptor> properties = {
164 DECLARE_NAPI_STATIC_PROPERTY(PROTOCOL_TLSV12, NapiUtils::CreateStringUtf8(env, PROTOCOL_TLS_V12)),
165 DECLARE_NAPI_STATIC_PROPERTY(PROTOCOL_TLSV13, NapiUtils::CreateStringUtf8(env, PROTOCOL_TLS_V13)),
166 };
167
168 napi_value protocol = NapiUtils::CreateObject(env);
169 NapiUtils::DefineProperties(env, protocol, properties);
170 NapiUtils::SetNamedProperty(env, exports, INTERFACE_PROTOCOL, protocol);
171 }
172
ConstructTLSSocketInstance(napi_env env,napi_callback_info info)173 napi_value TLSSocketModuleExports::ConstructTLSSocketInstance(napi_env env, napi_callback_info info)
174 {
175 return ModuleTemplate::NewInstance(env, info, INTERFACE_TLS_SOCKET, Finalize);
176 }
177
InitTLSSocketProperties(napi_env env,napi_value exports)178 void TLSSocketModuleExports::InitTLSSocketProperties(napi_env env, napi_value exports)
179 {
180 std::initializer_list<napi_property_descriptor> properties = {
181 DECLARE_NAPI_FUNCTION(FUNCTION_CONSTRUCTOR_TLS_SOCKET_INSTANCE, ConstructTLSSocketInstance),
182 };
183 NapiUtils::DefineProperties(env, exports, properties);
184 }
185
InitTLSSocketModule(napi_env env,napi_value exports)186 napi_value TLSSocketModuleExports::InitTLSSocketModule(napi_env env, napi_value exports)
187 {
188 DefineTLSSocketClass(env, exports);
189 InitTLSSocketProperties(env, exports);
190 InitProtocol(env, exports);
191 return exports;
192 }
193 } // namespace TlsSocket
194 } // namespace NetStack
195 } // namespace OHOS
196