qira/middleware/qira.py

#!/usr/bin/env python
import os
import argparse
import socket
import threading
import time

import qira_socat
import qira_program
import qira_webserver

if __name__ == '__main__':
  parser = argparse.ArgumentParser(description = 'Analyze binary.')
  parser.add_argument('-s', "--server", help="bind on port 4000. like socat", action="store_true")
  parser.add_argument('-t', "--tracelibraries", help="trace into all libraries", action="store_true")
  parser.add_argument('binary', help="path to the binary")
  parser.add_argument('args', nargs='*', help="arguments to the binary")
  args = parser.parse_args()

  # creates the file symlink, program is constant through server run
  program = qira_program.Program(args.binary, args.args)

  if args.tracelibraries:
    program.defaultargs.append("-tracelibraries")

  is_qira_running = 1
  try:
    socket.create_connection(('127.0.0.1', qira_webserver.QIRA_PORT))
    if args.server:
      raise Exception("can't run as server if QIRA is already running")
  except:
    is_qira_running = 0
    print "no qira server found, starting it"
    program.clear()

  # start the binary runner
  if args.server:
    qira_socat.start_bindserver(program, 4000, -1, 1, True)
  else:
    print "**** running "+program.program
    if is_qira_running or os.fork() == 0:   # cute?
      program.execqira()

  if not is_qira_running:
    # start the http server
    qira_webserver.run_server(program)