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 "updateconfiguration_fuzzer.h"
17 
18 #include <cstddef>
19 #include <cstdint>
20 
21 #include "app_mgr_client.h"
22 #include "configuration.h"
23 #include "securec.h"
24 
25 using namespace OHOS::AAFwk;
26 using namespace OHOS::AppExecFwk;
27 
28 namespace OHOS {
29 namespace {
30 constexpr size_t FOO_MAX_LEN = 1024;
31 constexpr size_t U32_AT_SIZE = 4;
32 }
DoSomethingInterestingWithMyAPI(const char * data,size_t size)33 bool DoSomethingInterestingWithMyAPI(const char* data, size_t size)
34 {
35     AppMgrClient* appMgrClient = new AppMgrClient();
36     if (!appMgrClient) {
37         return false;
38     }
39 
40     // fuzz for string, the value of colorMode is constrained
41     Configuration config;
42     if (!config.AddItem(AAFwk::GlobalConfigurationKey::SYSTEM_LANGUAGE, reinterpret_cast<const char*>(data))) {
43         return false;
44     }
45 
46     if (appMgrClient->UpdateConfiguration(config) != 0) {
47         return false;
48     }
49     if (appMgrClient) {
50         delete appMgrClient;
51         appMgrClient = nullptr;
52     }
53     return true;
54 }
55 }
56 
57 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)58 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
59 {
60     /* Run your code on data */
61     if (data == nullptr) {
62         std::cout << "invalid data" << std::endl;
63         return 0;
64     }
65 
66     /* Validate the length of size */
67     if (size > OHOS::FOO_MAX_LEN || size < OHOS::U32_AT_SIZE) {
68         return 0;
69     }
70 
71     char* ch = static_cast<char*>(malloc(size + 1));
72     if (ch == nullptr) {
73         std::cout << "malloc failed." << std::endl;
74         return 0;
75     }
76 
77     (void)memset_s(ch, size + 1, 0x00, size + 1);
78     if (memcpy_s(ch, size + 1, data, size) != EOK) {
79         std::cout << "copy failed." << std::endl;
80         free(ch);
81         ch = nullptr;
82         return 0;
83     }
84 
85     OHOS::DoSomethingInterestingWithMyAPI(ch, size);
86     free(ch);
87     ch = nullptr;
88     return 0;
89 }
90 
91