tiny cleanup

This commit is contained in:
Dominik Maier 2020-12-31 17:30:29 +01:00
parent 1a88cac145
commit ca1a15f8a0
2 changed files with 14 additions and 12 deletions

View File

@ -1,13 +1,15 @@
/*!
This shows how llmp can be used directly, without libafl abstractions
*/
extern crate alloc; extern crate alloc;
use core::convert::TryInto; use core::{convert::TryInto, time::Duration};
use core::time::Duration; use std::{thread, time};
use std::thread;
use std::time;
use afl::events::llmp; use afl::{
use afl::events::shmem::AflShmem; events::{llmp, shmem::AflShmem},
use afl::AflError; AflError,
};
const TAG_SIMPLE_U32_V1: u32 = 0x51300321; const TAG_SIMPLE_U32_V1: u32 = 0x51300321;
const TAG_MATH_RESULT_V1: u32 = 0x77474331; const TAG_MATH_RESULT_V1: u32 = 0x77474331;

View File

@ -22,7 +22,7 @@ void __sanitizer_cov_trace_pc_guard(uint32_t *guard) {
void __sanitizer_cov_trace_pc_guard_init(uint32_t *start, uint32_t *stop) { void __sanitizer_cov_trace_pc_guard_init(uint32_t *start, uint32_t *stop) {
if (start == stop || *start) return; if (start == stop || *start) { return; }
*(start++) = (++__lafl_max_edges_size) & (MAP_SIZE -1); *(start++) = (++__lafl_max_edges_size) & (MAP_SIZE -1);
@ -124,7 +124,6 @@ void __sanitizer_cov_trace_switch(uint64_t val, uint64_t *cases) {
} }
static void afl_libfuzzer_copy_args(int argc, char** argv, char** envp) { static void afl_libfuzzer_copy_args(int argc, char** argv, char** envp) {
orig_argc = argc; orig_argc = argc;
orig_argv = argv; orig_argv = argv;
@ -138,9 +137,10 @@ void afl_libfuzzer_main();
int afl_libfuzzer_init() { int afl_libfuzzer_init() {
if (LLVMFuzzerInitialize) if (LLVMFuzzerInitialize) {
return LLVMFuzzerInitialize(&orig_argc, &orig_argv); return LLVMFuzzerInitialize(&orig_argc, &orig_argv);
else } else {
return 0; return 0;
}
} }