about summary refs log tree commit diff
path: root/llvm_mode/LLVMInsTrim.so.cc
diff options
context:
space:
mode:
authorvan Hauser <vh@thc.org>2019-08-01 15:44:56 +0200
committerGitHub <noreply@github.com>2019-08-01 15:44:56 +0200
commit54bb9f4b55f6707f0f5b211ce2d28ffcee705102 (patch)
tree01b6b4e123d208a5dab9144f98c48daf613107eb /llvm_mode/LLVMInsTrim.so.cc
parent487a87df02c20784bb4ef272954300dd18c04d9a (diff)
parentaf823d6486f9848ee8844f23ed3baf27da45a107 (diff)
downloadafl++-54bb9f4b55f6707f0f5b211ce2d28ffcee705102.tar.gz
Merge pull request #36 from vanhauser-thc/map_1mb
map size fixes
Diffstat (limited to 'llvm_mode/LLVMInsTrim.so.cc')
-rw-r--r--llvm_mode/LLVMInsTrim.so.cc8
1 files changed, 3 insertions, 5 deletions
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;