实验python tcp-proxy无法正常工作

时间:2012-12-11 13:11:34

标签: python sockets tcp

我只是尝试实现一个像代理一样的python应用程序(用于端口22) 代码如下:

import socket
import select
import code
inputSock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
inputSock.bind( ('',9999) )
inputSock.listen(5)
socks = {} #a dictionary!

#Now, wait for an input!
while 1:
    #accept new conns
    (sread,swrite,sexec) = select.select([inputSock],[],[])
    for sock in sread:
        print("Got an input!")
        newsock,(remhost,remport) = sock.accept()
        sendsock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
        try:
            print("Now connecting to the centOS...")
            sendsock.connect( ('202.131.30.12',80))
        except:
            print("ERROR: cannot connect to host 192.168.52.128:22")
        print("successfully connected to Cent. Creating new socks")
        socks[id(sendsock)] = newsock
        socks[id(newsock)] = sendsock
        print("socks created!")
    #recv from socks, etc.
    (sread,swrite,sexec) = select.select(socks.values(),[],[])
    for sock in sread:
        print("got input!")
        data = sock.recv(700)
        print(data)
        if(data == ""):
            #remove that sock.
            id1 = id(sock)
            id2 = id(socks[id1])
            socks.pop(id1)
            socks.pop(id2)
            print("removed empty conns")
        else:
            #code.interact(globals=globals())
            sendSock = socks[id(sock)]
            sendSock.send(data)
            print("data sent!")

现在,当我运行脚本并将浏览器指向127.0.0.1:9999时,我的浏览器将无法获取网页(在某些“永远加载网页”状态)。这是我从python提示中得到的: http://imgur.com/CffOY

你能帮我找到我错的地方吗?修复这个实验性实施? (或者给我一些链接以获得一个想法?) 谢谢:))

1 个答案:

答案 0 :(得分:0)

您已经从浏览器中读取了一些(可能是部分的)HTTP请求数据,并将其发送到另一个套接字上的服务器。然后外部循环将你带回监听套接字上的select,阻塞等待新的连接,所以你被卡住了。

将对select的两次调用合并为一个。只需为侦听套接字做一个特殊情况。