1 /*
2 * Copyright (c) 2022 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 "queryaccountinfo_fuzzer.h"
17
18 #include "os_account_manager.h"
19 #include "account_log_wrapper.h"
20 #include "fuzz_data.h"
21 #include "os_account_constants.h"
22 #include <cstdint>
23 #include <string>
24 #include <vector>
25
26 using namespace std;
27 using namespace OHOS::AccountSA;
28
29 namespace OHOS {
QueryOsAccountConstraintSourceTypesFuzzTest(const uint8_t * data,size_t size)30 bool QueryOsAccountConstraintSourceTypesFuzzTest(const uint8_t* data, size_t size)
31 {
32 int32_t result = ERR_OK;
33 if ((data != nullptr) && (size != 0)) {
34 FuzzData fuzzData(data, size);
35 std::vector<ConstraintSourceTypeInfo> testConstraintSourceTypeInfos;
36 int32_t testId = fuzzData.GetData<int32_t>();
37 std::string testConstraint(fuzzData.GenerateString());
38 result = OsAccountManager::QueryOsAccountConstraintSourceTypes(
39 testId, testConstraint, testConstraintSourceTypeInfos);
40 }
41 return result == ERR_OK;
42 }
43
QueryOsAccountByIdFuzzTest(const uint8_t * data,size_t size)44 bool QueryOsAccountByIdFuzzTest(const uint8_t* data, size_t size)
45 {
46 int32_t result = ERR_OK;
47 if ((data != nullptr) && (size != 0)) {
48 FuzzData fuzzData(data, size);
49 int32_t testId = fuzzData.GetData<int32_t>();
50 OsAccountInfo osAccountInfo;
51 result = OsAccountManager::QueryOsAccountById(testId, osAccountInfo);
52 }
53 return result == ERR_OK;
54 }
55
QueryCurrentOsAccountFuzzTest(const uint8_t * data,size_t size)56 bool QueryCurrentOsAccountFuzzTest(const uint8_t* data, size_t size)
57 {
58 int32_t result = ERR_OK;
59 if ((data != nullptr) && (size != 0)) {
60 FuzzData fuzzData(data, size);
61 std::string testName(fuzzData.GenerateString());
62 OsAccountInfo osAccountInfo;
63 osAccountInfo.SetLocalName(testName);
64 result = OsAccountManager::QueryCurrentOsAccount(osAccountInfo);
65 }
66 return result == ERR_OK;
67 }
68 } // namespace
69
70 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)71 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
72 {
73 /* Run your code on data */
74 OHOS::QueryOsAccountConstraintSourceTypesFuzzTest(data, size);
75 OHOS::QueryOsAccountByIdFuzzTest(data, size);
76 OHOS::QueryCurrentOsAccountFuzzTest(data, size);
77 return 0;
78 }
79
80