Python实现简单的代理服务器

本文实例讲述了Python实现简单的代理服务器。分享给大家供大家参考。具体如下：

具备简单的管理功能，运行后 telnet localhost 9000 端口可以进行管理主要功能就是做包转发，如果有一个桥服务器，可以用来外网访问内网用，还是很管用的

    #/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("""-------------------------------------------\r
    exit\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()

希望本文所述对大家的Python程序设计有所帮助。