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程序设计有所帮助。