mirror of
https://github.com/AFLplusplus/AFLplusplus.git
synced 2025-06-18 20:48:07 +00:00
Add python scripts
Add python scripts
This commit is contained in:
64
utils/autodict_ql/strncmp-strings.py
Normal file
64
utils/autodict_ql/strncmp-strings.py
Normal file
@ -0,0 +1,64 @@
|
||||
#!/usr/bin/env python3
|
||||
# Autodict-QL - Optimal token generation for fuzzing
|
||||
# Part of AFL++ Project
|
||||
# Author : Microsvuln - Arash.vre@gmail.com
|
||||
|
||||
import os
|
||||
import string
|
||||
import binascii
|
||||
import codecs
|
||||
import errno
|
||||
import struct
|
||||
import argparse
|
||||
import re
|
||||
from binascii import unhexlify
|
||||
|
||||
def ensure_dir(dir):
|
||||
try:
|
||||
os.makedirs(dir)
|
||||
except OSError as e:
|
||||
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")
|
||||
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
def do_string_analysis(corpdir, infile1):
|
||||
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)
|
||||
str11 = str(num1)
|
||||
str11 = str11.replace("|","")
|
||||
str11 = str11.replace("\n","")
|
||||
str11 = str11.lstrip()
|
||||
str11 = str11.rstrip()
|
||||
print("all strings : %s" % str11)
|
||||
str11 = str(str11)
|
||||
if ((" " in str11 ) or (")" in str11) or ("(" in str11)):
|
||||
print("Space / Paranthesis String : %s" % str11)
|
||||
else :
|
||||
with open(corpdir+'/strncmp-str{0}'.format(n), 'w') as file:
|
||||
file.write(str11)
|
||||
print("Hahaha : %s" % str11)
|
||||
n=n+1
|
||||
|
||||
def main():
|
||||
args = parse_args()
|
||||
ensure_dir(args.corpdir)
|
||||
do_string_analysis(args.corpdir, args.infile)
|
||||
if __name__ == '__main__':
|
||||
main()
|
Reference in New Issue
Block a user