1 /*
2 * Copyright (c) 2024-2025 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 "asset_napi_query.h"
17
18 #include "asset_log.h"
19 #include "asset_system_api.h"
20 #include "asset_system_type.h"
21
22 #include "asset_napi_check.h"
23 #include "asset_napi_common.h"
24
25 namespace OHOS {
26 namespace Security {
27 namespace Asset {
28 namespace {
29 const uint32_t QUERY_ARG_COUNT = 1;
30 const uint32_t QUERY_ARG_COUNT_AS_USER = 2;
31
32 const std::vector<uint32_t> OPTIONAL_TAGS = {
33 SEC_ASSET_TAG_RETURN_LIMIT,
34 SEC_ASSET_TAG_RETURN_OFFSET,
35 SEC_ASSET_TAG_RETURN_ORDERED_BY,
36 SEC_ASSET_TAG_RETURN_TYPE,
37 SEC_ASSET_TAG_AUTH_TOKEN,
38 SEC_ASSET_TAG_AUTH_CHALLENGE,
39 };
40
CheckQueryArgs(const napi_env env,const std::vector<AssetAttr> & attrs)41 napi_status CheckQueryArgs(const napi_env env, const std::vector<AssetAttr> &attrs)
42 {
43 std::vector<uint32_t> validTags;
44 validTags.insert(validTags.end(), CRITICAL_LABEL_TAGS.begin(), CRITICAL_LABEL_TAGS.end());
45 validTags.insert(validTags.end(), NORMAL_LABEL_TAGS.begin(), NORMAL_LABEL_TAGS.end());
46 validTags.insert(validTags.end(), NORMAL_LOCAL_LABEL_TAGS.begin(), NORMAL_LOCAL_LABEL_TAGS.end());
47 validTags.insert(validTags.end(), ACCESS_CONTROL_TAGS.begin(), ACCESS_CONTROL_TAGS.end());
48 validTags.insert(validTags.end(), ASSET_SYNC_TAGS.begin(), ASSET_SYNC_TAGS.end());
49 validTags.insert(validTags.end(), OPTIONAL_TAGS.begin(), OPTIONAL_TAGS.end());
50 IF_ERROR_THROW_RETURN(env, CheckAssetTagValidity(env, attrs, validTags, SEC_ASSET_INVALID_ARGUMENT));
51 IF_ERROR_THROW_RETURN(env, CheckAssetValueValidity(env, attrs, SEC_ASSET_INVALID_ARGUMENT));
52 return napi_ok;
53 }
54
ParseAttrMap(napi_env env,napi_callback_info info,BaseContext * context)55 napi_status ParseAttrMap(napi_env env, napi_callback_info info, BaseContext *context)
56 {
57 napi_value argv[MAX_ARGS_NUM] = { 0 };
58 IF_ERR_RETURN(ParseJsArgs(env, info, argv, QUERY_ARG_COUNT));
59 IF_ERR_RETURN(ParseJsMap(env, argv[0], context->attrs));
60 IF_ERR_RETURN(CheckQueryArgs(env, context->attrs));
61 return napi_ok;
62 }
63
ParseAttrMapAsUser(napi_env env,napi_callback_info info,BaseContext * context)64 napi_status ParseAttrMapAsUser(napi_env env, napi_callback_info info, BaseContext *context)
65 {
66 napi_value argv[MAX_ARGS_NUM] = { 0 };
67 IF_ERR_RETURN(ParseJsArgs(env, info, argv, QUERY_ARG_COUNT_AS_USER));
68 uint32_t index = 0;
69 IF_ERR_RETURN(ParseJsUserId(env, argv[index++], context->attrs));
70 IF_ERR_RETURN(ParseJsMap(env, argv[index++], context->attrs));
71 IF_ERR_RETURN(CheckQueryArgs(env, context->attrs));
72 return napi_ok;
73 }
74 } // anonymous namespace
75
NapiQuery(const napi_env env,napi_callback_info info,bool asUser,bool async)76 napi_value NapiQuery(const napi_env env, napi_callback_info info, bool asUser, bool async)
77 {
78 auto context = std::unique_ptr<QueryContext>(new (std::nothrow)QueryContext());
79 NAPI_THROW(env, context == nullptr, SEC_ASSET_OUT_OF_MEMORY, "Unable to allocate memory for Context.");
80
81 context->parse = asUser ? ParseAttrMapAsUser : ParseAttrMap;
82 context->execute = [](napi_env env, void *data) {
83 QueryContext *context = static_cast<QueryContext *>(data);
84 context->result = AssetQuery(&context->attrs[0], context->attrs.size(), &context->resultSet);
85 };
86
87 context->resolve = [](napi_env env, BaseContext *baseContext) -> napi_value {
88 QueryContext *context = static_cast<QueryContext *>(baseContext);
89 return CreateJsMapArray(env, context->resultSet);
90 };
91
92 if (async) {
93 return CreateAsyncWork(env, info, std::move(context), __func__);
94 } else {
95 return CreateSyncWork(env, info, context.get());
96 }
97 }
98
NapiQuery(const napi_env env,napi_callback_info info)99 napi_value NapiQuery(const napi_env env, napi_callback_info info)
100 {
101 return NapiQuery(env, info, false, true);
102 }
103
NapiQueryAsUser(const napi_env env,napi_callback_info info)104 napi_value NapiQueryAsUser(const napi_env env, napi_callback_info info)
105 {
106 return NapiQuery(env, info, true, true);
107 }
108
NapiQuerySync(const napi_env env,napi_callback_info info)109 napi_value NapiQuerySync(const napi_env env, napi_callback_info info)
110 {
111 return NapiQuery(env, info, false, false);
112 }
113
114 } // Asset
115 } // Security
116 } // OHOS
117