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_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_RETURN_LIMIT,
40     SEC_ASSET_TAG_RETURN_OFFSET,
41     SEC_ASSET_TAG_RETURN_ORDERED_BY,
42     SEC_ASSET_TAG_RETURN_TYPE,
43     SEC_ASSET_TAG_AUTH_TOKEN,
44     SEC_ASSET_TAG_AUTH_CHALLENGE,
45 
46 };
47 
CheckQueryArgs(const napi_env env,const std::vector<AssetAttr> & attrs)48 napi_status CheckQueryArgs(const napi_env env, const std::vector<AssetAttr> &attrs)
49 {
50     std::vector<uint32_t> validTags;
51     validTags.insert(validTags.end(), CRITICAL_LABEL_TAGS.begin(), CRITICAL_LABEL_TAGS.end());
52     validTags.insert(validTags.end(), NORMAL_LABEL_TAGS.begin(), NORMAL_LABEL_TAGS.end());
53     validTags.insert(validTags.end(), NORMAL_LOCAL_LABEL_TAGS.begin(), NORMAL_LOCAL_LABEL_TAGS.end());
54     validTags.insert(validTags.end(), ACCESS_CONTROL_TAGS.begin(), ACCESS_CONTROL_TAGS.end());
55     validTags.insert(validTags.end(), ASSET_SYNC_TAGS.begin(), ASSET_SYNC_TAGS.end());
56     validTags.insert(validTags.end(), OPTIONAL_TAGS.begin(), OPTIONAL_TAGS.end());
57     IF_FALSE_RETURN(CheckAssetTagValidity(env, attrs, validTags), napi_invalid_arg);
58     IF_FALSE_RETURN(CheckAssetValueValidity(env, attrs), napi_invalid_arg);
59     return napi_ok;
60 }
61 
62 } // anonymous namespace
63 
NapiQuery(const napi_env env,napi_callback_info info,const NapiCallerArgs & args)64 napi_value NapiQuery(const napi_env env, napi_callback_info info, const NapiCallerArgs &args)
65 {
66     napi_async_execute_callback execute =
67         [](napi_env env, void *data) {
68             AsyncContext *context = static_cast<AsyncContext *>(data);
69             context->result = AssetQuery(&context->attrs[0], context->attrs.size(), &context->resultSet);
70         };
71     return NapiAsync(env, info, execute, args, &CheckQueryArgs);
72 }
73 
NapiQuery(const napi_env env,napi_callback_info info)74 napi_value NapiQuery(const napi_env env, napi_callback_info info)
75 {
76     NapiCallerArgs args = { .expectArgNum = NORMAL_ARGS_NUM, .isUpdate = false, .isAsUser = false };
77     return NapiQuery(env, info, args);
78 }
79 
NapiQueryAsUser(const napi_env env,napi_callback_info info)80 napi_value NapiQueryAsUser(const napi_env env, napi_callback_info info)
81 {
82     NapiCallerArgs args = { .expectArgNum = AS_USER_ARGS_NUM, .isUpdate = false, .isAsUser = true };
83     return NapiQuery(env, info, args);
84 }
85 
NapiQuerySync(const napi_env env,napi_callback_info info)86 napi_value NapiQuerySync(const napi_env env, napi_callback_info info)
87 {
88     std::vector<AssetAttr> attrs;
89     AssetResultSet resultSet = { 0 };
90     napi_value result = nullptr;
91     NapiCallerArgs args = { .expectArgNum = NORMAL_ARGS_NUM, .isUpdate = false, .isAsUser = false };
92     do {
93         if (ParseParam(env, info, args, attrs) != napi_ok) {
94             break;
95         }
96 
97         if (CheckQueryArgs(env, attrs) != napi_ok) {
98             break;
99         }
100 
101         int32_t res = AssetQuery(&attrs[0], attrs.size(), &resultSet);
102         CHECK_RESULT_BREAK(env, res);
103         result = CreateJsMapArray(env, resultSet);
104     } while (false);
105     AssetFreeResultSet(&resultSet);
106     FreeAssetAttrs(attrs);
107     return result;
108 }
109 
110 } // Asset
111 } // Security
112 } // OHOS
113