Pythonでシンプルなプロキシサーバーを実現するプログラム

2014年7月11日

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 接続した.’ % addr
ts = socket.socket()

try:
ts.connect((s_ip,s_port))
except:
cs.close()
print '%s %d 閉じた.’ % 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(“""トランスサーバー 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()