0x1949 Team - FAZEMRX - MANAGER
Edit File: amp.cpython-310.pyc
o �b�x � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZ dd lmZmZ dd lmZm Z m Z dd l m Z m Z m!Z! dd l"m#Z# dd l$m%Z% ddl&m'Z' ddl(m)Z)m*Z* ddl+m,Z,m-Z- ddl.m/Z0m1Z2 ddl3m4Z4 ddl5m6Z6 ddl7m8Z8 zddl9m:Z; e;j<r�ddl=m>Z>m?Z?m@Z@mAZA ndZ:W n eBy� dZ:Y nw e;Z:g d�ZCdZDdZEdZFdZGd ZHd ZId ZJd ZKd ZLd!ZMG d"d#� d#e�ZNG d$d%� d%e�ZOG d&d'� d'e�ZPG d(d)� d)e�ZQG d*d+� d+eR�ZSG d,d-� d-eR�ZTG d.d/� d/eS�ZUG d0d1� d1eS�ZVG d2d3� d3eS�ZWG d4d5� d5eS�ZXG d6d7� d7eS�ZYG d8d9� d9eS�ZZG d:d;� d;eZ�Z[G d<d=� d=eS�Z\G d>d?� d?eS�Z]G d@dA� dAeS�Z^eKe]iZ_G dBdC� dCe`�ZaeaZbG dDdE� dEea�ZcG dFdG� dGea�ZdeeP�G dHdI� dI��ZeG dJdK� dKef�ZgeeQ�G dLdM� dMegdN��ZheeQ�G dOdP� dP��Zig dQ�ZjdRdS� ZkeeN�G dTdU� dU��ZlG dVdW� dWel�ZmG dXdY� dYel�ZnG dZd[� d[el�ZoG d\d]� d]el�ZpG d^d_� d_en�ZqG d`da� daeq�ZrG dbdc� dcel�ZsG ddde� deel�ZtG dfdg� dgem�ZuG dhdi� dief�ZvG djdk� dkevdN�ZwG dldm� dm�ZxG dndo� doea�ZyG dpdq� dqen�ZzG drds� dsew�Z{G dtdu� duew�Z|ee#�G dvdw� dw��Z}eeO�G dxdy� dye*e)e}��Z~G dzd{� d{e~eeehei�ZG d|d}� d}�Z�e�j�Z�e�j�Z�d~d� Z�d�d�� Z�G d�d�� d�el�Z�G d�d�� d�el�Z�dS )�a� This module implements AMP, the Asynchronous Messaging Protocol. AMP is a protocol for sending multiple asynchronous request/response pairs over the same connection. Requests and responses are both collections of key/value pairs. AMP is a very simple protocol which is not an application. This module is a "protocol construction kit" of sorts; it attempts to be the simplest wire-level implementation of Deferreds. AMP provides the following base-level features: - Asynchronous request/response handling (hence the name) - Requests and responses are both key/value pairs - Binary transfer of all data: all data is length-prefixed. Your application will never need to worry about quoting. - Command dispatching (like HTTP Verbs): the protocol is extensible, and multiple AMP sub-protocols can be grouped together easily. The protocol implementation also provides a few additional features which are not part of the core wire protocol, but are nevertheless very useful: - Tight TLS integration, with an included StartTLS command. - Handshaking to other protocols: because AMP has well-defined message boundaries and maintains all incoming and outgoing requests for you, you can start a connection over AMP and then switch to another protocol. This makes it ideal for firewall-traversal applications where you may have only one forwarded port but multiple applications that want to use it. Using AMP with Twisted is simple. Each message is a command, with a response. You begin by defining a command type. Commands specify their input and output in terms of the types that they expect to see in the request and response key-value pairs. Here's an example of a command that adds two integers, 'a' and 'b':: class Sum(amp.Command): arguments = [('a', amp.Integer()), ('b', amp.Integer())] response = [('total', amp.Integer())] Once you have specified a command, you need to make it part of a protocol, and define a responder for it. Here's a 'JustSum' protocol that includes a responder for our 'Sum' command:: class JustSum(amp.AMP): def sum(self, a, b): total = a + b print 'Did a sum: %d + %d = %d' % (a, b, total) return {'total': total} Sum.responder(sum) Later, when you want to actually do a sum, the following expression will return a L{Deferred} which will fire with the result:: ClientCreator(reactor, amp.AMP).connectTCP(...).addCallback( lambda p: p.callRemote(Sum, a=13, b=81)).addCallback( lambda result: result['total']) Command responders may also return Deferreds, causing the response to be sent only once the Deferred fires:: class DelayedSum(amp.AMP): def slowSum(self, a, b): total = a + b result = defer.Deferred() reactor.callLater(3, result.callback, {'total': total}) return result Sum.responder(slowSum) This is transparent to the caller. You can also define the propagation of specific errors in AMP. For example, for the slightly more complicated case of division, we might have to deal with division by zero:: class Divide(amp.Command): arguments = [('numerator', amp.Integer()), ('denominator', amp.Integer())] response = [('result', amp.Float())] errors = {ZeroDivisionError: 'ZERO_DIVISION'} The 'errors' mapping here tells AMP that if a responder to Divide emits a L{ZeroDivisionError}, then the other side should be informed that an error of the type 'ZERO_DIVISION' has occurred. Writing a responder which takes advantage of this is very simple - just raise your exception normally:: class JustDivide(amp.AMP): def divide(self, numerator, denominator): result = numerator / denominator print 'Divided: %d / %d = %d' % (numerator, denominator, total) return {'result': result} Divide.responder(divide) On the client side, the errors mapping will be used to determine what the 'ZERO_DIVISION' error means, and translated into an asynchronous exception, which can be handled normally as any L{Deferred} would be:: def trapZero(result): result.trap(ZeroDivisionError) print "Divided by zero: returning INF" return 1e1000 ClientCreator(reactor, amp.AMP).connectTCP(...).addCallback( lambda p: p.callRemote(Divide, numerator=1234, denominator=0) ).addErrback(trapZero) For a complete, runnable example of both of these commands, see the files in the Twisted repository:: doc/core/examples/ampserver.py doc/core/examples/ampclient.py On the wire, AMP is a protocol which uses 2-byte lengths to prefix keys and values, and empty keys to separate messages:: <2-byte length><key><2-byte length><value> <2-byte length><key><2-byte length><value> ... <2-byte length><key><2-byte length><value> <NUL><NUL> # Empty Key == End of Message And so on. Because it's tedious to refer to lengths and NULs constantly, the documentation will refer to packets as if they were newline delimited, like so:: C: _command: sum C: _ask: ef639e5c892ccb54 C: a: 13 C: b: 81 S: _answer: ef639e5c892ccb54 S: total: 94 Notes: In general, the order of keys is arbitrary. Specific uses of AMP may impose an ordering requirement, but unless this is specified explicitly, any ordering may be generated and any ordering must be accepted. This applies to the command-related keys I{_command} and I{_ask} as well as any other keys. Values are limited to the maximum encodable size in a 16-bit length, 65535 bytes. Keys are limited to the maximum encodable size in a 8-bit length, 255 bytes. Note that we still use 2-byte lengths to encode keys. This small redundancy has several features: - If an implementation becomes confused and starts emitting corrupt data, or gets keys confused with values, many common errors will be signalled immediately instead of delivering obviously corrupt packets. - A single NUL will separate every key, and a double NUL separates messages. This provides some redundancy when debugging traffic dumps. - NULs will be present at regular intervals along the protocol, providing some padding for otherwise braindead C implementations of the protocol, so that <stdio.h> string functions will see the NUL and stop. - This makes it possible to run an AMP server on a port also used by a plain-text protocol, and easily distinguish between non-AMP clients (like web browsers) which issue non-NUL as the first byte, and AMP clients, which always issue NUL as the first byte. @var MAX_VALUE_LENGTH: The maximum length of a message. @type MAX_VALUE_LENGTH: L{int} @var ASK: Marker for an Ask packet. @type ASK: L{bytes} @var ANSWER: Marker for an Answer packet. @type ANSWER: L{bytes} @var COMMAND: Marker for a Command packet. @type COMMAND: L{bytes} @var ERROR: Marker for an AMP box of error type. @type ERROR: L{bytes} @var ERROR_CODE: Marker for an AMP box containing the code of an error. @type ERROR_CODE: L{bytes} @var ERROR_DESCRIPTION: Marker for an AMP box containing the description of the error. @type ERROR_DESCRIPTION: L{bytes} � N)�partial)�BytesIO)�count)�pack)� MethodType)�Any�Callable�Dict�List�Optional�Tuple�Type�Union)� Interface� implementer)�Deferred�fail� maybeDeferred)�ConnectionClosed�ConnectionLost�PeerVerifyError)�IFileDescriptorReceiver)�CONNECTION_LOST)�Protocol)�Int16StringReceiver�StatefulStringProtocol)�filepath�log)�UTC�FixedOffsetTimeZone)� nativeString)�Failure)�accumulateClassDict)�ssl)�DN� Certificate�CertificateOptions�KeyPair)5�AMP�ANSWER�ASK�AmpBox�AmpError�AmpList�Argument�BadLocalReturn�BinaryBoxProtocol�Boolean�Box� BoxDispatcher�COMMAND�Command�CommandLocator�Decimal� Descriptor�ERROR� ERROR_CODE�ERROR_DESCRIPTION�Float� IArgumentType� IBoxReceiver� IBoxSender�IResponderLocator�IncompatibleVersions�Integer�InvalidSignature�ListOf�MAX_KEY_LENGTH�MAX_VALUE_LENGTH�MalformedAmpBox� NoEmptyBoxes� OnlyOneTLS�PROTOCOL_ERRORS�PYTHON_KEYWORDS�Path�ProtocolSwitchCommand�ProtocolSwitched�QuitBox�RemoteAmpError�SimpleStringLocator�StartTLS�String�TooLong�UNHANDLED_ERROR_CODE�UNKNOWN_ERROR_CODE�UnhandledCommand�utc�Unicode�UnknownRemoteError�parse� parseStrings _asks _answers _commands _errors _error_codes _error_descriptions UNKNOWNs UNHANDLED� � c @ � e Zd ZdZdd� Zdd� ZdS )r= z� An L{IArgumentType} can serialize a Python object into an AMP box and deserialize information from an AMP box back into a Python object. @since: 9.0 c C � dS )a Given an argument name and an AMP box containing serialized values, extract one or more Python objects and add them to the C{objects} dictionary. @param name: The name associated with this argument. Most commonly this is the key which can be used to find a serialized value in C{strings}. @type name: C{bytes} @param strings: The AMP box from which to extract one or more values. @type strings: C{dict} @param objects: The output dictionary to populate with the value for this argument. The key used will be derived from C{name}. It may differ; in Python 3, for example, the key will be a Unicode/native string. See L{_wireNameToPythonIdentifier}. @type objects: C{dict} @param proto: The protocol instance which received the AMP box being interpreted. Most likely this is an instance of L{AMP}, but this is not guaranteed. @return: L{None} N� ��name�strings�objects�protora ra �7/usr/lib/python3/dist-packages/twisted/protocols/amp.py�fromBox; � zIArgumentType.fromBoxc C r` )a= Given an argument name and a dictionary containing structured Python objects, serialize values into one or more strings and add them to the C{strings} dictionary. @param name: The name associated with this argument. Most commonly this is the key in C{strings} to associate with a C{bytes} giving the serialized form of that object. @type name: C{bytes} @param strings: The AMP box into which to insert one or more strings. @type strings: C{dict} @param objects: The input dictionary from which to extract Python objects to serialize. The key used will be derived from C{name}. It may differ; in Python 3, for example, the key will be a Unicode/native string. See L{_wireNameToPythonIdentifier}. @type objects: C{dict} @param proto: The protocol instance which will send the AMP box once it is fully populated. Most likely this is an instance of L{AMP}, but this is not guaranteed. @return: L{None} Nra rb ra ra rg �toBoxW ri zIArgumentType.toBoxN)�__name__� __module__� __qualname__�__doc__rh rj ra ra ra rg r= 3 s r= c @ r_ )r? z7 A transport which can send L{AmpBox} objects. c C r` )z� Send an L{AmpBox}. @raise ProtocolSwitched: if the underlying protocol has been switched. @raise ConnectionLost: if the underlying connection has already been lost. Nra ��boxra ra rg �sendBoxx ri zIBoxSender.sendBoxc C r` )z� An unhandled error occurred in response to a box. Log it appropriately. @param failure: a L{Failure} describing the error that occurred. Nra )�failurera ra rg �unhandledError� ri zIBoxSender.unhandledErrorN)rk rl rm rn rq rs ra ra ra rg r? s s r? c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) r> zh An application object which can receive L{AmpBox} objects and dispatch them appropriately. c C r` )z� The L{IBoxReceiver.ampBoxReceived} method will start being called; boxes may be responded to by responding to the given L{IBoxSender}. @param boxSender: an L{IBoxSender} provider. Nra �� boxSenderra ra rg �startReceivingBoxes� ri z IBoxReceiver.startReceivingBoxesc C r` )zS A box was received from the transport; dispatch it appropriately. Nra ro ra ra rg �ampBoxReceived� ri zIBoxReceiver.ampBoxReceivedc C r` )zi No further boxes will be received on this connection. @type reason: L{Failure} Nra )�reasonra ra rg �stopReceivingBoxes� ri z IBoxReceiver.stopReceivingBoxesN)rk rl rm rn rw rx rz ra ra ra rg r> � s r> c @ � e Zd ZdZdd� ZdS )r@ ze An application object which can look up appropriate responder methods for AMP commands. c C r` )a� Locate a responder method appropriate for the named command. @param name: the wire-level name (commandName) of the AMP command to be responded to. @type name: C{bytes} @return: a 1-argument callable that takes an L{AmpBox} with argument values for the given command, and returns an L{AmpBox} containing argument values for the named command, or a L{Deferred} that fires the same. Nra )rc ra ra rg �locateResponder� ri z!IResponderLocator.locateResponderN)rk rl rm rn r| ra ra ra rg r@ � � r@ c @ � e Zd ZdZdS )r, z3 Base class of all Amp-related exceptions. N�rk rl rm rn ra ra ra rg r, � � r, c @ r~ )rN z� Connections which have been switched to other protocols can no longer accept traffic at the AMP level. This is raised when you try to send it. Nr ra ra ra rg rN � r� rN c @ r~ )rI z` This is an implementation limitation; TLS may only be started once per connection. Nr ra ra ra rg rI � r� rI c @ r~ )rH zt You can't have empty boxes on the connection. This is raised when you receive or attempt to send one. Nr ra ra ra rg rH � r� rH c @ r~ )rC z5 You didn't pass all the required arguments. Nr ra ra ra rg rC � r� rC c @ s( e Zd ZdZddd�Zdefdd�ZdS ) rT a One of the protocol's length limitations was violated. @ivar isKey: true if the string being encoded in a key position, false if it was in a value position. @ivar isLocal: Was the string encoded locally, or received too long from the network? (It's only physically possible to encode "too long" values on the network for keys.) @ivar value: The string that was too long. @ivar keyName: If the string being encoded was in a value position, what key was it being encoded for? Nc C s& t �| � || _|| _|| _|| _d S �N)r, �__init__�isKey�isLocal�value�keyName)�selfr� r� r� r� ra ra rg r� � s zTooLong.__init__�returnc C sH | j rdpd}| j s|dt| j� 7 }| jrdpd}d||t| j�f S )N�keyr� � �local�remotez%s %s too long: %d)r� �reprr� r� �lenr� )r� �hdr�lclra ra rg �__repr__� s zTooLong.__repr__r� )rk rl rm rn r� �strr� ra ra ra rg rT � s rT c @ s8 e Zd ZdZdededdfdd�Zdefdd �ZeZdS ) r/ zU A bad value was returned from a local command; we were unable to coerce it. �message�enclosedr� Nc C s t �| � || _|| _d S r� )r, r� r� r� )r� r� r� ra ra rg r� s zBadLocalReturn.__init__c C s | j d | j�� S )Nr� )r� r� �getBriefTraceback�r� ra ra rg r� s zBadLocalReturn.__repr__) rk rl rm rn r� r! r� r� �__str__ra ra ra rg r/ � s r/ c s" e Zd ZdZd� fdd� Z� ZS )rP z� This error indicates that something went wrong on the remote end of the connection, and the error was serialized and transmitted to you. FNc sz |r d}|� � }nd}d}d�dd� |D ��}|r"d�||||�}nd�|||�}t� �|� || _|| _|| _|| _dS )a� Create a remote error with an error code and description. @param errorCode: the AMP error code of this error. @type errorCode: C{bytes} @param description: some text to show to the user. @type description: C{str} @param fatal: a boolean, true if this error should terminate the connection. @param local: a local Failure, if one exists. z (local)� c s s, � | ]}|d krd|d��nt |�V qdS )� z\x�2xN)�chr)�.0�cra ra rg � <genexpr>* s � �z*RemoteAmpError.__init__.<locals>.<genexpr>zCode<{}>{}: {} {}zCode<{}>{}: {}N) r� �join�format�superr� r� � errorCode� description�fatal) r� r� r� r� r� � localwhat�othertb�errorCodeForMessager� �� __class__ra rg r� s. ��� zRemoteAmpError.__init__)FN)rk rl rm rn r� � __classcell__ra ra r� rg rP s rP c @ r{ )rZ zc This means that an error whose type we can't identify was raised from the other side. c C s t }t�| ||� d S r� )rV rP r� )r� r� r� ra ra rg r� G s zUnknownRemoteError.__init__N)rk rl rm rn r� ra ra ra rg rZ A r} rZ c @ r~ )rG zJ This error indicates that the wire-level protocol was malformed. Nr ra ra ra rg rG L r� rG c @ r~ )rW z= A command received via amp could not be dispatched. Nr ra ra ra rg rW R r� rW c @ r~ )rA zw It was impossible to negotiate a compatible version of the protocol with the other end of the connection. Nr ra ra ra rg rA X r� rA c sX e Zd ZU dZg Zee ed< � fdd�Zdd� Z dd� Z d d � Z d efd d �Z � Z S )r+ z\ I am a packet in the AMP protocol, much like a regular bytes:bytes dictionary. � __slots__c sF t � j|i |�� dd� | D �}|D ]}|�d�}| �|�| |< qdS )a Initialize a new L{AmpBox}. In Python 3, keyword arguments MUST be Unicode/native strings whereas in Python 2 they could be either byte strings or Unicode strings. However, all keys of an L{AmpBox} MUST be byte strings, or possible to transparently coerce into byte strings (i.e. Python 2). In Python 3, therefore, native string keys are coerced to byte strings by encoding as ASCII. This can result in C{UnicodeEncodeError} being raised. @param args: See C{dict}, but all keys and values should be C{bytes}. On Python 3, native strings may be used as keys provided they contain only ASCII characters. @param kw: See C{dict}, but all keys and values should be C{bytes}. On Python 3, native strings may be used as keys provided they contain only ASCII characters. @raise UnicodeEncodeError: When a native string key cannot be coerced to an ASCII byte string (Python 3 only). c S s g | ] }t |t�s|�qS ra �� isinstance�bytes)r� �nra ra rg � <listcomp>� � z#AmpBox.__init__.<locals>.<listcomp>�asciiN)r� r� �encode�pop)r� �args�kw� nonByteNames� nonByteName�byteNamer� ra rg r� l s �zAmpBox.__init__c C s | � � }|�| � |S )z5 Return another AmpBox just like me. )r� �update)r� �newBoxra ra rg �copy� s z AmpBox.copyc C s� t | �� �}g }|j}|D ]N\}}t|�tkr td| ��t|�tkr-td|�d|����t|�tkr:tdd|d��t|�t krGtdd||��||fD ]}|t dt|��� ||� qKq |t dd�� d � |�S ) z� Convert me into a wire-encoded string. @return: a C{bytes} encoded according to the rules described in the module docstring. zUnicode key not allowed: %rzUnicode value for key z not allowed: TNF�!Hr � ) �sorted�items�append�typer� � TypeErrorr� rE rT rF r r� )r� �i�L�w�k�v�kvra ra rg � serialize� s$ � zAmpBox.serializec C s |� | � dS )aW Serialize and send this box to an Amp instance. By the time it is being sent, several keys are required. I must have exactly ONE of:: _ask _answer _error If the '_ask' key is set, then the '_command' key must also be set. @param proto: an AMP instance. N)rq �r� rf ra ra rg �_sendTo� s zAmpBox._sendTor� c C s dt �| �� d�S )NzAmpBox(�))�dictr� r� ra ra rg r� � � zAmpBox.__repr__)rk rl rm rn r� r r� �__annotations__r� r� r� r� r� r� ra ra r� rg r+ b s r+ c sD e Zd ZU dZg Zee ed< def� fdd� Z� fdd�Z � Z S )rO zJ I am an AmpBox that, upon being sent, terminates the connection. r� r� c s dt � �� � d�S )Nz QuitBox(**r� )r� r� r� r� ra rg r� � r� zQuitBox.__repr__c s t � �|� |j�� dS )z@ Immediately call loseConnection after sending. N)r� r� � transport�loseConnectionr� r� ra rg r� � s zQuitBox._sendTo) rk rl rm rn r� r r� r� r� r� r� ra ra r� rg rO � s rO c s: e Zd ZdZ� fdd�Zdefdd�Z� fdd�Z� ZS ) � _SwitchBoxz| Implementation detail of ProtocolSwitchCommand: I am an AmpBox which sets up state for the protocol to switch. c s t � jdi |�� || _dS )z� Create a _SwitchBox with the protocol to switch to after being sent. @param innerProto: the protocol instance to switch to. @type innerProto: an IProtocol provider. Nra )r� r� � innerProto)r� r� r� r� ra rg r� � s z_SwitchBox.__init__r� c C s d� | jt�| ��S )Nz_SwitchBox({!r}, **{}))r� r� r� r� r� ra ra rg r� � s �z_SwitchBox.__repr__c s$ t � �|� |�� |�| j� dS )z{ Send me; I am the last box on the connection. All further traffic will be over the new protocol. N)r� r� �_lockForSwitch� _switchTor� r� r� ra rg r� � s z_SwitchBox._sendTo) rk rl rm rn r� r� r� r� r� ra ra r� rg r� � s r� c @ s� e Zd ZdZdZdZdZdZdd� Zdd� Z dd � Z d d � Z d d � Z d#dd�Z d#dd�Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� ZdS )$r3 a A L{BoxDispatcher} dispatches '_ask', '_answer', and '_error' L{AmpBox}es, both incoming and outgoing, to their appropriate destinations. Outgoing commands are converted into L{Deferred}s and outgoing boxes, and associated tracking state to fire those L{Deferred} when '_answer' boxes come back. Incoming '_answer' and '_error' boxes are converted into callbacks and errbacks on those L{Deferred}s, respectively. Incoming '_ask' boxes are converted into method calls on a supplied method locator. @ivar _outstandingRequests: a dictionary mapping request IDs to L{Deferred}s which were returned for those requests. @ivar locator: an object with a L{CommandLocator.locateResponder} method that locates a responder function that takes a Box and returns a result (either a Box or a Deferred which fires one). @ivar boxSender: an object which can send boxes, via the L{_sendBoxCommand} method, such as an L{AMP} instance. @type boxSender: L{IBoxSender} Nr c C s i | _ || _d S r� )�_outstandingRequests�locator)r� r� ra ra rg r� s zBoxDispatcher.__init__c C � || _ dS )z� The given boxSender is going to start calling boxReceived on this L{BoxDispatcher}. @param boxSender: The L{IBoxSender} to send command responses to. Nru )r� rv ra ra rg rw � z!BoxDispatcher.startReceivingBoxesc C s | � |� dS )z� No further boxes will be received here. Terminate all currently outstanding command deferreds with the given reason. N)�failAllOutgoing�r� ry ra ra rg rz % s z BoxDispatcher.stopReceivingBoxesc C s2 || _ | j�� }d| _|D ] \}}|�|� q dS )z� Call the errback on all outstanding requests awaiting responses. @param reason: the Failure instance to pass to those errbacks. N)�_failAllReasonr� r� �errback)r� ry �ORr� r� ra ra rg r� , s �z BoxDispatcher.failAllOutgoingc C s | j d7 _ d| j f S )z� Generate protocol-local serial numbers for _ask keys. @return: a string that has not yet been used on this connection. � s %x)�_counterr� ra ra rg �_nextTag8 s zBoxDispatcher._nextTagTc C sd | j dur|r t| j �S dS ||t< | �� }|r ||t< |�| j� |r.t� }| j|< |S d}|S )a� Send a command across the wire with the given C{amp.Box}. Mutate the given box to give it any additional keys (_command, _ask) required for the command and request/response machinery, then send it. If requiresAnswer is True, returns a C{Deferred} which fires when a response is received. The C{Deferred} is fired with an C{amp.Box} on success, or with an C{amp.RemoteAmpError} if an error is received. If the Deferred fails and the error is not handled by the caller of this method, the failure will be logged and the connection dropped. @param command: a C{bytes}, the name of the command to issue. @param box: an AmpBox with the arguments for the command. @param requiresAnswer: a boolean. Defaults to True. If True, return a Deferred which will fire when the other side responds to this command. If False, return None and do not ask the other side for acknowledgement. @return: a Deferred which fires the AmpBox that holds the response to this command, or None, as specified by requiresAnswer. @raise ProtocolSwitched: if the protocol has been switched. N) r� r r4 r� r* r� rv r r� )r� �commandrp �requiresAnswer�tag�resultra ra rg �_sendBoxCommandA s �z BoxDispatcher._sendBoxCommandc K s t |�}| �|||�S )at This is a low-level API, designed only for optimizing simple messages for which the overhead of parsing is too great. @param command: a C{bytes} naming the command. @param kw: arguments to the amp box. @param requiresAnswer: a boolean. Defaults to True. If True, return a Deferred which will fire when the other side responds to this command. If False, return None and do not ask the other side for acknowledgement. @return: a Deferred which fires the AmpBox that holds the response to this command, or None, as specified by requiresAnswer. )r2 r� )r� r� r� r� rp ra ra rg �callRemoteStringl s z BoxDispatcher.callRemoteStringc O s6 z ||i |��}W n t y t� Y S w |�| �S )ah This is the primary high-level API for sending messages via AMP. Invoke it with a command and appropriate arguments to send a message to this connection's peer. @param commandType: a subclass of Command. @type commandType: L{type} @param a: Positional (special) parameters taken by the command. Positional parameters will typically not be sent over the wire. The only command included with AMP which uses positional parameters is L{ProtocolSwitchCommand}, which takes the protocol that will be switched to as its first argument. @param kw: Keyword arguments taken by the command. These are the arguments declared in the command's 'arguments' attribute. They will be encoded and sent to the peer as arguments for the L{commandType}. @return: If L{commandType} has a C{requiresAnswer} attribute set to L{False}, then return L{None}. Otherwise, return a L{Deferred} which fires with a dictionary of objects representing the result of this call. Additionally, this L{Deferred} may fail with an exception representing a connection failure, with L{UnknownRemoteError} if the other end of the connection fails for an unknown reason, or with any error specified as a key in L{commandType}'s C{errors} dictionary. )� BaseExceptionr � _doCommand)r� � commandType�ar� �cora ra rg � callRemote s % � zBoxDispatcher.callRemotec C s | j �|�S )zy This is a terminal callback called after application code has had a chance to quash any errors. )rv rs �r� rr ra ra rg rs � s z BoxDispatcher.unhandledErrorc C s* | j �|t �}|�| j� |�|� dS )z� An AMP box was received that answered a command previously sent with L{callRemote}. @param box: an AmpBox with a value for its L{ANSWER} key. N)r� r� r) � addErrbackrs �callback)r� rp �questionra ra rg �_answerReceived� s z BoxDispatcher._answerReceivedc C sv | j �|t �}|�| j� |t }|t }t|t�r!|� dd�}|t v r-t | ||�}nt ||�}|� t |�� dS )z� An AMP box was received that answered a command previously sent with L{callRemote}, with an error. @param box: an L{AmpBox} with a value for its L{ERROR}, L{ERROR_CODE}, and L{ERROR_DESCRIPTION} keys. �utf-8�replaceN)r� r� r9 r� rs r: r; r� r� �decoderJ rP r� r! )r� rp r� r� r� �excra ra rg �_errorReceived� s z BoxDispatcher._errorReceivedc sR � fdd�}� fdd�}| � � �}t� v r!|�||� |�| j� |�| j� dS )zc @param box: an L{AmpBox} with a value for its L{COMMAND} and L{ASK} keys. c s � t | t< | S r� )r* r) )� answerBoxro ra rg � formatAnswer� s z4BoxDispatcher._commandReceived.<locals>.formatAnswerc s� | � t�r$| jj}| jj}t|t�r|�dd�}| jjr t � }nt � }n t � }t � | � t }d}� t |t< ||t< ||t< |S )Nr� r� s Unknown Error)�checkrP r� r� r� r� r� r� r� rO r+ r �errrV r* r9 r; r: )�error�code�desc�errorBoxro ra rg � formatError� s z3BoxDispatcher._commandReceived.<locals>.formatErrorN)�dispatchCommandr* � addCallbacks� addCallback� _safeEmitr� rs )r� rp r� r �deferredra ro rg �_commandReceived� s z BoxDispatcher._commandReceivedc C sJ t |v r | �|� dS t|v r| �|� dS t|v r!| �|� dS t|��)a� An AmpBox was received, representing a command, or an answer to a previously issued command (either successful or erroneous). Respond to it according to its contents. @param box: an AmpBox @raise NoEmptyBoxes: when a box is received that does not contain an '_answer', '_command' / '_ask', or '_error' key; i.e. one which does not fit into the command / response protocol defined by AMP. N)r) r� r9 r� r4 r rH �r� rp ra ra rg rx � s z BoxDispatcher.ampBoxReceivedc C s, z |� | j� W dS ttfy Y dS w )z� Emit a box, ignoring L{ProtocolSwitched} and L{ConnectionLost} errors which cannot be usefully handled. N)r� rv rN r )r� �aBoxra ra rg r s �zBoxDispatcher._safeEmitc C sJ |t }| j�|�}|du r d|��}ttt|dtt� �d��S t||�S )z� A box with a _command key was received. Dispatch it to a local handler call it. @param box: an AmpBox to be dispatched. NzUnhandled Command: F�r� ) r4 r� r| r rP rU r! rW r )r� rp �cmd� responderr� ra ra rg r s �� z BoxDispatcher.dispatchCommand)T)rk rl rm rn r� r� r� rv r� rw rz r� r� r� r� r� rs r� r� r rx r r ra ra ra rg r3 � s* ++ % r3 c @ s& e Zd ZU dZg Zded< dd� ZdS )�_CommandLocatorMetaa� This metaclass keeps track of all of the Command.responder-decorated methods defined since the last CommandLocator subclass was defined. It assumes (usually correctly, but unfortunately not necessarily so) that those commands responders were all declared as methods of the class being defined. Note that this list can be incorrect if users use the Command.responder decorator outside the context of a CommandLocator class declaration. Command responders defined on subclasses are given precedence over those inherited from a base class. The Command.responder decorator explicitly cooperates with this metaclass. z List[Tuple[Command, Callable]]�_currentClassCommandsc C s� | j d d � }g | j d d �< i }|d< t�| |||�}t|jdd � �}|�� |D ] }|�t|di �� q+|D ] \} } | | f|| j< q9|rT|j t j krTdd� } | |_ |S )N�_commandDispatchr� c S s t jdtdd� | �|�S )Nz-Override locateResponder, not lookupFunction.� ��category� stacklevel)�warnings�warn�PendingDeprecationWarning�lookupFunction�r� rc ra ra rg r| K s � z4_CommandLocatorMeta.__new__.<locals>.locateResponder) r r� �__new__�list�__mro__�reverser� �getattr� commandNamer r6 r| ) �clsrc �bases�attrs�commands�cd�subcls� ancestors�ancestor� commandClass� responderFuncr| ra ra rg r > s z_CommandLocatorMeta.__new__N)rk rl rm rn r r� r ra ra ra rg r + s r c @ rt ) r6 z� A L{CommandLocator} is a collection of responders to AMP L{Command}s, with the help of the L{Command.responder} decorator. c s � ��fdd�}|S )a< Wrap aCallable with its command's argument de-serialization and result serialization logic. @param aCallable: a callable with a 'command' attribute, designed to be called with keyword arguments. @param command: the command class whose serialization to use. @return: a 1-arg callable which, when invoked with an AmpBox, will deserialize the argument list and invoke appropriate user code for the callable's command, returning a Deferred which fires with the result or fails with an error. c sD �� | ��}�fdd�}� ��fdd�}t� fi |���|��|�S )Nc s: | j � j� }� j| }t| j�}tt|||� jv | d��S )Nr )�trap� allErrorsr� r� r! rP � fatalErrors)r� r� r� r� )r� ra rg �checkKnownErrorsq s �zMCommandLocator._wrapWithSerialization.<locals>.doit.<locals>.checkKnownErrorsc s8 z�� | ��W S ty t� }td� | �f |��w )Nz,%r returned %r and %r could not serialize it)� makeResponser� r! r/ )re �originalFailure�� aCallabler� r� ra rg �makeResponseFory s ���zLCommandLocator._wrapWithSerialization.<locals>.doit.<locals>.makeResponseFor)�parseArgumentsr r r� )rp r� r, r1 r/ ra rg �doitn s �z3CommandLocator._wrapWithSerialization.<locals>.doitra )r� r0 r� r3 ra r/ rg �_wrapWithSerialization^ s z%CommandLocator._wrapWithSerializationc C s4 | j jtjkr t�| |�S tjdtdd� | �|�S )zJ Deprecated synonym for L{CommandLocator.locateResponder} z)Call locateResponder, not lookupFunction.r r )r� r r6 r| r r r r ra ra rg r � s � z CommandLocator.lookupFunctionc C s4 | j }||v r|| \}}t|| �}| �||�S dS )a� Locate a callable to invoke when executing the named command. @param name: the normalized name (from the wire) of the command. @type name: C{bytes} @return: a 1-argument function that takes a Box and returns a box or a Deferred which fires a Box, for handling the command identified by the given name, or None, if no appropriate responder can be found. N)r r r4 )r� rc r# r' r( �responderMethodra ra rg r| � s �z CommandLocator.locateResponderN)rk rl rm rn r4 r r| ra ra ra rg r6 W s / r6 )� metaclassc @ s e Zd ZdZdZdd� ZdS )rQ z^ Implement the L{AMP.locateResponder} method to do simple, string-based dispatch. s amp_c C s t | j|�� �}t| |d�S )a� Locate a callable to invoke when executing the named command. @return: a function with the name C{"amp_" + name} on the same instance, or None if no such function exists. This function will then be called with the L{AmpBox} itself as an argument. @param name: the normalized name (from the wire) of the command. @type name: C{bytes} N)r �baseDispatchPrefix�upperr )r� rc �fNamera ra rg r| � s z#SimpleStringLocator.locateResponderN)rk rl rm rn r7 r| ra ra ra rg rQ � s rQ ) �and�del�for�is�raise�assert�elif�from�lambdar� �break�else�global�not�try�class�except�if�or�while�continue�exec�import�pass�yield�def�finally�in�printc C s$ t | �dd��}|tv r|�� S |S )a| (Private) Normalize an argument name from the wire for use with Python code. If the return value is going to be a python keyword it will be capitalized. If it contains any dashes they will be replaced with underscores. The rationale behind this method is that AMP should be an inherently multi-language protocol, so message keys may contain all manner of bizarre bytes. This is not a complete solution; there are still forms of arguments that this implementation will be unable to parse. However, Python identifiers share a huge raft of properties with identifiers from many other languages, so this is a 'good enough' effort for now. We deal explicitly with dashes because that is the most likely departure: Lisps commonly use dashes to separate method names, so protocols initially implemented in a lisp amp dialect may use dashes in argument or command names. @param key: a C{bytes}, looking something like 'foo-bar-baz' or 'from' @type key: C{bytes} @return: a native string which is a valid python identifier, looking something like 'foo_bar_baz' or 'From'. � -� _)r r� rK �title)r� �lkeyra ra rg �_wireNameToPythonIdentifier� s rZ c @ sV e Zd ZdZdZddd�Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dS )r. a� Base-class of all objects that take values from Amp packets and convert them into objects for Python functions. This implementation of L{IArgumentType} provides several higher-level hooks for subclasses to override. See L{toString} and L{fromString} which will be used to define the behavior of L{IArgumentType.toBox} and L{IArgumentType.fromBox}, respectively. Fc C r� )z� Create an Argument. @param optional: a boolean indicating whether this argument can be omitted in the protocol. N��optional)r� r\ ra ra rg r� r� zArgument.__init__c C s0 | j r|�|�}|dur||= |S |�|�}|S )a/ Retrieve the given key from the given dictionary, removing it if found. @param d: a dictionary. @param name: a key in I{d}. @param proto: an instance of an AMP. @raise KeyError: if I am not optional and no value was found. @return: d[name]. N)r\ �getr� )r� �drc rf r� ra ra rg �retrieve s �zArgument.retrievec C sD | � |||�}t|�}| jr|du rd||< dS | �||�||< dS )aG Populate an 'out' dictionary with mapping names to Python values decoded from an 'in' AmpBox mapping strings to string values. @param name: the argument name to retrieve @type name: C{bytes} @param strings: The AmpBox to read string(s) from, a mapping of argument names to string values. @type strings: AmpBox @param objects: The dictionary to write object(s) to, a mapping of names to Python objects. Keys will be native strings. @type objects: dict @param proto: an AMP instance. N)r_ rZ r\ �fromStringProto)r� rc rd re rf �st�nkra ra rg rh 4 s zArgument.fromBoxc C s8 | � |t|�|�}| jr|du rdS | �||�||< dS )a] Populate an 'out' AmpBox with strings encoded from an 'in' dictionary mapping names to Python values. @param name: the argument name to retrieve @type name: C{bytes} @param strings: The AmpBox to write string(s) to, a mapping of argument names to string values. @type strings: AmpBox @param objects: The dictionary to read object(s) from, a mapping of names to Python objects. Keys should be native strings. @type objects: dict @param proto: the protocol we are converting for. @type proto: AMP N)r_ rZ r\ � toStringProto)r� rc rd re rf �objra ra rg rj M s zArgument.toBoxc C � | � |�S )z� Convert a string to a Python value. @param inString: the string to convert. @type inString: C{bytes} @param proto: the protocol we are converting for. @type proto: AMP @return: a Python object. )� fromString�r� �inStringrf ra ra rg r` h s zArgument.fromStringProtoc C re )z� Convert a Python object to a string. @param inObject: the object to convert. @param proto: the protocol we are converting for. @type proto: AMP )�toString�r� �inObjectrf ra ra rg rc v s zArgument.toStringProtoc C r` )z� Convert a string to a Python object. Subclasses must implement this. @param inString: the string to convert. @type inString: C{bytes} @return: the decoded value from C{inString} Nra �r� rh ra ra rg rf � ri zArgument.fromStringc C r` )a Convert a Python object into a string for passing over the network. @param inObject: an object of the type that this Argument is intended to deal with. @return: the wire encoding of inObject @rtype: C{bytes} Nra �r� rk ra ra rg ri � ri zArgument.toStringN�F) rk rl rm rn r\ r� r_ rh rj r` rc rf ri ra ra ra rg r. s r. c @ � e Zd ZdZeZdd� ZdS )rB z� Encode any integer values of any size on the wire as the string representation. Example: C{123} becomes C{"123"} c C s d|f S )Ns %dra rm ra ra rg ri � � zInteger.toStringN)rk rl rm rn �intrf ri ra ra ra rg rB � s rB c @ r_ )rS zB Don't do any conversion at all; just pass through 'str'. c C � |S r� ra rm ra ra rg ri � � zString.toStringc C rr r� ra rl ra ra rg rf � rs zString.fromStringN�rk rl rm rn ri rf ra ra ra rg rS � � rS c @ ro )r< zA Encode floating-point values on the wire as their repr. c C s& t |t�s td|����t|��d�S )NzBad float value r� )r� �float� ValueErrorr� r� rl ra ra rg ri � s zFloat.toStringN)rk rl rm rn rv rf ri ra ra ra rg r< � s r< c @ r_ )r1 z@ Encode True or False as "True" or "False" on the wire. c C s& |dkrdS |dkr dS t d|����)N� TrueT� FalseFzBad boolean value: )r� rl ra ra rg rf � s zBoolean.fromStringc C s |rdS dS )Nrx ry ra rm ra ra rg ri � s zBoolean.toStringN�rk rl rm rn rf ri ra ra ra rg r1 � s r1 c @ r_ )rY z7 Encode a unicode string on the wire as UTF-8. c C s t �| |�d��S �Nr� )rS ri r� rm ra ra rg ri � r� zUnicode.toStringc C s t �| |��d�S r{ )rS rf r� rl ra ra rg rf � r� zUnicode.fromStringNrt ra ra ra rg rY � ru rY c @ r_ )rL aY Encode and decode L{filepath.FilePath} instances as paths on the wire. This is really intended for use with subprocess communication tools: exchanging pathnames on different machines over a network is not generally meaningful, but neither is it disallowed; you can use this to communicate about NFS paths, for example. c C s t �t�| |��S r� )r �FilePathrY rf rl ra ra rg rf � r� zPath.fromStringc C s t �| |�� j�S r� )rY ri � asTextMode�pathrm ra ra rg ri � r� z Path.toStringNrz ra ra ra rg rL � s rL c @ �* e Zd ZdZd dd�Zdd� Zdd� Zd S ) rD a� Encode and decode lists of instances of a single other argument type. For example, if you want to pass:: [3, 7, 9, 15] You can create an argument like this:: ListOf(Integer()) The serialized form of the entire list is subject to the limit imposed by L{MAX_VALUE_LENGTH}. List elements are represented as 16-bit length prefixed strings. The argument type passed to the L{ListOf} initializer is responsible for producing the serialized form of each element. @ivar elementType: The L{Argument} instance used to encode and decode list elements (note, not an arbitrary L{IArgumentType} implementation: arguments must be implemented using only the C{fromString} and C{toString} methods, not the C{fromBox} and C{toBox} methods). @param optional: a boolean indicating whether this argument can be omitted in the protocol. @since: 10.0 Fc C s || _ t�| |� d S r� )� elementTyper. r� )r� r� r\ ra ra rg r� s zListOf.__init__c s6 g }t � }|j|_|�|� | jj� � fdd�|D �S )zn Convert the serialized form of a list of instances of some type back into that list. c s g | ]}� |��qS ra ra )r� �string��elementFromStringra rg r� s z%ListOf.fromString.<locals>.<listcomp>)r r� �stringReceived� dataReceivedr� rf )r� rh rd �parserra r� rg rf s zListOf.fromStringc C sB g }|D ]}| j �|�}|�tdt|��� |�|� qd�|�S )zI Serialize the given list of objects to a single string. r� r� )r� ri r� r r� r� )r� rk rd rd � serializedra ra rg ri s zListOf.toStringNrn )rk rl rm rn r� rf ri ra ra ra rg rD � s rD c @ r ) r- aI Convert a list of dictionaries into a list of AMP boxes on the wire. For example, if you want to pass:: [{'a': 7, 'b': u'hello'}, {'a': 9, 'b': u'goodbye'}] You might use an AmpList like this in your arguments or response list:: AmpList([('a', Integer()), ('b', Unicode())]) Fc C s6 t dd� |D ��sJ d|f ��|| _t�| |� dS )ag Create an AmpList. @param subargs: a list of 2-tuples of ('name', argument) describing the schema of the dictionaries in the sequence of amp boxes. @type subargs: A C{list} of (C{bytes}, L{Argument}) tuples. @param optional: a boolean indicating whether this argument can be omitted in the protocol. c s s � | ] \}}t |t�V qd S r� r� )r� rc �_ra ra rg r� @ s � z#AmpList.__init__.<locals>.<genexpr>zgAmpList should be defined with a list of (name, argument) tuples where `name' is a byte string, got: %rN)�all�subargsr. r� )r� r� r\ ra ra rg r� 5 s ��zAmpList.__init__c s t |�}� �fdd�|D �}|S )Nc s g | ] }t |�j� ��qS ra )�_stringsToObjectsr� )r� rp �rf r� ra rg r� I r� z+AmpList.fromStringProto.<locals>.<listcomp>)r\ )r� rh rf �boxes�valuesra r� rg r` G s zAmpList.fromStringProtoc s d� � �fdd�|D ��S )Nr� c s" g | ] }t |�jt� � ��� �qS ra )�_objectsToStringsr� r2 r� )r� re r� ra rg r� N s ��z)AmpList.toStringProto.<locals>.<listcomp>)r� rj ra r� rg rc L s ��zAmpList.toStringProtoNrn )rk rl rm rn r� r` rc ra ra ra rg r- ' s r- c @ r_ )r8 aO Encode and decode file descriptors for exchange over a UNIX domain socket. This argument type requires an AMP connection set up over an L{IUNIXTransport<twisted.internet.interfaces.IUNIXTransport>} provider (for example, the kind of connection created by L{IReactorUNIX.connectUNIX<twisted.internet.interfaces.IReactorUNIX.connectUNIX>} and L{UNIXClientEndpoint<twisted.internet.endpoints.UNIXClientEndpoint>}). There is no correspondence between the integer value of the file descriptor on the sending and receiving sides, therefore an alternate approach is taken to matching up received descriptors with particular L{Descriptor} parameters. The argument is encoded to an ordinal (unique per connection) for inclusion in the AMP command or response box. The descriptor itself is sent using L{IUNIXTransport.sendFileDescriptor<twisted.internet.interfaces.IUNIXTransport.sendFileDescriptor>}. The receiver uses the order in which file descriptors are received and the ordinal value to come up with the received copy of the descriptor. c C s |� t|��S )a� Take a unique identifier associated with a file descriptor which must have been received by now and use it to look up that descriptor in a dictionary where they are kept. @param inString: The base representation (as a byte string) of an ordinal indicating which file descriptor corresponds to this usage of this argument. @type inString: C{str} @param proto: The protocol used to receive this descriptor. This protocol must be connected via a transport providing L{IUNIXTransport<twisted.internet.interfaces.IUNIXTransport>}. @type proto: L{BinaryBoxProtocol} @return: The file descriptor represented by C{inString}. @rtype: C{int} )�_getDescriptorrq rg ra ra rg r` j s zDescriptor.fromStringProtoc C s |� |�}t�| ||�}|S )a� Send C{inObject}, an integer file descriptor, over C{proto}'s connection and return a unique identifier which will allow the receiver to associate the file descriptor with this argument. @param inObject: A file descriptor to duplicate over an AMP connection as the value for this argument. @type inObject: C{int} @param proto: The protocol which will be used to send this descriptor. This protocol must be connected via a transport providing L{IUNIXTransport<twisted.internet.interfaces.IUNIXTransport>}. @return: A byte string which can be used by the receiver to reconstruct the file descriptor. @rtype: C{bytes} )�_sendFileDescriptorrB rc )r� rk rf � identifier� outStringra ra rg rc s zDescriptor.toStringProtoN)rk rl rm rn r` rc ra ra ra rg r8 U s r8 c @ r{ )� _CommandMetazh Metaclass hack to establish reverse-mappings for 'errors' and 'fatalErrors' as class vars. c C s� i }|d< i }|d< d|vr|� d�|d< t�| |||�}t|jt�s-td�|j���|jD ]\}}t|t�s@td|����q0|j D ]\}}t|t�sTtd|����qDi }i } t |d|� t |d | � t|j t �sqt |j �|_ t|j t �s}t |j �|_ |�� D ] \} } | || < | || < q�| �� D ] \} } | || < | || < q�|j �� D ]\}}t|t�s�td |����q�|j �� D ]\}}t|t�s�td |����q�|S ) N� reverseErrorsr* r r� z-Command names must be byte strings, got: {!r}z*Argument names must be byte strings, got: z*Response names must be byte strings, got: �errorsr+ z'Error names must be byte strings, got: z-Fatal error names must be byte strings, got: )r� r� r r� r r� r� r� � arguments�responser" r� r� r+ r� ) r rc r r! r� �er�newtyper� r� r+ r� r� ra ra rg r � sX �� � � � ��z_CommandMeta.__new__N)rk rl rm rn r ra ra ra rg r� � r} r� c @ s� e Zd ZU dZg Zeeeef e d< g Z eeeef e d< g Z ee e d< i Z eee ef e d< i Zeee ef e d< eZde d< eZee e d < d Zd d � Zed d� �Zedd� �Zedd� �Zedd� �Z edd� �Z dd� Z dS )r5 a� Subclass me to specify an AMP Command. @cvar arguments: A list of 2-tuples of (name, Argument-subclass-instance), specifying the names and values of the parameters which are required for this command. @cvar response: A list like L{arguments}, but instead used for the return value. @cvar errors: A mapping of subclasses of L{Exception} to wire-protocol tags for errors represented as L{str}s. Responders which raise keys from this dictionary will have the error translated to the corresponding tag on the wire. Invokers which receive Deferreds from invoking this command with L{BoxDispatcher.callRemote} will potentially receive Failures with keys from this mapping as their value. This mapping is inherited; if you declare a command which handles C{FooError} as 'FOO_ERROR', then subclass it and specify C{BarError} as 'BAR_ERROR', responders to the subclass may raise either C{FooError} or C{BarError}, and invokers must be able to deal with either of those exceptions. @cvar fatalErrors: like 'errors', but errors in this list will always terminate the connection, despite being of a recognizable error type. @cvar commandType: The type of Box used to issue commands; useful only for protocol-modifying behavior like startTLS or protocol switching. Defaults to a plain vanilla L{Box}. @cvar responseType: The type of Box used to respond to this command; only useful for protocol-modifying behavior like startTLS or protocol switching. Defaults to a plain vanilla L{Box}. @ivar requiresAnswer: a boolean; defaults to True. Set it to False on your subclass if you want callRemote to return None. Note: this is a hint only to the client side of the protocol. The return-type of a command responder method must always be a dictionary adhering to the contract specified by L{response}, because clients are always free to request a response if they want one. r� r� �extrar� r+ z Union[Type[Command], Type[Box]]r� � responseTypeTc K s` || _ g }| jD ]\}}t|�}|| j vr |js |�|� q|r,td�d�|�| j���g }dS )a5 Create an instance of this command with specified values for its parameters. In Python 3, keyword arguments MUST be Unicode/native strings whereas in Python 2 they could be either byte strings or Unicode strings. A L{Command}'s arguments are defined in its schema using C{bytes} names. The values for those arguments are plucked from the keyword arguments using the name returned from L{_wireNameToPythonIdentifier}. In other words, keyword arguments should be named using the Python-side equivalent of the on-wire (C{bytes}) name. @param kw: a dict containing an appropriate value for each name specified in the L{arguments} attribute of my class. @raise InvalidSignature: if you forgot any required arguments. zforgot {} for {}z, N) � structuredr� rZ r\ r� rC r� r� r )r� r� � forgottenrc �arg� pythonNamera ra rg r� s ��zCommand.__init__c C s6 z| � � }W n ty t� Y S w t|| j||�S )a` Serialize a mapping of arguments using this L{Command}'s response schema. @param objects: a dict with keys matching the names specified in self.response, having values of the types that the Argument objects in self.response can format. @param proto: an L{AMP}. @return: an L{AmpBox}. )r� r� r r� r� )r re rf r� ra ra rg r- " s �zCommand.makeResponsec C sX t � }| jD ] \}}|�t|�� q|D ] }||vr!t|� d���qt|| j| �� |�S )a� Serialize a mapping of arguments using this L{Command}'s argument schema. @param objects: a dict with keys similar to the names specified in self.arguments, having values of the types that the Argument objects in self.arguments can parse. @param proto: an L{AMP}. @return: An instance of this L{Command}'s C{commandType}. z is not a valid argument)�setr� �addrZ rC r� r� )r re rf � allowedNames�argName�ignored� intendedArgra ra rg � makeArguments6 s �zCommand.makeArgumentsc C � t || j|�S )aZ Parse a mapping of serialized arguments using this L{Command}'s response schema. @param box: A mapping of response-argument names to the serialized forms of those arguments. @param protocol: The L{AMP} protocol. @return: A mapping of response-argument names to the parsed forms. )r� r� �r rp �protocolra ra rg � parseResponseM s zCommand.parseResponsec C r� )a? Parse a mapping of serialized arguments using this L{Command}'s argument schema. @param box: A mapping of argument names to the seralized forms of those arguments. @param protocol: The L{AMP} protocol. @return: A mapping of argument names to the parsed forms. )r� r� r� ra ra rg r2 \ s zCommand.parseArgumentsc C s t j�| |f� |S )a Declare a method to be a responder for a particular command. This is a decorator. Use like so:: class MyCommand(Command): arguments = [('a', ...), ('b', ...)] class MyProto(AMP): def myFunMethod(self, a, b): ... MyCommand.responder(myFunMethod) Notes: Although decorator syntax is not used within Twisted, this function returns its argument and is therefore safe to use with decorator syntax. This is not thread safe. Don't declare AMP subclasses in other threads. Don't declare responders outside the scope of AMP subclasses; the behavior is undefined. @param methodfunc: A function which will later become a method, which has a keyword signature compatible with this command's L{arguments} list and returns a dictionary with a set of keys compatible with this command's L{response} list. @return: the methodfunc parameter. )r6 r r� )r � methodfuncra ra rg r j s zCommand.responderc sJ � fdd�}|� � j� �� j|�� j�}� jr#|�� j|� |�|� |S )aL Encode and send this Command to the given protocol. @param proto: an AMP, representing the connection to send to. @return: a Deferred which will fire or error appropriately when the other side responds to the command (or error if the connection is lost before it is responded to). c s. | � t� | j}� j�|jt�}t||j��S r� ) r) rP r� r� r] r� rZ r! r� )r� �rje� errorTyper� ra rg � _massageError� s z)Command._doCommand.<locals>._massageError)r� r r� r� r� r r� r� )r� rf r� r^ ra r� rg r� � s � zCommand._doCommandN) rk rl rm rn r� r r r� r. r� r� r� r r� r r � Exceptionr+ r2 r� r� r+ r� r� � classmethodr- r� r� r2 r r� ra ra ra rg r5 � s, * #r5 c @ r_ )�_NoCertificatea This is for peers which don't want to use a local certificate. Used by AMP because AMP's internal language is all about certificates and this duck-types in the appropriate place; this API isn't really stable though, so it's not exposed anywhere public. For clients, it will use ephemeral DH keys, or whatever the default is for certificate-less clients in OpenSSL. For servers, it will generate a temporary self-signed certificate with garbage values in the DN and use that. c C r� )aT Create a _NoCertificate which either is or isn't for the client side of the connection. @param client: True if we are a client and should truly have no certificate and be anonymous, False if we are a server and actually have to generate a temporary certificate. @type client: bool N)�client)r� r� ra ra rg r� � s z_NoCertificate.__init__c G s� | j s%tdd�}t�� }|�|�}|�||dd� d�}|�|�}|j|� S t� }|r9|� tdddd� |D �d �� t d i |��}|S ) zT Behaves like L{twisted.internet.ssl.PrivateCertificate.options}(). zTEMPORARY CERTIFICATE)�CNc S r` �NTra )�dnra ra rg �<lambda>� ri z(_NoCertificate.options.<locals>.<lambda>r� Tc S s g | ]}|j �qS ra )�original)r� �authra ra rg r� � s z*_NoCertificate.options.<locals>.<listcomp>)�verify�requireCertificate�caCertsNra ) r� r$ r' �generate�certificateRequest�signCertificateRequest�newCertificate�optionsr� r� r& ) r� � authorities�sharedDNr� �cr�sscrd�certr� �occora ra rg r� � s$ ��z_NoCertificate.optionsN)rk rl rm rn r� r� ra ra ra rg r� � s r� c @ sJ e Zd ZU dZg Zee ed< dd� Ze dd� �Z e dd� �Z d d � Z d S ) �_TLSBoxzK I am an AmpBox that, upon being sent, initiates a TLS connection. r� c C s t d u r tdd��t�| � d S )Ns TLS_ERRORzTLS not available)r# rP r+ r� r� ra ra rg r� � s z_TLSBox.__init__c C s | � dtd��S )N� tls_localCertificateF)r] r� r� ra ra rg � certificate� s z_TLSBox.certificatec C s | � dd �S )N� tls_verifyAuthorities)r] r� ra ra rg r� � s z_TLSBox.verifyc C s< t | �}dD ]}|�|d� q|�|� |�| j| j� dS )zK Send my encoded value to the protocol, then initiate TLS. )r� r� N)r+ r� r� � _startTLSr� r� )r� rf �abr� ra ra rg r� � s z_TLSBox._sendToN) rk rl rm rn r� r r� r� r� �propertyr� r� r� ra ra ra rg r� � s r� c @ r{ )�_LocalArgumentz� Local arguments are never actually relayed across the wire. This is just a shim so that StartTLS can pretend to have some arguments: if arguments acquire documentation properties, replace this with something nicer later. c C s d S r� ra )r� rc rd re rf ra ra rg rh rs z_LocalArgument.fromBoxN)rk rl rm rn rh ra ra ra rg r� s r� c @ sd e Zd ZdZdedd�fdedd�fgZdedd�fdedd�fgZeZddd�dd �Z d d � Z dS ) rR az Use, or subclass, me to implement a command that starts TLS. Callers of StartTLS may pass several special arguments, which affect the TLS negotiation: - tls_localCertificate: This is a twisted.internet.ssl.PrivateCertificate which will be used to secure the side of the connection it is returned on. - tls_verifyAuthorities: This is a list of twisted.internet.ssl.Certificate objects that will be used as the certificate authorities to verify our peer's certificate. Each of those special parameters may also be present as a key in the response dictionary. r� Tr[ r� N)�tls_localCertificate�tls_verifyAuthoritiesc K sB t du rtd��|du rtd�n|| _|| _tj| fi |�� dS )a� Create a StartTLS command. (This is private. Use AMP.callRemote.) @param tls_localCertificate: the PrivateCertificate object to use to secure the connection. If it's L{None}, or unspecified, an ephemeral DH key is used instead. @param tls_verifyAuthorities: a list of Certificate objects which represent root certificates to verify our peer with. NzTLS not available.T)r# � RuntimeErrorr� r� r� r5 r� )r� r� r� r� ra ra rg r� - s ��zStartTLS.__init__c s8 t ��� �}� ��j�j� � �fdd�}|�|� |S )z� When a StartTLS command is sent, prepare to start TLS, but don't actually do it; wait for the acknowledgement, then initiate the TLS handshake. c s � � �j�j� | S r� )r� r� r� )r� r� ra rg � actuallystartK s z*StartTLS._doCommand.<locals>.actuallystart)r5 r� � _prepareTLSr� r� r )r� rf r^ r� ra r� rg r� B s zStartTLS._doCommand) rk rl rm rn r� r� r� r� r� r� r� ra ra ra rg rR s � � rR c s8 e Zd ZdZ� fdd�Zedd� �Z� fdd�Z� ZS )rM a� Use this command to switch from something Amp-derived to a different protocol mid-connection. This can be useful to use amp as the connection-startup negotiation phase. Since TLS is a different layer entirely, you can use Amp to negotiate the security parameters of your connection, then switch to a different protocol, and the connection will remain secured. c s || _ t� jdi |�� dS )a Create a ProtocolSwitchCommand. @param _protoToSwitchToFactory: a ProtocolFactory which will generate the Protocol to switch to. @param kw: Keyword arguments, encoded and handled normally as L{Command} would. Nra )�protoToSwitchToFactoryr� r� )r� �_protoToSwitchToFactoryr� r� ra rg r� ] s z ProtocolSwitchCommand.__init__c C s t |�S r� )r� )r r� rf ra ra rg r- k s z"ProtocolSwitchCommand.makeResponsec s< t � �� �}� �� � �fdd�}� �fdd�}|�||�S )z� When we emit a ProtocolSwitchCommand, lock the protocol, but don't actually switch to the new protocol unless an acknowledgement is received. If an error is received, switch back. c s$ �j �� j�� �}� �|�j � | S r� )r� � buildProtocolr� �getPeerr� )�ignr� r� ra rg � switchNowx s �z3ProtocolSwitchCommand._doCommand.<locals>.switchNowc s � � � �j�d tt�� | S r� )�_unlockFromSwitchr� �clientConnectionFailedr! r )r� r� ra rg �handle s �z0ProtocolSwitchCommand._doCommand.<locals>.handle)r� r� r� r )r� rf r^ r� r� r� r� rg r� o s z ProtocolSwitchCommand._doCommand) rk rl rm rn r� r� r- r� r� ra ra r� rg rM S s rM c @ rt ) �_DescriptorExchangera L{_DescriptorExchanger} is a mixin for L{BinaryBoxProtocol} which adds support for receiving file descriptors, a feature offered by L{IUNIXTransport<twisted.internet.interfaces.IUNIXTransport>}. @ivar _descriptors: Temporary storage for all file descriptors received. Values in this dictionary are the file descriptors (as integers). Keys in this dictionary are ordinals giving the order in which each descriptor was received. The ordering information is used to allow L{Descriptor} to determine which is the correct descriptor for any particular usage of that argument type. @type _descriptors: C{dict} @ivar _sendingDescriptorCounter: A no-argument callable which returns the ordinals, starting from 0. This is used to construct values for C{_sendFileDescriptor}. @ivar _receivingDescriptorCounter: A no-argument callable which returns the ordinals, starting from 0. This is used to construct values for C{fileDescriptorReceived}. c C s0 i | _ | j j| _ttt� �| _ttt� �| _d S r� )� _descriptorsr� r� r �nextr �_sendingDescriptorCounter�_receivingDescriptorCounterr� ra ra rg r� � s z _DescriptorExchanger.__init__c C s | j �|� | �� S )z� Assign and return the next ordinal to the given descriptor after sending the descriptor over this protocol's transport. )r� �sendFileDescriptorr� �r� � descriptorra ra rg r� � s z(_DescriptorExchanger._sendFileDescriptorc C s || j | �� < dS )z� Collect received file descriptors to be claimed later by L{Descriptor}. @param descriptor: The received file descriptor. @type descriptor: C{int} N)r� r� r� ra ra rg �fileDescriptorReceived� s z+_DescriptorExchanger.fileDescriptorReceivedN)rk rl rm rn r� r� r� ra ra ra rg r� � s r� c @ s� e Zd ZU dZdZdZdZdZdZdZ dZ dZ dZ e e ed< dZdd� Zd)dd�Zd d � Zd d � Zd d� Zdd� ZdZdZeZdd� Zdd� Zdd� Z dd� Z dd � Z d d � Z d d � Z d!d"� Z!e"d#d$� �Z#d%d&� Z$d'd(� Z%e&�'e%� dS )*r0 a� A protocol for receiving L{AmpBox}es - key/value pairs - via length-prefixed strings. A box is composed of: - any number of key-value pairs, described by: - a 2-byte network-endian packed key length (of which the first byte must be null, and the second must be non-null: i.e. the value of the length must be 1-255) - a key, comprised of that many bytes - a 2-byte network-endian unsigned value length (up to the maximum of 65535) - a value, comprised of that many bytes - 2 null bytes In other words, an even number of strings prefixed with packed unsigned 16-bit integers, and then a 0-length string to indicate the end of the box. This protocol also implements 2 extra private bits of functionality related to the byte boundaries between messages; it can start TLS between two given boxes or switch to an entirely different protocol. However, due to some tricky elements of the implementation, the public interface to this functionality is L{ProtocolSwitchCommand} and L{StartTLS}. @ivar _keyLengthLimitExceeded: A flag which is only true when the connection is being closed because a key length prefix which was longer than allowed by the protocol was received. @ivar boxReceiver: an L{IBoxReceiver} provider, whose L{IBoxReceiver.ampBoxReceived} method will be invoked for each L{AmpBox} that is received. FN� innerProtocolc C s t �| � || _d S r� )r� r� � boxReceiver)r� r� ra ra rg r� � s zBinaryBoxProtocol.__init__c C s: | j }d| _ || _|| _|�| j� |r|�|� dS dS )a� Switch this BinaryBoxProtocol's transport to a new protocol. You need to do this 'simultaneously' on both ends of a connection; the easiest way to do this is to use a subclass of ProtocolSwitchCommand. @param newProto: the new protocol instance to switch to. @param clientFactory: the ClientFactory to send the L{twisted.internet.protocol.ClientFactory.clientConnectionLost} notification to. r� N)�recvdr� �innerProtocolClientFactory�makeConnectionr� r� )r� �newProto� clientFactory� newProtoDatara ra rg r� � s �zBinaryBoxProtocol._switchToc C sL | j rtd��| jdu rt� �| jdur | j�|� dS | j�|�� � dS )a/ Send a amp.Box to my peer. Note: transport.write is never called outside of this method. @param box: an AmpBox. @raise ProtocolSwitched: if the protocol has previously been switched. @raise ConnectionLost: if the connection has previously been lost. z5This connection has switched: no AMP traffic allowed.N)�_lockedrN r� r �_startingTLSBufferr� �writer� r ra ra rg rq s � zBinaryBoxProtocol.sendBoxc C s || _ | j�| � | �� dS )z� Notify L{boxReceiver} that it is about to receive boxes from this protocol by invoking L{IBoxReceiver.startReceivingBoxes}. N)r� r� rw �connectionMade�r� r� ra ra rg r� " s z BinaryBoxProtocol.makeConnectionc C s2 | j rd| _ | jdur| j�|� dS t�| |�S )zg Either parse incoming data as L{AmpBox}es or relay it to our nested protocol. FN)�_justStartedTLSr� r� r )r� �datara ra rg r� + s z BinaryBoxProtocol.dataReceivedc C sv | j dur| j �|� | jdur| j�d|� | jr$ttdddd��}n|�t�r1| j r1t d�}n|}| j � |� dS )zF The connection was lost; notify any nested protocol. NTFz4Peer rejected our certificate for an unknown reason.) r� �connectionLostr� �clientConnectionLost�_keyLengthLimitExceededr! rT r� r r� r r� rz )r� ry � failReasonra ra rg r� 9 s �z BinaryBoxProtocol.connectionLostr] r^ c C s t � | _| �|�S )z6 String received in the 'init' state. )r+ � _currentBox� proto_key�r� r� ra ra rg � proto_initY s z BinaryBoxProtocol.proto_initc C s. |r || _ | j| _dS | j�| j� d| _dS )zu String received in the 'key' state. If the key is empty, a complete box has been received. r� N�init)� _currentKey�_MAX_VALUE_LENGTH� MAX_LENGTHr� rx r� r� ra ra rg r� ` s zBinaryBoxProtocol.proto_keyc C s || j | j< d| _| j| _dS )z7 String received in the 'value' state. Nr� )r� r� �_MAX_KEY_LENGTHr r� ra ra rg � proto_valuen s z BinaryBoxProtocol.proto_valuec C s d| _ | j�� dS )z� The key length limit was exceeded. Disconnect the transport and make sure a meaningful exception is reported. TN)r� r� r� )r� �lengthra ra rg �lengthLimitExceededw s z%BinaryBoxProtocol.lengthLimitExceededc C s d| _ dS )a Lock this binary protocol so that no further boxes may be sent. This is used when sending a request to switch underlying protocols. You probably want to subclass ProtocolSwitchCommand rather than calling this directly. TN)r� r� ra ra rg r� r� z BinaryBoxProtocol._lockForSwitchc C s | j dur td��d| _dS )z� Unlock this locked binary protocol so that further boxes may be sent again. This is used after an attempt to switch protocols has failed for some reason. Nz*Protocol already switched. Cannot unlock.F)r� rN r� r� ra ra rg r� � s z#BinaryBoxProtocol._unlockFromSwitchc C s. g | _ | jdurtd| j| j||ff ��dS )z� Used by StartTLSCommand to put us into the state where we don't actually send things that get sent, instead we buffer them. see L{_sendBoxCommand}. NzUPreviously authenticated connection between %s and %s is trying to re-establish as %s)r� �hostCertificaterI �peerCertificate)r� r� �verifyAuthoritiesra ra rg r� � s ����z BinaryBoxProtocol._prepareTLSc C sZ || _ d| _|du r d}| j�|j|� � | j}|dur)d| _|D ] }| �|� q!dS dS )a< Used by TLSBox to initiate the SSL handshake. @param certificate: a L{twisted.internet.ssl.PrivateCertificate} for use locally. @param verifyAuthorities: L{twisted.internet.ssl.Certificate} instances representing certificate authorities which will verify our peer. TNra )r r� r� �startTLSr� r� rq )r� r� r �stlsbrp ra ra rg r� � s �zBinaryBoxProtocol._startTLSc C s | j rd S t�| j�S r� )�noPeerCertificater% �peerFromTransportr� r� ra ra rg r � s z!BinaryBoxProtocol.peerCertificatec C s( t �|d� | jdur| j�� dS dS )zv The buck stops here. This error was completely unhandled, time to terminate the connection. z�Amp server or network failure unhandled by client application. Dropping connection! To avoid, add errbacks to ALL remote commands!N)r r� r� r� r� ra ra rg rs � s � �z BinaryBoxProtocol.unhandledErrorc C s i S )aX The default TLS responder doesn't specify any certificate or anything. From a security perspective, it's little better than a plain-text connection - but it is still a *bit* better, so it's included for convenience. You probably want to override this by providing your own StartTLS.responder. ra r� ra ra rg �_defaultStartTLSResponder� s z+BinaryBoxProtocol._defaultStartTLSResponderr� )(rk rl rm rn r� r� r� r� r� r� r r r� r r r� r� r� r� rq r� r� r� r r r r� r� r r r� r� r� r� r� r rs r rR r ra ra ra rg r0 � sD r0 c @ sD e Zd ZdZdZddd�Zdd� Zdefd d �Zd d � Z d d� Z dS )r( za This protocol is an AMP connection. See the module docstring for protocol details. FNc C s: d| _ |d u r | }|d u r| }t�| |� t�| |� d S r� )�_ampInitializedr3 r� r0 )r� r� r� ra ra rg r� � s z AMP.__init__c C s( t �| |�}|dur |S t�| |�}|S )z� Unify the implementations of L{CommandLocator} and L{SimpleStringLocator} to perform both kinds of dispatch, preferring L{CommandLocator}. @type name: C{bytes} N)r6 r| rQ )r� rc �firstResponder�secondResponderra ra rg r| � s zAMP.locateResponderr� c C s< | j dur d| j ��}nd}d| jj� |� dt| �d�d�S )zo A verbose string representation which gives us information about this AMP connection. Nz inner r� �<z at 0x�x�>)r� r� rk �id)r� � innerReprra ra rg r� s z AMP.__repr__c C sP | j st�| � |�� | _|�� | _t�d| j j | j| jf � t � | |� dS )zI Emit a helpful log message when the connection is made. z+%s connection established (HOST:%s PEER:%s)N) r r( r� r� �_transportPeer�getHost�_transportHostr �msgr� rk r0 r� r� ra ra rg r� s ��zAMP.makeConnectionc C s2 t �d| jj| j| jf � t�| |� d| _dS )zI Emit a helpful log message when the connection is lost. z$%s connection lost (HOST:%s PEER:%s)N) r r r� rk r r r0 r� r� r� ra ra rg r� s �� zAMP.connectionLost)NN) rk rl rm rn r r� r| r� r� r� r� ra ra ra rg r( � s r( c @ sT e Zd ZdZdd� Zdd� Zdd� ZdZd d � Zd d � Z e d d� �Z e dd� �Z dS )� _ParserHelperz: A box receiver which records all boxes received. c C s g | _ d S r� )r� r� ra ra rg r� . rp z_ParserHelper.__init__c C r` �Nr� ra r� ra ra rg r� 1 rs z_ParserHelper.getPeerc C r` r ra r� ra ra rg r 4 rs z_ParserHelper.getHostFc C r` )z0 No initialization is required. Nra )r� �senderra ra rg rw 9 ri z!_ParserHelper.startReceivingBoxesc C s | j �|� d S r� )r� r� r ra ra rg rx > s z _ParserHelper.ampBoxReceivedc C s. | � }t |d�}|�|� |�|�� � |jS )z� Parse some amp data stored in a file. @param fileObj: a file-like object. @return: a list of AmpBoxes encoded in the given file. )r� )r0 r� r� �readr� )r �fileObj� parserHelper�bbpra ra rg r[ B s z_ParserHelper.parsec C s | � t|��S )z� Parse some amp data stored in a string. @param data: a str holding some amp-encoded data. @return: a list of AmpBoxes encoded in the given string. )r[ r )r r� ra ra rg r\ Q s z_ParserHelper.parseStringN) rk rl rm rn r� r� r � disconnectingrw rx r� r[ r\ ra ra ra rg r ) s r c C s. i }| � � }|D ] \}}|�||||� q|S )a~ Convert an AmpBox to a dictionary of python objects, converting through a given arglist. @param strings: an AmpBox (or dict of strings) @param arglist: a list of 2-tuples of strings and Argument objects, as described in L{Command.arguments}. @param proto: an L{AMP} instance. @return: the converted dictionary mapping names to argument objects. )r� rh )rd �arglistrf re � myStrings�argname� argparserra ra rg r� a s r� c C s* | � � }|D ] \}}|�||||� q|S )a- Convert a dictionary of python objects to an AmpBox, converting through a given arglist. @param objects: a dict mapping names to python objects @param arglist: a list of 2-tuples of strings and Argument objects, as described in L{Command.arguments}. @param strings: [OUT PARAMETER] An object providing the L{dict} interface which will be populated with serialized data. @param proto: an L{AMP} instance. @return: The converted dictionary mapping names to encoded argument strings (identical to C{strings}). )r� rj )re r" rd rf � myObjectsr$ r% ra ra rg r� v s r� c @ r_ )r7 a� Encodes C{decimal.Decimal} instances. There are several ways in which a decimal value might be encoded. Special values are encoded as special strings:: - Positive infinity is encoded as C{"Infinity"} - Negative infinity is encoded as C{"-Infinity"} - Quiet not-a-number is encoded as either C{"NaN"} or C{"-NaN"} - Signalling not-a-number is encoded as either C{"sNaN"} or C{"-sNaN"} Normal values are encoded using the base ten string representation, using engineering notation to indicate magnitude without precision, and "normal" digits to indicate precision. For example:: - C{"1"} represents the value I{1} with precision to one place. - C{"-1"} represents the value I{-1} with precision to one place. - C{"1.0"} represents the value I{1} with precision to two places. - C{"10"} represents the value I{10} with precision to two places. - C{"1E+2"} represents the value I{10} with precision to one place. - C{"1E-1"} represents the value I{0.1} with precision to one place. - C{"1.5E+2"} represents the value I{15} with precision to two places. U{http://speleotrove.com/decimal/} should be considered the authoritative specification for the format. c C s t |�}t�|�S r� )r �decimalr7 rl ra ra rg rf � s zDecimal.fromStringc C s" t |tj�r t|��d�S td��)zW Serialize a C{decimal.Decimal} instance to the specified wire format. r� z8amp.Decimal can only encode instances of decimal.Decimal)r� r' r7 r� r� rw rm ra ra rg ri � s zDecimal.toStringNrz ra ra ra rg r7 � s r7 c @ sl e Zd ZdZedd�edd�edd�edd �ed d �ed d �edd�edd�edd�g Zdd� Zdd� ZdS )�DateTimeaO Encodes C{datetime.datetime} instances. Wire format: '%04i-%02i-%02iT%02i:%02i:%02i.%06i%s%02i:%02i'. Fields in order are: year, month, day, hour, minute, second, microsecond, timezone direction (+ or -), timezone hour, timezone minute. Encoded string is always exactly 32 characters long. This format is compatible with ISO 8601, but that does not mean all ISO 8601 dates can be accepted. Also, note that the datetime module's notion of a "timezone" can be complex, but the wire format includes only a fixed offset, so the conversion is not lossless. A lossless transmission of a C{datetime} instance is not feasible since the receiving end would require a Python interpreter. @ivar _positions: A sequence of slices giving the positions of various interesting parts of the wire format. r � � � � � � � � � � � � � � � � � c sp t � �� t� �dkrtd� ����� fdd�| jD �}� d }tj|g|dd� �R � }|g|dd�< tj|� S )z| Parse a string containing a date and time in the wire format into a C{datetime.datetime} instance. r9 zinvalid date format c s g | ]}t � | ��qS ra )rq )r� �p��sra rg r� � s z'DateTime.fromString.<locals>.<listcomp>r5 r+ N)r r� rw � _positions�_FixedOffsetTZInfo�fromSignHoursMinutes�datetime)r� r< r� �sign�timezonera r; rg rf � s zDateTime.fromStringc C s� |� � }|du r td��|jd |j d }|dkr d}nd}d|j|j|j|j|j|j |j |t |�d t |�d f }|� d �S ) zm Serialize a C{datetime.datetime} instance to a string in the specified wire format. NzUamp.DateTime cannot serialize naive datetime instances. You may find amp.utc useful.i�Q �< r �+�-z-%04i-%02i-%02iT%02i:%02i:%02i.%06i%s%02i:%02ir� ) � utcoffsetrw �days�seconds�year�month�day�hour�minute�second� microsecond�absr� )r� r� �offset� minutesOffsetrA �packedra ra rg ri � s, � � zDateTime.toStringN)rk rl rm rn �slicer= rf ri ra ra ra rg r( � s � r( )�rn r@ r' r � functoolsr �ior � itertoolsr �structr �typesr �typingr r r r r r r r �zope.interfacer r �twisted.internet.deferr r r �twisted.internet.errorr r r �twisted.internet.interfacesr �twisted.internet.mainr �twisted.internet.protocolr �twisted.protocols.basicr r �twisted.pythonr r �twisted.python._tzhelperr rX r r> �twisted.python.compatr �twisted.python.failurer! �twisted.python.reflectr" �twisted.internetr# �_ssl� supported�twisted.internet.sslr$ r% r& r' � ImportError�__all__r* r) r4 r9 r: r; rV rU rE rF r= r? r> r@ r� r, rN rI rH rC rT r/ rP rZ rG rW rA rJ r� r+ r2 rO r� r3 r� r r6 rQ rK rZ r. rB rS r< r1 rY rL rD r- r8 r� r5 r� r� r� rR rM r� r0 r( r r[ r\ r� r� r7 r( ra ra ra rg �<module> s� @ ( � �9@ 4 `" 3,W! 8.A7 `8 E6/ � $M4+