Python实现简单的代理服务器

#/bin/env python #coding:utf-8 import socket,select,sys,time import thread s_list = [] def loop(cs,addr,s_ip,s_port):  print '%s %d connected.' % addr  ts = socket.socket()  try:   ts.connect((s_ip,s_port))  except:   cs.close()   print '%s %d closed.' % addr   sys.exit(0)  while True:   rl,wl,xl = select.select([cs.fileno(),ts.fileno()],[],[cs.fileno(),ts.fileno()])   if len(xl) > 0:    cs.close()    ts.close()    print '%s %d closed.' % addr    sys.exit(0)   if len(rl) > 0:    if rl[0] == cs.fileno():     rs = ts     ws = cs    else:     rs = cs     ws = ts    try:     buffer = ws.recv(10000)     if len(buffer) == 0:      raise     rs.send(buffer)    except:     rs.close()     ts.close()     print '%s %d closed.' % addr     sys.exit(0) def mainserver(l_port,s_ip,s_port):  global s_list  try:   ss = socket.socket()   ss.bind(('0.0.0.0',l_port))   ss.listen(10)   s_list.append((l_port,s_ip,s_port))    except:   sys.exit(0)  while True:   cs,addr = ss.accept()   thread.start_new_thread(loop,(cs,addr,s_ip,s_port)) def manager(l_port):  global start,s_list  ss = socket.socket()  ss.bind(('0.0.0.0',l_port))  ss.listen(10)  while True:   cs,addr = ss.accept()   cs.send("""trans server 1.0/r/ntype 'help' to get help/r/n""")  buffer = ''   while True:    buf = cs.recv(10000)    if len(buf) == 0:     cs.close()     break    if buf[-1] not in ('/r','/n'):     buffer += buf     continue    buffer += buf    cmd = buffer.strip()    buffer = ''    if cmd == 'exit':     cs.close()     break    elif cmd == 'stop':     start = 0     cs.close()     sys.exit(0)    elif cmd == 'list':     b = ''     for l in s_list:      b += '%4d %s:%d/r/n' % l     if len(b) > 0:      cs.send(b)    elif cmd in ('help','?'):     cs.send("""-------------------------------------------/rexit/r  exit telnet/r start localport serverip:serverport/r  start a new server/r list/r  list all server/r -------------------------------------------/r """)    else:     cmds = cmd.split(" ",1)     if len(cmds) > 1 and cmds[0] == 'start':      args = cmds[1].strip().split(" ",1)      if len(args) != 2:       cs.send('start localport serverip:serverport/r/n')       continue      arg = args[1].split(":",1)      if len(arg) != 2:       cs.send('start localport serverip:serverport/r/n')       continue      try:       l_port = int(args[0])       s_ip = arg[0]       s_port = int(arg[1])      except:       cs.send('start localport serverip:serverport/r/n')       continue      thread.start_new_thread(mainserver,(l_port,s_ip,s_port))      cs.send('start OK!/r/n')     else:      cs.send('no command [%s]/r/n' % cmd)      continue def main():  global start  if len(sys.argv) == 3:   try:    l_port = int(sys.argv[1])    s_ip,s_port = sys.argv[2].split(":")    s_port = int(s_port)    thread.start_new_thread(mainserver,(l_port,s_ip,s_port))   except:    pass  start = 1  thread.start_new_thread(manager,(9000,))  while start:   time.sleep(1) if __name__ == '__main__':  start = 0  main()