lldbsh.py

# Code is available under MIT license

import lldb
import subprocess

def sh(debugger, command, result, internal_dict):
  commands = command.split("|", 1)
  lldb_cmd = commands[0]

  if lldb_cmd == '':
      print("No LLDB command given. Use: 'sh <LLDB command> | <shell command>'")
      return

  interpreter = debugger.GetCommandInterpreter()
  rto = lldb.SBCommandReturnObject()
  interpreter.HandleCommand(lldb_cmd, rto)

  if not rto.Succeeded():
      raise Exception("failed to execute lldb command!")
  if not rto.GetOutput():
      return

  if len(commands) == 1:
    # `sh command` is equivalent to `command`
    print(rto.GetOutput())
    return
  elif len(commands) == 2:
    # TODO support chaining?
    shell_cmd = commands[1]

  # TODO: Not sure what the underlying buffer is for HandleCommand, but
  # ideally do some check/optim for really large output.
  output = rto.GetOutput().encode('UTF-8')
  proc = subprocess.Popen(shell_cmd, shell=True, stdin=subprocess.PIPE)

  # TODO?
  # Not using a timer here allows piping to a pager. However,
  # it also means that the shell-out process may not be killable.
  try:
      out, errs = proc.communicate(output)
  except:
      print("Unexpected exit, killing subprocess")
      proc.kill()
      outs, errs = proc.communicate()

def __lldb_init_module (debugger, internal_dict):
  res = lldb.SBCommandReturnObject()
  interpreter = debugger.GetCommandInterpreter()
  interpreter.HandleCommand('command script delete sh', res)
  debugger.HandleCommand('command script add -f lldbsh.sh sh ')