/usr/share/pyshared/txzookeeper/tests/proxy.py is in python-txzookeeper 0.9.8-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | #
# Copyright (C) 2010-2011 Canonical Ltd. All Rights Reserved
#
# This file is part of txzookeeper.
#
# Authors:
# Kapil Thangavelu
#
# txzookeeper is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# txzookeeper is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with txzookeeper. If not, see <http://www.gnu.org/licenses/>.
#
from twisted.protocols import portforward
class Blockable(object):
_blocked = None
def set_blocked(self, value):
value = bool(value)
self._blocked = value
if self.transport and not self._blocked:
self.transport.resumeProducing()
class ProxyClient(portforward.ProxyClient, Blockable):
def dataReceived(self, data):
if self._blocked:
return
portforward.ProxyClient.dataReceived(self, data)
def setServer(self, server):
server.set_blocked(self._blocked)
super(ProxyClient, self).setServer(server)
def connectionMade(self):
self.peer.setPeer(self)
if not self._blocked:
# The server waits till the client is connected
self.peer.transport.resumeProducing()
else:
self.transport.pauseProducing()
class ProxyClientFactory(portforward.ProxyClientFactory):
protocol = ProxyClient
class ProxyServer(portforward.ProxyServer, Blockable):
clientProtocolFactory = ProxyClientFactory
def dataReceived(self, data):
if self._blocked:
return
portforward.ProxyServer.dataReceived(self, data)
class ProxyFactory(portforward.ProxyFactory):
protocol = ProxyServer
instance = _blocked = False
def lose_connection(self):
"""Terminate both ends of the proxy connection."""
if self.instance:
self.instance.transport.loseConnection()
if self.instance.peer:
self.instance.peer.transport.loseConnection()
def set_blocked(self, value):
self._blocked = bool(value)
if self.instance:
self.instance.set_blocked(self._blocked)
if self.instance.peer:
self.instance.peer.set_blocked(self._blocked)
def buildProtocol(self, addr):
# Track last protocol used, on reconnect any pauses are disabled.
self.instance = portforward.ProxyFactory.buildProtocol(self, addr)
# Propogate the value, the client will aggressively try to
# reconnect else.
self.instance.set_blocked(self._blocked)
return self.instance
|