Python簡單實現的代理服務器端口映射功能示例

更新時間：2018年04月08日 14:21:26 作者：chengqiuming

這篇文章主要介紹了Python簡單實現的代理服務器端口映射功能,結合實例形式分析了Python模擬服務器、代理服務器及客戶端訪問的相關操作技巧,需要的朋友可以參考下

本文實例講述了Python簡單實現的代理服務器端口映射功能。分享給大家供大家參考，具體如下：

一代碼

1、模擬服務端代碼

import sys
import socket
import threading
#回復消息，原樣返回
def replyMessage(conn):
  while True:
    data = conn.recv(1024)
    conn.send(data)
    if data.decode().lower() == 'bye':
      break
  conn.close()
def main():
  sockScr = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  sockScr.bind(('', port))
  sockScr.listen(200)
  while True:
    try:
      conn, addr = sockScr.accept()
      #只允許特定主機訪問本服務器
      if addr[0] != onlyYou:
        conn.close()
        continue
      #創(chuàng)建并啟動線程
      t = threading.Thread(target=replyMessage, args=(conn,))
      t.start()
    except:
      print('error')
if __name__ == '__main__':
  try:
    #獲取命令行參數
    port = int(sys.argv[1])
    onlyYou = sys.argv[2]
    main()
  except:
    print('Must give me a number as port')

2、模擬代理服務器代碼

import sys
import socket
import threading
def middle(conn, addr):
  #面向服務器的Socket
  sockDst = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  sockDst.connect((ipServer,portServer))
  while True:
    data = conn.recv(1024).decode()
    print('收到客戶端消息：'+data)
    if data == '不要發(fā)給服務器':
      conn.send('該消息已被代理服務器過濾'.encode())
      print('該消息已過濾')
    elif data.lower() == 'bye':
      print(str(addr)+'客戶端關閉連接')
      break
    else:
      sockDst.send(data.encode())
      print('已轉發(fā)服務器')
      data_fromServer = sockDst.recv(1024).decode()
      print('收到服務器回復的消息：'+data_fromServer)
      if data_fromServer == '不要發(fā)給客戶端':
        conn.send('該消息已被代理服務器修改'.encode())
        print('消息已被篡改')
      else:
        conn.send(b'Server reply:'+data_fromServer.encode())
        print('已轉發(fā)服務器消息給客戶端')
  conn.close()
  sockDst.close()
def main():
  sockScr = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  sockScr.bind(('', portScr))
  sockScr.listen(200)
  print('代理已啟動')
  while True:
    try:
      conn, addr = sockScr.accept()
      t = threading.Thread(target=middle, args=(conn, addr))
      t.start()
      print('新客戶：'+str(addr))
    except:
      pass
if __name__ == '__main__':
  try:
    #(本機IP地址,portScr)<==>(ipServer,portServer)
    #代理服務器監(jiān)聽端口
    portScr = int(sys.argv[1])
    #服務器IP地址與端口號
    ipServer = sys.argv[2]
    portServer = int(sys.argv[3])
    main()
  except:
    print('Sth error')

3、模擬客戶端代碼

import sys
import socket
def main():
  sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  sock.connect((ip, port))
  while True:
    data = input('What do you want to ask:')
    sock.send(data.encode())
    print(sock.recv(1024).decode())
    if data.lower() == 'bye':
      break
  sock.close()
if __name__ == '__main__':
  try:
    #代理服務器的IP地址和端口號
    ip = sys.argv[1]
    port = int(sys.argv[2])
    main()
  except:
    print('Sth error')

二運行結果