extras: afl_realloc -> ck_realloc

This commit is contained in:
van Hauser
2020-10-30 09:40:51 +01:00
parent bdadba60f7
commit a2739ef5ff
2 changed files with 15 additions and 5 deletions

View File

@ -984,7 +984,7 @@ void afl_fsrv_write_to_testcase(afl_forkserver_t *fsrv, u8 *buf, size_t len) {
if (fd < 0) { PFATAL("Unable to create '%s'", fsrv->out_file); }
} else if (unlikely(!fd)) {
// We should never have stdin as fd here, 0 is likely unset.
FATAL("Nowhere to write output to (neither out_fd nor out_file set)");

View File

@ -423,8 +423,8 @@ void dedup_extras(afl_state_t *afl) {
}
if (afl->extras_cnt != orig_cnt)
afl->extras = afl_realloc((void **)&afl->extras,
afl->extras_cnt * sizeof(struct extra_data));
afl->extras = ck_realloc((void **)&afl->extras,
afl->extras_cnt * sizeof(struct extra_data));
}
@ -462,8 +462,18 @@ void add_extra(afl_state_t *afl, u8 *mem, u32 len) {
}
afl->extras = afl_realloc((void **)&afl->extras,
(afl->extras_cnt + 1) * sizeof(struct extra_data));
if (afl->extra) {
afl->extras = ck_realloc((void **)&afl->extras,
(afl->extras_cnt + 1) * sizeof(struct extra_data));
} else {
afl->extras = ck_alloc((void **)&afl->extras,
(afl->extras_cnt + 1) * sizeof(struct extra_data));
}
if (unlikely(!afl->extras)) { PFATAL("alloc"); }
afl->extras[afl->extras_cnt].data = ck_alloc(len);