def run_ropgadget_on_pair(bench_name, prog_before, prog_after, opts, \
logfile, outfd, errfd):
outfd.seek(0, 0)
errfd.seek(0, 0)
res_before, res_after = None, None
args = [get_ropgadget(), '--binary', prog_before] + opts
returncode, _, _, _ = cmd.run_limited_cmd(args, outfd, errfd,
bench_name)
# ROPgadget returns 1 if success
if returncode <> 1:
cmd.warning("something failed while running \"" + ' '.join(args) +
"\"")
else:
res_before = read_ropgadget_output(logfile)
outfd.seek(0, 0)
errfd.seek(0, 0)
args = [get_ropgadget(), '--binary', prog_after] + opts
returncode, _, _, _ = cmd.run_limited_cmd(args, outfd, errfd,
bench_name)
# ROPgadget returns 1 if success
if returncode <> 1:
cmd.warning("something failed while running \"" +
' '.join(args) + "\"")
else:
res_after = read_ropgadget_output(logfile)
return (res_before, res_after)
def run_occam(dirname, execname, workdir, cpu, mem, slash_opts=[]):
#benchmark_name = os.path.basename(os.path.normpath(dirname))
benchmark_name = execname
outfile = benchmark_name + ".occam.out"
errfile = benchmark_name + ".occam.err"
outfd = open(os.path.join(workdir, outfile), "w")
errfd = open(os.path.join(workdir, errfile), "w")
res_before, res_after = None, None
#1. Generate bitcode: run `make`
returncode, _, _, _ = cmd.run_limited_cmd(['make'], outfd, errfd,
benchmark_name, dirname)
if returncode <> 0:
cmd.warning("something failed while running \"make\"" + benchmark_name + "\n" + \
"Read logs " + outfile + " and " + errfile)
else:
#2. Run slash (OCCAM) on it: `build.sh opts`
slash_args = ['./build.sh']
slash_args.extend(slash_opts)
print "Running slash with options " + str(slash_args)
returncode,_,_,_ = \
cmd.run_limited_cmd(slash_args, outfd, errfd, benchmark_name, dirname, cpu, mem)
if returncode <> 0:
cmd.warning("something failed while running \"" + ' '.join(slash_args) + \
"\"" + benchmark_name + "\n" + \
"Read logs " + outfile + " and " + errfile)
outfd.close()
errfd.close()
if returncode == 0:
#All OCCAM output is redirected to error
logfile = os.path.join(workdir, errfile)
(res_before, res_after) = read_occam_output(logfile)
return (benchmark_name, res_before, res_after)
def run_occam(dirname, execname, workdir, cpu, mem):
#benchmark_name = os.path.basename(os.path.normpath(dirname))
benchmark_name = execname
outfile = benchmark_name + ".occam.out"
errfile = benchmark_name + ".occam.err"
outfd = open(os.path.join(workdir, outfile), "w")
errfd = open(os.path.join(workdir, errfile), "w")
res_before, res_after = None, None
#Generate bitcode
returncode,_,_,_ = cmd.run_limited_cmd(['make'], outfd, errfd, benchmark_name, dirname)
if returncode <> 0:
cmd.warning("something failed while running \"make\"" + benchmark_name + "\n" + \
"Read logs " + outfile + " and " + errfile)
else:
#Run slash (OCCAM) on it
returncode,_,_,_ = cmd.run_limited_cmd(['make','slash'], outfd, errfd, benchmark_name, dirname, cpu, mem)
if returncode <> 0:
cmd.warning("something failed while running \"make slash\"" + benchmark_name + "\n" + \
"Read logs " + outfile + " and " + errfile)
outfd.close()
errfd.close()
if returncode == 0:
#All OCCAM output is redirected to error
logfile = os.path.join(workdir, errfile)
(res_before, res_after) = read_occam_output(logfile)
return (benchmark_name, res_before, res_after)
def run_ropgadget_on_pair(bench_name, prog_before, prog_after, opts, \
logfile, outfd, errfd):
outfd.seek(0,0)
errfd.seek(0,0)
res_before, res_after = None, None
args = [get_ropgadget(), '--binary', prog_before] + opts
returncode,_,_,_ = cmd.run_limited_cmd(args, outfd, errfd, bench_name)
# ROPgadget returns 1 if success
if returncode <> 1:
cmd.warning("something failed while running \"" + ' '.join(args) + "\"")
else:
res_before = read_ropgadget_output(logfile)
outfd.seek(0,0)
errfd.seek(0,0)
args = [get_ropgadget(), '--binary', prog_after] + opts
returncode,_,_,_ = cmd.run_limited_cmd(args, outfd, errfd, bench_name)
# ROPgadget returns 1 if success
if returncode <> 1:
cmd.warning("something failed while running \"" + ' '.join(args) + "\"")
else:
res_after = read_ropgadget_output(logfile)
return (res_before, res_after)
def run_ropgadget(dirname, execname, workdir, cpu, mem):
def run_ropgadget_on_pair(bench_name, prog_before, prog_after, opts, \
logfile, outfd, errfd):
outfd.seek(0, 0)
errfd.seek(0, 0)
res_before, res_after = None, None
args = [get_ropgadget(), '--binary', prog_before] + opts
returncode, _, _, _ = cmd.run_limited_cmd(args, outfd, errfd,
bench_name)
# ROPgadget returns 1 if success
if returncode <> 1:
cmd.warning("something failed while running \"" + ' '.join(args) +
"\"")
else:
res_before = read_ropgadget_output(logfile)
outfd.seek(0, 0)
errfd.seek(0, 0)
args = [get_ropgadget(), '--binary', prog_after] + opts
returncode, _, _, _ = cmd.run_limited_cmd(args, outfd, errfd,
bench_name)
# ROPgadget returns 1 if success
if returncode <> 1:
cmd.warning("something failed while running \"" +
' '.join(args) + "\"")
else:
res_after = read_ropgadget_output(logfile)
return (res_before, res_after)
#benchname = os.path.basename(os.path.normpath(dirname))
benchname = execname
outfile = benchname + ".ropgadget.out"
errfile = benchname + ".ropgadget.err"
outfd = open(os.path.join(workdir, outfile), "w")
errfd = open(os.path.join(workdir, errfile), "w")
prog_before = os.path.join(dirname, benchname + "_orig")
prog_after = os.path.join(dirname, benchname + "_occamized")
total_before, total_after = None, None
rop_before, rop_after = None, None
jop_before, jop_after = None, None
sys_before, sys_after = None, None
if os.path.exists(prog_before) and os.path.exists(prog_after):
# total gadgets: ropgadget --binary prog
# only jop: ropgadget --binary --norop --nosys
# only sys: ropgadget --binary --nojop --norop
# only rop: ropgadget --binary --nojop --nosys
logfile = os.path.join(workdir, outfile)
opts = []
(total_before, total_after) = run_ropgadget_on_pair(benchname, prog_before, prog_after, \
opts, logfile, outfd, errfd)
opts = ['--norop', '--nosys']
(jop_before, jop_after) = run_ropgadget_on_pair(benchname, prog_before, prog_after, \
opts, logfile, outfd, errfd)
opts = ['--nojop', '--norop']
(sys_before, sys_after) = run_ropgadget_on_pair(benchname, prog_before, prog_after, \
opts, logfile, outfd, errfd)
opts = ['--nojop', '--nosys']
(rop_before, rop_after) = run_ropgadget_on_pair(benchname, prog_before, prog_after, \
opts, logfile, outfd, errfd)
else:
if not os.path.exists(prog_before):
cmd.warning(prog_before + " does not exist")
if not os.path.exists(prog_after):
cmd.warning(prog_after + " does not exist")
outfd.close()
errfd.close()
return (benchname, \
total_before, total_after, \
jop_before, jop_after, \
sys_before, sys_after, \
rop_before, rop_after)
def run_ropgadget(dirname, execname, workdir, cpu, mem):
def run_ropgadget_on_pair(bench_name, prog_before, prog_after, opts, \
logfile, outfd, errfd):
outfd.seek(0,0)
errfd.seek(0,0)
res_before, res_after = None, None
args = [get_ropgadget(), '--binary', prog_before] + opts
returncode,_,_,_ = cmd.run_limited_cmd(args, outfd, errfd, bench_name)
# ROPgadget returns 1 if success
if returncode <> 1:
cmd.warning("something failed while running \"" + ' '.join(args) + "\"")
else:
res_before = read_ropgadget_output(logfile)
outfd.seek(0,0)
errfd.seek(0,0)
args = [get_ropgadget(), '--binary', prog_after] + opts
returncode,_,_,_ = cmd.run_limited_cmd(args, outfd, errfd, bench_name)
# ROPgadget returns 1 if success
if returncode <> 1:
cmd.warning("something failed while running \"" + ' '.join(args) + "\"")
else:
res_after = read_ropgadget_output(logfile)
return (res_before, res_after)
#benchname = os.path.basename(os.path.normpath(dirname))
benchname = execname
outfile = benchname + ".ropgadget.out"
errfile = benchname + ".ropgadget.err"
outfd = open(os.path.join(workdir, outfile), "w")
errfd = open(os.path.join(workdir, errfile), "w")
prog_before = os.path.join(dirname, benchname + "_orig")
prog_after = os.path.join(dirname, benchname + "_slashed")
total_before, total_after = None, None
rop_before, rop_after = None, None
jop_before, jop_after = None, None
sys_before, sys_after = None, None
if os.path.exists(prog_before) and os.path.exists(prog_after):
# total gadgets: ropgadget --binary prog
# only jop: ropgadget --binary --norop --nosys
# only sys: ropgadget --binary --nojop --norop
# only rop: ropgadget --binary --nojop --nosys
logfile = os.path.join(workdir, outfile)
opts = []
(total_before, total_after) = run_ropgadget_on_pair(benchname, prog_before, prog_after, \
opts, logfile, outfd, errfd)
opts = ['--norop','--nosys']
(jop_before, jop_after) = run_ropgadget_on_pair(benchname, prog_before, prog_after, \
opts, logfile, outfd, errfd)
opts = ['--nojop', '--norop']
(sys_before, sys_after) = run_ropgadget_on_pair(benchname, prog_before, prog_after, \
opts, logfile, outfd, errfd)
opts = ['--nojop', '--nosys']
(rop_before, rop_after) = run_ropgadget_on_pair(benchname, prog_before, prog_after, \
opts, logfile, outfd, errfd)
else:
if not os.path.exists(prog_before):
cmd.warning(prog_before + " does not exist")
if not os.path.exists(prog_after):
cmd.warning(prog_after + " does not exist")
outfd.close()
errfd.close()
return (benchname, \
total_before, total_after, \
jop_before, jop_after, \
sys_before, sys_after, \
rop_before, rop_after)
