1 /*
2  * Copyright (c) 2022-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 "setremotehaptokeninfo_fuzzer.h"
17 
18 #include <string>
19 #include <vector>
20 #include <thread>
21 #include "accesstoken_fuzzdata.h"
22 #undef private
23 #include "accesstoken_kit.h"
24 
25 using namespace std;
26 using namespace OHOS::Security::AccessToken;
27 
28 namespace OHOS {
SetRemoteHapTokenInfoFuzzTest(const uint8_t* data, size_t size)29     bool SetRemoteHapTokenInfoFuzzTest(const uint8_t* data, size_t size)
30     {
31 #ifdef TOKEN_SYNC_ENABLE
32         if ((data == nullptr) || (size == 0)) {
33             return false;
34         }
35 
36         AccessTokenFuzzData fuzzData(data, size);
37         HapTokenInfo baseInfo = {
38             .apl = APL_NORMAL,
39             .ver = 1,
40             .userID = 1,
41             .bundleName = fuzzData.GenerateRandomString(),
42             .instIndex = 1,
43             .appID = fuzzData.GenerateRandomString(),
44             .deviceID = fuzzData.GenerateRandomString(),
45             .tokenID = fuzzData.GetData<AccessTokenID>(),
46             .tokenAttr = 0
47         };
48         PermissionStateFull infoManagerTestState = {
49             .grantFlags = {PermissionFlag::PERMISSION_SYSTEM_FIXED},
50             .grantStatus = {PermissionState::PERMISSION_GRANTED},
51             .isGeneral = true,
52             .permissionName = fuzzData.GenerateRandomString(),
53             .resDeviceID = {fuzzData.GenerateRandomString()}};
54         std::vector<PermissionStateFull> permStateList;
55         permStateList.emplace_back(infoManagerTestState);
56         HapTokenInfoForSync remoteTokenInfo = {
57             .baseInfo = baseInfo,
58             .permStateList = permStateList
59         };
60 
61         int32_t result = AccessTokenKit::SetRemoteHapTokenInfo(fuzzData.GenerateRandomString(), remoteTokenInfo);
62         return result == RET_SUCCESS;
63 #else
64         return true;
65 #endif
66     }
67 }
68 
69 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)70 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
71 {
72     /* Run your code on data */
73     OHOS::SetRemoteHapTokenInfoFuzzTest(data, size);
74     return 0;
75 }
76 
77