ref: 5fa6c44a61fbf946646899d9db24d92cdce478ac
dir: /tests/libfuzzer/alloc_fail.c/
/* * Copyright © 2018, VideoLAN and dav1d authors * Copyright © 2018, Janne Grunau * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "config.h" #include <stddef.h> #include <stdlib.h> #include <errno.h> #include "alloc_fail.h" static int fail_probability; void dav1d_setup_alloc_fail(unsigned seed, unsigned probability) { srand(seed); while (probability >= RAND_MAX) probability >>= 1; fail_probability = probability; } void * __wrap_malloc(size_t); void * __wrap_malloc(size_t sz) { if (rand() < fail_probability) return NULL; return malloc(sz); } #if defined(HAVE_POSIX_MEMALIGN) int __wrap_posix_memalign(void **memptr, size_t alignment, size_t size); int __wrap_posix_memalign(void **memptr, size_t alignment, size_t size) { if (rand() < fail_probability) return ENOMEM; return posix_memalign(memptr, alignment, size); } #else #error "HAVE_POSIX_MEMALIGN required" #endif