diff options
author | hexcoder- <heiko@hexco.de> | 2020-04-02 17:00:58 +0200 |
---|---|---|
committer | hexcoder- <heiko@hexco.de> | 2020-04-02 17:00:58 +0200 |
commit | 71f0bd003fed0867b1d5e8d3e75dd8a4721cd00a (patch) | |
tree | 83c9596a0a70bb46e1cbe86c37ffac8f1492bf04 /include/afl-fuzz.h | |
parent | cc65e91eeba541aa9e3b42e81df3d752814eff21 (diff) | |
parent | 2ae0208d3be8ffba76f6cf370c99efa05429ad69 (diff) | |
download | afl++-71f0bd003fed0867b1d5e8d3e75dd8a4721cd00a.tar.gz |
Merge branch 'dev' of https://github.com/vanhauser-thc/AFLplusplus into dev
Diffstat (limited to 'include/afl-fuzz.h')
-rw-r--r-- | include/afl-fuzz.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/include/afl-fuzz.h b/include/afl-fuzz.h index a9165064..56135d0e 100644 --- a/include/afl-fuzz.h +++ b/include/afl-fuzz.h @@ -935,13 +935,13 @@ u8 input_to_state_stage(afl_state_t *afl, u8 *orig_buf, u8 *buf, u32 len, static inline u32 rand_below(afl_state_t *afl, u32 limit) { #ifdef HAVE_ARC4RANDOM - if (afl->fixed_seed) { return random() % limit; } + if (unlikely(afl->fixed_seed)) { return random() % limit; } /* The boundary not being necessarily a power of 2, we need to ensure the result uniformity. */ return arc4random_uniform(limit); #else - if (!afl->fixed_seed && unlikely(!afl->rand_cnt--)) { + if (unlikely(!afl->rand_cnt--) && likely(!afl->fixed_seed)) { ck_read(afl->fsrv.dev_urandom_fd, &afl->rand_seed, sizeof(afl->rand_seed), "/dev/urandom"); @@ -957,7 +957,7 @@ static inline u32 rand_below(afl_state_t *afl, u32 limit) { static inline u32 get_rand_seed(afl_state_t *afl) { - if (afl->fixed_seed) return (u32)afl->init_seed; + if (unlikely(afl->fixed_seed)) return (u32)afl->init_seed; return afl->rand_seed[0]; } |