0x1949 Team - FAZEMRX - MANAGER
Edit File: pb.cpython-310.pyc
o �b_� � @ s� d Z ddlZddlmZ ddlmZmZ ddlmZm Z m Z m Z ddl m Z ddlmZmZ ddlmZ dd lmZmZmZ dd lmZmZ dd lmZ dd l m Z dd l m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2 ddl3m4Z4m5Z5 ddl6m7Z7m8Z8m9Z9m:Z: dZ;dZ<G dd� de=�Z>G dd� de>�Z?G dd� de=�Z@G dd� de=�ZAeG dd� d��ZBG d d � d e=�ZCG d d � d e�ZDeeD�G d d!� d!��ZEG d"d#� d#e$�ZFee5�eG d$d%� d%e)ejG���ZHe1d&eH� G d'd(� d(�ZIG d)d*� d*ejJe �ZKG d+d,� d,e'ejJ�ZLe1eKeL� dDd-d.�ZMG d/d0� d0e jN�ZOd1d2� ZPd3d4� ZQG d5d6� d6ejR�ZSG d7d8� d8ejT�ZUG d9d:� d:e �ZVee!�G d;d<� d<��ZWeeWe e!� G d=d>� d>�ZXG d?d@� d@e$eX�ZYee eV�G dAdB� dBe$eX��ZZg dC�Z[dS )Ea� Perspective Broker "This isn't a professional opinion, but it's probably got enough internet to kill you." --glyph Introduction ============ This is a broker for proxies for and copies of objects. It provides a translucent interface layer to those proxies. The protocol is not opaque, because it provides objects which represent the remote proxies and require no context (server references, IDs) to operate on. It is not transparent because it does I{not} attempt to make remote objects behave identically, or even similarly, to local objects. Method calls are invoked asynchronously, and specific rules are applied when serializing arguments. To get started, begin with L{PBClientFactory} and L{PBServerFactory}. @author: Glyph Lefkowitz � N)�md5)� Interface� implementer)� Anonymous� IAnonymous� ICredentials�IUsernameHashedPassword)�Portal)�defer�protocol)�styles)�failure�log�reflect)�cmp� comparable)�registerAdapter)�banana)� Cacheable�Copyable�IPBRoot� Jellyable� NoSuchMethod� Referenceable� RemoteCache�RemoteCacheObserver� RemoteCopy�Root� Serializable�Viewable� ViewPoint�copyTags�setCopierForClass�setCopierForClassTree�setFactoryForClass� setUnjellyableFactoryForClass�setUnjellyableForClass�setUnjellyableForClassTree)� IJellyable� IUnjellyable)� _newInstance�globalSecurity�jelly�unjellyi iS" c @ � e Zd ZdZdS )� ProtocolErrorzN This error is raised when an invalid protocol statement is received. N��__name__� __module__� __qualname__�__doc__� r5 r5 �3/usr/lib/python3/dist-packages/twisted/spread/pb.pyr/ T � r/ c @ r. )�DeadReferenceErrorzx This error is raised when a method is called on a dead reference (one whose broker has been disconnected). Nr0 r5 r5 r5 r6 r8 Z r7 r8 c @ r. )�Errora This error can be raised to generate known error conditions. When a PB callable method (perspective_, remote_, view_) raises this error, it indicates that a traceback should not be printed, but instead, the string representation of the exception should be sent. Nr0 r5 r5 r5 r6 r9 a r7 r9 c @ � e Zd ZdZdd� ZdS )� RemoteErrora� This class is used to wrap a string-ified exception from the remote side to be able to reraise it. (Raising string exceptions is no longer possible in Python 2.6+) The value of this exception will be a str() representation of the remote value. @ivar remoteType: The full import path of the exception class which was raised on the remote end. @type remoteType: C{str} @ivar remoteTraceback: The remote traceback. @type remoteTraceback: C{str} @note: It's not possible to include the remoteTraceback if this exception is thrown into a generator. It must be accessed as an attribute. c C s t �| |� || _|| _d S �N)� Exception�__init__� remoteType�remoteTraceback)�selfr? �valuer@ r5 r5 r6 r> � s zRemoteError.__init__N�r1 r2 r3 r4 r> r5 r5 r5 r6 r; l s r; c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � RemoteMethodz> This is a translucent reference to a remote message. c C s || _ || _dS )zT Initialize with a L{RemoteReference} and the name of this message. N)�obj�name)rA rF rG r5 r5 r6 r> � s zRemoteMethod.__init__c C s t | j| jf|�S r< )r rF rG �rA �otherr5 r5 r6 �__cmp__� s zRemoteMethod.__cmp__c C s t | j| jf�S r< )�hashrF rG �rA r5 r5 r6 �__hash__� s zRemoteMethod.__hash__c O s( | j j�d| j j| j j| j�d�||�S )z8 Asynchronously invoke a remote method. � �utf-8)rF �broker� _sendMessage� perspective�luidrG �encode)rA �args�kwr5 r5 r6 �__call__� s �zRemoteMethod.__call__N)r1 r2 r3 r4 r> rJ rM rW r5 r5 r5 r6 rE � s rE c @ s e Zd ZdS )�PBConnectionLostN)r1 r2 r3 r5 r5 r5 r6 rX � s rX c @ r: )� IPerspectivea per*spec*tive, n. : The relationship of aspects of a subject to each other and to a whole: 'a perspective of history'; 'a need to view the problem in the proper perspective'. This is a Perspective Broker-specific wrapper for an avatar. That is to say, a PB-published view on to the business logic for the system's concept of a 'user'. The concept of attached/detached is no longer implemented by the framework. The realm is expected to implement such semantics if needed. c C � dS )a� This method is called when a network message is received. @arg broker: The Perspective Broker. @type message: str @arg message: The name of the method called by the other end. @type args: list in jelly format @arg args: The arguments that were passed by the other end. It is recommend that you use the `unserialize' method of the broker to decode this. @type kwargs: dict in jelly format @arg kwargs: The keyword arguments that were passed by the other end. It is recommended that you use the `unserialize' method of the broker to decode this. @rtype: A jelly list. @return: It is recommended that you use the `serialize' method of the broker on whatever object you need to return to generate the return value. Nr5 )rP �messagerU �kwargsr5 r5 r6 �perspectiveMessageReceived� � z'IPerspective.perspectiveMessageReceivedN�r1 r2 r3 r4 r] r5 r5 r5 r6 rY � s rY c @ r: )�Avatara A default IPerspective implementor. This class is intended to be subclassed, and a realm should return an instance of such a subclass when IPerspective is requested of it. A peer requesting a perspective will receive only a L{RemoteReference} to a pb.Avatar. When a method is called on that L{RemoteReference}, it will translate to a method on the remote perspective named 'perspective_methodname'. (For more information on invoking methods on other objects, see L{flavors.ViewPoint}.) c C sv |� || �}|� || �}t| d| �}z ||i |��}W n ty1 t�|� d|� d|� �� � w |�|| |||�S )ai This method is called when a network message is received. This will call:: self.perspective_%(message)s(*broker.unserialize(args), **broker.unserialize(kw)) to handle the method; subclasses of Avatar are expected to implement methods using this naming convention. zperspective_%sz didn't accept z and )� unserialize�getattr� TypeErrorr �msg� serialize)rA rP r[ rU rV �method�stater5 r5 r6 r] � s �z!Avatar.perspectiveMessageReceivedNr_ r5 r5 r5 r6 r` � s r` c @ s e Zd ZdZddd�ZdS )�AsReferenceablez6 A reference directed towards another object. �remotec C s t ||d �| _d S )N�MessageReceived)rb �remoteMessageReceived)rA �object� messageTyper5 r5 r6 r> s zAsReferenceable.__init__N)ri rC r5 r5 r5 r6 rh � s rh c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )�RemoteReferencea] A translucent reference to a remote object. I may be a reference to a L{flavors.ViewPoint}, a L{flavors.Referenceable}, or an L{IPerspective} implementer (e.g., pb.Avatar). From the client's perspective, it is not possible to tell which except by convention. I am a "translucent" reference because although no additional bookkeeping overhead is given to the application programmer for manipulating a reference, return values are asynchronous. See also L{twisted.internet.defer}. @ivar broker: The broker I am obtained through. @type broker: L{Broker} c C s" || _ || _|| _|| _g | _dS )z�(internal) Initialize me with a broker and a locally-unique ID. The ID is unique only to the particular Perspective Broker instance. N)rS rP � doRefCountrR �disconnectCallbacks)rA rR rP rS ro r5 r5 r6 r> s zRemoteReference.__init__c C s< t |�sJ �| j�|� t| j�dkr | j�| j� dS dS )z� Register a callback to be called if our broker gets disconnected. @param callback: a callable which will be called with one argument, this instance. � N)�callablerp �append�lenrP �notifyOnDisconnect� _disconnected�rA �callbackr5 r5 r6 ru ' s �z"RemoteReference.notifyOnDisconnectc C s( | j �|� | j s| j�| j� dS dS )zu Remove a callback that was registered with notifyOnDisconnect. @param callback: a callable N)rp �removerP �dontNotifyOnDisconnectrv rw r5 r5 r6 rz 3 s �z&RemoteReference.dontNotifyOnDisconnectc C s | j D ]}|| � qd| _ dS )zN Called if we are disconnected and have callbacks registered. N)rp rw r5 r5 r6 rv = s z RemoteReference._disconnectedc C s( |j r| j|j ks J d��d| jfS dS )zc If I am being sent back to where I came from, serialize as a local backreference. z6Can't send references to brokers other than their own.s local)s unpersistablez References cannot be serialized)�invokerrP rS )rA �jellierr5 r5 r6 �jellyForE s � zRemoteReference.jellyForc C s | � |jj|j|d d� | S )Nrq )r> r{ �unserializingPerspective)rA � unjellier� unjellyListr5 r5 r6 � unjellyForQ s �zRemoteReference.unjellyForc O s. t |t�s |�d�}| j�d| j| j|||�S )a� Asynchronously invoke a remote method. @type _name: L{str} @param _name: the name of the remote method to invoke @param args: arguments to serialize for the remote function @param kw: keyword arguments to serialize for the remote function. @rtype: L{twisted.internet.defer.Deferred} @returns: a Deferred which will be fired when the result of this remote call is received. �utf8rN )� isinstance�bytesrT rP rQ rR rS )rA �_namerU rV r5 r5 r6 � callRemoteZ s �zRemoteReference.callRemotec C s t | |�S )zX @param key: The key. @return: A L{RemoteMethod} for this key. )rE )rA �keyr5 r5 r6 � remoteMethodp s z RemoteReference.remoteMethodc C s0 t |t�r|j| jkrt| j|j�S t| j|�S )zM @param other: another L{RemoteReference} to compare me to. )r� rn rP r rS rH r5 r5 r6 rJ x s zRemoteReference.__cmp__c C s | j S )z Hash me. )rS rL r5 r5 r6 rM � s zRemoteReference.__hash__c C s | j r | j�| j� dS dS )zD Do distributed reference counting on finalization. N)ro rP � sendDecRefrS rL r5 r5 r6 �__del__� s �zRemoteReference.__del__N)r1 r2 r3 r4 r> ru rz rv r} r� r� r� rJ rM r� r5 r5 r5 r6 rn s rn ri c @ s8 e Zd ZdZd dd�Zdefdd�Zdd � Zd d � ZdS ) �Localz3 (internal) A reference to a local object. Nc C s || _ || _d| _dS )z Initialize. rq N)rl rR �refcount)rA rl rR r5 r5 r6 r> � s zLocal.__init__�returnc C s d| j �d| j� d�S )Nz <pb.Local z ref:�>)rl r� rL r5 r5 r6 �__repr__� s zLocal.__repr__c C � | j d | _ | j S )zi Increment the reference count. @return: the reference count after incrementing rq �r� rL r5 r5 r6 �incref� � z Local.increfc C s | j d | _ | j S )zi Decrement the reference count. @return: the reference count after decrementing rq r� rL r5 r5 r6 �decref� r� z Local.decrefr< ) r1 r2 r3 r4 r> �strr� r� r� r5 r5 r5 r6 r� � s r� c @ s e Zd ZdZdZdd� ZdS )�CopyableFailurez} A L{flavors.RemoteCopy} and L{flavors.Copyable} version of L{twisted.python.failure.Failure} for serialization. r c C s� | j �� }d|d< g |d< g |d< t| j�|d< t| jt�r$| j|d< n t�| j�� d�|d< | j r:| � � |d< |S d |d< |S ) z� Collect state related to the exception which occurred, discarding state which cannot reasonably be serialized. N�tb�frames�stackrB �typerO � tracebackzTraceback unavailable ) �__dict__�copyr� rB r� r� r� r �qualrT �unsafeTracebacks� getTraceback)rA rg r5 r5 r6 �getStateToCopy� s �z CopyableFailure.getStateToCopyN)r1 r2 r3 r4 r� r� r5 r5 r5 r6 r� � s r� c @ s* e Zd ZdZd dd�Zdd� ZeZeZdS ) � CopiedFailurea� A L{CopiedFailure} is a L{pb.RemoteCopy} of a L{failure.Failure} transferred via PB. @ivar type: The full import path of the exception class which was raised on the remote end. @type type: C{str} @ivar value: A str() representation of the remote value. @type value: L{CopiedFailure} or C{str} @ivar traceback: The remote traceback. @type traceback: C{str} Nr �defaultc C sT |d u rt j}| j}t|t�s|�d�}|�d� |�|d | j � |�d� d S )NrO z Traceback from remote host -- z: � )r �logfiler� r� r� �decode�writerB )rA �file�elideFrameworkCode�detail� failureTyper5 r5 r6 �printTraceback� s z CopiedFailure.printTracebackc C s |� t| j| j| j��S )a� Throw the original exception into the given generator, preserving traceback information if available. In the case of a L{CopiedFailure} where the exception type is a string, a L{pb.RemoteError} is thrown instead. @return: The next value yielded from the generator. @raise StopIteration: If there are no more values in the generator. @raise RemoteError: The wrapped remote exception. )�throwr; r� rB r� )rA �gr5 r5 r6 �throwExceptionIntoGenerator� s z)CopiedFailure.throwExceptionIntoGenerator)Nr r� )r1 r2 r3 r4 r� r� �printBriefTraceback�printDetailedTracebackr5 r5 r5 r6 r� � s r� c C s t t| j�}||_|S r< )r* r� r� r� )�failr� �fr5 r5 r6 �failure2Copyable s r� c @ s� e Zd ZdZdZdZdZdefdd�Zdd� Z d d � Z d d � Z d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� ZdZdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'Zd(d)� Zd*d+� Z d,d-� Z d^d.d/�Z d0d1� Z d2d3� Z d4d5� Z!d6d7� Z"d_d8d9�Z#d`d:d;�Z$d<d=� Z%d>d?� Z&d@dA� Z'dBdC� Z(dDdE� Z)dFdG� Z*dHdI� Z+dJdK� Z,dLdM� Z-dNdO� Z.dPdQ� Z/dRdS� Z0dTdU� Z1dVdW� Z2dXdY� Z3dZd[� Z4d\d]� Z5dS )a�Brokerz$ I am a broker for objects. � Nrq c C sr t j�| |� d| _g | _g | _g | _i | _|| _g | _ d| _ d| _ d | _ i | _ i | _i | _i | _i | _i | _d S �Nr )r �Bananar> � disconnected� disconnects�failures�connects� localObjects�security� pageProducers�currentRequestID�currentLocalIDr~ �luids�remotelyCachedObjects�remotelyCachedLUIDs�locallyCachedObjects�waitingForAnswers� _localCleanup)rA �isClientr� r5 r5 r6 r> s" zBroker.__init__c C sV t t| j�d dd�D ]}| j| }|�� |�� s | j|= q | js)| j�� dS dS )zM Called when the consumer attached to me runs out of buffer. rq ���N)�rangert r� � sendNextPage� stillPaging� transport�unregisterProducer)rA �pageridx�pagerr5 r5 r6 �resumeProducing7 s ��zBroker.resumeProducingc C � d S r< r5 rL r5 r5 r6 �pauseProducingD � zBroker.pauseProducingc C r� r< r5 rL r5 r5 r6 � stopProducingH r� zBroker.stopProducingc C s0 | j �|� t| j �dkr| j�| d� d S d S )Nrq r )r� rs rt r� �registerProducer)rA r� r5 r5 r6 �registerPageProducerL s �zBroker.registerPageProducerc C sj t |t�r1|d }t |t�s|�d�}d| }t| |d�}|r)||dd� � dS | �d|� dS td��)z: Evaluate an expression as it's received. r r� zproto_%sNrq s didNotUnderstandz Non-list expression received.)r� �listr� r� rb �sendCallr/ )rA �sexp�command� methodNamerf r5 r5 r6 �expressionReceivedQ s zBroker.expressionReceivedc C s$ || j krtd| j � d|� ���dS )z� Protocol message: (version version-number) Check to make sure that both ends of the protocol are speaking the same version dialect. @param vnum: The version number. zVersion Incompatibility: � N)�versionr/ )rA �vnumr5 r5 r6 � proto_versione s �zBroker.proto_versionc G s | � |� dS )z� Utility method to send an expression to the other side of the connection. @param exp: The expression. N)� sendEncoded)rA �expr5 r5 r6 r� r s zBroker.sendCallc C s t �d| � dS )a Respond to stock 'C{didNotUnderstand}' message. Log the command that was not understood and continue. (Note: this will probably be changed to close the connection or raise an exception in the future.) @param command: The command to log. z Didn't understand command: %rN)r rd )rA r� r5 r5 r6 �proto_didNotUnderstandz s z Broker.proto_didNotUnderstandc C sT | � d| j� | jD ]}z|� W q ty t�� Y q w d| _| j�| � dS )zF Initialize. Called after Banana negotiation is done. s versionN)r� r� r� � BaseExceptionr �deferr�factory�clientConnectionMade�rA �notifierr5 r5 r6 �connectionReady� s �zBroker.connectionReadyc C s: | j D ]}z|� W q ty t�� Y qw d | _ d S r< )r� r� r r� r� r5 r5 r6 �connectionFailed� s � zBroker.connectionFailedc C s d| _ d| _| jr*| j�� D ]}z |�t�t|��� W q ty) t � � Y qw | j �� D ]!}|j }|j }z |�|t| ||�� W q/ tyP t � � Y q/w | jdd� D ]}z|� W qX tyl t � � Y qXw d| _d| _d| _d| _d| _ d| _d| _d| _dS )zg The connection was lost. @param reason: message to put in L{failure.Failure} rq N)r� r� r� �values�errbackr �FailurerX r� r r� r� rl rR �stoppedObservingr r� � localSecurity�remoteSecurityr� r� r� )rA �reason�d�lobj� cacheablerR r� r5 r5 r6 �connectionLost� sB � � � � zBroker.connectionLostc C � t |�sJ �| j�|� dS )zQ @param notifier: callback to call when the Broker disconnects. N)rr r� rs r� r5 r5 r6 ru � � zBroker.notifyOnDisconnectc C r� )zT @param notifier: callback to call if the Broker fails to connect. N)rr r� rs r� r5 r5 r6 � notifyOnFail� r� zBroker.notifyOnFailc C sP t |�sJ �| jdu r z|� W dS ty t�� Y dS w | j�|� dS )zN @param notifier: callback to call when the Broker connects. N)rr r� r� r �errrs r� r5 r5 r6 �notifyOnConnect� s �zBroker.notifyOnConnectc C s( z | j �|� W dS ty Y dS w )zY @param notifier: callback to remove from list of disconnect callbacks. N)r� ry � ValueErrorr� r5 r5 r6 rz � s �z Broker.dontNotifyOnDisconnectc C s2 t |t�r |�d�}| j�|�}|du rdS |jS )z� Get a local object for a locally unique ID. @return: An object previously stored with L{registerReference} or L{None} if there is no object which corresponds to the given identifier. r� N)r� r� rT r� �getrl )rA rS �lobr5 r5 r6 �localObjectForID� s zBroker.localObjectForIDr c C s� |dusJ �|� � }| j�|�}|du rEt| j�tkr3| jd | _| jdkr/| j�� t d��t d��| � � }t |�| j|< || j|< |S | j| � � |S )z� Store a persistent reference to a local object and map its id() to a generated, session-unique ID. @param object: a local object @return: the generated ID Nrq � z.Maximum PB reference count exceeded. Goodbye.z$Maximum PB reference count exceeded.) �processUniqueIDr� r� rt r� �MAX_BROKER_REFS�maxBrokerRefsViolationsr� �loseConnectionr9 � newLocalIDr� r� )rA rl �puidrS r5 r5 r6 �registerReference s �zBroker.registerReferencec C s2 t |t�r |�d�}|dusJ �t|�| j|< dS )z� Store a special (string) ID for this object. This is how you specify a 'base' set of objects that the remote protocol can connect to. @param name: An ID. @param object: The object. r� N)r� r� rT r� r� )rA rG rl r5 r5 r6 �setNameForLocal s zBroker.setNameForLocalc C s" t |t�r |�d�}td| |d�S )a Returns an object from the remote name mapping. Note that this does not check the validity of the name, only creates a translucent reference for it. @param name: The name to look up. @return: An object which maps to the name. r� Nr )r� r� rT rn )rA rG r5 r5 r6 � remoteForName, s zBroker.remoteForNamec C s2 |� � }| j�|�}|dur|r| j| �� |S )a @param instance: The instance to look up. @param incref: Flag to specify whether to increment the reference. @return: An ID that says what this instance is cached as remotely, or L{None} if it's not. N)r r� r� r� r� )rA �instancer� r rS r5 r5 r6 �cachedRemotelyAs; s zBroker.cachedRemotelyAsc C s | j | jS )zk @param luid: The LUID to look up. @return: An instance which is cached remotely. )r� rl )rA rS r5 r5 r6 �remotelyCachedForLUIDK s z Broker.remotelyCachedForLUIDc C sn |� � }| �� }t| j�tkr'| jd | _| jdkr#| j�� td��td��|| j |< t || j �| j|< |S )z3 XXX @return: A new LUID. rq r z*Maximum PB cache count exceeded. Goodbye.z Maximum PB cache count exceeded.) r r rt r� r r r� r r9 r� r� �serializingPerspective)rA r r rS r5 r5 r6 � cacheRemotelyS s zBroker.cacheRemotelyc C s || j |< dS )zL(internal) Store a non-filled-out cached instance locally. N�r� �rA �cidr r5 r5 r6 � cacheLocallyh s zBroker.cacheLocallyc C s | j | }|S r< r r r5 r5 r6 �cachedLocallyAso s zBroker.cachedLocallyAsc C s� t |tj�r|j| jdd� ||||d�d� |S || _|| _|| _|| _zt || j d| �W d| _d| _d| _d| _S d| _d| _d| _d| _w )a Jelly an object according to the remote security rules for this broker. @param object: The object to jelly. @param perspective: The perspective. @param method: The method. @param args: Arguments. @param kw: Keyword arguments. c S s | S r< r5 )�xr5 r5 r6 �<lambda>� r^ z"Broker.serialize.<locals>.<lambda>)rR rf rU rV )�callbackKeywordsN) r� r �Deferred� addCallbacksre r � jellyMethod� jellyArgs�jellyKwr, r� )rA rl rR rf rU rV r5 r5 r6 re s s2 �� �zBroker.serializec C s( || _ z t|| jd| �W d| _ S d| _ w )z� Unjelly an sexp according to the local security rules for this broker. @param sexp: The object to unjelly. @param perspective: The perspective. N)r~ r- r� )rA r� rR r5 r5 r6 ra � s zBroker.unserializec C r� )z3 @return: A newly generated LUID. rq )r� rL r5 r5 r6 r � � zBroker.newLocalIDc C r� )z9 @return: A newly generated request ID. rq )r� rL r5 r5 r6 � newRequestID� r zBroker.newRequestIDc C s d }d }d} d|v r|d }|d= d|v r |d }|d= d|v r/|s$|r(J d��|d } |d= | j r6td��z| j|||d�} | j|||d�} W n tyX t�t�� � Y S w | �� } | rxt� � } | | j | < |sl|rwt � d� | � ||� nd } | �|d | ||| | | � | S ) Nrq � pbcallback� pberrback�pbanswerz*You can't specify a no-answer requirement.zCalling Stale Broker)rR rf z&warning! using deprecated "pbcallback"s message)r� r8 re r� r r� r r� r r r� r rd r r� )rA �prefixrR �objectIDr[ rU rV �pbc�pbe�answerRequired�netArgs�netKw� requestID�rvalr5 r5 r6 rQ � sP � �� zBroker._sendMessagec C � | � | j||||||� d S r< )� _recvMessager �rA r) r# r[ r& r' r( r5 r5 r6 � proto_message� � �zBroker.proto_messagec C r+ r< )r, r r- r5 r5 r6 �proto_cachemessage� r/ zBroker.proto_cachemessagec C s@ t |t�s |�d�}z||�}|du rtd��|�| |||�} W n] ty^ } z2|rLt | t�s6| j�| j�r=| � | |� n| � t | �|� W Y d} ~ dS W Y d} ~ dS W Y d} ~ dS d} ~ w t y} |rvt j ddd� t � } | � | |� t �� Y dS w |r�t | tj�r�|f} | j| j| j| | d� dS | �| |� dS dS )a� Received a message-send. Look up message based on object, unserialize the arguments, and invoke it with args, and send an 'answer' or 'error' response. @param findObjMethod: A callable which takes C{objectID} as argument. @param requestID: The requiest ID. @param objectID: The object ID. @param message: The message. @param answerRequired: @param netArgs: Arguments. @param netKw: Keyword arguments. r� NzInvalid Object ID�)Peer will receive following PB traceback:T)�isError)� callbackArgs� errbackArgs)r� r� r� r9 rk r r� �isClassAllowed� __class__� _sendErrorr� r� r rd r� r r r � _sendAnswer�_sendFailureOrError) rA � findObjMethodr) r# r[ r& r' r( rl � netResult�er� rU r5 r5 r6 r, � sF � � � � ��zBroker._recvMessagec C s | � d||� dS )z� (internal) Send an answer to a previously sent message. @param netResult: The answer. @param requestID: The request ID. s answerN�r� )rA r; r) r5 r5 r6 r8 @ � zBroker._sendAnswerc C �& | j | }| j |= |�| �|�� dS )z� (internal) Got an answer to a previously sent message. Look up the appropriate callback and call it. @param requestID: The request ID. @param netResult: The answer. N)r� rx ra )rA r) r; r� r5 r5 r6 � proto_answerI s zBroker.proto_answerc C s. |� t�du r| �||� dS | �||� dS )z� Call L{_sendError} or L{_sendFailure}, depending on whether C{fail} represents an L{Error} subclass or not. @param fail: The failure. @param requestID: The request ID. N)�checkr9 � _sendFailurer7 �rA r� r) r5 r5 r6 r9 V s zBroker._sendFailureOrErrorc C s$ t �d� t �|� | �||� dS )zz Log error and then send it. @param fail: The failure. @param requestID: The request ID. r1 N)r rd r� r7 rC r5 r5 r6 rB c s zBroker._sendFailurec C st t |tj�r$t |jt�s| j�|jj�r|j}n t |t�s$t || j j �}t |t�r.| j j |_ | � d|| � |�� dS )z� (internal) Send an error for a previously sent message. @param fail: The failure. @param requestID: The request ID. s errorN)r� r r� rB r r� r5 r6 r� r� r� r� r� re rC r5 r5 r6 r7 n s � zBroker._sendErrorc C r? )z} (internal) Deal with an error. @param requestID: The request ID. @param fail: The failure. N)r� r� ra )rA r) r� r� r5 r5 r6 � proto_error� s zBroker.proto_errorc C � | � d|� dS )z^ (internal) Send a DECREF directive. @param objectID: The object ID. s decrefNr= �rA r# r5 r5 r6 r� � � zBroker.sendDecRefc C sf t |t�r |�d�}| j| �� }|dkr1| j| j�� }| j|= | j|= | j� |dd� �� dS dS )z� (internal) Decrement the reference count of an object. If the reference count is zero, it will free the reference to this object. @param objectID: The object ID. r� r c S r� r< r5 r5 r5 r5 r6 r � r^ z%Broker.proto_decref.<locals>.<lambda>N) r� r� rT r� r� rl r r� r� �pop)rA r# �refsr r5 r5 r6 � proto_decref� s �zBroker.proto_decrefc C rE )z_ (internal) Send a DECACHE directive. @param objectID: The object ID. s decacheNr= rF r5 r5 r6 � decCacheRef� rG zBroker.decCacheRefc C s� | j | �� }|dkrD| j | }|j}|j}z |�|t| ||�� W n ty/ t�� Y nw |� � }| j |= | j |= | � d|� dS dS )z� (internal) Decrement the reference count of a cached object. If the reference count is zero, free the reference, then send an 'uncached' directive. @param objectID: The object ID. r s uncacheN) r� r� rl rR r� r r� r r� r r� r� )rA r# rI r� r� rR r r5 r5 r6 � proto_decache� s" � ��zBroker.proto_decachec C s | j | }d|_| j |= dS )zx (internal) Tell the client it is now OK to uncache an object. @param objectID: The object ID. N)r� rP )rA r# rF r5 r5 r6 � proto_uncache� s zBroker.proto_uncache�r )NNNNr< )6r1 r2 r3 r4 r� �usernamer� r+ r> r� r� r� r� r� r� r� r� r� r� r� r� ru r� r� rz r r r r r r r r r r re ra r r rQ r. r0 r, r8 r@ r9 rB r7 rD r� rJ rK rL rM r5 r5 r5 r6 r� sf & ) *) B r� c C s> t � }|�|� |�� }t � }|�|� |�| � |�� }|S )z� Respond to a challenge. This is useful for challenge/response authentication. @param challenge: A challenge. @param password: A password. @return: The password hashed twice. )r �update�digest)� challenge�password�m�hashedPassword�doubleHashedPasswordr5 r5 r6 �respond� s rW c C s. t dd� tt�dd��D ��} t| ��� } | S )z% @return: Some random data. c s s � | ] }t �d d�V qdS )�A �Z N)�random�randint)�.0r r5 r5 r6 � <genexpr>� s � z challenge.<locals>.<genexpr>� � )r� r� rZ � randranger rQ )�crapr5 r5 r6 rR � s rR c @ s� e Zd ZdZeZdZdefdd�Zdd� Z dd� Z d d � Z d d � Z d dd�Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d d �Zd S )!�PBClientFactoryz� Client factory for PB brokers. As with all client factories, use with reactor.connectTCP/SSL/etc.. getPerspective and getRootObject can be called either before or after the connect. Fc C s || _ || _| �� dS )aE @param unsafeTracebacks: if set, tracebacks for exceptions will be sent over the wire. @type unsafeTracebacks: C{bool} @param security: security options used by the broker, default to C{globalSecurity}. @type security: L{twisted.spread.jelly.SecurityOptions} N)r� r� �_reset)rA r� r� r5 r5 r6 r> s zPBClientFactory.__init__c C s | j d| jd�}| |_|S )zP Build the broker instance, passing the security options to it. T�r� r� )r r� r� )rA �addr�pr5 r5 r6 � buildProtocol s z PBClientFactory.buildProtocolc C s g | _ d | _d | _d S r< )�rootObjectRequests�_broker�_rootrL r5 r5 r6 rc s zPBClientFactory._resetc C s& | j }| �� |D ]}|�|� q d S r< )rh rc r� )rA r� � deferredsr� r5 r5 r6 �_failAll s �zPBClientFactory._failAllc C s | � |� d S r< )rl )rA � connectorr� r5 r5 r6 �clientConnectionFailed% s z&PBClientFactory.clientConnectionFailedr c C s" |r d| _ d| _dS | �|� dS )zJ Reconnecting subclasses should call with reconnecting=1. N)ri rj rl )rA rm r� � reconnectingr5 r5 r6 �clientConnectionLost( s z$PBClientFactory.clientConnectionLostc C s8 || _ |�d�| _| j}g | _|D ]}|�| j� qd S )N�root)ri r rj rh rx )rA rP �dsr� r5 r5 r6 r� 4 s �z$PBClientFactory.clientConnectionMadec C s2 | j r | j js t�| j�S t�� }| j�|� |S )ze Get root object of remote PB server. @return: Deferred of the root object. )ri r� r �succeedrj r rh rs )rA r� r5 r5 r6 � getRootObject< s z PBClientFactory.getRootObjectc C s | j r | j j�� dS dS )z� If the factory is connected, close the connection. Note that if you set up the factory to reconnect, you will need to implement extra logic to prevent automatic reconnection after this is called. N)ri r� r rL r5 r5 r6 � disconnectH s �zPBClientFactory.disconnectc C s |� d|��| j||�S )N�login)r� � addCallback� _cbResponse)rA rq rO rS �clientr5 r5 r6 �_cbSendUsernameS s �z PBClientFactory._cbSendUsernamec C s |\}}|� dt||�|�S )NrW )r� rW )rA � challengesrS ry rR � challengerr5 r5 r6 rx X s zPBClientFactory._cbResponsec C s |� d|�S )ab Attempt an anonymous login on the given remote root object. @type root: L{RemoteReference} @param root: The object on which to attempt the login, most likely returned by a call to L{PBClientFactory.getRootObject}. @param client: A jellyable object which will be used as the I{mind} parameter for the login attempt. @rtype: L{Deferred} @return: A L{Deferred} which will be called back with a L{RemoteReference} to an avatar when anonymous login succeeds, or which will errback if anonymous login fails. �loginAnonymous)r� )rA rq ry r5 r5 r6 �_cbLoginAnonymous\ s z!PBClientFactory._cbLoginAnonymousNc C s> | � � }t�|�r|�| j|� |S |�| j|j|j|� |S )a� Login and get perspective from remote PB server. Currently the following credentials are supported:: L{twisted.cred.credentials.IUsernamePassword} L{twisted.cred.credentials.IAnonymous} @rtype: L{Deferred} @return: A L{Deferred} which will be called back with a L{RemoteReference} for the avatar logged in to, or which will errback if login fails. )rt r � providedByrw r~ rz rO rS )rA � credentialsry r� r5 r5 r6 rv n s ��zPBClientFactory.loginrN r< )r1 r2 r3 r4 r� r r� r+ r> rg rc rl rn rp r� rt ru rz rx r~ rv r5 r5 r5 r6 rb � s" rb c @ s6 e Zd ZdZdZeZdefdd�Zdd� Z dd� Z d S ) �PBServerFactorya� Server factory for perspective broker. Login is done using a Portal object, whose realm is expected to return avatars implementing IPerspective. The credential checkers in the portal should accept IUsernameHashedPassword or IUsernameMD5Password. Alternatively, any object providing or adaptable to L{IPBRoot} can be used instead of a portal to provide the root object of the PB server. Fc C s t |�| _|| _|| _dS )a� @param root: factory providing the root Referenceable used by the broker. @type root: object providing or adaptable to L{IPBRoot}. @param unsafeTracebacks: if set, tracebacks for exceptions will be sent over the wire. @type unsafeTracebacks: C{bool} @param security: security options used by the broker, default to C{globalSecurity}. @type security: L{twisted.spread.jelly.SecurityOptions} N)r rq r� r� )rA rq r� r� r5 r5 r6 r> � s zPBServerFactory.__init__c C s. | j d| jd�}| |_|�d| j�|�� |S )zT Return a Broker attached to the factory (as the service provider). Frd rq )r r� r� r rq � rootObject)rA re �protor5 r5 r6 rg � s z PBServerFactory.buildProtocolc C r� r< r5 )rA r r5 r5 r6 r� � r� z$PBServerFactory.clientConnectionMadeN) r1 r2 r3 r4 r� r� r r+ r> rg r� r5 r5 r5 r6 r� � s r� c @ � e Zd ZdZdd� Zdd� ZdS )�IUsernameMD5Passwordab I encapsulate a username and a hashed password. This credential is used for username/password over PB. CredentialCheckers which check this kind of credential must store the passwords in plaintext form or as a MD5 digest. @type username: C{str} or C{Deferred} @ivar username: The username associated with these credentials. c C rZ )a� Validate these credentials against the correct password. @type password: C{str} @param password: The correct, plaintext password against which to check. @rtype: C{bool} or L{Deferred} @return: C{True} if the credentials represented by this object match the given password, C{False} if they do not, or a L{Deferred} which will be called back with one of these values. Nr5 �rS r5 r5 r6 � checkPassword� r^ z"IUsernameMD5Password.checkPasswordc C rZ )a� Validate these credentials against the correct MD5 digest of the password. @type password: C{str} @param password: The correct MD5 digest of a password against which to check. @rtype: C{bool} or L{Deferred} @return: C{True} if the credentials represented by this object match the given digest, C{False} if they do not, or a L{Deferred} which will be called back with one of these values. Nr5 r� r5 r5 r6 �checkMD5Password� r^ z%IUsernameMD5Password.checkMD5PasswordN)r1 r2 r3 r4 r� r� r5 r5 r5 r6 r� � s r� c @ r� )� _PortalRootz/ Root object, used to login to portal. c C s || _ d S r< )�portal)rA r� r5 r5 r6 r> � s z_PortalRoot.__init__c C s t | j|�S r< )�_PortalWrapperr� )rA rP r5 r5 r6 r� � s z_PortalRoot.rootObjectN)r1 r2 r3 r4 r> r� r5 r5 r5 r6 r� � s r� c @ r: )�_JellyableAvatarMixinzk Helper class for code which deals with avatars which PB must be capable of sending to a peer. c sT |\}}� t �|�st|d�}|�� }� g� � fdd�}|| jj|< | j�|� |S )z� Ensure that the avatar to be returned to the client is jellyable and set up disconnection notification to call the realm's logout object. rR c s � sd S � d } � d= | � d S r� r5 )�fn��logoutr5 r6 � maybeLogout s z3_JellyableAvatarMixin._cbLogin.<locals>.maybeLogout)r( r rh r rP r� ru )rA �result� interface�avatarr r� r5 r� r6 �_cbLogin� s z _JellyableAvatarMixin._cbLoginN)r1 r2 r3 r4 r� r5 r5 r5 r6 r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r� z= Root Referenceable object, used to login to portal. c C s || _ || _d S r< )r� rP )rA r� rP r5 r5 r6 r> s z_PortalWrapper.__init__c C s t � }|t| j| j||�fS )z[ Start of username/password login. @param username: The username. )rR �_PortalAuthChallengerr� rP )rA rO �cr5 r5 r6 � remote_login s z_PortalWrapper.remote_loginc C s" | j �t� |t�}|�| j� |S )aU Attempt an anonymous login. @param mind: An object to use as the mind parameter to the portal login call (possibly None). @rtype: L{Deferred} @return: A Deferred which will be called back with an avatar when login succeeds or which will be errbacked if login fails somehow. )r� rv r rY rw r� )rA �mindr� r5 r5 r6 �remote_loginAnonymous& s z$_PortalWrapper.remote_loginAnonymousN)r1 r2 r3 r4 r> r� r� r5 r5 r5 r6 r� s r� c @ rD ) r� z5 Called with response to password challenge. c C s || _ || _|| _|| _d S r< )r� rP rO rR )rA r� rP rO rR r5 r5 r6 r> < s z _PortalAuthChallenger.__init__c C s& || _ | j�| |t�}|�| j� |S r< )�responser� rv rY rw r� )rA r� r� r� r5 r5 r6 �remote_respondB s z$_PortalAuthChallenger.remote_respondc C s | � t|��� �S )z� L{IUsernameHashedPassword} @param password: The password. @return: L{_PortalAuthChallenger.checkMD5Password} )r� r rQ )rA rS r5 r5 r6 r� H r> z#_PortalAuthChallenger.checkPasswordc C s. t � }|�|� |�| j� |�� }| j|kS )z� L{IUsernameMD5Password} @param md5Password: @rtype: L{bool} @return: L{True} if password matches. )r rP rR rQ r� )rA � md5Password�md�correctr5 r5 r6 r� Q s z&_PortalAuthChallenger.checkMD5PasswordN)r1 r2 r3 r4 r> r� r� r� r5 r5 r5 r6 r� 6 s r� )(r r r r r r r r r r r r r r! r& r% r' r" r$ r# r �portnor/ r8 r9 rX rE rY r` rh rn r� r� r� r� rW rR rb r� r� rN )\r4 rZ �hashlibr �zope.interfacer r �twisted.cred.credentialsr r r r �twisted.cred.portalr �twisted.internetr r �twisted.persistedr �twisted.pythonr r r �twisted.python.compatr r �twisted.python.componentsr �twisted.spreadr �twisted.spread.flavorsr r r r r r r r r r r r r r! r" r# r$ r% r& r' �twisted.spread.interfacesr( r) �twisted.spread.jellyr* r+ r, r- r r� r= r/ r8 r9 r; rE rX rY r` rh � Ephemeralrn r� r� r� r� r� r� r� rW rR � ClientFactoryrb � ServerFactoryr� r� r� r� r� r� �__all__r5 r5 r5 r6 �<module> st X )( $ + W 0* #" )