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 "sethtmltext_fuzzer.h"
17
18 #include <fuzzer/FuzzedDataProvider.h>
19
20 #define private public
21 #include "pasteboard_client_adapter_impl.h"
22 #undef private
23
24 using namespace OHOS::NWeb;
25 using namespace OHOS::MiscServices;
26
27 namespace OHOS {
28 namespace {
29 constexpr uint8_t TEST_STRING_LENGTH = 8;
30 }
SetHtmlTextFuzzTest(const uint8_t * data,size_t size)31 bool SetHtmlTextFuzzTest(const uint8_t* data, size_t size)
32 {
33 if ((data == nullptr) || (size == 0)) {
34 return false;
35 }
36 std::shared_ptr<PasteDataRecord> record = std::make_shared<PasteDataRecord>();
37 std::shared_ptr<PasteDataRecordAdapterImpl> dataRecordAdapterImpl =
38 std::make_shared<PasteDataRecordAdapterImpl>(record);
39 dataRecordAdapterImpl->record_ = nullptr;
40
41 FuzzedDataProvider dataProvider(data, size);
42 std::string text = dataProvider.ConsumeRandomLengthString(TEST_STRING_LENGTH);
43 std::shared_ptr<PasteDataRecordAdapter> dataRecordAdapter = PasteDataRecordAdapter::NewRecord(text);
44 std::shared_ptr<std::string> htmlText = std::make_shared<std::string>(text);
45 dataRecordAdapterImpl->SetHtmlText(htmlText);
46 dataRecordAdapterImpl->SetHtmlText(nullptr);
47
48 dataRecordAdapterImpl->SetUri(text);
49 dataRecordAdapterImpl->SetUri("");
50
51 PasteCustomData testData;
52 dataRecordAdapterImpl->SetCustomData(testData);
53 std::vector<uint8_t> customText = dataProvider.ConsumeBytes<uint8_t>(TEST_STRING_LENGTH);
54 testData[text] = customText;
55 dataRecordAdapterImpl->SetCustomData(testData);
56
57 dataRecordAdapterImpl->GetMimeType();
58 dataRecordAdapterImpl->GetHtmlText();
59 dataRecordAdapterImpl->GetPlainText();
60 dataRecordAdapterImpl->GetPixelMap();
61 dataRecordAdapterImpl->GetImgData(nullptr);
62 dataRecordAdapterImpl->GetUri();
63 dataRecordAdapterImpl->GetCustomData();
64 return true;
65 }
66
SetHtmlText1FuzzTest(const uint8_t * data,size_t size)67 bool SetHtmlText1FuzzTest(const uint8_t* data, size_t size)
68 {
69 if ((data == nullptr) || (size == 0)) {
70 return false;
71 }
72 FuzzedDataProvider dataProvider(data, size);
73 std::string text = dataProvider.ConsumeRandomLengthString(TEST_STRING_LENGTH);
74 std::shared_ptr<PasteDataRecordAdapter> dataRecordAdapter = PasteDataRecordAdapter::NewRecord(text);
75 std::shared_ptr<std::string> htmlText = std::make_shared<std::string>(text);
76 dataRecordAdapter->SetHtmlText(htmlText);
77 dataRecordAdapter->SetHtmlText(nullptr);
78 return true;
79 }
80 } // namespace OHOS
81
82 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)83 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
84 {
85 /* Run your code on data */
86 OHOS::SetHtmlTextFuzzTest(data, size);
87 OHOS::SetHtmlText1FuzzTest(data, size);
88 return 0;
89 }
90