From af823d6486f9848ee8844f23ed3baf27da45a107 Mon Sep 17 00:00:00 2001
From: van Hauser <vh@thc.org>
Date: Thu, 1 Aug 2019 15:44:10 +0200
Subject: map size fixes

---
 llvm_mode/LLVMInsTrim.so.cc | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

(limited to 'llvm_mode/LLVMInsTrim.so.cc')

diff --git a/llvm_mode/LLVMInsTrim.so.cc b/llvm_mode/LLVMInsTrim.so.cc
index 81cf98c4..148f94c3 100644
--- a/llvm_mode/LLVMInsTrim.so.cc
+++ b/llvm_mode/LLVMInsTrim.so.cc
@@ -46,15 +46,13 @@ namespace {
     std::mt19937 generator;
     int total_instr = 0;
 
-    unsigned genLabel() {
-      return generator() % 65536;
+    unsigned int genLabel() {
+      return generator() &= MAP_SIZE - 1;
     }
 
   public:
     static char ID;
-    InsTrim() : ModulePass(ID), generator(0) {//}
-    
-//    AFLCoverage() : ModulePass(ID) {
+    InsTrim() : ModulePass(ID), generator(0) {
       char* instWhiteListFilename = getenv("AFL_LLVM_WHITELIST");
       if (instWhiteListFilename) {
         std::string line;
-- 
cgit 1.4.1