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 <cstdint>
17 #include <vector>
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_check.h"
30 #include "asset_napi_common.h"
31 #include "asset_napi_pre_query.h"
32 
33 namespace OHOS {
34 namespace Security {
35 namespace Asset {
36 namespace {
37 
38 const std::vector<uint32_t> OPTIONAL_TAGS = {
39     SEC_ASSET_TAG_AUTH_VALIDITY_PERIOD
40 };
41 
CheckPreQueryArgs(const napi_env env,const std::vector<AssetAttr> & attrs)42 napi_status CheckPreQueryArgs(const napi_env env, const std::vector<AssetAttr> &attrs)
43 {
44     std::vector<uint32_t> validTags;
45     validTags.insert(validTags.end(), CRITICAL_LABEL_TAGS.begin(), CRITICAL_LABEL_TAGS.end());
46     validTags.insert(validTags.end(), NORMAL_LABEL_TAGS.begin(), NORMAL_LABEL_TAGS.end());
47     validTags.insert(validTags.end(), NORMAL_LOCAL_LABEL_TAGS.begin(), NORMAL_LOCAL_LABEL_TAGS.end());
48     validTags.insert(validTags.end(), ACCESS_CONTROL_TAGS.begin(), ACCESS_CONTROL_TAGS.end());
49     validTags.insert(validTags.end(), OPTIONAL_TAGS.begin(), OPTIONAL_TAGS.end());
50     IF_FALSE_RETURN(CheckAssetTagValidity(env, attrs, validTags), napi_invalid_arg);
51     IF_FALSE_RETURN(CheckAssetValueValidity(env, attrs), napi_invalid_arg);
52     return napi_ok;
53 }
54 
55 } // anonymous namespace
56 
NapiPreQuery(const napi_env env,napi_callback_info info,const NapiCallerArgs & args)57 napi_value NapiPreQuery(const napi_env env, napi_callback_info info, const NapiCallerArgs &args)
58 {
59     napi_async_execute_callback execute =
60         [](napi_env env, void *data) {
61             AsyncContext *context = static_cast<AsyncContext *>(data);
62             context->result = AssetPreQuery(&context->attrs[0], context->attrs.size(), &context->challenge);
63         };
64     return NapiAsync(env, info, execute, args, &CheckPreQueryArgs);
65 }
66 
NapiPreQuery(const napi_env env,napi_callback_info info)67 napi_value NapiPreQuery(const napi_env env, napi_callback_info info)
68 {
69     NapiCallerArgs args = { .expectArgNum = NORMAL_ARGS_NUM, .isUpdate = false, .isAsUser = false };
70     return NapiPreQuery(env, info, args);
71 }
72 
NapiPreQueryAsUser(const napi_env env,napi_callback_info info)73 napi_value NapiPreQueryAsUser(const napi_env env, napi_callback_info info)
74 {
75     NapiCallerArgs args = { .expectArgNum = AS_USER_ARGS_NUM, .isUpdate = false, .isAsUser = true };
76     return NapiPreQuery(env, info, args);
77 }
78 
NapiPreQuerySync(const napi_env env,napi_callback_info info)79 napi_value NapiPreQuerySync(const napi_env env, napi_callback_info info)
80 {
81     std::vector<AssetAttr> attrs;
82     AssetBlob challenge = { 0 };
83     napi_value result = nullptr;
84     NapiCallerArgs args = { .expectArgNum = NORMAL_ARGS_NUM, .isUpdate = false, .isAsUser = false };
85     do {
86         if (ParseParam(env, info, args, attrs) != napi_ok) {
87             break;
88         }
89 
90         if (CheckPreQueryArgs(env, attrs) != napi_ok) {
91             break;
92         }
93 
94         int32_t res = AssetPreQuery(&attrs[0], attrs.size(), &challenge);
95         CHECK_RESULT_BREAK(env, res);
96         result = CreateJsUint8Array(env, challenge);
97     } while (false);
98     AssetFreeBlob(&challenge);
99     FreeAssetAttrs(attrs);
100     return result;
101 }
102 
103 } // Asset
104 } // Security
105 } // OHOS
106