You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
3.7 KiB
119 lines
3.7 KiB
#!/usr/bin/env python2 |
|
|
|
import argparse |
|
import cmd2 |
|
import crochet |
|
import imp |
|
import os |
|
import schema.update |
|
import shutil |
|
import sys |
|
import sqlite3 |
|
import tempfile |
|
from pappyproxy import console |
|
from pappyproxy import config |
|
from pappyproxy import comm |
|
from pappyproxy import http |
|
from pappyproxy import context |
|
from pappyproxy import proxy |
|
from twisted.enterprise import adbapi |
|
from twisted.internet import reactor, defer |
|
from twisted.internet.threads import deferToThread |
|
from twisted.internet.protocol import ServerFactory |
|
|
|
|
|
crochet.no_setup() |
|
|
|
def parse_args(): |
|
# parses sys.argv and returns a settings dictionary |
|
|
|
parser = argparse.ArgumentParser(description='An intercepting proxy for testing web applications.') |
|
parser.add_argument('-l', '--lite', help='Run the proxy in "lite" mode', action='store_true') |
|
|
|
args = parser.parse_args(sys.argv[1:]) |
|
settings = {} |
|
|
|
if args.lite: |
|
settings['lite'] = True |
|
else: |
|
settings['lite'] = False |
|
|
|
return settings |
|
|
|
def set_text_factory(conn): |
|
conn.text_factory = str |
|
|
|
def delete_datafile(): |
|
print 'Deleting temporary datafile' |
|
os.remove(config.DATAFILE) |
|
|
|
@defer.inlineCallbacks |
|
def main(): |
|
settings = parse_args() |
|
|
|
if settings['lite']: |
|
conf_settings = config.get_default_config() |
|
conf_settings['debug_dir'] = None |
|
conf_settings['debug_to_file'] = False |
|
with tempfile.NamedTemporaryFile(delete=False) as tf: |
|
conf_settings['data_file'] = tf.name |
|
print 'Temporary datafile is %s' % tf.name |
|
delete_data_on_quit = True |
|
config.load_settings(conf_settings) |
|
else: |
|
# Initialize config |
|
config.load_from_file('./config.json') |
|
delete_data_on_quit = False |
|
|
|
# If the data file doesn't exist, create it with restricted permissions |
|
if not os.path.isfile(config.DATAFILE): |
|
with os.fdopen(os.open(config.DATAFILE, os.O_CREAT, 0o0600), 'r') as f: |
|
pass |
|
|
|
dbpool = adbapi.ConnectionPool("sqlite3", config.DATAFILE, |
|
check_same_thread=False, |
|
cp_openfun=set_text_factory, |
|
cp_max=1) |
|
yield schema.update.update_schema(dbpool) |
|
http.init(dbpool) |
|
yield context.init() |
|
|
|
# Run the proxy |
|
if config.DEBUG_DIR and os.path.exists(config.DEBUG_DIR): |
|
shutil.rmtree(config.DEBUG_DIR) |
|
print 'Removing old debugging output' |
|
factory = ServerFactory() |
|
factory.protocol = proxy.ProxyServer |
|
listen_strs = [] |
|
for listener in config.LISTENERS: |
|
reactor.listenTCP(listener[0], factory, interface=listener[1]) |
|
listener_str = 'port %d' % listener[0] |
|
if listener[1] not in ('127.0.0.1', 'localhost'): |
|
listener_str += ' (bound to %s)' % listener[1] |
|
listen_strs.append(listener_str) |
|
if listen_strs: |
|
print 'Proxy is listening on %s' % (', '.join(listen_strs)) |
|
|
|
com_factory = ServerFactory() |
|
com_factory.protocol = comm.CommServer |
|
# Make the port different for every instance of pappy, then pass it to |
|
# anything we run. Otherwise we can only have it running once on a machine |
|
comm_port = reactor.listenTCP(0, com_factory, interface='127.0.0.1') |
|
comm.set_comm_port(comm_port.getHost().port) |
|
|
|
sys.argv = [sys.argv[0]] # cmd2 tries to parse args |
|
d = deferToThread(console.ProxyCmd().cmdloop) |
|
d.addCallback(lambda ignored: reactor.stop()) |
|
if delete_data_on_quit: |
|
d.addCallback(lambda ignored: delete_datafile()) |
|
|
|
# Load the scope |
|
yield context.load_scope(http.dbpool) |
|
context.reset_to_scope() |
|
|
|
def start(): |
|
reactor.callWhenRunning(main) |
|
reactor.run() |
|
|
|
if __name__ == '__main__': |
|
start()
|
|
|