mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 11:13:28 +01:00
bc5eec4c75
Make sure all print statements are compatible with Python 2 and Python3 using the `from __future__ import print_function` statement. Differential Revision: https://reviews.llvm.org/D56249 llvm-svn: 350307
51 lines
1.4 KiB
Python
Executable File
51 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""A tool for looking for indirect jumps and calls in x86 binaries.
|
|
|
|
Helpful to verify whether or not retpoline mitigations are catching
|
|
all of the indirect branches in a binary and telling you which
|
|
functions the remaining ones are in (assembly, etc).
|
|
|
|
Depends on llvm-objdump being in your path and is tied to the
|
|
dump format.
|
|
"""
|
|
|
|
from __future__ import print_function
|
|
|
|
import os
|
|
import sys
|
|
import re
|
|
import subprocess
|
|
import optparse
|
|
|
|
# Look for indirect calls/jmps in a binary. re: (call|jmp).*\*
|
|
def look_for_indirect(file):
|
|
args = ['llvm-objdump']
|
|
args.extend(["-d"])
|
|
args.extend([file])
|
|
|
|
p = subprocess.Popen(args=args, stdin=None, stderr=subprocess.PIPE, stdout=subprocess.PIPE)
|
|
(stdout,stderr) = p.communicate()
|
|
|
|
function = ""
|
|
for line in stdout.splitlines():
|
|
if line.startswith(' ') == False:
|
|
function = line
|
|
result = re.search('(call|jmp).*\*', line)
|
|
if result != None:
|
|
# TODO: Perhaps use cxxfilt to demangle functions?
|
|
print(function)
|
|
print(line)
|
|
return
|
|
|
|
def main(args):
|
|
# No options currently other than the binary.
|
|
parser = optparse.OptionParser("%prog [options] <binary>")
|
|
(opts, args) = parser.parse_args(args)
|
|
if len(args) != 2:
|
|
parser.error("invalid number of arguments: %s" % len(args))
|
|
look_for_indirect(args[1])
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv)
|