From 52a3cfebf2bdde8e331949e75737f55b1a84b46c Mon Sep 17 00:00:00 2001
From: Andrea Fioraldi <andreafioraldi@gmail.com>
Date: Tue, 5 Sep 2023 10:56:11 +0200
Subject: [PATCH] Use memop_size in libafl_gen_read/write

---
 accel/tcg/translate-all.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/accel/tcg/translate-all.c b/accel/tcg/translate-all.c
index 0323137c23..f977cb3e9f 100644
--- a/accel/tcg/translate-all.c
+++ b/accel/tcg/translate-all.c
@@ -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)
 {
-    size_t size = 1 << (oi & MO_SIZE);
+    size_t size = memop_size(oi);
 
     struct libafl_rw_hook* hook = libafl_read_hooks;
     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)
 {
-    size_t size = 1 << (oi & MO_SIZE);
+    size_t size = memop_size(oi);
 
     struct libafl_rw_hook* hook = libafl_write_hooks;
     while (hook) {