Python implements port forwarding and redirection examples under TCP/IP protocol-Python Tutorial-php.cn

Python implements port forwarding and redirection examples under TCP/IP protocol

高洛峰

Release： 2016-10-17 16:34:32

Original

3036 people have browsed it

First, we use webpy to write a simple website, listen to port 8080, and return the "Hello, EverET.org" page.

Then we use our forwarding.py to establish two communication pipes between port 80 and port 8080 for two-way communication.

At this point, we access our server through port 80.

The browser gets:

Python implements port forwarding and redirection examples under TCP/IP protocol

Then, we can see the communication content between the browser and webpy in the output of forwarding.py.

Python implements port forwarding and redirection examples under TCP/IP protocol

Code:

#!/usr/bin/env python
import sys, socket, time, threading
loglock = threading.Lock()
def log(msg):
  loglock.acquire()
  try:
    print &#39;[%s]: \n%s\n&#39; % (time.ctime(), msg.strip())
    sys.stdout.flush()
  finally:
    loglock.release()
class PipeThread(threading.Thread):
  def __init__(self, source, target):
    threading.Thread.__init__(self)
    self.source = source
    self.target = target
  def run(self):
    while True:
      try:
        data = self.source.recv(1024)
        log(data)
        if not data: break
        self.target.send(data)
      except:
        break
    log(&#39;PipeThread done&#39;)
class Forwarding(threading.Thread):
  def __init__(self, port, targethost, targetport):
    threading.Thread.__init__(self)
    self.targethost = targethost
    self.targetport = targetport
    self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    self.sock.bind((&#39;0.0.0.0&#39;, port))
    self.sock.listen(10)
  def run(self):
    while True:
      client_fd, client_addr = self.sock.accept()
      target_fd = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
      target_fd.connect((self.targethost, self.targetport))
      log(&#39;new connect&#39;)
      # two direct pipe
      PipeThread(target_fd, client_fd).start()
      PipeThread(client_fd, target_fd).start()
if __name__ == &#39;__main__&#39;:
  print &#39;Starting&#39;
  import sys
  try:
    port = int(sys.argv[1])
    targethost = sys.argv[2]
    try: targetport = int(sys.argv[3])
    except IndexError: targetport = port
  except (ValueError, IndexError):
    print &#39;Usage: %s port targethost [targetport]&#39; % sys.argv[0]
    sys.exit(1)
  #sys.stdout = open(&#39;forwaring.log&#39;, &#39;w&#39;)
  Forwarding(port, targethost, targetport).start()

Copy after login