alloc-inl.h/ck_maybe_grow() back to size_t, reimplement overflow check

This commit is contained in:
hexcoder-
2020-03-28 11:01:29 +01:00
committed by Dominik Maier
parent 9d7ac3d99f
commit 1119a2e185
2 changed files with 8 additions and 9 deletions

View File

@ -36,7 +36,7 @@ static void *unsupported(afl_state_t *afl, unsigned int seed) {
}
/* sorry for this makro...
it just filles in `&py_mutator->something_buf, &py_mutator->something_size`. */
it just fills in `&py_mutator->something_buf, &py_mutator->something_size`. */
#define BUF_PARAMS(name) \
(void **)&((py_mutator_t *)py_mutator)->name##_buf, \
&((py_mutator_t *)py_mutator)->name##_size
@ -371,8 +371,7 @@ size_t pre_save_py(void *py_mutator, u8 *buf, size_t buf_size, u8 **out_buf) {
py_out_buf_size = PyByteArray_Size(py_value);
ck_maybe_grow((void **)&py->pre_save_buf, &py->pre_save_size,
py_out_buf_size);
ck_maybe_grow(BUF_PARAMS(pre_save), py_out_buf_size);
memcpy(py->pre_save_buf, PyByteArray_AsString(py_value), py_out_buf_size);
Py_DECREF(py_value);