1 /*
2 * Copyright (c) 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 "distributedschedserviceseven_fuzzer.h"
17
18 #include <cstddef>
19 #include <cstdint>
20 #include <fuzzer/FuzzedDataProvider.h>
21 #include <singleton.h>
22
23 #include "distributed_sched_interface.h"
24 #include "distributed_sched_service.h"
25 #include "distributed_sched_stub.h"
26 #include "distributedWant/distributed_want.h"
27 #include "mock_fuzz_util.h"
28 #include "mock_distributed_sched.h"
29 #include "parcel_helper.h"
30 #include "dms_continue_time_dumper.h"
31
32 using namespace OHOS::AAFwk;
33 using namespace OHOS::AppExecFwk;
34
35 namespace OHOS {
36 namespace DistributedSchedule {
37
38 std::string GetDExtensionName(std::string bundleName, int32_t userId);
39 std::string GetDExtensionProcess(std::string bundleName, int32_t userId);
40
RemoveContinuationTimeoutFuzzTest(const uint8_t * data,size_t size)41 void RemoveContinuationTimeoutFuzzTest(const uint8_t* data, size_t size)
42 {
43 if (data == nullptr || size < sizeof(int32_t)) {
44 return;
45 }
46
47 FuzzedDataProvider fdp(data, size);
48 int32_t missionId = fdp.ConsumeIntegral<int32_t>();
49 DistributedSchedService::GetInstance().RemoveContinuationTimeout(missionId);
50 }
51
SetContinuationTimeoutFuzzTest(const uint8_t * data,size_t size)52 void SetContinuationTimeoutFuzzTest(const uint8_t* data, size_t size)
53 {
54 if (data == nullptr || size < sizeof(int32_t) + sizeof(int32_t)) {
55 return;
56 }
57
58 FuzzedDataProvider fdp(data, size);
59 int32_t missionId = fdp.ConsumeIntegral<int32_t>();
60 int32_t timeout = fdp.ConsumeIntegral<int32_t>();
61 DistributedSchedService::GetInstance().SetContinuationTimeout(missionId, timeout);
62 }
63
GetContinuationDeviceFuzzTest(const uint8_t * data,size_t size)64 void GetContinuationDeviceFuzzTest(const uint8_t* data, size_t size)
65 {
66 if (data == nullptr || size < sizeof(int32_t)) {
67 return;
68 }
69
70 FuzzedDataProvider fdp(data, size);
71 int32_t missionId = fdp.ConsumeIntegral<int32_t>();
72 std::string result = DistributedSchedService::GetInstance().GetContinuaitonDevice(missionId);
73 }
74
SetWantForContinuationFuzzTest(const uint8_t * data,size_t size)75 void SetWantForContinuationFuzzTest(const uint8_t* data, size_t size)
76 {
77 if (data == nullptr || size < sizeof(int32_t)) {
78 return;
79 }
80
81 FuzzedDataProvider fdp(data, size);
82 AAFwk::Want newWant;
83 std::string deviceId = fdp.ConsumeRandomLengthString();
84 std::string bundleName = fdp.ConsumeRandomLengthString();
85 std::string abilityName = fdp.ConsumeRandomLengthString();
86 AppExecFwk::ElementName element(deviceId, bundleName, abilityName);
87 newWant.SetElement(element);
88 newWant.SetParam("ohos.extra.param.key.supportContinuePageStack", fdp.ConsumeBool());
89 newWant.SetParam("ohos.extra.param.key.supportContinueModuleNameUpdate", fdp.ConsumeRandomLengthString());
90 int32_t missionId = fdp.ConsumeIntegral<int32_t>();
91 DistributedSchedService::GetInstance().SetWantForContinuation(newWant, missionId);
92 }
93
DealDSchedEventResultFuzzTest(const uint8_t * data,size_t size)94 void DealDSchedEventResultFuzzTest(const uint8_t* data, size_t size)
95 {
96 if (data == nullptr || size < sizeof(int32_t)) {
97 return;
98 }
99
100 FuzzedDataProvider fdp(data, size);
101 AAFwk::Want want;
102 std::string deviceId = fdp.ConsumeRandomLengthString();
103 std::string bundleName = fdp.ConsumeRandomLengthString();
104 std::string abilityName = fdp.ConsumeRandomLengthString();
105 AppExecFwk::ElementName element(deviceId, bundleName, abilityName);
106 want.SetElement(element);
107 int32_t status = fdp.ConsumeIntegral<int32_t>();
108 DistributedSchedService::GetInstance().DealDSchedEventResult(want, status);
109 }
110 }
111 }
112 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)113 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
114 {
115 OHOS::DistributedSchedule::RemoveContinuationTimeoutFuzzTest(data, size);
116 OHOS::DistributedSchedule::SetContinuationTimeoutFuzzTest(data, size);
117 OHOS::DistributedSchedule::GetContinuationDeviceFuzzTest(data, size);
118 OHOS::DistributedSchedule::SetWantForContinuationFuzzTest(data, size);
119 OHOS::DistributedSchedule::DealDSchedEventResultFuzzTest(data, size);
120 return 0;
121 }
122