[Python-Twisted] Twisted入門之端口轉發服務器 .

Twisted 是Python界很有名的一個基於異步事件的網絡I/O框架，性能棒棒的，經歷過BT的考驗。本人垂涎很久了，於是寫了一個端口轉發服務器，純練手~~~

 

需求：將windows的遠程桌面做一個端口轉發。

即：有三台機器分別為A B C.在C上打開遠程桌面服務，開啟3389端口。

      在B上運行端口轉發程序，將發往B的1099端口的數據發送到C的3389.

     這樣在A上通過遠程桌面客戶端訪問B的1099端口就可以遠程訪問C的機器。

Understand?

let's go !!

Code  ：

from twisted.internet.protocol import Protocol,ClientCreator

from twisted.internet import reactor
from twisted.protocols.basic import LineReceiver
from twisted.internet.protocol import Factory,ClientFactory
class Transfer(Protocol):
        def __init__(self):
                pass
        def connectionMade(self):
                c = ClientCreator(reactor,Clienttransfer)
                c.connectTCP("10.61.1.243",3389).addCallback(self.set_protocol)
                self.transport.pauseProducing()

        def set_protocol(self,p):
                self.server = p
                p.set_protocol(self)

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

        def connectionLost(self,reason):
                self.transport.loseConnection()
                self.server.transport.loseConnection()

class Clienttransfer(Protocol):
        def __init__(self):
                pass

        def set_protocol(self,p):
                self.server = p
                self.server.transport.resumeProducing()
                pass
        def dataReceived(self,data):
                self.server.transport.write(data)
                pass

factory = Factory()
factory.protocol = Transfer
reactor.listenTCP(1099,factory)
reactor.run()
寫得比較倉促紅色代碼為轉發的目的地址以及端口。

小測試了下。呵呵還行 遠程桌面使用正常。我這個的實現比較惡，毫無構架擴展性而言,慢慢來 呵呵。

這個程序挺適合入門練手的，包括了服務端和客戶端的寫法，其中注意綠色標記的地方，我剛開始沒有注意這個，

導致 self.server.transport.write(data)中的server未初始化就開始寫數據。

o~了。。

寫完后看了看twisted的源碼，發現在twisted.protocols.base 下有一個portforword.py。

貼出來：
# Copyright (c) 2001-2004 Twisted Matrix Laboratories.
# See LICENSE for details.

"""
A simple port forwarder.
"""

# Twisted imports
from twisted.internet import protocol
from twisted.python import log

class Proxy(protocol.Protocol):
    noisy = True

    peer = None

    def setPeer(self, peer):
        self.peer = peer

    def connectionLost(self, reason):
        if self.peer is not None:
            self.peer.transport.loseConnection()
            self.peer = None
        elif self.noisy:
            log.msg("Unable to connect to peer: %s" % (reason,))

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

class ProxyClient(Proxy):
    def connectionMade(self):
        self.peer.setPeer(self)
        # We're connected, everybody can read to their hearts content.
        self.peer.transport.resumeProducing()

class ProxyClientFactory(protocol.ClientFactory):

    protocol = ProxyClient

    def setServer(self, server):
        self.server = server

    def buildProtocol(self, *args, **kw):
        prot = protocol.ClientFactory.buildProtocol(self, *args, **kw)
        prot.setPeer(self.server)
        return prot

    def clientConnectionFailed(self, connector, reason):
        self.server.transport.loseConnection()

class ProxyServer(Proxy):

    clientProtocolFactory = ProxyClientFactory

    def connectionMade(self):
        # Don't read anything from the connecting client until we have
        # somewhere to send it to.
        self.transport.pauseProducing()

        client = self.clientProtocolFactory()
        client.setServer(self)

        from twisted.internet import reactor
        reactor.connectTCP(self.factory.host, self.factory.port, client)

class ProxyFactory(protocol.Factory):
    """Factory for port forwarder."""

    protocol = ProxyServer

    def __init__(self, host, port):
        self.host = host
        self.port = port