/usr/lib/python2.7/dist-packages/foolscap/pb.py is in python-foolscap 0.13.1-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 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 | # -*- test-case-name: foolscap.test.test_pb -*-
import os.path, weakref, binascii, re
from warnings import warn
from zope.interface import implements
from twisted.internet import (reactor, defer, protocol, error, interfaces,
endpoints)
from twisted.application import service
from twisted.python.failure import Failure
from twisted.python.deprecate import deprecated
from twisted.python.versions import Version
from foolscap import ipb, base32, negotiate, broker, eventual, storage
from foolscap import connection, util, info
from foolscap.connections import tcp
from foolscap.referenceable import SturdyRef
from .furl import BadFURLError
from foolscap.tokens import PBError, BananaError, WrongTubIdError, \
WrongNameError, NoLocationError
from foolscap.reconnector import Reconnector
from foolscap.logging import log as flog
from foolscap.logging import log
from foolscap.logging import publish as flog_publish
from foolscap.logging.log import UNUSUAL
from foolscap import crypto
class Listener(protocol.ServerFactory, service.Service):
"""I am responsible for a single listening port, which connects to a
single Tub. I listen on an Endpoint, and can be constructed with either
the Endpoint, or a string (which I will pass to serverFromString())."""
# this also serves as the ServerFactory
def __init__(self, tub, endpoint_or_description, _test_options={},
negotiationClass=negotiate.Negotiation):
assert isinstance(tub, Tub)
self._tub = tub
if interfaces.IStreamServerEndpoint.providedBy(endpoint_or_description):
self._ep = endpoint_or_description
elif isinstance(endpoint_or_description, str):
self._ep = endpoints.serverFromString(reactor,
endpoint_or_description)
else:
raise TypeError("I require an endpoint, or a string description that can be turned into one")
self._lp = None
self._test_options = _test_options
self._negotiationClass = negotiationClass
self._redirects = {}
def startService(self):
service.Service.startService(self)
d = self._ep.listen(self)
def _listening(lp):
self._lp = lp
d.addCallback(_listening)
def stopService(self):
service.Service.stopService(self)
if self._lp:
return self._lp.stopListening()
@deprecated(Version("Foolscap", 0, 12, 0),
# "please use .."
"pre-allocated port numbers")
def getPortnum(self):
"""When this Listener was created with a port string of '0' or
'tcp:0' (meaning 'please allocate me something'), and if the Listener
is active (it is attached to a Tub which is in the 'running' state),
this method will return the port number that was allocated. This is
useful for the following pattern::
t = Tub()
l = t.listenOn('tcp:0')
t.setLocation('localhost:%d' % l.getPortnum())
"""
assert self._lp
return self._lp.getHost().port
def __repr__(self):
return ("<Listener at 0x%x on %s with tub %s>" %
(abs(id(self)), str(self._ep), str(self._tub.tubID)))
def addRedirect(self, tubID, location):
assert tubID is not None
self._redirects[tubID] = location
def removeRedirect(self, tubID):
del self._redirects[tubID]
def buildProtocol(self, addr):
"""Return a Broker attached to me (as the service provider).
"""
lp = log.msg("%s accepting connection from %s" % (self, addr),
addr=(addr.host, addr.port),
facility="foolscap.listener")
proto = self._negotiationClass(logparent=lp)
ci = info.ConnectionInfo()
ci._set_listener_description(self._describe())
ci._set_listener_status("negotiating")
proto.initServer(self, ci)
proto.factory = self
return proto
def lookupTubID(self, tubID):
tub = None
if tubID == self._tub.tubID:
tub = self._tub
return (tub, self._redirects.get(tubID))
def _describe(self):
desc = "Listener"
if self._lp:
desc += " on %s" % str(self._lp.getHost())
return desc
def generateSwissnumber(bits):
bytes = os.urandom(bits/8)
return base32.encode(bytes)
class Tub(service.MultiService):
"""I am a presence in the PB universe, also known as a Tub.
I am a Service (in the twisted.application.service.Service sense),
so you either need to call my startService() method before using me,
or setServiceParent() me to a running service.
This is the primary entry point for all PB-using applications, both
clients and servers.
I am known to the outside world by a base URL, which may include
authentication information (a yURL). This is my 'TubID'.
I contain Referenceables, and manage RemoteReferences to Referenceables
that live in other Tubs.
@param certData: if provided, use it as a certificate rather than
generating a new one. This is a PEM-encoded
private/public keypair, as returned by Tub.getCertData()
@param certFile: if provided, the Tub will store its certificate in
this file. If the file does not exist when the Tub is
created, the Tub will generate a new certificate and
store it here. If the file does exist, the certificate
will be loaded from this file.
The simplest way to use the Tub is to choose a long-term
location for the certificate, use certFile= to tell the
Tub about it, and then let the Tub manage its own
certificate.
You may provide certData, or certFile, (or neither), but
not both.
@param _test_options: a dictionary of options that can influence
connection connection negotiation. Currently
defined keys are:
- debug_slow: if True, wait half a second between
each negotiation response
@ivar brokers: maps TubIDs to L{Broker} instances
@ivar referenceToName: maps Referenceable to a name
@ivar nameToReference: maps name to Referenceable
@type tubID: string
@ivar tubID: a global identifier for this Tub, possibly including
authentication information, hash of SSL certificate
"""
implements(ipb.ITub)
unsafeTracebacks = True # TODO: better way to enable this
logLocalFailures = False
logRemoteFailures = False
debugBanana = False
NAMEBITS = 160 # length of swissnumber for each reference
TUBIDBITS = 16 # length of non-crypto tubID
negotiationClass = negotiate.Negotiation
brokerClass = broker.Broker
keepaliveTimeout = 4*60 # ping when connection has been idle this long
disconnectTimeout = None # disconnect after this much idle time
tubID = None
def __init__(self, certData=None, certFile=None, _test_options={}):
service.MultiService.__init__(self)
self.setup(_test_options)
if certFile:
self.setupEncryptionFile(certFile)
else:
self.setupEncryption(certData)
def __repr__(self):
return "<Tub id=%s>" % self.tubID
def setupEncryptionFile(self, certFile):
try:
certData = open(certFile, "rb").read()
except EnvironmentError:
certData = None
self.setupEncryption(certData)
if certData is None:
f = open(certFile, "wb")
f.write(self.getCertData())
f.close()
def setupEncryption(self, certData):
if certData:
cert = crypto.loadCertificate(certData)
else:
cert = self.createCertificate()
self.myCertificate = cert
self.tubID = crypto.digest32(cert.digest("sha1"))
def make_incarnation(self):
unique = binascii.b2a_hex(os.urandom(8))
# TODO: it'd be nice to have a sequential component, so incarnations
# could be ordered, but it requires disk space
sequential = None
self.incarnation = (unique, sequential)
self.incarnation_string = unique
def getIncarnationString(self):
return self.incarnation_string
def setup(self, _test_options):
self._test_options = _test_options
self.logger = flog.theLogger
self.listeners = []
self.locationHints = []
# duplicate-connection management
self.make_incarnation()
# the master_table records the master-seqnum we used for the last
# established connection with the given tubid. It only contains
# entries for which we were the master.
self.master_table = {} # k:tubid, v:seqnum
# the slave_table records the (master-IR,master-seqnum) pair for the
# last established connection with the given tubid. It only contains
# entries for which we were the slave.
self.slave_table = {} # k:tubid, v:(master-IR,seqnum)
# local Referenceables
self.nameToReference = weakref.WeakValueDictionary()
self.referenceToName = weakref.WeakKeyDictionary()
self.strongReferences = []
self.nameLookupHandlers = []
# remote stuff. Most of these use a TubRef as a dictionary key
self.tubConnectors = {} # maps TubRef to a TubConnector
self.waitingForBrokers = {} # maps TubRef to list of Deferreds
self.brokers = {} # maps TubRef to a Broker that connects to them
self.reconnectors = []
self._connectionHandlers = {"tcp": tcp.default()}
self._activeConnectors = []
self._pending_getReferences = [] # list of (d, furl) pairs
self._logport = None
self._logport_furl = None
self._logport_furlfile = None
self._log_gatherer_furls = []
self._log_gatherer_furlfile = None
self._log_gatherer_connectors = {} # maps furl to reconnector
self._handle_old_duplicate_connections = False
self._expose_remote_exception_types = True
self.accept_gifts = True
def setOption(self, name, value):
if name == "logLocalFailures":
# log (with log.err) any exceptions that occur during the
# execution of a local Referenceable's method, which is invoked
# on behalf of a remote caller. These exceptions are reported to
# the remote caller through their callRemote's Deferred as usual:
# this option enables logging on the callee's side (i.e. our
# side) as well.
#
# TODO: This does not yet include Violations which were raised
# because the inbound callRemote had arguments that didn't meet
# our specifications. But it should.
self.logLocalFailures = value
elif name == "logRemoteFailures":
# log (with log.err) any exceptions that occur during the
# execution of a remote Referenceabe's method, invoked on behalf
# of a local RemoteReference.callRemote(). These exceptions are
# reported to our local caller through the usual Deferred.errback
# mechanism: this enables logging on the caller's side (i.e. our
# side) as well.
self.logRemoteFailures = value
elif name == "keepaliveTimeout":
self.keepaliveTimeout = value
elif name == "disconnectTimeout":
self.disconnectTimeout = value
elif name == "logport-furlfile":
self.setLogPortFURLFile(value)
elif name == "log-gatherer-furl":
self.setLogGathererFURL(value)
elif name == "log-gatherer-furlfile":
self.setLogGathererFURLFile(value)
elif name == "bridge-twisted-logs":
assert value is not False, "cannot unbridge twisted logs"
if value is True:
return flog.bridgeLogsFromTwisted(self.tubID)
else:
# for tests, bridge logs from a specific twisted LogPublisher
return flog.bridgeLogsFromTwisted(self.tubID,
twisted_logger=value)
elif name == "handle-old-duplicate-connections":
if value is True:
value = 60
self._handle_old_duplicate_connections = int(value)
elif name == "expose-remote-exception-types":
self._expose_remote_exception_types = bool(value)
elif name == "accept-gifts":
self.accept_gifts = bool(value)
else:
raise KeyError("unknown option name '%s'" % name)
def removeAllConnectionHintHandlers(self):
self._connectionHandlers = {}
def addConnectionHintHandler(self, hint_type, handler):
assert ipb.IConnectionHintHandler.providedBy(handler)
self._connectionHandlers[hint_type] = handler
def setLogGathererFURL(self, gatherer_furl_or_furls):
assert not self._log_gatherer_furls
if isinstance(gatherer_furl_or_furls, basestring):
self._log_gatherer_furls.append(gatherer_furl_or_furls)
else:
self._log_gatherer_furls.extend(gatherer_furl_or_furls)
self._maybeConnectToGatherer()
def setLogGathererFURLFile(self, gatherer_furlfile):
assert not self._log_gatherer_furlfile
self._log_gatherer_furlfile = gatherer_furlfile
self._maybeConnectToGatherer()
def _maybeConnectToGatherer(self):
if not self.locationHints:
return
furls = []
if self._log_gatherer_furls:
furls.extend(self._log_gatherer_furls)
if self._log_gatherer_furlfile:
try:
# allow multiple lines
for line in open(self._log_gatherer_furlfile, "r").readlines():
furl = line.strip()
if furl:
furls.append(furl)
except EnvironmentError:
pass
for f in furls:
if f in self._log_gatherer_connectors:
continue
connector = self.connectTo(f, self._log_gatherer_connected)
self._log_gatherer_connectors[f] = connector
def _log_gatherer_connected(self, rref):
# we want the logport's furl to be nailed down now, so we'll use the
# right (persistent) name even if the user never calls
# tub.getLogPortFURL() directly.
ignored = self.getLogPortFURL()
del ignored
tubID = self.tubID
rref.callRemoteOnly('logport', tubID, self.getLogPort())
def getLogPort(self):
if not self.locationHints:
raise NoLocationError
return self._maybeCreateLogPort()
def _maybeCreateLogPort(self):
if not self._logport:
self._logport = flog_publish.LogPublisher(self.logger)
return self._logport
def setLogPortFURLFile(self, furlfile):
self._logport_furlfile = furlfile
self._maybeCreateLogPortFURLFile()
def _maybeCreateLogPortFURLFile(self):
if not self._logport_furlfile:
return
if not self.locationHints:
return
# getLogPortFURL() creates the logport-furlfile as a side-effect
ignored = self.getLogPortFURL()
del ignored
def getLogPortFURL(self):
if not self.locationHints:
raise NoLocationError
if self._logport_furl:
return self._logport_furl
furlfile = self._logport_furlfile
# the Tub must be running and configured (setLocation) by now
self._logport_furl = self.registerReference(self.getLogPort(),
furlFile=furlfile)
return self._logport_furl
def log(self, *args, **kwargs):
kwargs['tubID'] = self.tubID
return log.msg(*args, **kwargs)
def createCertificate(self):
return crypto.createCertificate()
def getCertData(self):
# the string returned by this method can be used as the certData=
# argument to create a new Tub with the same identity. TODO: actually
# test this, I don't know if dump/keypair.newCertificate is the right
# pair of methods.
return self.myCertificate.dumpPEM()
def setLocation(self, *hints):
"""Tell this service what its location is: a host:port description of
how to reach it from the outside world. You need to use this because
the Tub can't do it without help. If you do a
C{s.listenOn('tcp:1234')}, and the host is known as
C{foo.example.com}, then it would be appropriate to do::
s.setLocation('foo.example.com:1234')
You must set the location before you can register any references.
Tubs can have multiple location hints, just provide multiple
arguments. """
if self.locationHints:
raise PBError("Tub.setLocation() can only be called once")
self.locationHints = hints
self._maybeCreateLogPortFURLFile()
self._maybeConnectToGatherer()
@deprecated(Version("Foolscap", 0, 12, 0),
# "please use .."
"user-provided hostnames")
def setLocationAutomatically(self, *extra_addresses):
"""Determine one of this host's publically-visible IP addresses and
use it to set our location. This uses whatever source address would
be used to get to a well-known public host (A.ROOT-SERVERS.NET),
which is effectively the interface on which a default route lives.
This is neither very pretty (IP address instead of hostname) nor
guaranteed to work (it may very well be a 192.168 'private' address),
but for publically-visible hosts this will probably produce a useable
FURL.
This method returns a Deferred that will fire once the location is
actually established. Calls to registerReference() must be put off
until the location has been set. And of course, you must call
listenOn() before calling setLocationAutomatically()."""
# first, make sure the reactor is actually running, by using the
# eventual-send queue
d = eventual.fireEventually()
def _reactor_running(res):
assert self.running
# we can't use get_local_ip_for until the reactor is running
return util.get_local_ip_for()
d.addCallback(_reactor_running)
def _got_local_ip(local_address):
local_addresses = set(extra_addresses)
if local_address:
local_addresses.add(local_address)
local_addresses.add("127.0.0.1")
locations = set()
for l in self.getListeners():
portnum = l.getPortnum()
for addr in local_addresses:
locations.add("%s:%d" % (addr, portnum))
locations = list(locations)
locations.sort()
assert len(locations) >= 1
location = ",".join(locations)
self.setLocation(location)
d.addCallback(_got_local_ip)
return d
def listenOn(self, what, _test_options={}):
"""Start listening for connections.
@type what: string
@param what: a L{twisted.internet.endpoints.serverFromString} -style
description
@param _test_options: a dictionary of options that can influence
connection negotiation before the target Tub
has been determined
@return: The Listener object that was created. This can be used to
stop listening later on."""
if what in ("0", "tcp:0"):
warningString = ("Tub.listenOn('tcp:0') was deprecated "
"in Foolscap 0.12.0; please use pre-allocated "
"port numbers instead")
warn(warningString, DeprecationWarning, stacklevel=2)
if isinstance(what, str) and re.search(r"^\d+$", what):
warn("Tub.listenOn('12345') was deprecated "
"in Foolscap 0.12.0; please use qualified endpoint "
"descriptions like 'tcp:12345'",
DeprecationWarning, stacklevel=2)
what = "tcp:%s" % what
l = Listener(self, what, _test_options, self.negotiationClass)
self.listeners.append(l)
l.setServiceParent(self)
return l
def stopListeningOn(self, l):
# this returns a Deferred when the port is shut down
self.listeners.remove(l)
return l.disownServiceParent()
def getListeners(self):
"""Return the set of Listener objects that allow the outside world to
connect to this Tub."""
return self.listeners[:]
def getTubID(self):
return self.tubID
def getShortTubID(self):
return self.tubID[:4]
def getConnectionInfoForFURL(self, furl):
try:
tubref = SturdyRef(furl).getTubRef()
except (ValueError, BadFURLError):
return None # unparseable FURL
return self._getConnectionInfoForTubRef(tubref)
def _getConnectionInfoForTubRef(self, tubref):
if tubref in self.brokers:
return self.brokers[tubref].getConnectionInfo()
if tubref in self.tubConnectors:
return self.tubConnectors[tubref].getConnectionInfo()
return None # currently have no established or in-progress connection
def connectorStarted(self, c):
assert self.running
# TODO: why a list? shouldn't there only ever be one TubConnector?
self._activeConnectors.append(c)
def connectorFinished(self, c):
if c in self._activeConnectors:
self._activeConnectors.remove(c)
def startService(self):
service.MultiService.startService(self)
for d,sturdy in self._pending_getReferences:
d1 = eventual.fireEventually(sturdy)
d1.addCallback(self.getReference)
d1.addBoth(lambda res, d=d: d.callback(res))
del self._pending_getReferences
for rc in self.reconnectors:
eventual.eventually(rc.startConnecting, self)
def _tubsAreNotRestartable(self, *args, **kwargs):
raise RuntimeError("Sorry, but Tubs cannot be restarted.")
def _tubHasBeenShutDown(self, *args, **kwargs):
raise RuntimeError("Sorry, but this Tub has been shut down.")
def stopService(self):
# note that once you stopService a Tub, I cannot be restarted. (at
# least this code is not designed to make that possible.. it might be
# doable in the future).
assert self.running
self.startService = self._tubsAreNotRestartable
self.getReference = self._tubHasBeenShutDown
self.connectTo = self._tubHasBeenShutDown
# Tell everything to shut down now. We assume that it will stop
# twitching by the next tick, so Trial unit tests won't complain
# about a dirty reactor. We wait on a few things that might not
# behave.
dl = []
for rc in list(self.reconnectors):
rc.stopConnecting()
del self.reconnectors
for c in list(self._activeConnectors):
c.shutdown()
why = Failure(error.ConnectionDone("Tub.stopService was called"))
for b in self.brokers.values():
broker_disconnected = defer.Deferred()
dl.append(broker_disconnected)
b._notifyOnConnectionLost(
lambda d=broker_disconnected: d.callback(None)
)
b.shutdown(why, fireDisconnectWatchers=False)
d = defer.DeferredList(dl)
d.addCallback(lambda _: service.MultiService.stopService(self))
d.addCallback(eventual.fireEventually)
return d
def generateSwissnumber(self, bits):
return generateSwissnumber(bits)
def buildURL(self, name):
# TODO: IPv6 dotted-quad addresses have colons, but need to have
# host:port
hints = ",".join(self.locationHints)
return "pb://" + self.tubID + "@" + hints + "/" + name
def registerReference(self, ref, name=None, furlFile=None):
"""Make a Referenceable available to the outside world. A URL is
returned which can be used to access this object. This registration
will remain in effect (and the Tub will retain a reference to the
object to keep it meaningful) until explicitly unregistered, or the
Tub is shut down.
@type name: string (optional)
@param name: if provided, the object will be registered with this
name. If not, a random (unguessable) string will be
used.
@param furlFile: if provided, get the name from this file (if
it exists), and write the new FURL to this file.
If 'name=' is also provided, it is used for the
name, but the FURL is still written to this file.
@rtype: string
@return: the URL which points to this object. This URL can be passed
to Tub.getReference() in any Tub on any host which can reach this
one.
"""
if not self.locationHints:
raise NoLocationError("you must setLocation() before "
"you can registerReference()")
oldfurl = None
if furlFile:
try:
oldfurl = open(furlFile, "r").read().strip()
except EnvironmentError:
pass
if oldfurl:
sr = SturdyRef(oldfurl)
if name is None:
name = sr.name
if self.tubID != sr.tubID:
raise WrongTubIdError("I cannot keep using the old FURL from %s"
" because it does not have the same"
" TubID as I do (%s)" %
(furlFile, self.tubID))
if name != sr.name:
raise WrongNameError("I cannot keep using the old FURL from %s"
" because you called registerReference"
" with a new name (%s)" %
(furlFile, name))
name = self._assignName(ref, name)
assert name
if ref not in self.strongReferences:
self.strongReferences.append(ref)
furl = self.buildURL(name)
if furlFile:
need_to_chmod = not os.path.exists(furlFile)
f = open(furlFile, "w")
f.write(furl + "\n")
f.close()
if need_to_chmod:
# XXX: open-to-chmod race here
os.chmod(furlFile, 0600)
return furl
# this is called by either registerReference or by
# getOrCreateURLForReference
def _assignName(self, ref, preferred_name=None):
"""Make a Referenceable available to the outside world, but do not
retain a strong reference to it. If we must create a new name, use
preferred_name. If that is None, use a random unguessable name.
"""
if not self.locationHints:
# without a location, there is no point in giving it a name
return None
if self.referenceToName.has_key(ref):
return self.referenceToName[ref]
name = preferred_name
if not name:
name = self.generateSwissnumber(self.NAMEBITS)
self.referenceToName[ref] = name
self.nameToReference[name] = ref
return name
def getReferenceForName(self, name):
if name in self.nameToReference:
return self.nameToReference[name]
for lookup in self.nameLookupHandlers:
ref = lookup(name)
if ref:
if ref not in self.referenceToName:
self.referenceToName[ref] = name
return ref
# don't reveal the full swissnum
hint = name[:2]
raise KeyError("unable to find reference for name starting with '%s'"
% hint)
def getReferenceForURL(self, url):
# TODO: who should this be used by?
sturdy = SturdyRef(url)
assert sturdy.tubID == self.tubID
return self.getReferenceForName(sturdy.name)
def getOrCreateURLForReference(self, ref):
"""Return the global URL for the reference, if there is one, or None
if there is not."""
name = self._assignName(ref)
if name:
return self.buildURL(name)
return None
def revokeReference(self, ref):
# TODO
pass
def unregisterURL(self, url):
sturdy = SturdyRef(url)
name = sturdy.name
ref = self.nameToReference[name]
del self.nameToReference[name]
del self.referenceToName[ref]
self.revokeReference(ref)
def unregisterReference(self, ref):
name = self.referenceToName[ref]
url = self.buildURL(name)
sturdy = SturdyRef(url)
name = sturdy.name
del self.nameToReference[name]
del self.referenceToName[ref]
if ref in self.strongReferences:
self.strongReferences.remove(ref)
self.revokeReference(ref)
def registerNameLookupHandler(self, lookup):
"""Add a function to help convert names to Referenceables.
When remote systems pass a FURL to their Tub.getReference(), our Tub
will be asked to locate a Referenceable for the name inside that
furl. The normal mechanism for this is to look at the table
maintained by registerReference() and unregisterReference(). If the
name does not exist in that table, other 'lookup handler' functions
are given a chance. Each lookup handler is asked in turn, and the
first which returns a non-None value wins.
This may be useful for cases where the furl represents an object that
lives on disk, or is generated on demand: rather than creating all
possible Referenceables at startup, the lookup handler can create or
retrieve the objects only when someone asks for them.
Note that constructing the FURLs of these objects may be non-trivial.
It is safe to create an object, use tub.registerReference in one
invocation of a program to obtain (and publish) the furl, parse the
furl to extract the name, save the contents of the object on disk,
then in a later invocation of the program use a lookup handler to
retrieve the object from disk. This approach means the objects that
are created in a given invocation stick around (inside
tub.strongReferences) for the rest of that invocation. An alternatve
approach is to create the object but *not* use tub.registerReference,
but in that case you have to construct the FURL yourself, and the Tub
does not currently provide any support for doing this robustly.
@param lookup: a callable which accepts a name (as a string) and
returns either a Referenceable or None. Note that
these strings should not contain a slash, a question
mark, or an ampersand, as these are reserved in the
FURL for later expansion (to add parameters beyond the
object name)
"""
self.nameLookupHandlers.append(lookup)
def unregisterNameLookupHandler(self, lookup):
self.nameLookupHandlers.remove(lookup)
def getReference(self, sturdyOrURL):
"""Acquire a RemoteReference for the given SturdyRef/URL.
The Tub must be running (i.e. Tub.startService()) when this is
invoked. Future releases may relax this requirement.
@return: a Deferred that fires with the RemoteReference. Any failures
are returned asynchronously.
"""
return defer.maybeDeferred(self._getReference, sturdyOrURL)
def _getReference(self, sturdyOrURL):
if isinstance(sturdyOrURL, SturdyRef):
sturdy = sturdyOrURL
else:
sturdy = SturdyRef(sturdyOrURL)
if not self.running:
# queue their request for service once the Tub actually starts
log.msg("Tub.getReference(%s) queued until Tub.startService called"
% sturdy, facility="foolscap.tub")
d = defer.Deferred()
self._pending_getReferences.append((d, sturdy))
return d
name = sturdy.name
d = self.getBrokerForTubRef(sturdy.getTubRef())
d.addCallback(lambda b: b.getYourReferenceByName(name))
return d
def connectTo(self, _sturdyOrURL, _cb, *args, **kwargs):
"""Establish (and maintain) a connection to a given PBURL.
I establish a connection to the PBURL and run a callback to inform
the caller about the newly-available RemoteReference. If the
connection is lost, I schedule a reconnection attempt for the near
future. If that one fails, I keep trying at longer and longer
intervals (exponential backoff).
I accept a callback which will be fired each time a connection
attempt succeeds. This callback is run with the new RemoteReference
and any additional args/kwargs provided to me. The callback should
then use rref.notifyOnDisconnect() to get a message when the
connection goes away. At some point after it goes away, the
Reconnector will reconnect.
The Tub must be running (i.e. Tub.startService()) when this is
invoked. Future releases may relax this requirement.
I return a Reconnector object. When you no longer want to maintain
this connection, call the stopConnecting() method on the Reconnector.
I promise to not invoke your callback after you've called
stopConnecting(), even if there was already a connection attempt in
progress. If you had an active connection before calling
stopConnecting(), you will still have access to it, until it breaks
on its own. (I will not attempt to break existing connections, I will
merely stop trying to create new ones). All my Reconnector objects
will be shut down when the Tub is stopped.
Usage::
def _got_ref(rref, arg1, arg2):
rref.callRemote('hello again')
# etc
rc = tub.connectTo(_got_ref, 'arg1', 'arg2')
...
rc.stopConnecting() # later
"""
rc = Reconnector(_sturdyOrURL, _cb, args, kwargs)
if self.running:
rc.startConnecting(self)
else:
self.log("Tub.connectTo(%s) queued until Tub.startService called"
% _sturdyOrURL, level=UNUSUAL)
self.reconnectors.append(rc)
return rc
def serialize(self, obj):
b = broker.StorageBroker(None)
b.setTub(self)
d = storage.serialize(obj, banana=b)
return d
def unserialize(self, data):
b = broker.StorageBroker(None)
b.setTub(self)
d = storage.unserialize(data, banana=b)
assert isinstance(d, defer.Deferred)
return d
# beyond here are internal methods, not for use by application code
# _removeReconnector is called by the Reconnector
def _removeReconnector(self, rc):
self.reconnectors.remove(rc)
def getBrokerForTubRef(self, tubref):
if tubref in self.brokers:
return defer.succeed(self.brokers[tubref])
if tubref.getTubID() == self.tubID:
b = self._createLoopbackBroker(tubref)
# _createLoopbackBroker will call brokerAttached, which will add
# it to self.brokers
# TODO: stash this in self.brokers, so we don't create multiples
return defer.succeed(b)
d = defer.Deferred()
if tubref not in self.waitingForBrokers:
self.waitingForBrokers[tubref] = []
self.waitingForBrokers[tubref].append(d)
if tubref not in self.tubConnectors:
# the TubConnector will call our brokerAttached when it finishes
# negotiation, which will fire waitingForBrokers[tubref].
c = connection.TubConnector(self, tubref, self._connectionHandlers)
self.tubConnectors[tubref] = c
c.connect()
return d
def _createLoopbackBroker(self, tubref):
t1,t2 = broker.LoopbackTransport(), broker.LoopbackTransport()
t1.setPeer(t2); t2.setPeer(t1)
n = negotiate.Negotiation()
params = n.loopbackDecision()
ci = info.ConnectionInfo()
b1 = self.brokerClass(tubref, params, connectionInfo=ci)
b2 = self.brokerClass(tubref, params)
# we treat b1 as "our" broker, and b2 as "theirs", and we pretend
# that b2 has just connected to us. We keep track of b1, and b2 keeps
# track of us.
b1.setTub(self)
b2.setTub(self)
t1.protocol = b1; t2.protocol = b2
b1.makeConnection(t1); b2.makeConnection(t2)
ci._set_connected(True)
ci._set_winning_hint("loopback")
ci._set_connection_status("loopback", "connected")
ci._set_established_at(b1.creation_timestamp)
self.brokerAttached(tubref, b1, False)
return b1
def connectionFailed(self, tubref, why):
# we previously initiated an outbound TubConnector to this tubref, but
# it was unable to establish a connection. 'why' is the most useful
# Failure that occurred (i.e. it is a NegotiationError if we made it
# that far, otherwise it's a ConnectionFailed).
if tubref in self.tubConnectors:
del self.tubConnectors[tubref]
if tubref in self.brokers:
# oh, but fortunately an inbound connection must have succeeded.
# Nevermind.
return
# inform hopeful Broker-waiters that they aren't getting one
if tubref in self.waitingForBrokers:
waiting = self.waitingForBrokers[tubref]
del self.waitingForBrokers[tubref]
for d in waiting:
d.errback(why)
def brokerAttached(self, tubref, broker, isClient):
assert self.running
assert tubref
if tubref in self.tubConnectors:
# we initiated an outbound connection to this tubref
if not isClient:
# however, the connection we got was from an inbound
# connection. The completed (inbound) connection wins, so
# abandon the outbound TubConnector
self.tubConnectors[tubref].shutdown()
# we don't need the TubConnector any more
del self.tubConnectors[tubref]
if tubref in self.brokers:
# this shouldn't happen: acceptDecision is supposed to drop any
# existing old connection first.
self.log("ERROR: unexpected duplicate connection from %s" % tubref)
raise BananaError("unexpected duplicate connection")
self.brokers[tubref] = broker
# now inform everyone who's been waiting on it
if tubref in self.waitingForBrokers:
for d in self.waitingForBrokers[tubref]:
eventual.eventually(d.callback, broker)
del self.waitingForBrokers[tubref]
def brokerDetached(self, broker, why):
# a loopback connection will produce two Brokers that both use the
# same tubref. Both will shut down about the same time. Make sure
# this doesn't confuse us.
# the Broker will have already severed all active references
for tubref in self.brokers.keys():
if self.brokers[tubref] is broker:
del self.brokers[tubref]
def debug_listBrokers(self):
# return a list of (tubref, inbound, outbound) tuples. The tubref
# tells you which broker this is, 'inbound' is a list of
# InboundDelivery objects (one per outstanding inbound message), and
# 'outbound' is a list of PendingRequest objects (one per message
# that's waiting on a remote broker to complete).
output = []
all_brokers = self.brokers.items()
for tubref,_broker in all_brokers:
inbound = _broker.inboundDeliveryQueue[:]
outbound = [pr
for (reqID, pr) in
sorted(_broker.waitingForAnswers.items()) ]
output.append( (str(tubref), inbound, outbound) )
output.sort(lambda x,y: cmp( (len(x[1]), len(x[2])),
(len(y[1]), len(y[2])) ))
return output
|