def main():
afl.init()
try:
decode_rfc6979_signature(sys.stdin.read())
except ValueError:
pass
def main() -> None:
afl.init()
with open(sys.argv[1]) as f:
# afl-fuzz will often generate invalid Unicode and count that as a
# crash. See
# https://barro.github.io/2018/01/taking-a-look-at-python-afl/
try:
file_info = FileInfo(file_path="<stdin>", source_code=f.read())
except UnicodeDecodeError:
pass
else:
check_for_buggy_parse(file_info)
os._exit(0)
def main():
global swarm, showActions, noCheck, alwaysSave, noSave, sut
if "--help" in sys.argv:
print(
"Usage: tstl_afl [--noCheck] [--swarm] [--verbose] [--showActions] [--noSave] [--alwaysSave]"
)
print("Options:")
print(" --noCheck: do not run property checks")
print(
" --swarm use first four bytes to determine a swarm configuration"
)
print(" --verbose: make actions verbose")
print(" --showActions: show actions in test")
print(
" --noSave: don't save failing tests as standard TSTL tests")
print(" --alwaysSave: save even non-failing tests")
sys.exit(0)
sut = SUT.sut()
try:
sut.stopCoverage()
except BaseException:
pass
sut.restart()
swarm = "--swarm" in sys.argv
showActions = "--showActions" in sys.argv
if "--verbose" in sys.argv:
sut.verbose(True)
noSave = "--noSave" in sys.argv
alwaysSave = "--alwaysSave" in sys.argv
noCheck = "--noCheck" in sys.argv
persist = "--persist" in sys.argv
if not persist:
afl.init()
runTest()
else:
while afl.loop():
runTest()
sut.restart()
os._exit(0)
def main():
global args, device, script, session, pid, sock, code, app_name
opt = argparse.ArgumentParser(
description=DESCR, formatter_class=argparse.RawTextHelpFormatter)
opt.add_argument('-l',
action='store',
default='proxy/proxy.js',
help='Script filename')
opt.add_argument('-U',
action='store_true',
default=False,
help='Connect to USB')
opt.add_argument('-s',
action='store_true',
default=False,
help='Spawn instead of attach')
opt.add_argument(
'-t',
action='store',
help='file to store target program/pid (and arguments if spwaning)')
opt.add_argument('input',
nargs=argparse.REMAINDER,
help='Input corpus file')
args = opt.parse_args()
app_name = args.t
try:
app_name = int(app_name)
pid = app_name
except:
pass
with open(args.l) as f:
code = f.read()
afl.init(remote_trace=True)
fuzz()
sys.stdin.read()
p[0] = Vdecl(p[1], p[2])
# empty production
def p_empty(p):
'empty :'
pass
# Error rule for syntax errors
def p_error(p):
print("Syntax error in input!")
precedence = (
('left', 'OR', 'AND'),
('nonassoc', 'LTH', 'BTH', 'EQU'), # Nonassociative operators
('left', 'PLUS', 'MINUS'),
('left', 'TIMES', 'DIVIDE'),
('right', 'NEG'), # Unary minus operator
)
afl.init()
parser = yacc.yacc()
#data = sys.stdin.read()
result = parser.parse(sys.stdin.read())
print(result)
def test_double_init():
afl.init()
with assert_raises_regex(RuntimeError, '^AFL already initialized$'):
afl.init()
def test_double_init():
afl.init()
with assert_raises_regex(RuntimeError, '^AFL already initialized$'):
afl.init()
#!/usr/bin/env python
from __future__ import print_function
from PrussianBlue.ccproto import ChatCommand, ProtocolError, CLIENT_COMMANDS
import afl
import sys
afl.init()
try:
msg = unicode(sys.stdin.read(), 'utf-8').rstrip('\r\n')
cmd = ChatCommand.cc_decode(CLIENT_COMMANDS, msg)
res = cmd.cc_encode()
assert msg == res, "%r != %r" % (msg, res)
print("%r roundtrips." % msg)
except UnicodeError:
pass
except ProtocolError:
print("%r raises a ProtocolError." % msg)
def main():
if "--help" in sys.argv:
print(
"Usage: tstl_afl [--noCheck] [--swarm] [--verbose] [--showActions] [--noSave] [--alwaysSave]"
)
print("Options:")
print(" --noCheck: do not run property checks")
print(
" --swarm use first four bytes to determine a swarm configuration"
)
print(" --verbose: make actions verbose")
print(" --showActions: show actions in test")
print(
" --noSave: don't save failing tests as standard TSTL tests")
print(" --alwaysSave: save even non-failing tests")
sys.exit(0)
sut = SUT.sut()
try:
sut.stopCoverage()
except:
pass
sut.restart()
if "--swarm" in sys.argv:
swarm = True
R = random.Random()
else:
swarm = False
showActions = "--showActions" in sys.argv
if "--verbose" in sys.argv:
sut.verbose(True)
noSave = "--noSave" in sys.argv
alwaysSave = "--alwaysSave" in sys.argv
noCheck = "--noCheck" in sys.argv
afl.init()
bytesin = sys.stdin.read()
if len(bytesin) < 4:
os._exit(0)
if swarm:
R.seed(struct.unpack("<L", bytesin[0:4])[0])
sut.standardSwarm(R)
bytesin = bytesin[4:]
alen = len(sut.actions())
test = sut.bytesToTest(bytesin)
for a in test:
if a[1]():
if showActions:
print(sut.prettyName(a[0]))
ok = sut.safely(a)
if (not noSave) and not ok:
i = 0
saveFile = "aflfail." + str(
os.getpid()) + "." + str(i) + ".test"
while os.path.exists(saveFile):
i += 1
saveFile = "aflfail." + str(
os.getpid()) + "." + str(i) + ".test"
sut.saveTest(sut.test(), saveFile)
assert (ok)
if not noCheck:
checkResult = sut.check()
if (not noSave) and not checkResult:
i = 0
saveFile = "aflfail." + str(
os.getpid()) + "." + str(i) + ".test"
while os.path.exists(saveFile):
i += 1
saveFile = "aflfail." + str(
os.getpid()) + "." + str(i) + ".test"
sut.saveTest(sut.test(), saveFile)
assert (checkResult)
if alwaysSave:
i = 0
saveFile = "afltest." + str(os.getpid()) + "." + str(i) + ".test"
while os.path.exists(saveFile):
i += 1
saveFile = "afltest." + str(os.getpid()) + "." + str(i) + ".test"
sut.saveTest(sut.test(), saveFile)
os._exit(0)
