首頁 > 後端開發 > Python教學 > Python實作TCP/IP協定下的連接埠轉送及重定向範例

Python實作TCP/IP協定下的連接埠轉送及重定向範例

高洛峰
發布: 2016-10-17 16:34:32
原創
3100 人瀏覽過

首先,我們用webpy寫一個簡單的網站,監聽8080端口,回到「Hello, EverET.org」的頁面。

然後我們使用我們的forwarding.py,在80埠和8080埠中間建立兩個通訊管道用於雙向通訊。

此時,我們透過80連接埠存取我們的伺服器。

瀏覽器得到:

Python實作TCP/IP協定下的連接埠轉送及重定向範例

然後,我們在forwarding.py的輸出結果中可以看到瀏覽器和webpy之間的通訊內容。

Python實作TCP/IP協定下的連接埠轉送及重定向範例

代碼:

#!/usr/bin/env python
import sys, socket, time, threading
loglock = threading.Lock()
def log(msg):
  loglock.acquire()
  try:
    print '[%s]: \n%s\n' % (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('PipeThread done')
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(('0.0.0.0', 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('new connect')
      # two direct pipe
      PipeThread(target_fd, client_fd).start()
      PipeThread(client_fd, target_fd).start()
if __name__ == '__main__':
  print 'Starting'
  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 'Usage: %s port targethost [targetport]' % sys.argv[0]
    sys.exit(1)
  #sys.stdout = open('forwaring.log', 'w')
  Forwarding(port, targethost, targetport).start()
登入後複製


來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板