Merge pull request #5 from 5angjun/master

Fix mutate_length & Add unit test

Author: 5angjun

build/lib.linux-x86_64-3.10/kafl_fuzzer/native/bitmap.cpython-310-x86_64-linux-gnu.so

@@ -1,12 +1,11 @@
-Metadata-Version: 1.2
+Metadata-Version: 2.1
 Name: kafl-fuzzer
 Version: 0.6
 Summary: kAFL/Nyx Kernel Fuzzer
 Home-page: https://github.com/IntelLabs/kAFL
 Maintainer: Steffen Schulz
 Maintainer-email: steffen.schulz@intel.com
 License: UNKNOWN
-Description: UNKNOWN
 Platform: UNKNOWN
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
@@ -16,3 +15,7 @@ Classifier: License :: OSI Approved :: GNU Affero General Public License v3
 Classifier: Operating System :: POSIX :: Linux
 Classifier: Programming Language :: Python
 Classifier: Topic :: Security
+License-File: LICENSE.md
+
+UNKNOWN
+

build/temp.linux-x86_64-3.10/kafl_fuzzer/native/bitmap.o

@@ -1,10 +1,11 @@
+LICENSE.md
 README.md
 setup.cfg
 setup.py
 kafl_fuzzer/__init__.py
 kafl_fuzzer/__main__.py
+kafl_fuzzer/logging.yaml
 kafl_fuzzer/mcat.py
-kafl_fuzzer/test.py
 kafl_fuzzer.egg-info/PKG-INFO
 kafl_fuzzer.egg-info/SOURCES.txt
 kafl_fuzzer.egg-info/dependency_links.txt
@@ -19,17 +20,17 @@ kafl_fuzzer/common/self_check.py
 kafl_fuzzer/common/util.py
 kafl_fuzzer/common/config/__init__.py
 kafl_fuzzer/common/config/cmdline.py
+kafl_fuzzer/common/config/default_settings.yaml
 kafl_fuzzer/common/config/settings.py
 kafl_fuzzer/coverage/__init__.py
-kafl_fuzzer/debug/__init__.py
-kafl_fuzzer/debug/core.py
 kafl_fuzzer/gui/__init__.py
 kafl_fuzzer/manager/__init__.py
 kafl_fuzzer/manager/bitmap.py
 kafl_fuzzer/manager/communicator.py
 kafl_fuzzer/manager/core.py
 kafl_fuzzer/manager/manager.py
 kafl_fuzzer/manager/node.py
+kafl_fuzzer/manager/playMaker.py
 kafl_fuzzer/manager/queue.py
 kafl_fuzzer/manager/scheduler.py
 kafl_fuzzer/manager/statistics.py
@@ -41,14 +42,10 @@ kafl_fuzzer/plot/__init__.py
 kafl_fuzzer/technique/__init__.py
 kafl_fuzzer/technique/arithmetic.py
 kafl_fuzzer/technique/bitflip.py
-kafl_fuzzer/technique/grimoire_inference.py
-kafl_fuzzer/technique/grimoire_mutations.py
 kafl_fuzzer/technique/havoc.py
 kafl_fuzzer/technique/havoc_handler.py
 kafl_fuzzer/technique/helper.py
 kafl_fuzzer/technique/interesting_values.py
-kafl_fuzzer/technique/radamsa.py
-kafl_fuzzer/technique/trim.py
 kafl_fuzzer/technique/redqueen/__init__.py
 kafl_fuzzer/technique/redqueen/cmp.py
 kafl_fuzzer/technique/redqueen/colorize.py
@@ -60,7 +57,6 @@ kafl_fuzzer/technique/redqueen/parser.py
 kafl_fuzzer/technique/redqueen/workdir.py
 kafl_fuzzer/tests/__init__.py
 kafl_fuzzer/tests/helper.py
-kafl_fuzzer/tests/redqueen_mut.py
 kafl_fuzzer/tests/test_deterministic.py
 kafl_fuzzer/tests/test_havoc_handler.py
 kafl_fuzzer/tests/test_random.py

kafl_fuzzer.egg-info/PKG-INFO

@@ -1,11 +1,11 @@
-PyYAML==6.0
-appdirs
+PyYAML==6.0.1
+appdirs==1.4.4
 dynaconf==3.2.0
-fastrand
-inotify
-lz4
-mmh3
-msgpack
-psutil
-six
-tqdm
+fastrand==1.8.0
+inotify==0.2.10
+lz4==4.3.3
+mmh3==4.1.0
+msgpack==1.0.7
+psutil==5.9.8
+six==1.16.0
+tqdm==4.66.1

kafl_fuzzer.egg-info/SOURCES.txt

@@ -25,7 +25,6 @@ def load_dict(file_name):
     f.close()
     return dict_entries
 
-
 def init_havoc(config):
     global location_corpus,location_dependency
     if config.dict:
@@ -46,7 +45,6 @@ def havoc_range(perf_score):
 
     return max_iterations
 
-
 def mutate_seq_havoc_array(irp_list, index, func, max_iterations, resize=False):
     # if resize:
     #     data = data + data
@@ -72,7 +70,6 @@ def mutate_seq_havoc_array(irp_list, index, func, max_iterations, resize=False):
 
             func(irp_list)
 
-
 def mutate_seq_splice_array(irp_list, index, func, max_iterations, resize=False):
     global location_corpus
     havoc_rounds = 4
@@ -164,11 +161,6 @@ def add_insns(irp_list, func):
         new_irp_list.clear()
 
     return
-    
-
-    
-
-
 
 def mutate_random_sequence(irp_list, index, func):
     x = rand.int(10)
@@ -179,7 +171,6 @@ def mutate_random_sequence(irp_list, index, func):
         replace_insns(irp_list, func)
     else:
         add_insns(irp_list, func)
-        
 
 def mutate_length(irp_list, index, func):
 
@@ -233,25 +224,23 @@ def get_valid_length(target, IoControlCode):
                 chosen = MAX_PAYLOAD_LEN
         return chosen
 
-
     retry = 8
 
     for _ in range(retry):
         chosen = get_valid_length("InBuffer",IoControlCode)
         if chosen is not None:
-            target.InBuffer_length = chosen
-
+        
             if chosen > target.InBuffer_length:
-                target.InBuffer.ljust(chosen,b"\xff")
+                target.InBuffer = target.InBuffer.ljust(chosen, b"\xff")
             else:
                 target.InBuffer = target.InBuffer[:chosen]
 
+            target.InBuffer_length = chosen
 
         chosen = get_valid_length("OutBuffer",IoControlCode)
         if chosen is not None:
             target.OutBuffer_length = chosen
 
-
         func(irp_list)
 
     target.InBuffer_length = origin_InBufferLength

kafl_fuzzer.egg-info/requires.txt

@@ -0,0 +1,24 @@
+# Copyright (C) 2020 Intel Corporation
+# SPDX-License-Identifier: AGPL-3.0-or-later
+
+"""
+Helper functions for kAFL tests
+"""
+
+def ham_weight(x):
+    _x = bytearray(x)
+    weight = 0
+    for byte in _x:
+        weight += bin(byte).count("1")
+    return weight
+
+def ham_distance(a,b):
+    return ham_weight(bytes(x ^ y for (x, y) in zip(a, b)))
+
+def bindiff(a,b):
+    res = bytearray()
+    for (x, y) in zip(a, b):
+        r = bytearray([x^y])
+        if r != b'\x00':
+            res += r
+    return res