Your IP : 216.73.216.213
B
:��]�� � 6 @ s� d Z ddlZddlZddlZddlmZmZ ddlmZ ddl m
Z
ddl m Z m
Z
mZ ddlmZ ddlmZ dd lmZmZmZ d
Zd d d
ddddddd� Zi Zx e�� D ]\ZZ eee < q�W [[ dddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHd'dI�5Z dJdKdLdM�Z dNdO� Z dPdQ� Z e!� Z"G dRdS� dSe!�Z#dTdU� Z$G dVdW� dW�Z%d|dXdY�Z&dZd[� Z'd}d\d]�Z(G d^d_� d_e)�Z*G d`da� dae*�Z+G dbdc� dc�Z,G ddde� dee,�Z-G dfdg� dge,�Z.G dhdi� diej/�Z0G djdk� dke j1�Z2G dldm� dme�Z3G dndo� do�Z4G dpdq� dqe�Z5G drds� dse�Z6G dtdu� due2�Z7G dvdw� dwe�Z8G dxdy� dye7�Z9ee5e6�G dzd{� d{��Z:dS )~zP
Session Initialization Protocol.
Documented in RFC 2543.
[Superseded by 3261]
� N)� implementer� Interface)� OrderedDict)�cred)�protocol�defer�reactor)�basic)�log)�_PY3� iteritems�unicodei� �i�m�e�l�c�f�s�t�v) zcall-id�contactzcontent-encodingzcontent-lengthz content-type�fromZsubject�to�viaZTryingZRingingzCall Is Being ForwardedZQueuedzSession ProgressZOKzMultiple ChoiceszMoved PermanentlyzMoved Temporarilyz See Otherz Use ProxyzAlternative Servicez Bad RequestZ UnauthorizedzPayment RequiredZ Forbiddenz Not FoundzMethod Not AllowedzNot Acceptablez Proxy Authentication RequiredzRequest TimeoutZConflictZGonezLength RequiredzRequest Entity Too LargezRequest-URI Too LargezUnsupported Media TypezUnsupported URI Schemez
Bad ExtensionzExtension RequiredzInterval Too BriefzTemporarily Unavailablez Call/Transaction Does Not Existz
Loop Detectedz
Too Many HopszAddress IncompleteZ Ambiguousz Busy HerezRequest TerminatedzNot Acceptable HerezRequest PendingZUndecipherablezInternal Server ErrorzNot Implementedz Bad GatewayzService UnavailablezServer Time-outzSIP Version not supportedzMessage Too LargezBusy EverywhereZDeclinezDoes not exist anywhere)5�d � � � � �� i, i- i. i/ i1 i| i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i� i iX i[ i\ i^ ZCSeqzCall-IDzWWW-Authenticate)�cseqzcall-idzwww-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�xr# r# �H/opt/alt/python37/lib64/python3.7/site-packages/twisted/protocols/sip.py�
<listcomp>t s z"dashCapitalize.<locals>.<listcomp>)�join�split)r r# r# r' �dashCapitalizep s r+ c C s0 | d | d kr dkr,n n | dd� S | S )Nr ����"� r# )r r# r# r' �unqx s r/ c @ sL e Zd ZdZeddddeddfdd�Zdd� Zd d
� Ze ee�Z
d 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}
ZUDPNFc
K s� || _ || _|| _|| _|| _|| _|dkrJtjdtdd� d| _ d| _
n8|dkr`d| _ d| _
n"|t krvd| _ 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) �selfr4 r5 r3 r6 r7 r8 �rportr? r@ �kwr# r# r' �__init__� s0 z Via.__init__c C s& | j dkrdS | jdk r | jS dS dS )zG
Returns the rport value expected by the old SIP code.
TN)r= r< )rB r# r# r' � _getrport� s
z
Via._getrportc 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)r< r= )rB ZnewRPortr# r# r' � _setrport� s z
Via._setrportc C s� d| j | j| jf }| jr"|d7 }x.dD ]&}t| |�}|dk r(|d||f 7 }q(W | jrb|d7 }n| jdk r||d| jf 7 }t| j �
� �}x6|D ].\}}|dkr�|d| 7 }q�|d||f 7 }q�W |S ) zI
Serialize this header for use in a request or response.
zSIP/2.0/%s %s:%sz;hidden)r6 r? r@ r8 Nz;%s=%sz;rportz ;rport=%s�;) r3 r4 r5 r7 �getattrr= r< rC �sortedrA �items)rB r �n�value�etc�kr r# r# r' �toString� s"
z Via.toString) �__name__�
__module__� __qualname__�__doc__�PORTr> rE rF rG �propertyrC rP r# r# r# r' r0 � s *
r0 c C sP | � d�}|d |dd� }}|� dd�\}}|�� }i }|� d�\}}} |dks\|dkrjtd | f ��| |d
< d |kr�|� d �\}
} t| �|d < |
|d
< n||d
< x�|D ]�} | �� � dd�} t| �dkr�| d d } }
n| \} }
| dkr�d|d< q�| � dd�}t|�dk�r |d d }} n|\}} |dk�r:t| �} | ||< q�W tf |�S )za
Parse a Via header.
@return: The parsed version of this header.
@rtype: L{Via}
rH r r. N� �/ZSIPz2.0z wrong protocol or version: %rr3 �:r5 r4 � r7 T�=)rC r6 )r* �strip�
ValueError�int�lenr0 )rM �parts�sent�paramsZ protocolinfoZby�resultZpnameZpversionr3 r4 r5 �p�comment�namer# r# r' �parseViaHeader s<
rg c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) �URLz
A SIP URL.
Nc
C sl || _ || _|| _|| _|| _|| _|| _|
| _|| _| | _ | d krLg | _
n| | _
| d krbi | _ n| | _ d S )N) �usernamer4 �passwordr5 r3 �usertype�method�tagr6 r@ �other�headers)
rB r4 ri rj r5 r3 rk rl r6 r@ rm rn ro r# r# r' rE 2 s z URL.__init__c C s g }|j }|d� | jd krF|| j� | jd kr>|d| j � |d� || j� | jd krh|d| j � | jd kr�|d| j � x.dD ]&}t| |�}|d kr�|d||f � q�W x| jD ]}|d| � q�W | jr�|d � |d
� d d � | j�
� D ��� d
� |�S )Nzsip:z:%s�@z:%dz;user=%s)r3 r6 r@ rl rm z;%s=%sz;%s�?�&c S s* g | ]"\}}d t �|�p t|�|f �qS )z%s=%s)� specialCases�getr+ )r% �hr r# r# r' r( ^ s z URL.toString.<locals>.<listcomp>rZ ) �appendri rj r4 r5 rk rI rn ro r) rK )rB r �wrL r r# r# r' rP I s.
z URL.toStringc C s | � � S )N)rP )rB r# r# r' �__str__b s z URL.__str__c C s d| j | j| j| j| jf S )Nz<URL %s:%s@%s:%r/%s>)ri rj r4 r5 r3 )rB r# r# r' �__repr__f s z URL.__repr__) NNNNNNNNNNN)rQ rR rS rT rE rP rx ry r# r# r# r' rh - s
rh c C s i }| � d�s"td| dd� ��| dd� �d�}|d |dd� }}|�dd�}t|�d kr�|\}} |�d
d�}
t|
�dkr�|
d |d < q�|
d |d < |
d |d < n|d } | �d
d�} t| �dkr�| d |d
< n | d |d
< t| d �|d< |dk�r||d
< |dk�r||d< x�|D ]�} | |d k�r�d| k�r�i |d< }
| �dd�\} }x*|�d�D ] }|�d�\}}||
|< �qbW | �dd�}t|�dk�r�|�dg ��| � �q |\}}|dk�r�||d< n8|dk�r�|dk�r�t|�}|||< n|�dg ��| � �q W tf |�S )zV
Return string into URL object.
URIs are of form 'sip:user@example.com'.
zsip:zunsupported scheme: N� rH r r. rp r1 rY ri rj r4 r5 r, rq ro rr r[ rn �userrk )r3 r6 r@ rl rm r6 )�
startswithr] r* r_ r^ �
setdefaultrv rh )�urlr4 r5 �dr` Z
userdomainrb ZudpartsZuserpassZhostportZuppartsZhppartsrd ru ro �headerrO r �nvrf rM r# r# r' �parseURLk sV
r� c C s d| _ d| _d| _i | _dS )z*
Clean a URL from a Request line.
N)r3 r@ r6 ro )r~ r# r# r' �cleanRequestURL� s r� c C s� | � � } | �d�r&dt| ||d�i fS i }| �dd�\}}|� � }|�d�rX|dd� }|�d�rn|dd� }|�d d�\}}t|||d�}|� � }|r�x.|�d
�D ] }|s�q�|�d �\} }
|
|| < q�W |r�d|_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.
zsip:rZ )r4 r5 �<r. r- Nr, �>rH r[ ) r\ r| r� r* �endswithr6 ro r3 r@ ) �addressr4 r5 �cleanrb rf r~ Z paramstringr rO r r# r# r' � parseAddress� s2
r� c @ s e Zd Zddd�ZdS )�SIPErrorNc C s4 |d krt | }t�| d||f � || _|| _d S )NzSIP error (%d): %s)� statusCodes� ExceptionrE �code�phrase)rB r� r� r# r# r' rE � s
zSIPError.__init__)N)rQ rR rS rE r# r# r# r' r� � s r� c @ s e Zd ZdZdS )�RegistrationErrorz(
Registration was not possible.
N)rQ rR rS rT r# r# r# r' r� � s 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 )NrZ r )r ro �body�finished)rB r# r# r' rE � s zMessage.__init__c C s> |� � }t�||�}|dkr&t|�| _| j�|g ��|� d S )Nzcontent-length)�lower� longHeadersrt r^ �lengthro r} rv )rB rf rM r# r# r' � addHeader� s
zMessage.addHeaderc C s | j |7 _ d S )N)r� )rB �datar# r# r' �bodyDataReceived� s zMessage.bodyDataReceivedc C s, | j d kr"| j t| j�kr"td��d| _d S )Nzwrong body lengthr. )r� r_ r� r] r� )rB r# r# r' �creationFinished� s zMessage.creationFinishedc C sf d| � � }xB| j�� D ]4\}}x*|D ]"}|dt�|�p>t|�|f 7 }q&W qW |d7 }|| j7 }|S )Nz%s
z%s: %s
z
)�_getHeaderLinero rK rs rt r+ r� )rB r rL �vsr r# r# r' rP s
&
zMessage.toStringc C s t �d S )N)�NotImplementedError)rB r# r# r' r� s zMessage._getHeaderLine) rQ rR rS rT r� rE r� r� r� rP r� r# r# r# r' r� � s
r� c @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) �Requestz
A Request for a URI
�SIP/2.0c C s: t �| � || _t|t�r"|| _nt|�| _t| j� d S )N)r� rE rl �
isinstancerh �urir� r� )rB rl r� �versionr# r# r' rE s
zRequest.__init__c C s dt | �| j| j�� f S )Nz<SIP Request %d:%s %s>)�idrl r� rP )rB r# r# r' ry s zRequest.__repr__c C s d| j | j�� f S )Nz
%s %s SIP/2.0)rl r� rP )rB r# r# r' r� $ s zRequest._getHeaderLineN)r� )rQ rR rS rT rE ry r� r# r# r# r' r� s
r� c @ s* e Zd ZdZd
dd�Zdd� Zdd � ZdS ) �Responsez%
A Response to a URI Request
N�SIP/2.0c C s* t �| � || _|d kr t| }|| _d S )N)r� rE r� r� r� )rB r� r� r� r# r# r' rE . s
zResponse.__init__c C s dt | �| jf S )Nz<SIP Response %d:%s>)r� r� )rB r# r# r' ry 6 s zResponse.__repr__c C s d| j | jf S )Nz
SIP/2.0 %s %s)r� r� )rB r# r# r' r� : s zResponse._getHeaderLine)Nr� )rQ rR rS rT rE ry r� r# r# r# r' r� ) 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.
zSIP/2.0r. � firstliner c C s || _ | �� d S )N)�messageReceived�reset)rB ZmessageReceivedCallbackr# r# r' rE N s zMessagesParser.__init__rZ c C s, d| _ d | _d| _d | _d | _| �|� d S )Nr� r )�stater� � bodyReceived�messager� Z setLineMode)rB �
remainingDatar# r# r' r� S s zMessagesParser.resetc C s d| _ | �� d S )N�invalid)r� �
setRawMode)rB r# r# r' �invalidMessage\ s z MessagesParser.invalidMessagec C sb | � � | jdkrdS | jdkr,| �� dS | jdkr@| �� n | j| jk rV| �� ntd��dS )zI
Clear out any buffered data that may be hanging around.
r� Nr� zthis should never happen)ZclearLineBufferr� r� r� � messageDoner� � RuntimeError)rB r# r# r' �dataDonea s
zMessagesParser.dataDonec C sH y&t |t�r|�d�}tj�| |� W n t�� | �� Y nX d S )Nzutf-8) r� r
�encoder � LineReceiver� dataReceivedr
�errr� )rB r� r# r# r' r� v s
zMessagesParser.dataReceivedc C s t �dS )z2
Expected to create self.message.
N)r� )rB �liner# r# r' �handleFirstLine� s z MessagesParser.handleFirstLinec C s | � � d S )N)r� )rB r� r# r# r' �lineLengthExceeded� s z!MessagesParser.lineLengthExceededc C s0 t rt|t�r|�d�}| jdk� r�x$|�d�s:|�d�rH|dd � }q&W |sRd S y|�dd�\}}}W n tk
r� | �� d S X |dkr�| j r�y t
|�}W n tk
r� | �� d S X t ||�| _ n(|dkr�| j
r�t||�| _ n | �� d S d | _d S | jd k�st�|�r�|�d��s,|�d
��rJ| j\}}|||�� f| _n�| j�rf| j j| j� d | _y|�d d�\}}W n tk
�r� | �� d S X ||�� f| _|�� d k�r,yt
|�� �| _W n tk
�r� | �� d S X nBd
| _| j�r | j j| j� d | _| jdk�r$| �� d S | �� d S )Nzutf-8r� �
�
r. rW r1 zSIP/2.0ro � rY zcontent-lengthr� r )r r� �bytes�decoder� r| r* r] r� �acceptResponsesr^ r� r� �acceptRequestsr� �AssertionErrorr� �lstripr� r� r� r� r� )rB r� �a�br r� rf rM r# r# r' � lineReceived� sj
zMessagesParser.lineReceivedc C s2 | j dkst�| j�� | �| j� | �|� d S )Nr� )r� r� r� r� r� r� )rB r� r# r# r' r� � s
zMessagesParser.messageDonec C s� | j dkst�tr&t|t�r&|�d�}| j dkr4d S | jd krL| j�|� ntt |�}| j| j
}||kr�| j�|d |� � | � ||d � � d S | j
|7 _
| j�|� | j
| jkr�| � � d S )N)r� r� zutf-8r� ) r� r� r r� r� r� r� r� r� r_ r� r� )rB r� ZdataLenZ expectedLenr# r# r' �rawDataReceived� s"
z MessagesParser.rawDataReceivedN)rZ )rZ )rQ rR rS rT r� r� r� r� �debugrE r� r� r� r� r� r� r� r� r� r# r# r# r' r� ? s
D
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 )N)�messagesr� �
addMessage�parser)rB r# r# r' rE � s z
Base.__init__c C s | j �|� d S )N)r� rv )rB �msgr# r# r' r� � s zBase.addMessagec C s� | j �|� | j �� xZ| jD ]P}| �||� | jrJt�d|�� |f � t |t
�rb| � ||� q | � ||� q W g | jd d �< d S )NzReceived %r from %r)
r� r� r� r� � _fixupNATr� r
r� rP r� r� �handle_request�handle_response)rB r� �addrr r# r# r' �datagramReceived� s
zBase.datagramReceivedc C sz |\}}t |jd d �}|j|krN||_|j|kr:||_|�� |jd d< n(|jdkrv||_||_|�� |jd d< d S )Nr r T)rg ro r4 r8 r5 rC rP )rB r� �
sourcePeer�srcHost�srcPortZ senderViar# r# r' r� s
zBase._fixupNATc C sL t |jd d �}|jp |j}|jp.|jp.| j}t||d�}| �||� dS )zX
Deliver response.
Destination is based on topmost Via header.
r r )r4 r5 N) rg ro r8 r4 rC r5 rU rh � sendMessage)rB �responseMessage�destViar4 r5 �destAddrr# r# r' �deliverResponse s
zBase.deliverResponsec C s6 t |�}x(dD ] }|j�|g �dd� |j|< qW |S )z9
Create a response to a request message.
)r r r zcall-idr! N)r� ro rt )rB r� �request�responserf r# r# r' �responseFromRequest) s
zBase.responseFromRequestc C sj |j dkrtd��| jr.t�d|�� |f � |�� }t|t�rJ|�d�}| j � ||j
|j p`| 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.
)ZudpNz only UDP currently supportedzSending %r to %rzutf-8N)
r3 r� r� r
r� rP r� r
r� �writer4 r5 rU )rB ZdestURLr� r� r# r# r' r� 4 s
zBase.sendMessagec C s t �dS )z�
Override to define behavior for requests received
@type message: C{Message}
@type addr: C{tuple}
N)r� )rB r� r� r# r# r' r� E s zBase.handle_requestc C s t �dS )z�
Override to define behavior for responses received.
@type message: C{Message}
@type addr: C{tuple}
N)r� )rB r� r� r# r# r' r� O s zBase.handle_responseN)rQ rR rS rT rU r� rE r� r� r� r� r� r� r� r� r# r# r# r' r� � s
r� c @ s e Zd ZdZdS )�IContactz(
A user of a registrar or proxy
N)rQ rR rS rT r# r# r# r' r� Z s r� c @ s e Zd Zdd� ZdS )� Registrationc C s || _ || _d S )N)�secondsToExpiry�
contactURL)rB r� r� r# r# r' rE b s zRegistration.__init__N)rQ rR rS rE r# r# r# r' r� a s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � IRegistryz?
Allows registration of logical->physical URL mapping.
c C s dS )z�
Register the physical address of a logical URL.
@return: Deferred of C{Registration} or failure with RegistrationError.
Nr# )� domainURL�
logicalURL� physicalURLr# r# r' �registerAddressm s zIRegistry.registerAddressc C s dS )z�
Unregister the physical address of a logical URL.
@return: Deferred of C{Registration} or failure with RegistrationError.
Nr# )r� r� r� r# r# r' �unregisterAddressu s zIRegistry.unregisterAddressc C s dS )z�
Get registration info for logical URL.
@return: Deferred of C{Registration} object or failure of LookupError.
Nr# )r� r# r# r' �getRegistrationInfo} s z IRegistry.getRegistrationInfoN)rQ rR rS rT r� r� r� r# r# r# r' r� h s r� c @ s e Zd ZdZdd� ZdS )�ILocatorz<
Allow looking up physical address for logical URL.
c C s dS )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.
Nr# )r� r# r# r' �
getAddress� s zILocator.getAddressN)rQ rR rS rT r� r# r# r# r' r� � s r� 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" |p
t �� | _|| _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�getfqdnr4 r5 r� rE )rB r4 r5 r# r# r' rE � s zProxy.__init__c C s t | j| jd�S )z<
Return value of Via header for this proxy.
)r4 r5 )r0 r4 r5 )rB r# r# r' �getVia� s z Proxy.getViac
s� t �d� j d �}|d kr �j}y|� |�}W nZ tk
rd } z����|j� �� W d d }~X Y nB t�� ����d� �� Y n X |d k r�|� � �fdd�� d S )Nzhandle_%s_requesti� c s �� ��| j� ��S )N)r� r� r� )r )r� rB r# r' �<lambda>� s z&Proxy.handle_request.<locals>.<lambda>)
rI rl �handle_request_defaultr� r� r� r� r
r� �
addErrback)rB r� r� r r r r# )r� rB r' r� � s $zProxy.handle_requestc s� |\}}� fdd�}� � � }|�� |jd kr<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 )rP ro r� )r� r� )rB r# r' �_mungContactHeader� s z8Proxy.handle_request_default.<locals>._mungContactHeaderr zDropping looped message.Nr r r. )r� )
r� rP ro r
r� �insertr� �locatorr� � addCallbackr� r� �_cantForwardRequest) rB r� r� r� r� r� Z viaHeaderrf r� Ztagsr r# )rB r' r� � s
z Proxy.handle_request_defaultc C s, |� t� |jd d= | �| �d|�� d S )Nr r i� )�trap� LookupErrorro r� r� )rB �errorr� r# r# r' r� � s
zProxy._cantForwardRequestc C sL t |jd d �}|jp |j}|jp.|jp.| j}t||d�}| �||� dS )zX
Deliver response.
Destination is based on topmost Via header.
r r )r4 r5 N) rg ro r8 r4 rC r5 rU rh r� )rB r� r� r4 r5 r� r# r# r' r� � s
zProxy.deliverResponsec C s6 t |�}x(dD ] }|j�|g �dd� |j|< qW |S )z9
Create a response to a request message.
)r r r zcall-idr! N)r� ro rt )rB r� r� r� rf r# r# r' r� � s
zProxy.responseFromRequestc C sl t |jd d �}|j|jf| j| jfkr8t�d� dS |jd d= |jd s^| �||� dS | �|� dS )z+
Default response handler.
r r z&Dropping incorrectly addressed messageN)rg ro r4 r5 r
r� � gotResponser� )rB r� r� r r# r# r' r� s
zProxy.handle_responsec C s dS )zJ
Called with responses that are addressed at this server.
Nr# )rB r� r� r# r# r' r� s zProxy.gotResponse)rQ rR rS rT rU r� rE r� r� r� r� r� r� r� r� r# r# r# r' r� � s
r� c @ s e Zd Zdd� Zdd� ZdS )� IAuthorizerc C s dS )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
Nr# )Zpeerr# r# r' � getChallenge s zIAuthorizer.getChallengec C s dS )zf
Create a credentials object from the given response.
@type response: C{str}
Nr# )r� r# r# r' r� * s zIAuthorizer.decodeN)rQ rR rS r� r� r# r# r# r' r� s r� c @ s� e Zd ZdZdZdZi Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � ZdS ) �
RegisterProxyzk
A proxy that allows registration for a specific domain.
Unregistered users won't be handled.
Nc O s t j| f|�|� i | _d S )N)r� rE ZliveChallenges)rB �argsrD r# r# r' rE @ s zRegisterProxy.__init__c C s |\}}d S )Nr# )rB r� � host_portr4 r5 r# r# r' �handle_ACK_requestE s z RegisterProxy.handle_ACK_requestc C sL |\}}| j dkr"| �|||� n&d|jkr:| �|||�S | �|||�S dS )z`
Handle a registration request.
Currently registration is not proxied.
N�
authorization)�portal�registerro � unauthorized�login)rB r� r� r4 r5 r# r# r' �handle_REGISTER_requestO s
z%RegisterProxy.handle_REGISTER_requestc C s� | � d|�}xht| j�D ]Z\}}|�||f�}|d krJd|�� | jf }nd|�� || jf }|j�dg ��|� qW | � |� d S )Ni� z
%s realm="%s"z%s %s,realm="%s"zwww-authenticate)
r� r � authorizersr� �titler4 ro r} rv r� ) rB r� r4 r5 r �schemeZauthZchalrM r# r# r' r a s zRegisterProxy.unauthorizedc C s� |j d d �d d�}| j�|d �� �}|r�y|�|d �}W n: tk
rV � Y q� t�� | � | �
d|�� Y q�X | j d| j 7 _ | j
�|d t��| j|||��| j|||��tj� n| � | �
d|�� d S )Nr r r. i� rp i� )ro r* r rt r� r� r� r
r� r� r� ri r4 r r r� r� �_cbLoginr� �_ebLogin)rB r� r4 r5 r` r� r r# r# r' r m s
zRegisterProxy.loginc C s |\}}}| � |||� d S )N)r )rB Zi_a_lr� r4 r5 r r� r r# r# r' r
� s
zRegisterProxy._cbLoginc C s |� tjj� | �|||� d S )N)r� r r� ZUnauthorizedLoginr )rB Zfailurer� r4 r5 r# r# r' r � s zRegisterProxy._ebLoginc
C s� t |jd d dd�\}}}d}d|jkr8|jd d }|j�ddg�d dkr`| �|||� nV|dk r�t |||d �\}}}| j�|j||�} n | j�|�} | j| j | j
|f|fd
� dS ) z-
Allow all users to register
r r r. )r� Nr �expires�0)r4 r5 )Z callbackArgsZ errbackArgs) r� ro rt �
unregister�registryr� r� r� Z addCallbacks� _cbRegister� _ebRegister)
rB r� r4 r5 rf �toURLrb r r� r r# r# r' r � s
zRegisterProxy.registerc C sT | � d|�}|jd kr:|�d|j�� � |�dd|j � |�dd� | �|� d S )Nr r r z%dzcontent-lengthr
)r� r� r� rP r� r� )rB � registrationr� r� r# r# r' r � s
zRegisterProxy._cbRegisterc C s |� tt� d S )N)r� r� r� )rB r� r� r# r# r' r � s zRegisterProxy._ebRegisterc C s� yt |jd d �}W n& tk
r< | �| �d|�� Y nPX |dkr�|dkrTd}nt|�\}}}| j�|j||�}|� | j
|�� | j |� d S )Nr r i� �*)
r^ ro r] r� r� r� r r� r� r� �
_cbUnregisterr� �
_ebUnregister) rB r� r r r r� rf rb r r# r# r' r � s zRegisterProxy.unregisterc C s@ | � d|�}|j�dg ��|j�� � |�dd� | �|� d S )Nr r r r
)r� ro r} rv r� rP r� r� )rB r r� r� r# r# r' r � s zRegisterProxy._cbUnregisterc C s d S )Nr# )rB r r� r# r# r' r � s zRegisterProxy._ebUnregister)rQ rR rS rT r r r rE r r r r r
r r r r r r r r# r# r# r' r� 3 s"
r� 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 )N)�domain�users)rB r r# r# r' rE � s zInMemoryRegistry.__init__c C sR |j | jkrt�td��S |j| jkr@| j|j \}}t�|�S t�td��S d S )Nzunknown domainz no such user)r4 r r �failr� ri r �succeed)rB �userURI�dcr~ r# r# r' r� � s
zInMemoryRegistry.getAddressc C sh |j | jkrt�td��S |j| jkrV| j|j \}}t�tt |�
� t � � �|��S t�td��S d S )Nzunknown domainz no such user) r4 r r r r� ri r r r� r^ �getTime�time)rB r r r~ r# r# r' r� � s z$InMemoryRegistry.getRegistrationInfoc C sR y| j | \}}W n tk
r0 t�td��S X |�� | j |= t�td|��S )Nz no such userr )r �KeyErrorr r r� �cancelr r� )rB ri r r~ r# r# r' �_expireRegistration� s z$InMemoryRegistry._expireRegistrationc C s� |j | jkr$t�d� t�td��S |j | jkrHt�d� t�td��S |j| jkrp| j|j \}}|� d� nt
� d| j |j�}t�d|�
� |�
� f � ||f| j|j< t�tt|�� t�� �|��S )Nz(Registration for domain we don't handle.i� i zRegistered %s at %s)r4 r r
r� r r r� ri r r� r Z callLaterr" rP r r� r^ r r )rB r� r� r� r �oldr# r# r' r� � s
z InMemoryRegistry.registerAddressc C s | � |j�S )N)r" ri )rB r� r� r� r# r# r' r�
s z"InMemoryRegistry.unregisterAddressN)
rQ rR rS rT rE r� r� r" r� r� r# r# r# r' r � s
r )NN)NNr );rT r� r r9 Zzope.interfacer r � collectionsr Ztwistedr Ztwisted.internetr r r Ztwisted.protocolsr Ztwisted.pythonr
Ztwisted.python.compatr r r
rU Z shortHeadersr� rK rO r r� rs r+ r/ �objectr> r0 rg rh r� r� r� r� r� r� r� r� r� r� r� ZDatagramProtocolr� r� r� r� r� r� r� r� r r# r# r# r' �<module>
s� ,>
8
$
. /m !