Python写的Socks5协议代理服务器

直接上代码：

#!/usr/bin/python
# Filename s5.py
# Python Dynamic Socks5 Proxy
# Usage: python s5.py 1080
# Background Run: nohup python s5.py 1080 &

import socket, sys, select, SocketServer, struct, time

class ThreadingTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer): pass
class Socks5Server(SocketServer.StreamRequestHandler):
 def handle_tcp(self, sock, remote):
   fdset = [sock, remote]
   while True:
     r, w, e = select.select(fdset, [], [])
     if sock in r:
       if remote.send(sock.recv(4096)) <= 0: break
     if remote in r:
       if sock.send(remote.recv(4096)) <= 0: break
 def handle(self):
   try:
     pass # print 'from ', self.client_address nothing to do.
     sock = self.connection
     # 1. Version
     sock.recv(262)
     sock.send("\x05\x00");
     # 2. Request
     data = self.rfile.read(4)
     mode = ord(data[1])
     addrtype = ord(data[3])
     if addrtype == 1:    # IPv4
       addr = socket.inet_ntoa(self.rfile.read(4))
     elif addrtype == 3:   # Domain name
       addr = self.rfile.read(ord(sock.recv(1)[0]))
     port = struct.unpack('>H', self.rfile.read(2))
     reply = "\x05\x00\x00\x01"
     try:
       if mode == 1: # 1. Tcp connect
         remote = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         remote.connect((addr, port[0]))
         pass # print 'To', addr, port[0] nothing do to.
       else:
         reply = "\x05\x07\x00\x01" # Command not supported
       local = remote.getsockname()
       reply += socket.inet_aton(local[0]) + struct.pack(">H", local[1])
     except socket.error:
       # Connection refused
       reply = '\x05\x05\x00\x01\x00\x00\x00\x00\x00\x00'
     sock.send(reply)
     # 3. Transfering
     if reply[1] == '\x00': # Success
       if mode == 1:  # 1. Tcp connect
         self.handle_tcp(sock, remote)
   except socket.error:
     pass #print 'error' nothing to do .
   except IndexError:
     pass
def main():
 filename = sys.argv[0];
 if len(sys.argv)<2:
   print 'usage: ' + filename + ' port'
   sys.exit()
 socks_port = int(sys.argv[1]);  
 server = ThreadingTCPServer(('', socks_port), Socks5Server)
 print 'bind port: ％d' ％ socks_port + ' ok!'
 server.serve_forever()
if __name__ == '__main__':
 main()