1*01826a49SYabin Cui /*
2*01826a49SYabin Cui * Copyright (c) Meta Platforms, Inc. and affiliates.
3*01826a49SYabin Cui * All rights reserved.
4*01826a49SYabin Cui *
5*01826a49SYabin Cui * This source code is licensed under both the BSD-style license (found in the
6*01826a49SYabin Cui * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7*01826a49SYabin Cui * in the COPYING file in the root directory of this source tree).
8*01826a49SYabin Cui * You may select, at your option, one of the above-listed licenses.
9*01826a49SYabin Cui */
10*01826a49SYabin Cui #include "fuzz_helpers.h"
11*01826a49SYabin Cui
12*01826a49SYabin Cui #include <stddef.h>
13*01826a49SYabin Cui #include <stdlib.h>
14*01826a49SYabin Cui #include <string.h>
15*01826a49SYabin Cui
FUZZ_malloc(size_t size)16*01826a49SYabin Cui void* FUZZ_malloc(size_t size)
17*01826a49SYabin Cui {
18*01826a49SYabin Cui if (size > 0) {
19*01826a49SYabin Cui void* const mem = malloc(size);
20*01826a49SYabin Cui FUZZ_ASSERT(mem);
21*01826a49SYabin Cui return mem;
22*01826a49SYabin Cui }
23*01826a49SYabin Cui return NULL;
24*01826a49SYabin Cui }
25*01826a49SYabin Cui
FUZZ_malloc_rand(size_t size,FUZZ_dataProducer_t * producer)26*01826a49SYabin Cui void* FUZZ_malloc_rand(size_t size, FUZZ_dataProducer_t *producer)
27*01826a49SYabin Cui {
28*01826a49SYabin Cui if (size > 0) {
29*01826a49SYabin Cui void* const mem = malloc(size);
30*01826a49SYabin Cui FUZZ_ASSERT(mem);
31*01826a49SYabin Cui return mem;
32*01826a49SYabin Cui } else {
33*01826a49SYabin Cui uintptr_t ptr = 0;
34*01826a49SYabin Cui /* Add +- 1M 50% of the time */
35*01826a49SYabin Cui if (FUZZ_dataProducer_uint32Range(producer, 0, 1))
36*01826a49SYabin Cui FUZZ_dataProducer_int32Range(producer, -1000000, 1000000);
37*01826a49SYabin Cui return (void*)ptr;
38*01826a49SYabin Cui }
39*01826a49SYabin Cui
40*01826a49SYabin Cui }
41*01826a49SYabin Cui
FUZZ_memcmp(void const * lhs,void const * rhs,size_t size)42*01826a49SYabin Cui int FUZZ_memcmp(void const* lhs, void const* rhs, size_t size)
43*01826a49SYabin Cui {
44*01826a49SYabin Cui if (size == 0) {
45*01826a49SYabin Cui return 0;
46*01826a49SYabin Cui }
47*01826a49SYabin Cui return memcmp(lhs, rhs, size);
48*01826a49SYabin Cui }
49