File: //lib/python3/dist-packages/twisted/protocols/__pycache__/sip.cpython-310.pyc
o
�bH� �
@ sd d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
m Z ddl m
Z
ddlmZmZmZ ddlmZ dd lmZ d
Zd d d
ddddddd� Zi Ze�� D ]\ZZeee< qT[[i dd�dd�dd�dd �d d �d d �d!d"�d#d$�d%d&�d'd(�d)d*�d+d,�d-d.�d/d0�d1d2�d3d4�d5d6�i d7d8�d9d:�d;d<�d=d>�d?d@�dAdB�dCdD�dEdF�dGdH�dIdJ�dKdL�dMdN�dOdP�dQdR�dSdT�dUdV�dWdX��i dYdZ�d[d\�d]d^�d_d`�dadb�dcdd�dedf�dgdh�didj�dkdl�dmdn�dodp�dqdr�dsdt�dudv�dwdx�dydz��d{d:d|��Z d}d~dd��Z d�d�� Z d�d�� Z e � Z!G d�d�� d��Z"d�d�� Z#G d�d�� d��Z$d�d�d��Z%d�d�� Z&d�d�d��Z'G d�d�� d�e(�Z)G d�d�� d�e)�Z*G d�d�� d��Z+G d�d�� d�e+�Z,G d�d�� d�e+�Z-G d�d�� d�ej.�Z/G d�d�� d�ej0�Z1G d�d�� d�e
�Z2G d�d�� d��Z3G d�d�� d�e
�Z4G d�d�� d�e
�Z5G d�d�� d�e1�Z6G d�d�� d�e
�Z7G d�d�� d�e6�Z8e e4e5�G d�d�� d���Z9dS )�zP
Session Initialization Protocol.
Documented in RFC 2543.
[Superseded by 3261]
� N)� OrderedDict)�Dict�List)� Interface� implementer)�cred)�defer�protocol�reactor)�basic)�logi� �i�m�e�l�c�f�s�t�v) �call-id�contactzcontent-encoding�content-lengthz content-type�from�subject�to�via�d �Trying� �Ringing� zCall Is Being Forwarded� �Queued� zSession Progress�� �OKi, zMultiple Choicesi- zMoved Permanentlyi. zMoved Temporarilyi/ z See Otheri1 z Use Proxyi| zAlternative Service� z Bad Request� � Unauthorizedi� zPayment Requiredi� � Forbidden� z Not Foundi� zMethod Not Allowedi� zNot Acceptablei� z Proxy Authentication Requiredi� zRequest Timeouti� �Conflicti� �Gonei� zLength Requiredi� zRequest Entity Too Largei� zRequest-URI Too Largei� zUnsupported Media Typei� zUnsupported URI Schemei� z
Bad Extensioni� zExtension Requiredi� zInterval Too Briefi� zTemporarily Unavailablei� z Call/Transaction Does Not Existi� z
Loop Detectedi� z
Too Many Hopsi� zAddress Incompletei� � Ambiguousi� z Busy Herei� zRequest Terminatedi� zNot Acceptable Herei� zRequest Pendingi� �Undecipherable�� zInternal Server Error� zNot Implementedi� z Bad Gatewayi� zService Unavailablei� zServer Time-outi� zSIP Version not supportedi zMessage Too LargeiX zBusy Everywherei[ �DeclinezDoes not exist anywhere)i\ i^ �CSeqzCall-IDzWWW-Authenticate)�cseqr �www-authenticatec C s d� dd� | �d�D ��S )zK
Capitalize a string, making sure to treat '-' as a word separator
�-c S s g | ]}|� � �qS � )�
capitalize)�.0�xr7 r7 �7/usr/lib/python3/dist-packages/twisted/protocols/sip.py�
<listcomp>p s z"dashCapitalize.<locals>.<listcomp>)�join�split�r r7 r7 r; �dashCapitalizel s r@ c C s2 | d | d krdkrn | S | dd� S | S )Nr ����"� r7 r? r7 r7 r; �unqs s �rD c @ sL e Zd ZdZeddddeddfdd�Zedd� �Zej d d� �Zd
d � Z
dS ) �Viaa�
A L{Via} is a SIP Via header, representing a segment of the path taken by
the request.
See RFC 3261, sections 8.1.1.7, 18.2.2, and 20.42.
@ivar transport: Network protocol used for this leg. (Probably either "TCP"
or "UDP".)
@type transport: C{str}
@ivar branch: Unique identifier for this request.
@type branch: C{str}
@ivar host: Hostname or IP for this leg.
@type host: C{str}
@ivar port: Port used for this leg.
@type port C{int}, or None.
@ivar rportRequested: Whether to request RFC 3581 client processing or not.
@type rportRequested: C{bool}
@ivar rportValue: Servers wishing to honor requests for RFC 3581 processing
should set this parameter to the source port the request was received
from.
@type rportValue: C{int}, or None.
@ivar ttl: Time-to-live for requests on multicast paths.
@type ttl: C{int}, or None.
@ivar maddr: The destination multicast address, if any.
@type maddr: C{str}, or None.
@ivar hidden: Obsolete in SIP 2.0.
@type hidden: C{bool}
@ivar otherParams: Any other parameters in the header.
@type otherParams: C{dict}
�UDPNFc
K s� || _ || _|| _|| _|| _|| _|du r%tjdtdd� d| _ d| _
n |du r0d| _ d| _
n|t u r;d| _ d| _
n|| _ d| _
|| _ | | _
|
| _dS )a(
Set parameters of this Via header. All arguments correspond to
attributes of the same name.
To maintain compatibility with old SIP
code, the 'rport' argument is used to determine the values of
C{rportRequested} and C{rportValue}. If None, C{rportRequested} is set
to True. (The deprecated method for doing this is to pass True.) If an
integer, C{rportValue} is set to the given value.
Any arguments not explicitly named here are collected into the
C{otherParams} dict.
Tz+rport=True is deprecated since Twisted 9.0.� )�
stacklevelNF)� transport�host�port�ttl�hidden�received�warnings�warn�DeprecationWarning�
rportValue�rportRequested�_absent�branch�maddr� otherParams) �selfrJ rK rI rL rM rN �rportrU rV �kwr7 r7 r; �__init__� s2 �
z Via.__init__c C s" | j dkrdS | jdur| jS dS )zG
Returns the rport value expected by the old SIP code.
TN)rS rR �rX r7 r7 r; rY � s
z Via.rportc C s || _ d| _dS )z�
L{Base._fixupNAT} sets C{rport} directly, so this method sets
C{rportValue} based on that.
@param newRPort: The new rport value.
@type newRPort: C{int}
FN)rR rS )rX �newRPortr7 r7 r; rY � s
c C s� d| j � d| j� d| j� �}| jr|d7 }dD ]}t| |�}|dur,|d|� d|� �7 }q| jr5|d 7 }n
| jdurB|d
| j� �7 }t| j �
� �}|D ]\}}|du rZ|d| 7 }qK|d|� d|� �7 }qK|S ) zI
Serialize this header for use in a request or response.
zSIP/2.0/� �:z;hidden)rL rU rV rN N�;�=z;rportz;rport=) rI rJ rK rM �getattrrS rR rY �sortedrW �items)rX r �n�value�etc�kr r7 r7 r; �toString� s$
�
z Via.toString) �__name__�
__module__� __qualname__�__doc__�PORTrT r[ �propertyrY �setterri r7 r7 r7 r; rE | s #
�6
rE c C sL | � d�}|d |dd� }}|� dd�\}}|�� }i }|� d�\}}} |dks.|dkr5td | ����| |d
< d |v rO|� d �\}
} t| �|d < |
|d
< n||d
< |D ]I} | �� � dd�} t| �dkrm| d d} }
n| \} }
| dkrzd|d< qU| � dd�}t|�dkr�|d d}} n |\}} |dv r�t| �} | ||< qUtdi |��S )za
Parse a Via header.
@return: The parsed version of this header.
@rtype: L{Via}
r` r rC Nr^ �/�SIPz2.0zwrong protocol or version: rI r_ rK rJ � rM Tra )rY rL r7 )r> �strip�
ValueError�int�lenrE )rf �parts�sent�params� protocolinfo�by�result�pname�pversionrI rJ rK �p�comment�namer7 r7 r; �parseViaHeader s<
r� c @ sZ e Zd ZdZ d dd�Zdefdd�Zdefdd �Zdefd
d �ZdS )
�URLz
A SIP URL.
Nc
C sn || _ || _|| _|| _|| _|| _|| _|
| _|| _| | _ | d kr&g | _
n| | _
| d kr2i | _ d S | | _ d S �N) �usernamerJ �passwordrK rI �usertype�method�tagrL rV �other�headers)
rX rJ r� r� rK rI r� r� rL rV r� r� r� r7 r7 r; r[ 4 s
z URL.__init__�returnc C s� g }|j }|d� | jd kr#|| j� | jd kr |d| j � |d� || j� | jd kr4|d| j � | jd kr@|d| j � dD ]}t| |�}|d krW|d|� d|� �� qB| jD ]}|d | � q[| jrz|d
� |d � d d
� | j�
� D ��� d� |�S )N�sip:z:%s�@z:%dz;user=%s�rI rL rV r� r� r` ra z;%s�?�&c S s, g | ]\}}t �|�pt|�� d |� ��qS )ra )� specialCases�getr@ )r9 �hr r7 r7 r; r< n s ��z URL.toString.<locals>.<listcomp>rs ) �appendr� r� rJ rK r� rb r� r� r= rd )rX r �wre r r7 r7 r; ri V s<
�
���
z URL.toStringc C s | � � S r� )ri r\ r7 r7 r; �__str__v s z URL.__str__c C s d� | j| j| j| j| j�S )Nz<URL {}:{}@{}:{!r}/{}>)�formatr� r� rJ rK rI r\ r7 r7 r; �__repr__y s �z URL.__repr__) NNNNNNNNNNN) rj rk rl rm r[ �strri r� r� r7 r7 r7 r; r� / s"
�" r� c C s� i }| � d�std| dd� ��| dd� �d�}|d |dd� }}|�dd�}t|�d krU|\}} |�d
d�}
t|
�dkrH|
d |d < n|
d |d < |
d |d < n|d } | �d
d�} t| �dkrl| d |d
< n| d |d
< t| d �|d< |dkr�||d
< |dkr�||d< |D ]j} | |d kr�d| v r�i |d< }
| �dd�\} }|�d�D ]
}|�d�\}}||
|< q�| �dd�}t|�dkr�|�dg ��| � q�|\}}|dkr�||d< q�|dv r�|dkr�t|�}|||< q�|�dg ��| � q�tdi |��S )zV
Return string into URL object.
URIs are of form 'sip:user@example.com'.
r� zunsupported scheme: N� r` r rC r� rG r_ r� r� rJ rK rA r� r� r� ra r� �userr� r� rL r7 )�
startswithru r> rw rv �
setdefaultr� r� )�urlrJ rK �drx �
userdomainrz �udparts�userpass�hostport�upparts�hppartsr� r� r� �headerrh r �nvr� rf r7 r7 r; �parseURL� sV
r� c C s d| _ d| _d| _i | _dS )z*
Clean a URL from a Request line.
N)rI rV rL r� )r� r7 r7 r; �cleanRequestURL� s
r� c C s� | � � } | �d�rdt| ||d�i fS i }| �dd�\}}|� � }|�d�r,|dd� }|�d�r7|dd� }|�d d�\}}t|||d�}|� � }|rb|�d
�D ]}|sVqQ|�d �\} }
|
|| < qQ|rpd|_i |_d|_d|_|||fS ) z�
Return (name, uri, params) for From/To/Contact header.
@param clean: remove unnecessary info, usually for From and To headers.
r� rs �rJ rK �<rC rB NrA �>r` ra ) rt r� r� r> �endswithrL r� rI rV ) �addressrJ rK �cleanrz r� r� � paramstringr rh r r7 r7 r; � parseAddress� s2
r� c @ s e Zd Zddd�ZdS )�SIPErrorNc C s4 |d u rt | }t�| d||f � || _|| _d S )NzSIP error (%d): %s)� statusCodes� Exceptionr[ �code�phrase)rX r� r� r7 r7 r; r[ � s
zSIPError.__init__r� �rj rk rl r[ r7 r7 r7 r; r� � s r� c @ � e Zd ZdZdS )�RegistrationErrorz(
Registration was not possible.
N�rj rk rl rm r7 r7 r7 r; r� � � r� c @ sD e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dS )�Messagez
A SIP message.
Nc C s t � | _d| _d| _d S )Nrs r )r r� �body�finishedr\ r7 r7 r; r[ � s
zMessage.__init__c C s> |� � }t�||�}|dkrt|�| _| j�|g ��|� d S )Nr )�lower� longHeadersr� rv �lengthr� r� r� )rX r� rf r7 r7 r; � addHeader s
zMessage.addHeaderc C s | j |7 _ d S r� )r� �rX �datar7 r7 r; �bodyDataReceived � zMessage.bodyDataReceivedc C s, | j d kr| j t| j�krtd��d| _d S )Nzwrong body lengthrC )r� rw r� ru r� r\ r7 r7 r; �creationFinished s
zMessage.creationFinishedc C sb d| � � }| j�� D ]\}}|D ]}|t�|�p t|�� d|� d�7 }qq |d7 }|| j7 }|S )Nz%s
z: z
)�_getHeaderLiner� rd r� r� r@ r� )rX r re �vsr r7 r7 r; ri s $�
zMessage.toStringc C � t �r� ��NotImplementedErrorr\ r7 r7 r; r� � zMessage._getHeaderLine) rj rk rl rm r� r[ r� r� r� ri r� r7 r7 r7 r; r� � s r� c @ s0 e Zd ZdZd dd�Zdefdd�Zdd � Zd
S ) �Requestz
A Request for a URI
�SIP/2.0c C s< t �| � || _t|t�r|| _d S t|�| _t| j� d S r� )r� r[ r� �
isinstancer� �urir� r� )rX r� r� �versionr7 r7 r; r[ # s
zRequest.__init__r� c C s dt | �| j| j�� f S )Nz<SIP Request %d:%s %s>)�idr� r� ri r\ r7 r7 r; r� , s zRequest.__repr__c C s | j � d| j�� � d�S )Nr^ z SIP/2.0)r� r� ri r\ r7 r7 r; r� / s zRequest._getHeaderLineN)r� �rj rk rl rm r[ r� r� r� r7 r7 r7 r; r� s
r� c @ s0 e Zd ZdZd dd�Zdefdd�Zd d
� ZdS ) �Responsez%
A Response to a URI Request
Nr� c C s* t �| � || _|d krt| }|| _d S r� )r� r[ r� r� r� )rX r� r� r� r7 r7 r; r[ 8 s
zResponse.__init__r� c C s dt | �| jf S )Nz<SIP Response %d:%s>)r� r� r\ r7 r7 r; r� ? r� zResponse.__repr__c C s d| j � d| j� �S )NzSIP/2.0 r^ )r� r� r\ r7 r7 r; r� B s zResponse._getHeaderLine)Nr� r� r7 r7 r7 r; r� 3 s
r� c @ sx e Zd ZdZdZdZdZdZdZdd� Z d d d
�Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zd dd�Zdd� ZdS ) �MessagesParserz�
A SIP messages parser.
Expects dataReceived, dataDone repeatedly,
in that order. Shouldn't be connected to actual transport.
r� rC � firstliner c C s || _ | �� d S r� )�messageReceived�reset)rX �messageReceivedCallbackr7 r7 r; r[ U � zMessagesParser.__init__rs c C s, d| _ d | _d| _d | _d | _| �|� d S )Nr� r )�stater� � bodyReceived�messager� � setLineMode�rX �
remainingDatar7 r7 r; r� Y s zMessagesParser.resetc C s d| _ | �� d S )N�invalid)r� �
setRawModer\ r7 r7 r; �invalidMessagea r� z MessagesParser.invalidMessagec C sb | � � | jdkr dS | jdkr| �� dS | jdkr!| �� dS | j| jk r-| �� dS td��)zI
Clear out any buffered data that may be hanging around.
r� Nr� zthis should never happen)�clearLineBufferr� r� r� � messageDoner� � RuntimeErrorr\ r7 r7 r; �dataDonee s
zMessagesParser.dataDonec C sN zt |t�r |�d�}tj�| |� W d S ty& t�� | � � Y d S w )N�utf-8)
r� r� �encoder � LineReceiver� dataReceivedr� r �errr� r� r7 r7 r; r� y s
�zMessagesParser.dataReceivedc C r� )z2
Expected to create self.message.
r� �rX �liner7 r7 r; �handleFirstLine� � z MessagesParser.handleFirstLinec C s | � � d S r� )r� r� r7 r7 r; �lineLengthExceeded� � z!MessagesParser.lineLengthExceededc C s. t |t�r
|�d�}| jdkr�|�d�s|�d�r)|dd � }|�d�s|�d�s|s-d S z |�dd�\}}}W n tyF | �� Y d S w |dkrj| jrjzt |�}W n tyb | �� Y d S w t
||�| _ n|dkrx| j rxt
||�| _ n| �� d S d | _d S | jd ks�J �|r�|�d�s�|�d
�r�| j\}}|||�� f| _d S | jr�| j j| j� d | _z
|�d d�\}}W n ty� | �� Y d S w ||�� f| _|�� d kr�z
t |�� �| _W d S ty� | �� Y d S w d S d
| _| j�r| j j| j� d | _| jdk�r| �� d S | �� d S )Nr� r� �
�
rC r^ rG r� r� � r_ r r� r )r� �bytes�decoder� r� r> ru r� �acceptResponsesrv r� r� �acceptRequestsr� r� �lstripr� r� r� r� r� )rX r� �a�br r� r� rf r7 r7 r; � lineReceived� sv
� � �
� ��
zMessagesParser.lineReceivedc C s2 | j dksJ �| j�� | �| j� | �|� d S )Nr� )r� r� r� r� r� r� r7 r7 r; r� � s
zMessagesParser.messageDonec C s� | j dv sJ �t|t�r|�d�}| j dkrd S | jd kr%| j�|� d S t|�}| j| j }||krH| j�|d |� � | � ||d � � d S | j|7 _| j�|� | j| jkra| � � d S d S )N)r� r� r� r� )
r� r� r� r� r� r� r� rw r� r� )rX r� �dataLen� expectedLenr7 r7 r; �rawDataReceived� s$
�z MessagesParser.rawDataReceivedN)rs )rj rk rl rm r� r� r� r� �debugr[ r� r� r� r� r� r� r r� r r7 r7 r7 r; r� F s"
C r� c @ s` e Zd ZdZeZdZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� ZdS )�Basez1
Base class for SIP clients and servers.
Fc C s g | _ t| j�| _d S r� )�messagesr� �
addMessage�parserr\ r7 r7 r; r[ � s z
Base.__init__c C s | j �|� d S r� )r r� )rX �msgr7 r7 r; r � � zBase.addMessagec C s� | j �|� | j �� | jD ]*}| �||� | jr&t�d|�� �d|��� t |t
�r2| � ||� q| � ||� qg | jd d �< d S )Nz Received z from )
r r� r� r � _fixupNATr r r ri r� r� �handle_request�handle_response)rX r� �addrr r7 r7 r; �datagramReceived� s
zBase.datagramReceivedc C s� |\}}t |jd d �}|j|kr(||_|j|kr ||_|�� |jd d< d S |jdkr>||_||_|�� |jd d< d S d S )Nr r T)r� r� rJ rN rK rY ri )rX r� �
sourcePeer�srcHost�srcPort� senderViar7 r7 r; r s
�zBase._fixupNATc C �L t |jd d �}|jp|j}|jp|jp| j}t||d�}| �||� dS �zX
Deliver response.
Destination is based on topmost Via header.
r r r� N� r� r� rN rJ rY rK rn r� � sendMessage�rX �responseMessage�destViarJ rK �destAddrr7 r7 r; �deliverResponse s
zBase.deliverResponsec C �2 t |�}dD ]}|j�|g �dd� |j|< q|S �z9
Create a response to a request message.
)r r r r r4 N�r� r� r� �rX r� �request�responser� r7 r7 r; �responseFromRequest! s zBase.responseFromRequestc C sn |j dvr td��| jrt�d|�� �d|��� |�� }t|t�r'|�d�}| j � ||j
|j p2| j f� dS )z�
Send a message.
@param destURL: C{URL}. This should be a *physical* URL, not a logical one.
@param message: The message to send.
)�udpNz only UDP currently supportedzSending z to r� N)
rI r� r r r ri r� r� r� �writerJ rK rn )rX �destURLr� r� r7 r7 r; r + s
zBase.sendMessagec C r� )z�
Override to define behavior for requests received
@type message: C{Message}
@type addr: C{tuple}
r� �rX r� r r7 r7 r; r ; � zBase.handle_requestc C r� )z�
Override to define behavior for responses received.
@type message: C{Message}
@type addr: C{tuple}
r� r' r7 r7 r; r
D r( zBase.handle_responseN)rj rk rl rm rn r r[ r r r r r# r r r
r7 r7 r7 r; r � s
r c @ r� )�IContactz(
A user of a registrar or proxy
Nr� r7 r7 r7 r; r) N r� r) c @ s e Zd Zdd� ZdS )� Registrationc C s || _ || _d S r� )�secondsToExpiry�
contactURL)rX r+ r, r7 r7 r; r[ U �
zRegistration.__init__Nr� r7 r7 r7 r; r* T s r* c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � IRegistryz?
Allows registration of logical->physical URL mapping.
c C � dS )z�
Register the physical address of a logical URL.
@return: Deferred of C{Registration} or failure with RegistrationError.
Nr7 �� domainURL�
logicalURL� physicalURLr7 r7 r; �registerAddress_ � zIRegistry.registerAddressc C r/ )z�
Unregister the physical address of a logical URL.
@return: Deferred of C{Registration} or failure with RegistrationError.
Nr7 r0 r7 r7 r; �unregisterAddressf r5 zIRegistry.unregisterAddressc C r/ )z�
Get registration info for logical URL.
@return: Deferred of C{Registration} object or failure of LookupError.
Nr7 �r2 r7 r7 r; �getRegistrationInfom r5 z IRegistry.getRegistrationInfoN)rj rk rl rm r4 r6 r8 r7 r7 r7 r; r. Z s
r. c @ s e Zd ZdZdd� ZdS )�ILocatorz<
Allow looking up physical address for logical URL.
c C r/ )z�
Return physical URL of server for logical URL of user.
@param logicalURL: a logical C{URL}.
@return: Deferred which becomes URL or fails with LookupError.
Nr7 r7 r7 r7 r; �
getAddressz r5 zILocator.getAddressN)rj rk rl rm r: r7 r7 r7 r; r9 u s r9 c @ sf e Zd ZdZeZdZdefdd�Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� ZdS )�Proxyz
SIP proxy.
Nc C s" |pt �� | _|| _t�| � dS )z�
Create new instance.
@param host: our hostname/IP as set in Via headers.
@param port: our port as set in Via headers.
N)�socket�getfqdnrJ rK r r[ )rX rJ rK r7 r7 r; r[ � s zProxy.__init__c C s t | j| jd�S )z<
Return value of Via header for this proxy.
r� )rE rJ rK r\ r7 r7 r; �getVia� s z Proxy.getViac
s� t �d� j d �}|d u r�j}z|� |�}W n3 ty4 } z����|j� �� W Y d }~d S d }~w tyJ t� � ����d� �� Y d S w |d ur[|�
� �fdd�� d S d S )Nzhandle_%s_requestr0 c s �� ��| j� ��S r� )r r# r� )r �r� rX r7 r; �<lambda>� s �z&Proxy.handle_request.<locals>.<lambda>) rb r� �handle_request_defaultr� r r# r� �
BaseExceptionr r� �
addErrback)rX r� r r r� r r7 r? r; r � s" "� � ��zProxy.handle_requestc s� |\}}� fdd�}� � � }|�� |jd v r t�d� dS |jd �d|�� � t|jd d dd �\}}} � j�|�}
|
� � j
|� |
� � j |� dS )
z�
Default request handler.
Default behaviour for OPTIONS and unknown methods for proxies
is to forward message on to the client.
Since at the moment we are stateless proxy, that's basically
everything.
c s | � � |jd d< � �| |�S )Nr r )ri r� r )r� r� r\ r7 r; �_mungContactHeader� s z8Proxy.handle_request_default.<locals>._mungContactHeaderr zDropping looped message.Nr r rC �r� )
r> ri r� r r �insertr� �locatorr: � addCallbackr rC �_cantForwardRequest) rX r� r r r rD � viaHeaderr� r� �tagsr� r7 r\ r; rA � s
z Proxy.handle_request_defaultc C s, |� t� |jd d= | �| �d|�� d S )Nr r r+ )�trap� LookupErrorr� r r# �rX �errorr� r7 r7 r; rI � s
zProxy._cantForwardRequestc C r r r r r7 r7 r; r � s
zProxy.deliverResponsec C r r r r r7 r7 r; r# � s zProxy.responseFromRequestc C sl t |jd d �}|j|jf| j| jfkr t�d� dS |jd d= |jd s/| �||� dS | �|� dS )z+
Default response handler.
r r z&Dropping incorrectly addressed messageN)r� r� rJ rK r r � gotResponser )rX r� r r r7 r7 r; r
� s
zProxy.handle_responsec C r/ )zJ
Called with responses that are addressed at this server.
Nr7 r' r7 r7 r; rP � r� zProxy.gotResponse)rj rk rl rm rn rG r[ r> r rA rI r r# r
rP r7 r7 r7 r; r; � s r; c @ s e Zd Zdd� Zdd� ZdS )� IAuthorizerc C r/ )z�
Generate a challenge the client may respond to.
@type peer: C{tuple}
@param peer: The client's address
@rtype: C{str}
@return: The challenge string
Nr7 )�peerr7 r7 r; � getChallenge r5 zIAuthorizer.getChallengec C r/ )zf
Create a credentials object from the given response.
@type response: C{str}
Nr7 )r" r7 r7 r; r� r5 zIAuthorizer.decodeN)rj rk rl rS r� r7 r7 r7 r; rQ s rQ c @ s� e Zd ZU dZdZdZi Zeee f e
d< dd� Z dd� Z dd � Z
d
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � ZdS ) �
RegisterProxyzk
A proxy that allows registration for a specific domain.
Unregistered users won't be handled.
N� authorizersc O s" t j| g|�R i |�� i | _d S r� )r; r[ �liveChallenges)rX �argsrZ r7 r7 r; r[ & s
zRegisterProxy.__init__c C s |\}}d S r� r7 �rX r� � host_portrJ rK r7 r7 r; �handle_ACK_request* s z RegisterProxy.handle_ACK_requestc C sJ |\}}| j du r| �|||� dS d|jvr | �|||�S | �|||�S )z`
Handle a registration request.
Currently registration is not proxied.
N�
authorization)�portal�registerr� � unauthorized�loginrX r7 r7 r; �handle_REGISTER_request3 s
z%RegisterProxy.handle_REGISTER_requestc C s� | � d|�}| j�� D ]3\}}|�||f�}|d u r&|�� � d| j� d�}n|�� � d|� d| j� d�}|j�dg ��|� q | � |� d S )Nr( z realm="rB r^ z,realm="r5 )
r# rU rd rS �titlerJ r� r� r� r ) rX r� rJ rK r �scheme�auth�chalrf r7 r7 r; r^ D s zRegisterProxy.unauthorizedc C s� |j d d �d d�}| j�|d �� �}|rez |�|d �}W n ty( � ty> t� � | �
| � d|�� Y d S w | j d| j
7 _ | j�|d t��| j|||��| j|||��tj � d S | �
| � d|�� d S )Nr[ r rC r0 r� r1 )r� r> rU r� r� r� r� rB r r� r r# r� rJ r\ r_ r) rH �_cbLoginrC �_ebLogin)rX r� rJ rK rx r� r r7 r7 r; r_ O s" �
� zRegisterProxy.loginc C s |\}}}| � |||� d S r� )r] )rX �i_a_lr� rJ rK r
r� r r7 r7 r; re b s
zRegisterProxy._cbLoginc C s |� tjj� | �|||� d S r� )rL r rO �UnauthorizedLoginr^ )rX �failurer� rJ rK r7 r7 r; rf g s zRegisterProxy._ebLoginc
C s� t |jd d dd�\}}}d}d|jv r |jd d }|j�ddg�d dkr1| �|||� dS |durIt |||d �\}}}| j�|j||�} n| j�|�} | j| j | j
|f|fd
� dS ) z-
Allow all users to register
r r rC rE Nr �expires�0r� )� callbackArgs� errbackArgs) r� r� r� �
unregister�registryr4 r� r8 � addCallbacks� _cbRegister� _ebRegister)
rX r� rJ rK r� �toURLrz r r, r� r7 r7 r; r] k s
�zRegisterProxy.registerc C sT | � d|�}|jd kr |�d|j�� � |�dd|j � |�dd� | �|� d S )Nr% r rj z%dr rk )r# r, r� ri r+ r )rX � registrationr� r" r7 r7 r; rq � s
zRegisterProxy._cbRegisterc C s |� tt� d S r� )rL r� rM rN r7 r7 r; rr � r
zRegisterProxy._ebRegisterc C s� z t |jd d �}W n ty | �| �d|�� Y d S w |dkrH|dkr*d}nt|�\}}}| j�|j||�}|� | j
|�� | j |� d S d S )Nrj r r' �*)
rv r� ru r r# r� ro r6 r� rH �
_cbUnregisterrC �
_ebUnregister) rX r� rs r rj r, r� rz r� r7 r7 r; rn � s ���zRegisterProxy.unregisterc C s@ | � d|�}|j�dg ��|j�� � |�dd� | �|� d S )Nr% r rj rk )r# r� r� r� r, ri r� r )rX rt r� r r7 r7 r; rv � s zRegisterProxy._cbUnregisterc C s d S r� r7 )rX rt r� r7 r7 r; rw � r� zRegisterProxy._ebUnregister)rj rk rl rm r\ ro rU r r� rQ �__annotations__r[ rZ r` r^ r_ re rf r] rq rr rn rv rw r7 r7 r7 r; rT s$
rT c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�InMemoryRegistryz6
A simplistic registry for a specific domain.
c C s || _ i | _d S r� )�domain�users)rX rz r7 r7 r; r[ � r- zInMemoryRegistry.__init__c C sN |j | jkr
t�td��S |j| jv r | j|j \}}t�|�S t�td��S �Nzunknown domain� no such user)rJ rz r �failrM r� r{ �succeed�rX �userURI�dcr� r7 r7 r; r: � s
zInMemoryRegistry.getAddressc C sd |j | jkr
t�td��S |j| jv r+| j|j \}}t�tt |�
� t � � �|��S t�td��S r| ) rJ rz r r~ rM r� r{ r r* rv �getTime�timer� r7 r7 r; r8 � s z$InMemoryRegistry.getRegistrationInfoc C sT z | j | \}}W n ty t�td�� Y S w |�� | j |= t�td|��S )Nr} r )r{ �KeyErrorr r~ rM �cancelr r* )rX r� r� r� r7 r7 r; �_expireRegistration� s �z$InMemoryRegistry._expireRegistrationc C s� |j | jkrt�d� t�td��S |j | jkr$t�d� t�td��S |j| jv r8| j|j \}}|� d� n t
� d| j |j�}t�d|�
� � d|�
� � �� ||f| j|j< t�tt|�� t�� �|��S )Nz(Registration for domain we don't handle.r+ i z Registered z at )rJ rz r r r r~ r� r� r{ r� r
� callLaterr� ri r r* rv r� r� )rX r1 r2 r3 r� �oldr7 r7 r; r4 � s
z InMemoryRegistry.registerAddressc C s | � |j�S r� )r� r� )rX r1 r2 r3 r7 r7 r; r6 � r� z"InMemoryRegistry.unregisterAddressN)
rj rk rl rm r[ r: r8 r� r4 r6 r7 r7 r7 r; ry � s
ry )NN)NNr ):rm r< r� rO � collectionsr �typingr r �zope.interfacer r �twistedr �twisted.internetr r r
�twisted.protocolsr �twisted.pythonr rn � shortHeadersr� rd rh r r� r� r@ rD �objectrT rE r� r� r� r� r� r� r� r� r� r� r� r� r� �DatagramProtocolr r) r* r. r9 r; rQ rT ry r7 r7 r7 r; �<module> sJ �
�������� �
� � �
��������������� � � � � �!�"�#�$�%�&�'�(�)�*�+�,�-�.�/�0�1�2�3�4�9� +
T7
# ( %d