Das deutsche Python-Forum

Code: Alles auswählen

import SocketServer, select, socket, threading

class ConnectionHandler(SocketServer.BaseRequestHandler):
  BUFFERSIZE = 2**12

  def handle(self):
    src = self.request
    dest = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # hier brauch ich das verbindungsziel
    dest.connect((dest_ip_of_src, dest_port_of_src))
    pairs = {src:dest, dest:src}
    while pairs:
      read = select.select(pairs.keys(), [], [])[0]
      for s in read:
        data = s.recv(self.BUFFERSIZE)
        if data:
          pairs[s].sendall(data)
        else:
          pairs[s].shutdown(socket.SHUT_WR)
          s.shutdown(socket.SHUT_RD)
          del pairs[s]
    dest.close()

class Proxy(threading.Thread):
  def __init__(self, addr):
    threading.Thread.__init__(self)
    self.addr = addr

  def run(self):
    server = SocketServer.ThreadingTCPServer(self.addr, ConnectionHandler)
    server.serve_forever()

def start_proxy(addr):
  """start_proxy(address)
      Starts a proxy and binds it to ``address``"""
  p = Proxy(addr)
  p.setDaemon(1)
  p.start()

if __name__ == "__main__":
  print "Starting proxy at 0.0.0.0:8000"
  start_proxy(("", 8000))
  threading.Event().wait()