ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-13 02:28:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

autoformat with black

Browse Source
This commit is contained in:
Dominik Maier
2021-04-15 23:56:58 +02:00
parent 4f93220c4b
commit c8e96e5253
8 changed files with 408 additions and 259 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
utils/autodict_ql/strcmp-strings.py
View File

@ -5,7 +5,7 @@
import os
import string
import binascii
import binascii
import codecs
import errno
import struct
@ -13,6 +13,7 @@ import argparse
import re
from binascii import unhexlify
def ensure_dir(dir):
try:
os.makedirs(dir)
@ -20,44 +21,63 @@ def ensure_dir(dir):
if e.errno != errno.EEXIST:
raise
def parse_args():
parser = argparse.ArgumentParser(description=(
"Helper - Specify input file analysis and output folder to save corpus for strings in the overall project --------------------------------------------------------------------------- Example usage : python2 thisfile.py outdir str.txt" ))
parser.add_argument("corpdir",
help="The path to the corpus directory to generate strings.")
parser.add_argument("infile",
help="Specify file output of codeql analysis - ex. ooo-atr.txt, analysis take place on this file, example : python2 thisfile.py outdir strings.txt")
parser = argparse.ArgumentParser(
description=(
"Helper - Specify input file analysis and output folder to save corpus for strings in the overall project --------------------------------------------------------------------------- Example usage : python2 thisfile.py outdir str.txt"
)
)
parser.add_argument(
"corpdir", help="The path to the corpus directory to generate strings."
)
parser.add_argument(
"infile",
help="Specify file output of codeql analysis - ex. ooo-atr.txt, analysis take place on this file, example : python2 thisfile.py outdir strings.txt",
)
return parser.parse_args()
def do_string_analysis(corpdir, infile1):
with open(infile1, "r") as f1:
lines = f1.readlines()[1:]
f1.close()
with open(infile1, "r") as f1:
lines = f1.readlines()[1:]
f1.close()
new_lst1 = []
n = 1
for i, num1 in enumerate(lines):
if i != 0:
new_lst1.append(num1)
#print("num : %s" % num1)
# print("num : %s" % num1)
str11 = str(num1)
str11 = str11.replace("|","")
str11 = str11.replace("\n","")
str11 = str11.replace("|", "")
str11 = str11.replace("\n", "")
str11 = str11.lstrip()
str11 = str11.rstrip()
str11 = str(str11)
if ((" " in str11 ) or (")" in str11) or ("(" in str11) or ("<" in str11) or (">" in str11)) :
if (
(" " in str11)
or (")" in str11)
or ("(" in str11)
or ("<" in str11)
or (">" in str11)
):
print("Space / Paranthesis String : %s" % str11)
else :
with open(corpdir+'/strcmp-str{0}'.format(n), 'w') as file:
file.write(str11)
print("AFL++ Autodict-QL by Microsvuln : Writing Token : %s" % str11)
n=n+1
else:
with open(corpdir + "/strcmp-str{0}".format(n), "w") as file:
file.write(str11)
print(
"AFL++ Autodict-QL by Microsvuln : Writing Token : %s"
% str11
)
n = n + 1
def main():
args = parse_args()
args = parse_args()
ensure_dir(args.corpdir)
do_string_analysis(args.corpdir, args.infile)
if __name__ == '__main__':
main()
if __name__ == "__main__":
main()