1 /*
2  * Copyright (c) 2024 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 <vector>
17 #include <cstdint>
18 
19 #include "securec.h"
20 
21 #include "napi/native_api.h"
22 #include "napi/native_node_api.h"
23 
24 #include "asset_log.h"
25 #include "asset_mem.h"
26 #include "asset_system_api.h"
27 #include "asset_system_type.h"
28 
29 #include "asset_napi_add.h"
30 #include "asset_napi_check.h"
31 #include "asset_napi_common.h"
32 
33 namespace OHOS {
34 namespace Security {
35 namespace Asset {
36 namespace {
37 
38 const std::vector<uint32_t> REQUIRED_TAGS = {
39     SEC_ASSET_TAG_SECRET,
40     SEC_ASSET_TAG_ALIAS
41 };
42 
43 const std::vector<uint32_t> OPTIONAL_TAGS = {
44     SEC_ASSET_TAG_SECRET,
45     SEC_ASSET_TAG_CONFLICT_RESOLUTION
46 };
47 
CheckAddArgs(const napi_env env,const std::vector<AssetAttr> & attrs)48 napi_status CheckAddArgs(const napi_env env, const std::vector<AssetAttr> &attrs)
49 {
50     IF_FALSE_RETURN(CheckAssetRequiredTag(env, attrs, REQUIRED_TAGS), napi_invalid_arg);
51     std::vector<uint32_t> validTags;
52     validTags.insert(validTags.end(), CRITICAL_LABEL_TAGS.begin(), CRITICAL_LABEL_TAGS.end());
53     validTags.insert(validTags.end(), NORMAL_LABEL_TAGS.begin(), NORMAL_LABEL_TAGS.end());
54     validTags.insert(validTags.end(), NORMAL_LOCAL_LABEL_TAGS.begin(), NORMAL_LOCAL_LABEL_TAGS.end());
55     validTags.insert(validTags.end(), ACCESS_CONTROL_TAGS.begin(), ACCESS_CONTROL_TAGS.end());
56     validTags.insert(validTags.end(), ASSET_SYNC_TAGS.begin(), ASSET_SYNC_TAGS.end());
57     validTags.insert(validTags.end(), OPTIONAL_TAGS.begin(), OPTIONAL_TAGS.end());
58     IF_FALSE_RETURN(CheckAssetTagValidity(env, attrs, validTags), napi_invalid_arg);
59     IF_FALSE_RETURN(CheckAssetValueValidity(env, attrs), napi_invalid_arg);
60     return napi_ok;
61 }
62 
63 } // anonymous namespace
64 
NapiAdd(const napi_env env,napi_callback_info info,const NapiCallerArgs & args)65 napi_value NapiAdd(const napi_env env, napi_callback_info info, const NapiCallerArgs &args)
66 {
67     napi_async_execute_callback execute = [](napi_env env, void *data) {
68         AsyncContext *context = static_cast<AsyncContext *>(data);
69         context->result = AssetAdd(&context->attrs[0], context->attrs.size());
70     };
71     return NapiAsync(env, info, execute, args, &CheckAddArgs);
72 }
73 
NapiAdd(const napi_env env,napi_callback_info info)74 napi_value NapiAdd(const napi_env env, napi_callback_info info)
75 {
76     NapiCallerArgs args = { .expectArgNum = NORMAL_ARGS_NUM, .isUpdate = false, .isAsUser = false };
77     return NapiAdd(env, info, args);
78 }
79 
NapiAddAsUser(const napi_env env,napi_callback_info info)80 napi_value NapiAddAsUser(const napi_env env, napi_callback_info info)
81 {
82     NapiCallerArgs args = { .expectArgNum = AS_USER_ARGS_NUM, .isUpdate = false, .isAsUser = true };
83     return NapiAdd(env, info, args);
84 }
85 
NapiAddSync(const napi_env env,napi_callback_info info)86 napi_value NapiAddSync(const napi_env env, napi_callback_info info)
87 {
88     std::vector<AssetAttr> attrs;
89     NapiCallerArgs args = { .expectArgNum = NORMAL_ARGS_NUM, .isUpdate = false, .isAsUser = false };
90     do {
91         if (ParseParam(env, info, args, attrs) != napi_ok) {
92             break;
93         }
94 
95         if (CheckAddArgs(env, attrs) != napi_ok) {
96             break;
97         }
98 
99         int32_t result = AssetAdd(&attrs[0], attrs.size());
100         CHECK_RESULT_BREAK(env, result);
101     } while (false);
102     FreeAssetAttrs(attrs);
103     return nullptr;
104 }
105 
106 } // Asset
107 } // Security
108 } // OHOS
109