fix map size for nyx

1 files changed, 4 insertions, 2 deletions

diff --git a/src/afl-forkserver.c b/src/afl-forkserver.c
index 5ff82916..4d57b95d 100644
--- a/src/afl-forkserver.c
+++ b/src/afl-forkserver.c
@@ -387,7 +387,7 @@ void afl_fsrv_start(afl_forkserver_t *fsrv, char **argv,
   char *ignore_autodict = getenv("AFL_NO_AUTODICT");
 
 #ifdef __linux__
-  if (fsrv->nyx_mode) {
+  if (unlikely(fsrv->nyx_mode)) {
 
     if (fsrv->nyx_runner != NULL) { return; }
 
@@ -420,9 +420,11 @@ void afl_fsrv_start(afl_forkserver_t *fsrv, char **argv,
 
     if (fsrv->nyx_runner == NULL) { FATAL("Something went wrong ..."); }
 
-    fsrv->map_size =
+    u32 tmp_map_size =
         fsrv->nyx_handlers->nyx_get_bitmap_buffer_size(fsrv->nyx_runner);
     fsrv->real_map_size = fsrv->map_size;
+    fsrv->map_size = (((tmp_map_size + 63) >> 6) << 6);
+    if (!be_quiet) { ACTF("Target map size: %u", fsrv->real_map_size); }
 
     fsrv->trace_bits =
         fsrv->nyx_handlers->nyx_get_bitmap_buffer(fsrv->nyx_runner);