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 #include "valuesbucket_fuzzer.h"
16
17 #include <fuzzer/FuzzedDataProvider.h>
18
19 #include "value_object.h"
20 #include "values_bucket.h"
21
22
23 #define LOOPS_MIN 0
24 #define LOOPS_MAX 100
25
26 using namespace OHOS;
27 using namespace OHOS::NativeRdb;
28
29 namespace OHOS {
30
ValuesBucketPutStringFuzz(FuzzedDataProvider & provider)31 void ValuesBucketPutStringFuzz(FuzzedDataProvider &provider)
32 {
33 ValuesBucket bucket;
34 std::string columnName = provider.ConsumeRandomLengthString();
35 std::string value = provider.ConsumeRandomLengthString();
36 bucket.PutString(columnName, value);
37 }
38
ValuesBucketPutIntFuzz(FuzzedDataProvider & provider)39 void ValuesBucketPutIntFuzz(FuzzedDataProvider &provider)
40 {
41 ValuesBucket bucket;
42 std::string columnName = provider.ConsumeRandomLengthString();
43 int value = provider.ConsumeIntegral<int>();
44 bucket.PutInt(columnName, value);
45 }
46
ValuesBucketPutLongFuzz(FuzzedDataProvider & provider)47 void ValuesBucketPutLongFuzz(FuzzedDataProvider &provider)
48 {
49 ValuesBucket bucket;
50 std::string columnName = provider.ConsumeRandomLengthString();
51 int64_t value = provider.ConsumeIntegral<int64_t>();
52 bucket.PutLong(columnName, value);
53 }
54
ValuesBucketPutDoubleFuzz(FuzzedDataProvider & provider)55 void ValuesBucketPutDoubleFuzz(FuzzedDataProvider &provider)
56 {
57 ValuesBucket bucket;
58 std::string columnName = provider.ConsumeRandomLengthString();
59 double value = provider.ConsumeFloatingPoint<double>();
60 bucket.PutDouble(columnName, value);
61 }
62
ValuesBucketPutBoolFuzz(FuzzedDataProvider & provider)63 void ValuesBucketPutBoolFuzz(FuzzedDataProvider &provider)
64 {
65 ValuesBucket bucket;
66 std::string columnName = provider.ConsumeRandomLengthString();
67 bool value = provider.ConsumeBool();
68 bucket.PutBool(columnName, value);
69 }
70
ValuesBucketPutBlobFuzz(FuzzedDataProvider & provider)71 void ValuesBucketPutBlobFuzz(FuzzedDataProvider &provider)
72 {
73 ValuesBucket bucket;
74 std::string columnName = provider.ConsumeRandomLengthString();
75 size_t length = provider.ConsumeIntegralInRange<size_t>(LOOPS_MIN, LOOPS_MAX);
76 std::vector<uint8_t> value = provider.ConsumeBytes<uint8_t>(length);
77 bucket.PutBlob(columnName, value);
78 }
79
ValuesBucketPutNullFuzz(FuzzedDataProvider & provider)80 void ValuesBucketPutNullFuzz(FuzzedDataProvider &provider)
81 {
82 ValuesBucket bucket;
83 std::string columnName = provider.ConsumeRandomLengthString();
84 bucket.PutNull(columnName);
85 }
86
ValuesBucketPutValueObjectFuzz(FuzzedDataProvider & provider)87 void ValuesBucketPutValueObjectFuzz(FuzzedDataProvider &provider)
88 {
89 ValuesBucket bucket;
90 std::string columnName = provider.ConsumeRandomLengthString();
91 ValueObject value(provider.ConsumeIntegral<int32_t>());
92 bucket.Put(columnName, value);
93 }
94
ValuesBucketPutValueObjectMoveFuzz(FuzzedDataProvider & provider)95 void ValuesBucketPutValueObjectMoveFuzz(FuzzedDataProvider &provider)
96 {
97 ValuesBucket bucket;
98 std::string columnName = provider.ConsumeRandomLengthString();
99 ValueObject value(provider.ConsumeIntegral<int32_t>());
100 bucket.Put(columnName, std::move(value));
101 }
102
ValuesBucketDeleteFuzz(FuzzedDataProvider & provider)103 void ValuesBucketDeleteFuzz(FuzzedDataProvider &provider)
104 {
105 ValuesBucket bucket;
106 std::string columnName = provider.ConsumeRandomLengthString();
107 bucket.Delete(columnName);
108 }
109
ValuesBucketGetObjectFuzz(FuzzedDataProvider & provider)110 void ValuesBucketGetObjectFuzz(FuzzedDataProvider &provider)
111 {
112 ValuesBucket bucket;
113 std::string columnName = provider.ConsumeRandomLengthString();
114 ValueObject value(provider.ConsumeIntegral<int32_t>());
115 bucket.Put(columnName, value);
116 ValueObject output;
117 bucket.GetObject(columnName, output);
118 }
119
ValuesBucketHasColumnFuzz(FuzzedDataProvider & provider)120 void ValuesBucketHasColumnFuzz(FuzzedDataProvider &provider)
121 {
122 ValuesBucket bucket;
123 std::string columnName = provider.ConsumeRandomLengthString();
124 bucket.HasColumn(columnName);
125 }
126
ValuesBucketGetAllFuzz(FuzzedDataProvider & provider)127 void ValuesBucketGetAllFuzz(FuzzedDataProvider &provider)
128 {
129 ValuesBucket bucket;
130 std::string columnName = provider.ConsumeRandomLengthString();
131 ValueObject value(provider.ConsumeIntegral<int32_t>());
132 bucket.Put(columnName, value);
133 std::map<std::string, ValueObject> output;
134 bucket.GetAll(output);
135 }
136
137 } // namespace OHOS
138
139 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)140 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
141 {
142 FuzzedDataProvider provider(data, size);
143 OHOS::ValuesBucketPutStringFuzz(provider);
144 OHOS::ValuesBucketPutIntFuzz(provider);
145 OHOS::ValuesBucketPutLongFuzz(provider);
146 OHOS::ValuesBucketPutDoubleFuzz(provider);
147 OHOS::ValuesBucketPutBoolFuzz(provider);
148 OHOS::ValuesBucketPutBlobFuzz(provider);
149 OHOS::ValuesBucketPutNullFuzz(provider);
150 OHOS::ValuesBucketPutValueObjectFuzz(provider);
151 OHOS::ValuesBucketPutValueObjectMoveFuzz(provider);
152 OHOS::ValuesBucketDeleteFuzz(provider);
153 OHOS::ValuesBucketGetObjectFuzz(provider);
154 OHOS::ValuesBucketHasColumnFuzz(provider);
155 OHOS::ValuesBucketGetAllFuzz(provider);
156 return 0;
157 }
158