11306887291

from __future__ import absolute_import
from __future__ import division

import shutil
from argparse import ArgumentParser
from subprocess import CalledProcessError
from subprocess import check_output
from tempfile import NamedTemporaryFile

import pwnlib.args
pwnlib.args.free_form = False

from pwn import *
from pwnlib.commandline import common


def dump(objdump, path):
    n = NamedTemporaryFile(delete=False)
    o = check_output([objdump,'-d','-x','-s',path])
    n.write(o)
    n.flush()
    return n.name

def diff(a,b):
    try: return check_output(['diff',a,b], universal_newlines=True)
    except CalledProcessError as e:
        return e.output

p = common.parser_commands.add_parser(
    'elfdiff',
    help = 'Compare two ELF files',
    description = 'Compare two ELF files'
)

p.add_argument('a')
p.add_argument('b')

def main(a):
    with context.silent:
        x = ELF(a.a)
        y = ELF(a.b)

    if x.arch != y.arch:
        log.error("Architectures are not the same: %s vs %s" % (x.arch, y.arch))

    context.arch = x.arch

    objdump = pwnlib.asm.which_binutils('objdump')

    tmp = NamedTemporaryFile()
    name = tmp.name

    shutil.copy(x.path, name)
    x = dump(objdump, name)

    shutil.copy(y.path, name)
    y = dump(objdump, name)

    print(diff(x, y))

if __name__ == '__main__':
    pwnlib.commandline.common.main(__file__, main)

1	from __future__ import absolute_import	1✔
2	from __future__ import division	1✔
3
4	import shutil	1✔
5	from argparse import ArgumentParser	1✔
6	from subprocess import CalledProcessError	1✔
7	from subprocess import check_output	1✔
8	from tempfile import NamedTemporaryFile	1✔
9
10	import pwnlib.args	1✔
11	pwnlib.args.free_form = False	1✔
12
13	from pwn import *	1✔
14	from pwnlib.commandline import common	1✔
15
16
17	def dump(objdump, path):	1✔
18	n = NamedTemporaryFile(delete=False)	×
19	o = check_output([objdump,'-d','-x','-s',path])	×
20	n.write(o)	×
21	n.flush()	×
22	return n.name	×
23
24	def diff(a,b):	1✔
25	try: return check_output(['diff',a,b], universal_newlines=True)	×
26	except CalledProcessError as e:	×
27	return e.output	×
28
29	p = common.parser_commands.add_parser(	1✔
30	'elfdiff',
31	help = 'Compare two ELF files',
32	description = 'Compare two ELF files'
33	)
34
35	p.add_argument('a')	1✔
36	p.add_argument('b')	1✔
37
38	def main(a):	1✔
39	with context.silent:	×
40	x = ELF(a.a)	×
41	y = ELF(a.b)	×
42
43	if x.arch != y.arch:	×
44	log.error("Architectures are not the same: %s vs %s" % (x.arch, y.arch))	×
45
46	context.arch = x.arch	×
47
48	objdump = pwnlib.asm.which_binutils('objdump')	×
49
50	tmp = NamedTemporaryFile()	×
51	name = tmp.name	×
52
53	shutil.copy(x.path, name)	×
54	x = dump(objdump, name)	×
55
56	shutil.copy(y.path, name)	×
57	y = dump(objdump, name)	×
58
59	print(diff(x, y))	×
60
61	if __name__ == '__main__':	1!
NEW 62	pwnlib.commandline.common.main(__file__, main)	×

Gallopsled / pwntools / 11306887291

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous