Use memop_size in libafl_gen_read/write

This commit is contained in:
Andrea Fioraldi 2023-09-05 10:56:11 +02:00
parent 8cdd7b951c
commit 52a3cfebf2

View File

@ -301,7 +301,7 @@ void libafl_add_read_hook(uint64_t (*gen)(target_ulong pc, MemOpIdx oi, uint64_t
void libafl_gen_read(TCGTemp *addr, MemOpIdx oi) void libafl_gen_read(TCGTemp *addr, MemOpIdx oi)
{ {
size_t size = 1 << (oi & MO_SIZE); size_t size = memop_size(oi);
struct libafl_rw_hook* hook = libafl_read_hooks; struct libafl_rw_hook* hook = libafl_read_hooks;
while (hook) { while (hook) {
@ -406,7 +406,7 @@ void libafl_add_write_hook(uint64_t (*gen)(target_ulong pc, MemOpIdx oi, uint64_
void libafl_gen_write(TCGTemp *addr, MemOpIdx oi) void libafl_gen_write(TCGTemp *addr, MemOpIdx oi)
{ {
size_t size = 1 << (oi & MO_SIZE); size_t size = memop_size(oi);
struct libafl_rw_hook* hook = libafl_write_hooks; struct libafl_rw_hook* hook = libafl_write_hooks;
while (hook) { while (hook) {