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 "huks_napi_generate_key.h"
17
18 #include "securec.h"
19
20 #include "hks_api.h"
21 #include "hks_log.h"
22 #include "hks_mem.h"
23 #include "hks_param.h"
24 #include "hks_type.h"
25 #include "huks_napi_common.h"
26
27 namespace HuksNapi {
28 namespace {
29 constexpr int HUKS_NAPI_GENERATE_KEY_MIN_ARGS = 2;
30 constexpr int HUKS_NAPI_GENERATE_KEY_MAX_ARGS = 3;
31 } // namespace
32
33 struct GenerateKeyAsyncContextT {
34 napi_async_work asyncWork = nullptr;
35 napi_deferred deferred = nullptr;
36 napi_ref callback = nullptr;
37
38 int32_t result = 0;
39 struct HksBlob *keyAlias = nullptr;
40 struct HksParamSet *paramSetIn = nullptr;
41 struct HksParamSet *paramSetOut = nullptr;
42 };
43 using GenerateKeyAsyncContext = GenerateKeyAsyncContextT *;
44
CreateGenerateKeyAsyncContext()45 static GenerateKeyAsyncContext CreateGenerateKeyAsyncContext()
46 {
47 GenerateKeyAsyncContext context = static_cast<GenerateKeyAsyncContext>(HksMalloc(sizeof(GenerateKeyAsyncContextT)));
48 if (context != nullptr) {
49 (void)memset_s(context, sizeof(GenerateKeyAsyncContextT), 0, sizeof(GenerateKeyAsyncContextT));
50 }
51 return context;
52 }
53
DeleteGenerateKeyAsyncContext(napi_env env,GenerateKeyAsyncContext & context)54 static void DeleteGenerateKeyAsyncContext(napi_env env, GenerateKeyAsyncContext &context)
55 {
56 if (context == nullptr) {
57 return;
58 }
59 DeleteCommonAsyncContext(env, context->asyncWork, context->callback, context->keyAlias, context->paramSetIn);
60 if (context->paramSetOut != nullptr) {
61 HksFreeParamSet(&context->paramSetOut);
62 }
63 HKS_FREE(context);
64 context = nullptr;
65 }
66
GenerateKeyParseParams(napi_env env,napi_callback_info info,GenerateKeyAsyncContext context)67 static napi_value GenerateKeyParseParams(napi_env env, napi_callback_info info, GenerateKeyAsyncContext context)
68 {
69 size_t argc = HUKS_NAPI_GENERATE_KEY_MAX_ARGS;
70 napi_value argv[HUKS_NAPI_GENERATE_KEY_MAX_ARGS] = { 0 };
71 NAPI_CALL(env, napi_get_cb_info(env, info, &argc, argv, nullptr, nullptr));
72
73 if (argc < HUKS_NAPI_GENERATE_KEY_MIN_ARGS) {
74 napi_throw_error(env, nullptr, "invalid arguments");
75 HKS_LOG_E("no enough params");
76 return nullptr;
77 }
78
79 size_t index = 0;
80 napi_value result = ParseKeyAliasAndHksParamSet(env, argv, index, context->keyAlias, context->paramSetIn);
81 if (result == nullptr) {
82 HKS_LOG_E("generateKey parse params failed");
83 return nullptr;
84 }
85
86 index++;
87 if (index < argc) {
88 context->callback = GetCallback(env, argv[index]);
89 }
90
91 return GetInt32(env, 0);
92 }
93
GenerateKeyWriteResult(napi_env env,GenerateKeyAsyncContext context)94 static napi_value GenerateKeyWriteResult(napi_env env, GenerateKeyAsyncContext context)
95 {
96 return GenerateHksResult(env, context->result, nullptr, 0);
97 }
98
GenerateKeyAsyncWork(napi_env env,GenerateKeyAsyncContext & context)99 static napi_value GenerateKeyAsyncWork(napi_env env, GenerateKeyAsyncContext &context)
100 {
101 napi_value promise = nullptr;
102 if (context->callback == nullptr) {
103 NAPI_CALL(env, napi_create_promise(env, &context->deferred, &promise));
104 }
105
106 napi_value resourceName = nullptr;
107 napi_create_string_latin1(env, "generateKeyAsyncWork", NAPI_AUTO_LENGTH, &resourceName);
108
109 napi_create_async_work(
110 env,
111 nullptr,
112 resourceName,
113 [](napi_env env, void *data) {
114 GenerateKeyAsyncContext napiContext = static_cast<GenerateKeyAsyncContext>(data);
115
116 napiContext->result = HksGenerateKey(napiContext->keyAlias,
117 napiContext->paramSetIn, napiContext->paramSetOut);
118 },
119 [](napi_env env, napi_status status, void *data) {
120 GenerateKeyAsyncContext napiContext = static_cast<GenerateKeyAsyncContext>(data);
121 napi_value result = GenerateKeyWriteResult(env, napiContext);
122 if (napiContext->callback == nullptr) {
123 napi_resolve_deferred(env, napiContext->deferred, result);
124 } else if (result != nullptr) {
125 CallAsyncCallback(env, napiContext->callback, napiContext->result, result);
126 }
127 DeleteGenerateKeyAsyncContext(env, napiContext);
128 },
129 static_cast<void *>(context),
130 &context->asyncWork);
131
132 napi_status status = napi_queue_async_work(env, context->asyncWork);
133 if (status != napi_ok) {
134 GET_AND_THROW_LAST_ERROR((env));
135 DeleteGenerateKeyAsyncContext(env, context);
136 HKS_LOG_E("could not queue async work");
137 return nullptr;
138 }
139
140 if (context->callback == nullptr) {
141 return promise;
142 } else {
143 return GetNull(env);
144 }
145 }
146
HuksNapiGenerateKey(napi_env env,napi_callback_info info)147 napi_value HuksNapiGenerateKey(napi_env env, napi_callback_info info)
148 {
149 GenerateKeyAsyncContext context = CreateGenerateKeyAsyncContext();
150 if (context == nullptr) {
151 HKS_LOG_E("could not create context");
152 return nullptr;
153 }
154
155 napi_value result = GenerateKeyParseParams(env, info, context);
156 if (result == nullptr) {
157 HKS_LOG_E("could not parse params");
158 DeleteGenerateKeyAsyncContext(env, context);
159 return nullptr;
160 }
161
162 result = GenerateKeyAsyncWork(env, context);
163 if (result == nullptr) {
164 HKS_LOG_E("could not start async work");
165 DeleteGenerateKeyAsyncContext(env, context);
166 return nullptr;
167 }
168 return result;
169 }
170 } // namespace HuksNapi
171