High level version tcp proxy

simple_tcp_proxy.py

from socketserver import BaseRequestHandler, TCPServer
from socket import socket, AF_INET, SOCK_STREAM
from threading import Thread

BIND_HOST, BIND_PORT = "0.0.0.0", 22
HOST, PORT = "192.168.100.5", 22

def relay(fromSock, toSock):
  while True:
    try:
      d = fromSock.recv(4096)
      toSock.send(d)
    except:
      break
  return

class SockHandler(BaseRequestHandler):
  def handle(self):
    try:
      sock = socket(AF_INET, SOCK_STREAM)
      sock.connect((HOST, PORT))
    except Exception:
      self.request.close()
      return
    
    # self.request = client - proxy
    # sock = proxy - server

    th1 = Thread(target=relay, args=(self.request, sock))
    th2 = Thread(target=relay, args=(sock, self.request))

    th1.start()
    th2.start()

    th1.join()
    th2.join()

    self.request.close()
    sock.close()


if __name__ == '__main__':
  serve = TCPServer((BIND_HOST, BIND_PORT), SockHandler)
  serve.serve_forever()

https://github.com/robert/how-to-build-a-tcp-proxy