ssl deproxy client and server

framework/__init__.py

@@ -1,2 +1,3 @@
 __all__ = ['client', 'deproxy_client', 'deproxy_manager', 'deproxy_server',
-           'nginx_server', 'templates', 'tester', 'wrk_client', 'port_checks']
+           'nginx_server', 'templates', 'tester', 'wrk_client', 'port_checks',
+           'deproxy_client_v2']

framework/deproxy_client/client_proxy.py

@@ -0,0 +1,209 @@
+#!/usr/bin/python3
+
+import getopt
+import sys
+import io
+import os
+import time
+
+import signal
+import socket
+import http.server
+
+sock = None
+
+def connect(addr, port):
+    global sock
+    if sock != None:
+        raise Exception("already connected")
+    sock = socket.socket()
+    sock.connect((addr, int(port)))
+    sock.setblocking(False)
+
+def request(content):
+    global sock
+    if sock is None:
+        raise Exception("socket is closed")
+    sock.send(content)
+
+def read_response(maxlen):
+    global sock
+    if sock is None:
+        raise Exception("socket is closed")
+    try:
+        ans = sock.recv(int(maxlen))
+    except io.BlockingIOError:
+        return b""
+    return ans
+
+def finish():
+    global sock
+    if sock is None:
+        raise Exception("socket is closed")
+    sock.close()
+    sock = None
+
+class DeproxyHandler(http.server.BaseHTTPRequestHandler):
+    """ Handle commands from framework """
+    def do_POST(self):
+        try:
+            command = self.headers.get('command')
+            if command is None:
+                self.send_response(400)
+                self.send_header('Result', 'error')
+                self.end_headers()
+                self.wfile.write(b"No command specified\n")
+                return
+
+            if command == 'connect':
+                addr = self.headers.get('addr')
+                port = self.headers.get('port')
+                if addr is None:
+                    raise Exception("No addr header is specified")
+                if port is None:
+                    raise Exception("No port header is specified")
+                connect(addr, port)
+                self.send_response(200)
+                self.send_header('Result', 'ok')
+                self.end_headers()
+            elif command == 'request':
+                cl = self.headers.get('content-length')
+                if cl is None:
+                    raise Exception("No content-length header is specified")
+                content_len = int(cl)
+                content = self.rfile.read(content_len)
+                request(content)
+                self.send_response(200)
+                self.send_header('Result', 'ok')
+                self.end_headers()
+                return
+            elif command == 'read':
+                maxlen = self.headers.get('maxlen')
+                if maxlen is None:
+                    raise Exception("No maxlen header is specified")
+                data = read_response(maxlen)
+                self.send_response(200)
+                self.send_header('Result', 'ok')
+                self.send_header('Content-Length', len(data))
+                self.end_headers()
+                self.wfile.write(data)
+                return
+            elif command == 'finish':
+                finish()
+                self.send_response(200)
+                self.send_header('Result', 'ok')
+                self.end_headers()
+                return
+            else:
+                raise Exception("Unknown command %s" % command)
+        except Exception as e:
+            self.send_response(403)
+            self.send_header('Result', 'error')
+            self.end_headers()
+            self.wfile.write(bytes("Error: %s\n" % str(e), "utf-8"))
+            return
+        sys.stdout.flush()
+        sys.stderr.flush()
+
+listen = 7000
+
+try:
+    options, remainder = getopt.getopt(sys.argv[1:], 'l:',
+                                       ['listen='])
+
+    for opt, arg in options:
+        if opt in ('-l', '--listen'):
+            listen = int(arg)
+
+except getopt.GetoptError as e:
+    print(e)
+    sys.exit(2)
+except Exception as e:
+    print(e)
+    sys.exit(2)
+
+pidfile = 'proxy-%i.pid' % listen
+logfile = 'proxy-%i.log' % listen
+
+def wait_for_pidfile(timeout=2):
+    t0 = time.time()
+    while not os.path.exists(pidfile):
+        t = time.time()
+        if t - t0 > timeout:
+            return
+
+def fork(wait=True):
+    try:
+        pid = os.fork()
+        if pid > 0:
+            if wait:
+                wait_for_pidfile(1)
+            # wait while listen() starts
+            time.sleep(0.5)
+            sys.exit(0)
+    except OSError as e:
+        sys.stderr.write("Fork failed: %d (%s)\n" % (e.errno, e.strerror))
+        sys.exit(1)
+
+def exit_handler(signal, frame):
+    print("Exiting\n")
+    sys.stdout.flush()
+    os.remove(pidfile)
+    sys.exit(0)
+
+def test_pid_file(name):
+    if not os.path.exists(name):
+        return
+    pidf = open(name, 'r')
+    p = pidf.read()
+    pidf.close()
+    pid = None
+    try:
+        pid = int(p)
+        os.kill(pid, 0)
+    except:
+        return
+    print("Process %s,%s already exists\n" % (name, pid))
+    sys.exit(3)
+
+def daemonize():
+    fork(True)
+
+    os.setsid()
+    os.umask(0)
+
+    fork(False)
+
+    sys.stdout.flush()
+    sys.stderr.flush()
+
+    stdin = open('/dev/null', 'r')
+    os.dup2(stdin.fileno(), sys.stdin.fileno())
+
+    stdout = open(logfile, 'w')
+    os.dup2(stdout.fileno(), sys.stdout.fileno())
+    os.dup2(stdout.fileno(), sys.stderr.fileno())
+
+    pid = str(os.getpid())
+
+    pidf = open(pidfile,'w+')
+    pidf.write("%s\n" % pid)
+    pidf.close()
+
+    signal.signal(signal.SIGINT, exit_handler)
+    signal.signal(signal.SIGTERM, exit_handler)
+
+test_pid_file(pidfile)
+
+daemonize()
+
+print("Daemonized\n")
+sys.stdout.flush()
+
+try:
+    server_address = ('', listen)
+    httpd = http.server.HTTPServer(server_address, DeproxyHandler)
+    httpd.serve_forever()
+except Exception as e:
+    print("Error: %s\n" % e)
+    sys.exit(3)