From de267f0ea83f3828515543d9394d069e2378d808 Mon Sep 17 00:00:00 2001
From: James Bunton <jamesbunton@fastmail.fm>
Date: Mon, 9 Mar 2009 22:54:03 +1100
Subject: [PATCH] Handy little tcp proxy

---
 scripts/proxy.py | 98 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 98 insertions(+)
 create mode 100755 scripts/proxy.py

diff --git a/scripts/proxy.py b/scripts/proxy.py
new file mode 100755
index 0000000..ec32f41
--- /dev/null
+++ b/scripts/proxy.py
@@ -0,0 +1,98 @@
+#!/usr/bin/env python
+
+import asyncore
+import os
+import socket
+import sys
+
+class Proxy(asyncore.dispatcher):
+	def __init__(self, sock):
+		asyncore.dispatcher.__init__(self, sock)
+		self.other = None
+		self.towrite = ""
+	
+	def meet(self, other):
+		self.other = other
+		other.other = self
+	
+	def handle_error(self):
+		print >>sys.stderr, "Connection closed..."
+		self.handle_close()
+
+	def handle_read(self):
+		data = self.recv(8192)
+		self.other.towrite += data
+	
+	def handle_write(self):
+		sent = self.send(self.towrite)
+		self.towrite = self.towrite[sent:]
+	
+	def handle_close(self):
+		if not self.other:
+			return
+		self.other.close()
+		self.close()
+		self.other = None
+
+class Forwarder(asyncore.dispatcher):
+	def __init__(self, host, port):
+		asyncore.dispatcher.__init__(self)
+		self.host = host
+		self.port = port
+		self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
+		self.bind(("", port))
+		self.listen(5)
+	
+	def handle_error(self):
+		print >>sys.stderr, "Connection error!"
+	
+	def handle_accept(self):
+		# Get sockets
+		clientConnection, addr = self.accept()
+		print >>sys.stderr, "Accepted connection from", addr
+		serverConnection = socket.socket()
+		serverConnection.connect((self.host, self.port))
+
+		# Hook them up to the event loop
+		client = Proxy(clientConnection)
+		server = Proxy(serverConnection)
+		server.meet(client)
+
+def main(host, port):
+	proxy = Forwarder(host, port)
+	asyncore.loop()
+
+if __name__ == "__main__":
+	try:
+		if sys.argv[1] == "-d":
+			daemon = True
+			host, port = sys.argv[2:]
+		else:
+			host, port = sys.argv[1:]
+			daemon = False
+		port = int(port)
+	except ValueError:
+		print >> sys.stderr, "Usage: %s [-d] host port" % sys.argv[0]
+		sys.exit(1)
+
+	if not daemon:
+		try:
+			main(host, port)
+		except KeyboardInterrupt:
+			print
+	else:
+		sys.stdin.close()
+		sys.stdout.close()
+		sys.stderr.close()
+		sys.stdin = open("/dev/null")
+		sys.stdout = open("/dev/null")
+		sys.stderr = open("/dev/null")
+
+		if os.fork() == 0:
+			# We are the child
+			try:
+				main(host, port)
+			except KeyboardInterrupt:
+				print
+			sys.exit(0)
+
-- 
2.39.2