mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-27 17:56:06 +00:00
139e216e66
Add close_fd_mask functionality to AFL driver. Summary: Add support for env var AFL_DRIVER_CLOSE_FD_MASK which behaves the same as libFuzzer's -close_fd_mask=1. Also add tests. Reviewers: kcc, vitalybuka, morehouse Reviewed By: morehouse Subscribers: #sanitizers, llvm-commits Tags: #sanitizers, #llvm Differential Revision: https://reviews.llvm.org/D60334 llvm-svn: 358703
35 lines
1011 B
C++
35 lines
1011 B
C++
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
// Dummy functions used to avoid dependency on AFL.
|
|
extern "C" void __afl_manual_init() {}
|
|
|
|
extern "C" int __afl_persistent_loop(unsigned int N) {
|
|
static int Count = N;
|
|
fprintf(stderr, "__afl_persistent_loop called, Count = %d\n", Count);
|
|
return Count--;
|
|
}
|
|
|
|
// This declaration exists to prevent the Darwin linker
|
|
// from complaining about this being a missing weak symbol.
|
|
extern "C" int LLVMFuzzerInitialize(int *argc, char ***argv) {
|
|
return 0;
|
|
}
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
|
|
puts("STDOUT MESSAGE");
|
|
fflush(stdout);
|
|
fprintf(stderr, "STDERR MESSAGE\n"
|
|
"LLVMFuzzerTestOneInput called; Size = %zd\n",
|
|
Size);
|
|
if (Size < 4)
|
|
return 0;
|
|
|
|
return Data[Size];
|
|
}
|