mem error fix

This commit is contained in:
van Hauser
2020-12-18 10:20:43 +01:00
parent 4ff9eb0e67
commit ea9db86bb8
3 changed files with 6 additions and 5 deletions

View File

@ -1022,7 +1022,7 @@ u32 count_bytes(afl_state_t *, u8 *);
u32 count_non_255_bytes(afl_state_t *, u8 *); u32 count_non_255_bytes(afl_state_t *, u8 *);
void simplify_trace(afl_state_t *, u8 *); void simplify_trace(afl_state_t *, u8 *);
void classify_counts(afl_forkserver_t *); void classify_counts(afl_forkserver_t *);
u8 has_new_bits_unclassified(afl_state_t *, u8 *); void discover_word(u8 *ret, u64 *current, u64 *virgin);
void init_count_class16(void); void init_count_class16(void);
void minimize_bits(afl_state_t *, u8 *, u8 *); void minimize_bits(afl_state_t *, u8 *, u8 *);
#ifndef SIMPLE_FILES #ifndef SIMPLE_FILES
@ -1030,6 +1030,7 @@ u8 *describe_op(afl_state_t *, u8, size_t);
#endif #endif
u8 save_if_interesting(afl_state_t *, void *, u32, u8); u8 save_if_interesting(afl_state_t *, void *, u32, u8);
u8 has_new_bits(afl_state_t *, u8 *); u8 has_new_bits(afl_state_t *, u8 *);
u8 has_new_bits_unclassified(afl_state_t *, u8 *);
/* Extras */ /* Extras */

View File

@ -236,8 +236,8 @@ static void __afl_map_shm(void) {
if (__afl_final_loc) { if (__afl_final_loc) {
if (__afl_final_loc % 8) if (__afl_final_loc % 32)
__afl_final_loc = (((__afl_final_loc + 7) >> 3) << 3); __afl_final_loc = (((__afl_final_loc + 31) >> 5) << 5);
__afl_map_size = __afl_final_loc; __afl_map_size = __afl_final_loc;
if (__afl_final_loc > MAP_SIZE) { if (__afl_final_loc > MAP_SIZE) {

View File

@ -641,11 +641,11 @@ void afl_fsrv_start(afl_forkserver_t *fsrv, char **argv,
if (!fsrv->map_size) { fsrv->map_size = MAP_SIZE; } if (!fsrv->map_size) { fsrv->map_size = MAP_SIZE; }
if (unlikely(tmp_map_size % 8)) { if (unlikely(tmp_map_size % 32)) {
// should not happen // should not happen
WARNF("Target reported non-aligned map size of %u", tmp_map_size); WARNF("Target reported non-aligned map size of %u", tmp_map_size);
tmp_map_size = (((tmp_map_size + 8) >> 3) << 3); tmp_map_size = (((tmp_map_size + 31) >> 5) << 5);
} }