Home | History | Annotate | Download | only in python 
      1 #!/usr/bin/python
      2 
      3 #----------------------------------------------------------------------
      4 # Be sure to add the python path that points to the LLDB shared library.
      5 # On MacOSX csh, tcsh:
      6 #   setenv PYTHONPATH /Developer/Library/PrivateFrameworks/LLDB.framework/Resources/Python
      7 # On MacOSX sh, bash:
      8 #   export PYTHONPATH=/Developer/Library/PrivateFrameworks/LLDB.framework/Resources/Python
      9 #----------------------------------------------------------------------
     10 
     11 import lldb
     12 import os
     13 import sys
     14 
     15 def disassemble_instructions (insts):
     16     for i in insts:
     17         print i
     18 
     19 def usage():
     20     print "Usage: disasm.py [-n name] executable-image"
     21     print "       By default, it breaks at and disassembles the 'main' function."
     22     sys.exit(0)
     23 
     24 if len(sys.argv) == 2:
     25     fname = 'main'
     26     exe = sys.argv[1]
     27 elif len(sys.argv) == 4:
     28     if sys.argv[1] != '-n':
     29         usage()
     30     else:
     31         fname = sys.argv[2]
     32         exe = sys.argv[3]
     33 else:
     34     usage()
     35 
     36 # Create a new debugger instance
     37 debugger = lldb.SBDebugger.Create()
     38 
     39 # When we step or continue, don't return from the function until the process 
     40 # stops. We do this by setting the async mode to false.
     41 debugger.SetAsync (False)
     42 
     43 # Create a target from a file and arch
     44 print "Creating a target for '%s'" % exe
     45 
     46 target = debugger.CreateTargetWithFileAndArch (exe, lldb.LLDB_ARCH_DEFAULT)
     47 
     48 if target:
     49     # If the target is valid set a breakpoint at main
     50     main_bp = target.BreakpointCreateByName (fname, target.GetExecutable().GetFilename());
     51 
     52     print main_bp
     53 
     54     # Launch the process. Since we specified synchronous mode, we won't return
     55     # from this function until we hit the breakpoint at main
     56     process = target.LaunchSimple (None, None, os.getcwd())
     57     
     58     # Make sure the launch went ok
     59     if process:
     60         # Print some simple process info
     61         state = process.GetState ()
     62         print process
     63         if state == lldb.eStateStopped:
     64             # Get the first thread
     65             thread = process.GetThreadAtIndex (0)
     66             if thread:
     67                 # Print some simple thread info
     68                 print thread
     69                 # Get the first frame
     70                 frame = thread.GetFrameAtIndex (0)
     71                 if frame:
     72                     # Print some simple frame info
     73                     print frame
     74                     function = frame.GetFunction()
     75                     # See if we have debug info (a function)
     76                     if function:
     77                         # We do have a function, print some info for the function
     78                         print function
     79                         # Now get all instructions for this function and print them
     80                         insts = function.GetInstructions(target)
     81                         disassemble_instructions (insts)
     82                     else:
     83                         # See if we have a symbol in the symbol table for where we stopped
     84                         symbol = frame.GetSymbol();
     85                         if symbol:
     86                             # We do have a symbol, print some info for the symbol
     87                             print symbol
     88                             # Now get all instructions for this symbol and print them
     89                             insts = symbol.GetInstructions(target)
     90                             disassemble_instructions (insts)
     91 
     92                     registerList = frame.GetRegisters()
     93                     print "Frame registers (size of register set = %d):" % registerList.GetSize()
     94                     for value in registerList:
     95                         #print value
     96                         print "%s (number of children = %d):" % (value.GetName(), value.GetNumChildren())
     97                         for child in value:
     98                             print "Name: ", child.GetName(), " Value: ", child.GetValue()
     99 
    100             print "Hit the breakpoint at main, enter to continue and wait for program to exit or 'Ctrl-D'/'quit' to terminate the program"
    101             next = sys.stdin.readline()
    102             if not next or next.rstrip('\n') == 'quit':
    103                 print "Terminating the inferior process..."
    104                 process.Kill()
    105             else:
    106                 # Now continue to the program exit
    107                 process.Continue()
    108                 # When we return from the above function we will hopefully be at the
    109                 # program exit. Print out some process info
    110                 print process
    111         elif state == lldb.eStateExited:
    112             print "Didn't hit the breakpoint at main, program has exited..."
    113         else:
    114             print "Unexpected process state: %s, killing process..." % debugger.StateAsCString (state)
    115             process.Kill()
    116 
    117         
    118 
    119 lldb.SBDebugger.Terminate()
    120