本文实例讲述了python实现简单的tcp代理服务器的方法，分享给大家供大家参考。
具体实现代码如下：
# -*- coding: utf-8 -*-
”’
filename:rtcp.py
@desc:
利用python的socket端口转发，用于远程维护
如果连接不到远程，会sleep 36s，最多尝试200(即两小时)
@usage:
./rtcp.py stream1 stream2
stream为：l:port或c:host:port
l:port表示监听指定的本地端口
c:host:port表示监听远程指定的端口
@author: watercloud, zd, knownsec team
@web: www.knownsec.com, blog.knownsec.com
@date: 2009-7
”’
import socket
import sys
import threading
import time
streams = [none, none] # 存放需要进行数据转发的两个数据流（都是socketobj对象）
debug = 1 # 调试状态 0 or 1
def _usage():
print ‘usage: ./rtcp.py stream1 stream2\nstream : l:port or c:host:port’
def _get_another_stream(num):
”’
从streams获取另外一个流对象，如果当前为空，则等待
”’
if num == 0:
num = 1
elif num == 1:
num = 0
else:
raise “error”
while true:
if streams[num] == ‘quit’:
print(“can’t connect to the target, quit now!”)
sys.exit(1)
if streams[num] != none:
return streams[num]
else:
time.sleep(1)
def _xstream(num, s1, s2):
”’
交换两个流的数据
num为当前流编号,主要用于调试目的，区分两个回路状态用。
”’
try:
while true:
#注意，recv函数会阻塞，直到对端完全关闭（close后还需要一定时间才能关闭，最快关闭方法是shutdow）
buff = s1.recv(1024)
if debug > 0:
print num,”recv”
if len(buff) == 0: #对端关闭连接，读不到数据
print num,”one closed”
break
s2.sendall(buff)
if debug > 0:
print num,”sendall”
except :
print num,”one connect closed.”
try:
s1.shutdown(socket.shut_rdwr)
s1.close()
except:
pass
try:
s2.shutdown(socket.shut_rdwr)
s2.close()
except:
pass
streams[0] = none
streams[1] = none
print num, “closed”
def _server(port, num):
”’
处理服务情况,num为流编号（第0号还是第1号）
”’
srv = socket.socket(socket.af_inet, socket.sock_stream)
srv.bind((‘0.0.0.0′, port))
srv.listen(1)
while true:
conn, addr = srv.accept()
print “connected from:”, addr
streams[num] = conn # 放入本端流对象
s2 = _get_another_stream(num) # 获取另一端流对象
_xstream(num, conn, s2)
def _connect(host, port, num):
”’ 处理连接，num为流编号（第0号还是第1号）
@note: 如果连接不到远程，会sleep 36s，最多尝试200(即两小时)
”’
not_connet_time = 0
wait_time = 36
try_cnt = 199
while true:
if not_connet_time > try_cnt:
streams[num] = ‘quit’
print(‘not connected’)
return none
conn = socket.socket(socket.af_inet, socket.sock_stream)
try:
conn.connect((host, port))
except exception, e:
print (‘can not connect %s:%s!’ % (host, port))
not_connet_time += 1
time.sleep(wait_time)
continue
print “connected to %s:%i” % (host, port)
streams[num] = conn #放入本端流对象
s2 = _get_another_stream(num) #获取另一端流对象
_xstream(num, conn, s2)
if __name__ == ‘__main__’:
if len(sys.argv) != 3:
_usage()
sys.exit(1)
tlist = [] # 线程列表，最终存放两个线程对象
targv = [sys.argv[1], sys.argv[2] ]
for i in [0, 1]:
s = targv[i] # stream描述 c:ip:port 或 l:port
sl = s.split(‘:’)
if len(sl) == 2 and (sl[0] == ‘l’ or sl[0] == ‘l’): # l:port
t = threading.thread(target=_server, args=(int(sl[1]), i))
tlist.append(t)
elif len(sl) == 3 and (sl[0] == ‘c’ or sl[0] == ‘c’): # c:host:port
t = threading.thread(target=_connect, args=(sl[1], int(sl[2]), i))
tlist.append(t)
else:
_usage()
sys.exit(1)
for t in tlist:
t.start()
for t in tlist:
t.join()
sys.exit(0)
完整实例代码点击此处本站下载。
希望本文所述对大家的python程序设计有所帮助。