1 /*
2 * Copyright (c) 2023 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 "hiviewadapter_fuzzer.h"
17
18 #include <cstddef>
19 #include <cstdint>
20
21 #include "device_auth_defines.h"
22 #include "hc_time.h"
23 #include "hidump_adapter.h"
24 #include "performance_dumper.h"
25
26 namespace OHOS {
27 #define TEST_REQ_ID 123
28 #define TEST_REQ_ID1 124
29 #define TEST_REQ_ID2 125
30 #define TEST_REQ_ID3 126
31 #define TEST_TIME_INTERVAL1 1
32 #define TEST_TIME_INTERVAL2 2
33 #define TEST_TIME_INTERVAL3 3
34 #define TEST_TIME_INTERVAL4 4
35 #define TEST_TIME_INTERVAL5 5
36 #define TEST_TIME_INTERVAL6 6
37 #define TEST_TIME_INTERVAL7 7
38 #define TEST_TIME_INTERVAL8 8
39 #define TEST_TIME_INTERVAL9 9
40 #define TEST_TIME_INTERVAL10 10
41 #define MAX_SESSION_NUM 10
42 #define COUNT_TWO_NUM 2
43 #define COUNT_THREE_NUM 3
44 #define COUNT_FOUR_NUM 4
45 #define ENABLE_PERFORMANCE_DUMPER "--enable"
46 #define DISABLE_PERFORMANCE_DUMPER "--disable"
47 #define INVALID_DUMPER_ARG "--test"
48
EnablePerformDumper(void)49 static void EnablePerformDumper(void)
50 {
51 StringVector strArgVec = CreateStrVector();
52 HcString strArg1 = CreateString();
53 (void)StringSetPointer(&strArg1, PERFORM_DUMP_ARG);
54 (void)strArgVec.pushBackT(&strArgVec, strArg1);
55 HcString strArg2 = CreateString();
56 (void)StringSetPointer(&strArg2, ENABLE_PERFORMANCE_DUMPER);
57 (void)strArgVec.pushBackT(&strArgVec, strArg2);
58 DEV_AUTH_DUMP(0, &strArgVec);
59 DestroyStrVector(&strArgVec);
60 }
61
DumpByInvalidArg(void)62 static void DumpByInvalidArg(void)
63 {
64 StringVector strArgVec = CreateStrVector();
65 HcString strArg1 = CreateString();
66 (void)StringSetPointer(&strArg1, PERFORM_DUMP_ARG);
67 (void)strArgVec.pushBackT(&strArgVec, strArg1);
68 HcString strArg2 = CreateString();
69 (void)StringSetPointer(&strArg2, INVALID_DUMPER_ARG);
70 (void)strArgVec.pushBackT(&strArgVec, strArg2);
71 DEV_AUTH_DUMP(0, &strArgVec);
72 DestroyStrVector(&strArgVec);
73 }
74
DumpByInvalidArgNum(void)75 static void DumpByInvalidArgNum(void)
76 {
77 StringVector strArgVec = CreateStrVector();
78 HcString strArg1 = CreateString();
79 (void)StringSetPointer(&strArg1, PERFORM_DUMP_ARG);
80 (void)strArgVec.pushBackT(&strArgVec, strArg1);
81 HcString strArg2 = CreateString();
82 (void)StringSetPointer(&strArg2, ENABLE_PERFORMANCE_DUMPER);
83 (void)strArgVec.pushBackT(&strArgVec, strArg2);
84 HcString strArg3 = CreateString();
85 (void)StringSetPointer(&strArg3, ENABLE_PERFORMANCE_DUMPER);
86 (void)strArgVec.pushBackT(&strArgVec, strArg3);
87 DEV_AUTH_DUMP(0, &strArgVec);
88 DestroyStrVector(&strArgVec);
89 }
90
DisablePerformDumper(void)91 static void DisablePerformDumper(void)
92 {
93 StringVector strArgVec = CreateStrVector();
94 HcString strArg1 = CreateString();
95 (void)StringSetPointer(&strArg1, PERFORM_DUMP_ARG);
96 (void)strArgVec.pushBackT(&strArgVec, strArg1);
97 HcString strArg2 = CreateString();
98 (void)StringSetPointer(&strArg2, DISABLE_PERFORMANCE_DUMPER);
99 (void)strArgVec.pushBackT(&strArgVec, strArg2);
100 DEV_AUTH_DUMP(0, &strArgVec);
101 DestroyStrVector(&strArgVec);
102 }
103
DumpPerformData(void)104 static void DumpPerformData(void)
105 {
106 StringVector strArgVec = CreateStrVector();
107 HcString strArg = CreateString();
108 (void)StringSetPointer(&strArg, PERFORM_DUMP_ARG);
109 (void)strArgVec.pushBackT(&strArgVec, strArg);
110 DEV_AUTH_DUMP(0, &strArgVec);
111 DestroyStrVector(&strArgVec);
112 }
113
TestAddUpdatePerformData(int64_t reqId,bool isBind,bool isClient)114 static void TestAddUpdatePerformData(int64_t reqId, bool isBind, bool isClient)
115 {
116 int64_t curTimeInMillis = HcGetCurTimeInMillis();
117 ADD_PERFORM_DATA(reqId, isBind, isClient, curTimeInMillis);
118 UPDATE_PERFORM_DATA_BY_SELF_INDEX(reqId, curTimeInMillis + TEST_TIME_INTERVAL1);
119 UPDATE_PERFORM_DATA_BY_SELF_INDEX(reqId, curTimeInMillis + TEST_TIME_INTERVAL2);
120 UPDATE_PERFORM_DATA_BY_SELF_INDEX(reqId, curTimeInMillis + TEST_TIME_INTERVAL3);
121 UPDATE_PERFORM_DATA_BY_SELF_INDEX(reqId, curTimeInMillis + TEST_TIME_INTERVAL4);
122 UPDATE_PERFORM_DATA_BY_SELF_INDEX(reqId, curTimeInMillis + TEST_TIME_INTERVAL5);
123 UPDATE_PERFORM_DATA_BY_SELF_INDEX(reqId, curTimeInMillis + TEST_TIME_INTERVAL6);
124 UPDATE_PERFORM_DATA_BY_SELF_INDEX(reqId, curTimeInMillis + TEST_TIME_INTERVAL7);
125 UPDATE_PERFORM_DATA_BY_INPUT_INDEX(reqId, ON_SESSION_KEY_RETURN_TIME, curTimeInMillis + TEST_TIME_INTERVAL8);
126 UPDATE_PERFORM_DATA_BY_INPUT_INDEX(reqId, ON_FINISH_TIME, curTimeInMillis + TEST_TIME_INTERVAL9);
127 }
128
HiviewAdapterTest01(void)129 static void HiviewAdapterTest01(void)
130 {
131 // has not been initialized
132 int64_t curTimeInMillis = HcGetCurTimeInMillis();
133 ADD_PERFORM_DATA(TEST_REQ_ID, true, true, curTimeInMillis);
134 RESET_PERFORM_DATA(TEST_REQ_ID);
135 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis);
136 UPDATE_PERFORM_DATA_BY_INPUT_INDEX(TEST_REQ_ID, ON_SESSION_KEY_RETURN_TIME, curTimeInMillis);
137
138 INIT_PERFORMANCE_DUMPER();
139 INIT_PERFORMANCE_DUMPER(); // has been initialized
140
141 // dump by invalid args
142 DumpByInvalidArg();
143 DumpByInvalidArgNum();
144
145 // dump not enabled
146 DumpPerformData();
147 ADD_PERFORM_DATA(TEST_REQ_ID, true, true, curTimeInMillis);
148 RESET_PERFORM_DATA(TEST_REQ_ID);
149 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis);
150 UPDATE_PERFORM_DATA_BY_INPUT_INDEX(TEST_REQ_ID, ON_SESSION_KEY_RETURN_TIME, curTimeInMillis);
151
152 EnablePerformDumper();
153 ADD_PERFORM_DATA(TEST_REQ_ID, true, true, curTimeInMillis);
154 ADD_PERFORM_DATA(TEST_REQ_ID, true, true, curTimeInMillis); // remove existed request
155 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis + TEST_TIME_INTERVAL1);
156 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis + TEST_TIME_INTERVAL2);
157 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis + TEST_TIME_INTERVAL3);
158 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis + TEST_TIME_INTERVAL4);
159 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis + TEST_TIME_INTERVAL5);
160 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis + TEST_TIME_INTERVAL6);
161 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis + TEST_TIME_INTERVAL7);
162 UPDATE_PERFORM_DATA_BY_SELF_INDEX(TEST_REQ_ID, curTimeInMillis + TEST_TIME_INTERVAL8);
163 UPDATE_PERFORM_DATA_BY_INPUT_INDEX(TEST_REQ_ID, ON_SESSION_KEY_RETURN_TIME, curTimeInMillis + TEST_TIME_INTERVAL9);
164 UPDATE_PERFORM_DATA_BY_INPUT_INDEX(TEST_REQ_ID, ON_FINISH_TIME, curTimeInMillis + TEST_TIME_INTERVAL10);
165 UPDATE_PERFORM_DATA_BY_INPUT_INDEX(TEST_REQ_ID, FIRST_START_TIME, curTimeInMillis + TEST_TIME_INTERVAL10);
166 ADD_PERFORM_DATA(TEST_REQ_ID1, true, false, curTimeInMillis);
167 ADD_PERFORM_DATA(TEST_REQ_ID1, true, false, curTimeInMillis); // remove existed request
168 ADD_PERFORM_DATA(TEST_REQ_ID2, false, true, curTimeInMillis);
169 ADD_PERFORM_DATA(TEST_REQ_ID2, false, true, curTimeInMillis); // remove existed request
170 ADD_PERFORM_DATA(TEST_REQ_ID3, false, false, curTimeInMillis);
171 ADD_PERFORM_DATA(TEST_REQ_ID3, false, false, curTimeInMillis); // remove existed request
172 DumpPerformData();
173 RESET_PERFORM_DATA(TEST_REQ_ID);
174 DisablePerformDumper();
175
176 DESTROY_PERFORMANCE_DUMPER();
177 DESTROY_PERFORMANCE_DUMPER(); // has been deinitialized
178 }
179
HiviewAdapterTest02(void)180 static void HiviewAdapterTest02(void)
181 {
182 INIT_PERFORMANCE_DUMPER();
183 EnablePerformDumper();
184
185 int64_t curTimeInMillis = HcGetCurTimeInMillis();
186 for (uint32_t i = 0; i <= MAX_SESSION_NUM; i++) {
187 ADD_PERFORM_DATA(i, true, true, curTimeInMillis + i);
188 }
189
190 for (uint32_t i = MAX_SESSION_NUM; i <= MAX_SESSION_NUM * COUNT_TWO_NUM; i++) {
191 ADD_PERFORM_DATA(i, true, false, curTimeInMillis + i);
192 }
193
194 for (uint32_t i = MAX_SESSION_NUM * COUNT_TWO_NUM; i <= MAX_SESSION_NUM * COUNT_THREE_NUM; i++) {
195 ADD_PERFORM_DATA(i, false, true, curTimeInMillis + i);
196 }
197
198 for (uint32_t i = MAX_SESSION_NUM * COUNT_THREE_NUM; i <= MAX_SESSION_NUM * COUNT_FOUR_NUM; i++) {
199 ADD_PERFORM_DATA(i, false, false, curTimeInMillis + i);
200 }
201
202 DumpPerformData();
203
204 DisablePerformDumper();
205 DESTROY_PERFORMANCE_DUMPER();
206 }
207
HiviewAdapterTest03(void)208 static void HiviewAdapterTest03(void)
209 {
210 INIT_PERFORMANCE_DUMPER();
211 EnablePerformDumper();
212
213 TestAddUpdatePerformData(TEST_REQ_ID, true, false);
214 TestAddUpdatePerformData(TEST_REQ_ID1, false, true);
215 TestAddUpdatePerformData(TEST_REQ_ID2, false, false);
216
217 DumpPerformData();
218
219 DisablePerformDumper();
220 DESTROY_PERFORMANCE_DUMPER();
221 }
222
FuzzDoCallback(const uint8_t * data,size_t size)223 bool FuzzDoCallback(const uint8_t* data, size_t size)
224 {
225 (void)data;
226 (void)size;
227 (void)HiviewAdapterTest01();
228 (void)HiviewAdapterTest02();
229 (void)HiviewAdapterTest03();
230 return true;
231 }
232 }
233
234 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)235 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
236 {
237 /* Run your code on data */
238 OHOS::FuzzDoCallback(data, size);
239 return 0;
240 }
241
242