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 #include "clouddownloadcallbackstub_fuzzer.h"
16
17 #include <cstddef>
18 #include <cstdint>
19
20 #include "cloud_download_callback_client.h"
21 #include "cloud_download_callback_stub.h"
22 #include "cloud_fuzzer_helper.h"
23 #include "cloud_sync_common.h"
24
25 #include "message_parcel.h"
26 #include "utils_log.h"
27
28 namespace OHOS {
29 constexpr size_t U32_AT_SIZE = 4;
30 constexpr size_t U64_AT_SIZE = 8;
31 constexpr size_t SPLITE_SIZE = 3;
32
33 using namespace OHOS::FileManagement::CloudSync;
34 using namespace std;
35
36 class CloudDownloadCallbackTest : public CloudDownloadCallback {
37 public:
OnDownloadProcess(const DownloadProgressObj & progress)38 void OnDownloadProcess(const DownloadProgressObj &progress) override {}
39 };
40
WriteInterfaceToken(MessageParcel & data)41 bool WriteInterfaceToken(MessageParcel &data)
42 {
43 if (!data.WriteInterfaceToken(CloudDownloadCallbackStub::GetDescriptor())) {
44 LOGE("Write token failed.");
45 return false;
46 }
47 return true;
48 }
49
HandleOnProcessFuzzTest(std::shared_ptr<CloudDownloadCallbackStub> cloudDownloadCallbackStubStr,const uint8_t * data,size_t size)50 void HandleOnProcessFuzzTest(std::shared_ptr<CloudDownloadCallbackStub> cloudDownloadCallbackStubStr,
51 const uint8_t *data,
52 size_t size)
53 {
54 FuzzData fuzzData(data, size);
55 MessageParcel datas;
56 if (!WriteInterfaceToken(datas)) {
57 return;
58 }
59 DownloadProgressObj downloadProgressObj;
60 downloadProgressObj.refCount = fuzzData.GetData<uint32_t>();
61 downloadProgressObj.downloadId = fuzzData.GetData<int64_t>();
62 downloadProgressObj.state =
63 static_cast<DownloadProgressObj::Status>(fuzzData.GetData<int32_t>() % (SPLITE_SIZE + 1));
64 downloadProgressObj.downloadErrorType =
65 static_cast<DownloadProgressObj::DownloadErrorType>(fuzzData.GetData<int32_t>() % (SPLITE_SIZE + SPLITE_SIZE));
66 downloadProgressObj.downloadedSize = fuzzData.GetData<int64_t>();
67 downloadProgressObj.totalSize = fuzzData.GetData<int64_t>();
68 int len = static_cast<int>(size - (OHOS::U32_AT_SIZE * SPLITE_SIZE + OHOS::U64_AT_SIZE * SPLITE_SIZE));
69 downloadProgressObj.path = fuzzData.GetStringFromData(len);
70 datas.WriteParcelable(&downloadProgressObj);
71 datas.RewindRead(0);
72 // SERVICE_CMD_ON_PROCESS
73 uint32_t code = 0;
74 MessageParcel reply;
75 MessageOption option;
76
77 cloudDownloadCallbackStubStr->OnRemoteRequest(code, datas, reply, option);
78 }
79 } // namespace OHOS
80
81 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)82 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
83 {
84 /* Run your code on data */
85 if (data == nullptr || size < (OHOS::U32_AT_SIZE * OHOS::SPLITE_SIZE + OHOS::U64_AT_SIZE * OHOS::SPLITE_SIZE)) {
86 return 0;
87 }
88
89 auto cloudDownloadCallbackPtr = std::make_shared<OHOS::CloudDownloadCallbackTest>();
90 auto cloudDownloadCallbackStubStr =
91 std::make_shared<OHOS::FileManagement::CloudSync::CloudDownloadCallbackClient>(cloudDownloadCallbackPtr);
92 OHOS::HandleOnProcessFuzzTest(cloudDownloadCallbackStubStr, data, size);
93 return 0;
94 }
95