端口转发器

需要安装python2.5以上及twisted开发包
程序中的参数配置的函意是：126邮箱的中转服务器

1. #!/usr/bin/env python
2. # encoding: utf-8
3. #
4. # ################################################################
5. #
6. # 作者：黄江斌 (hjbcn@126.com)
7. #
8. # 版本：v1.0.0
9. #
10. # 来源：http://blog.csdn.net/zerodspace
11. #
12. # 工作模型
13. #
14. #         客户端             目标主机
15. #            ^                  ^
16. #            |                  |
17. #            v                  v
18. #         接收器  <--------> 转发器
19. #
20. # 映射表结构
21. #
22. #       ( "客户端IP", 本地端口, "目标主机IP", 目标主机端口 )
23. #
24. # ################################################################
25. from twisted.internet import protocol
26. from twisted.internet import reactor
28. TCPPortRedirectMap = [
29.     ( "*", 28, "smtp.126.com", 25 ),
30.     ( "*", 29, "pop.126.com", 110 ),
31. ]
33. class TCPPortRedirectorProtocol( protocol.Protocol ):
34.     def dataReceived( self, data ):
35.         # 所有接收到的数据都转发到“接收器”
36.         self.factory.recver.transport.write( data )
37.         print data
39.     def connectionLost( self, reason ):
40.         # 目标主机关闭连接后
41.         # 通知“接收器”关闭连接
42.         self.factory.recver.transport.loseConnection()
43.         
44. class TCPPortRedirectorFactory( protocol.ClientFactory ):
45.     protocol = TCPPortRedirectorProtocol
46.     def __init__( self, recver ):
47.         # 保存“接收器”对象
48.         self.recver = recver
49.         self.connector = None
50.         
51.     def startedConnecting( self, connector ):
52.         # 保存“目标主机”与“转发器”的“连接对象”
53.         # 供外部使用
54.         self.connector = connector
56.     
57. class TCPPortRedirectProtocol( protocol.Protocol ):
58.     def connectionMade( self ):
59.         # 获取连接信息
60.         #
61.         # 远程主机IP
62.         # 本地端口
63.         remote = self.transport.getPeer().host
64.         localhostPort = self.transport.getHost().port
65.         # 查找映射表
66.         # 确定转发目的地
67.         redirectMap = None
68.         for map in TCPPortRedirectMap:
69.             if ( map[0] == remote or map[0] == "*" ) and map[1] == localhostPort:
70.                 redirectMap = map
71.                 break
72.         # 如果没有找到转发关系
73.         # 关闭连接
74.         if not redirectMap:
75.             self.transport.loseConnection()
76.         # 建立转发器
77.         self.redirector = TCPPortRedirectorFactory( self )
78.         reactor.connectTCP( redirectMap[2], redirectMap[3], self.redirector )
79.         
80.     def dataReceived( self, data ):
81.         self.redirector.connector.transport.write( data )
82.         print data
84.     def connectionLost( self, reason ):
85.         self.redirector.connector.transport.loseConnection()
86.         
87.         
88. class TCPPortRedirectFactory( protocol.Factory ):
89.     protocol = TCPPortRedirectProtocol
91. def TCPPortRedirectStart():
92.     ports = []
93.     for map in TCPPortRedirectMap:
94.         if map[1] not in ports:
95.             ports.append( map[1] )
96.     for port in ports:
97.         reactor.listenTCP( port, TCPPortRedirectFactory() )
98.     reactor.run()
100. if __name__ == "__main__":
101.     TCPPortRedirectStart()