kmwebnet/OpenSK
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Resolved PR comments

Browse Source
This commit is contained in:
mingxguo27
2020-10-15 13:28:27 +00:00
parent 9bf4c16d81
commit 4edec261bc
2 changed files with 158 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
fuzz/make_corpus.py
View File

@@ -1,5 +1,5 @@
import json
import argparse import argparse
import json
import os.path import os.path
# Creates a directory containing seed inputs from a json file having # Creates a directory containing seed inputs from a json file having
@@ -20,6 +20,7 @@ import os.path
# python make_corpus.py ./corpus ./corpus_file.json # python make_corpus.py ./corpus ./corpus_file.json
# Creates a corpus directory to the given path from the given json file.
def make_corpus(corpus_dir, corpus_json): def make_corpus(corpus_dir, corpus_json):
if not os.path.exists(corpus_dir): if not os.path.exists(corpus_dir):
os.makedirs(corpus_dir) os.makedirs(corpus_dir)
@@ -28,24 +29,24 @@ def make_corpus(corpus_dir, corpus_json):
if os.path.isfile(corpus_json) and \ if os.path.isfile(corpus_json) and \
os.path.splitext(corpus_json)[-1] == ".json": os.path.splitext(corpus_json)[-1] == ".json":
with open(corpus_json) as file: with open(corpus_json) as corpus_file:
corpus = json.load(file) corpus = json.load(corpus_file)
else: else:
raise TypeError raise TypeError
for i, seed_file in enumerate(corpus): for i, seed_file in enumerate(corpus):
seed_file_name = "seed_file_" + str(i) seed_file_name = "seed_file_" + str(i)
raw_hex = seed_file['hex'].decode("hex") raw_hex = seed_file["hex"].decode("hex")
with open(os.path.join(corpus_dir, seed_file_name), 'wb') as f: with open(os.path.join(corpus_dir, seed_file_name), "wb") as f:
f.write(raw_hex) f.write(raw_hex)
def main(): def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(
'corpus_directory', help='the resulting corpus directory path') "corpus_directory", help="the resulting corpus directory path")
parser.add_argument( parser.add_argument(
'corpus_json', help='the json file path to make the corpus from') "corpus_json", help="the json file path to make the corpus from")
args = parser.parse_args() args = parser.parse_args()
try: try:
make_corpus(args.corpus_directory, args.corpus_json) make_corpus(args.corpus_directory, args.corpus_json)