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("x05x00"); 
 # 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 = "x05x00x00x01" 
 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 = "x05x07x00x01" # Command not supported 
 local = remote.getsockname() 
 reply += socket.inet_aton(local[0]) + struct.pack(">H", local[1])
 except socket.error: 
 # Connection refused 
 reply = 'x05x05x00x01x00x00x00x00x00x00' 
 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()