python扭曲了中间人的实现

时间:2013-03-26 15:14:36

标签: python twisted

我需要的是一种中间人实现:我需要一个从客户端接收连接的服务器(具有不同长度的二进制数据)并将流转发到它连接的服务器(充当客户端) ),然后将数据从其连接的服务器发回给客户端。

它实际上在客户端和服务器之间工作,并传递它们交换的数据(这是一个流,因此它不断从一侧传送并发送到另一侧)。
服务器是静态的,所以它总是一样的,它的地址甚至可以硬编码;但是,当客户端断开连接时,该服务器还必须断开与“真实”服务器的连接。

我一直在环顾四周,但找不到解决方案或这样一个简单问题的例子。

我实际制作的代码实际上是有效的,但我还没有设法找到如何将引用放入服务器部分,说明“这是你的指定客户端”,或者进入客户端说“这是你的服务器”。这是我的代码:

#!/usr/bin/env python

from twisted.internet import protocol, reactor
from twisted.protocols import basic

client = None
server = None

class ServerProtocol(protocol.Protocol):
    def connectionMade(self):
        global server
        factory = protocol.ClientFactory()
        factory.protocol = ClientProtocol
        server = self
        reactor.connectTCP('localhost', 1324, factory)

    def dataReceived(self, data):
        global client
        client.transport.write(data)

class ClientProtocol(protocol.Protocol):
    def connectionMade(self):
        global client
        # Here's the instance of the client
        client = self

    def dataReceived(self, data):
        global server
        server.transport.write(data)

def main():
    import sys
    from twisted.python import log
    log.startLogging(sys.stdout)
    factory = protocol.ServerFactory()
    factory.protocol = ServerProtocol

    # Here's the instance of the server
    server = ServerProtocol

    reactor.listenTCP(2593, factory)
    reactor.run()

if __name__ == '__main__':
    main()

现在,重点是实例不能包含在全局对象中,应该放在两个类中:如何?

2 个答案:

答案 0 :(得分:4)

我自己设法解决了这个问题,并且为了将来的参考(或帮助其他有这个问题的人),这里是我用来解决它的代码。
<登记/> 我认为我的解决方案和jedwards友善的工作方式都是如此;现在我只需要更多地学习自己,以确保我所做的是正确的:这是我第一次使用Twisted框架并研究其他人的解决方案是学习新东西的方法! :)

#!/usr/bin/env python

from twisted.internet import protocol, reactor
from twisted.protocols import basic

class ServerProtocol(protocol.Protocol):
    def __init__(self):
        self.buffer = None
        self.client = None

    def connectionMade(self):
        factory = protocol.ClientFactory()
        factory.protocol = ClientProtocol
        factory.server = self

        reactor.connectTCP('gameserver16.gamesnet.it', 2593, factory)

    def dataReceived(self, data):
        if (self.client != None):
            self.client.write(data)
        else:
            self.buffer = data

    def write(self, data):
        self.transport.write(data)
        print 'Server: ' + data.encode('hex')

class ClientProtocol(protocol.Protocol):
    def connectionMade(self):
        self.factory.server.client = self
        self.write(self.factory.server.buffer)
        self.factory.server.buffer = ''

    def dataReceived(self, data):
        self.factory.server.write(data)

    def write(self, data):
        self.transport.write(data)
        print 'Client: ' + data.encode('hex')

def main():
    import sys
    from twisted.python import log

    log.startLogging(sys.stdout)

    factory = protocol.ServerFactory()
    factory.protocol = ServerProtocol

    reactor.listenTCP(2593, factory)
    reactor.run()

if __name__ == '__main__':
    main()

答案 1 :(得分:3)

考虑这种方法

#!/usr/bin/env python

import sys
from twisted.internet import reactor
from twisted.internet.protocol import ServerFactory, ClientFactory, Protocol
from twisted.protocols import basic
from twisted.python import log

LISTEN_PORT = 2593
SERVER_PORT = 1234


class ServerProtocol(Protocol):
    def connectionMade(self):
        reactor.connectTCP('localhost', SERVER_PORT, MyClientFactory(self))

    def dataReceived(self, data):
        self.clientProtocol.transport.write(data)

class ClientProtocol(Protocol):
    def connectionMade(self):
        # Pass ServerProtocol a ref. to ClientProtocol
        self.serverProtocol.clientProtocol = self;  

    def dataReceived(self, data):
        self.serverProtocol.transport.write(data)

class MyServerFactory(ServerFactory):
    protocol = ServerProtocol
    def buildProtocol(self, addr):
        # Create ServerProtocol
        p = ServerFactory.buildProtocol(self, addr)
        return p

class MyClientFactory(ClientFactory):
    protocol = ClientProtocol
    def __init__(self, serverProtocol_):
        self.serverProtocol = serverProtocol_

    def buildProtocol(self, addr):
        # Create ClientProtocol
        p = ClientFactory.buildProtocol(self,addr)
        # Pass ClientProtocol a ref. to ServerProtocol
        p.serverProtocol = self.serverProtocol
        return p

def main():
    log.startLogging(sys.stdout)

    reactor.listenTCP(LISTEN_PORT, MyServerFactory())
    reactor.run()

if __name__ == '__main__':
    main()

ServerProtcol实例将自身的引用传递给MyClientFactory构造函数,然后设置它告诉ClientProtcol它与哪个ServerProtocol实例关联。

类似地,当建立ClientProtocol连接时,它使用它对ServerProtocol的引用来告诉ServerProtocol要使用的ClientProtocol。

注意:此代码中没有错误检查,因此如果出现问题,您可能会遇到有关NoneType的错误(例如,如果真实服务器没有收听)。

重要的是:

reactor.connectTCP('localhost', SERVER_PORT, MyClientFactory(self))
#...
def __init__(self, serverProtocol_):
    self.serverProtocol = serverProtocol_

在这里,您将对ServerProtocol的引用传递给MyClientFactory构造函数。它将此引用存储在成员变量中。这样做是为了在客户端工厂创建ClientProtocol时,它可以传递引用:

# Pass ClientProtocol a ref. to ServerProtocol
p.serverProtocol = self.serverProtocol

然后,一旦从脚本连接到真实的服务器,就会发生相反的情况。 ClientProtocol为ServerProtocol提供了对自身的引用:

# Pass ServerProtocol a ref. to ClientProtocol
self.serverProtocol.clientProtocol = self;

最后,两个协议使用彼此存储的引用在收到数据时发送数据:

def dataReceived(self, data):
    self.clientProtocol.transport.write(data)
#...
def dataReceived(self, data):
    self.serverProtocol.transport.write(data)