1 /*
2 * Copyright (C) 2022 Huawei Technologies Co., Ltd.
3 * Licensed under the Mulan PSL v2.
4 * You can use this software according to the terms and conditions of the Mulan PSL v2.
5 * You may obtain a copy of Mulan PSL v2 at:
6 * http://license.coscl.org.cn/MulanPSL2
7 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
8 * IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
9 * PURPOSE.
10 * See the Mulan PSL v2 for more details.
11 */
12
13 #include "teeclientallocatesharedmemory_fuzzer.h"
14
15 #include <cstddef>
16 #include <cstdint>
17 #include "tee_client_api.h"
18 #include "tee_client_constants.h"
19 #include "tee_client_type.h"
20
21 namespace OHOS {
TeeClientAllocateSharedMemoryFuzzTest(const uint8_t * data,size_t size)22 bool TeeClientAllocateSharedMemoryFuzzTest(const uint8_t *data, size_t size)
23 {
24 bool result = true;
25 if (size >= sizeof(TEEC_Context) + sizeof(TEEC_SharedMemory)) {
26 uint8_t *temp = const_cast<uint8_t *>(data);
27 TEEC_Context context = *reinterpret_cast<TEEC_Context *>(temp);
28 temp += sizeof(TEEC_Context);
29 TEEC_SharedMemory memory = *reinterpret_cast<TEEC_SharedMemory *>(temp);
30
31 TEEC_Result ret = TEEC_AllocateSharedMemory(&context, &memory);
32 if (ret == TEEC_SUCCESS) {
33 TEEC_ReleaseSharedMemory(&memory);
34 }
35 }
36 return result;
37 }
38 }
39
40 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)41 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
42 {
43 /* Run your code on data */
44 OHOS::TeeClientAllocateSharedMemoryFuzzTest(data, size);
45 return 0;
46 }