← Back
Editing: sip.cpython-38.pyc
U ��W[�� � 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 Ze�� D ]\ZZ eee < q�[[ 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) �call-id�contactzcontent-encoding�content-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 iX i[ i\ i^ ZCSeqzCall-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�xr+ r+ �7/usr/lib/python3/dist-packages/twisted/protocols/sip.py� <listcomp>t s z"dashCapitalize.<locals>.<listcomp>)�join�split�r r+ r+ r/ �dashCapitalizep s r4 c C s0 | d | d kr dkr,n n | dd� S | S )Nr ����"� r+ r3 r+ r+ r/ �unqx s r8 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) �selfr= r> r< r? r@ rA �rportrH rI �kwr+ r+ r/ �__init__� s2 �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)rF rE �rK 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)rE rF )rK ZnewRPortr+ r+ r/ � _setrport� s z Via._setrportc C s� d| j | j| jf }| jr"|d7 }dD ]&}t| |�}|dk r&|d||f 7 }q&| jr^|d7 }n| jdk rx|d| jf 7 }t| j � � �}|D ].\}}|dkr�|d| 7 }q�|d||f 7 }q�|S ) zI Serialize this header for use in a request or response. zSIP/2.0/%s %s:%sz;hidden)r? rH rI rA N�;%s=%sz;rportz ;rport=%s�;) r< r= r> r@ �getattrrF rE rL �sortedrJ �items)rK r �n�value�etc�kr r+ r+ r/ �toString� s" z Via.toString) �__name__� __module__� __qualname__�__doc__�PORTrG rN rP rQ �propertyrL r[ r+ r+ r+ r/ r9 � s � , r9 c C sL | � d�}|d |dd� }}|� dd�\}}|�� }i }|� d�\}}} |dks\|dkrjtd | f ��| |d < d |kr�|� d �\} } t| �|d < | |d < n||d < |D ]�} | �� � dd�} t| �dkr�| d d } } n| \} } | dkr�d|d< q�| � dd�}t|�dk�r |d d }} n|\}} |dk�r8t| �} | ||< q�tf |�S )za Parse a Via header. @return: The parsed version of this header. @rtype: L{Via} rS r r7 N� �/ZSIPz2.0z wrong protocol or version: %rr< �:r> r= � r@ T�=)rL r? )r2 �strip� ValueError�int�lenr9 )rX �partsZsent�paramsZ protocolinfoZby�resultZpnameZpversionr< r= r> �pZcomment�namer+ r+ r/ �parseViaHeader s< rp 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) �usernamer= �passwordr> r< �usertype�method�tagr? rI �other�headers) rK r= rs rt r> r< ru rv r? rI rw rx ry r+ r+ r/ rN 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 � dD ]&}t| |�}|d kr�|d||f � q�| jD ]}|d| � q�| jr�|d � |d � d d � | j� � D ��� d � |�S )N�sip:z:%s�@z:%dz;user=%s�r< r? rI rv rw rR z;%s�?�&c S s* g | ]"\}}d t �|�p t|�|f �qS )z%s=%s)� specialCases�getr4 )r- �hr r+ r+ r/ r0 ^ s z URL.toString.<locals>.<listcomp>re ) �appendrs rt r= r> ru rT rx ry r1 rV )rK r �wrW r r+ r+ r/ r[ I s. z URL.toStringc C s | � � S rr )r[ rO 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>)rs rt r= r> r< rO r+ r+ r/ �__repr__f s z URL.__repr__) NNNNNNNNNNN)r\ r] r^ r_ rN r[ r� r� r+ r+ r+ r/ rq - s � rq 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< |D ]�} | |d k�r|d| k�r|i |d< } | �dd�\} }|�d�D ] }|�d�\}}|| |< �q^| �dd�}t|�dk�r�|�dg ��| � �q |\}}|dk�r�||d< n8|dk�r�|dk�r�t|�}|||< n|�dg ��| � �q tf |�S )zV Return string into URL object. URIs are of form 'sip:user@example.com'. rz zunsupported scheme: N� rS r r7 r{ r: rd rs rt r= r> r5 r} ry r~ rf rx �userru r| r? )� startswithrh r2 rj ri � setdefaultr� rq )�urlr= r> �drk Z userdomainrl ZudpartsZuserpassZhostportZuppartsZhppartsrn r� ry �headerrZ r Znvro rX r+ r+ r/ �parseURLk sV r� c C s d| _ d| _d| _i | _dS )z* Clean a URL from a Request line. N)r< rI r? ry )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�|�d �D ] }|s�q�|�d �\} } | || < q�|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. rz re �r= r> �<r7 r6 Nr5 �>rS rf ) rg r� r� r2 �endswithr? ry r< rI ) Zaddressr= r> �cleanrl ro r� Z paramstringr rZ 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� ExceptionrN �code�phrase)rK r� r� r+ r+ r/ rN � s zSIPError.__init__)N�r\ r] r^ rN r+ r+ r+ r/ r� � s r� c @ s e Zd ZdZdS )�RegistrationErrorz( Registration was not possible. N�r\ r] r^ r_ 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 )Nre r )r ry �body�finishedrO r+ r+ r/ rN � s zMessage.__init__c C s> |� � }t�||�}|dkr&t|�| _| j�|g ��|� d S )Nr )�lower� longHeadersr� ri �lengthry r� r� )rK ro rX r+ r+ r/ � addHeader� s zMessage.addHeaderc C s | j |7 _ d S rr )r� �rK �datar+ r+ r/ �bodyDataReceived� s zMessage.bodyDataReceivedc C s, | j d kr"| j t| j�kr"td��d| _d S )Nzwrong body lengthr7 )r� rj r� rh r� rO r+ r+ r/ �creationFinished� s zMessage.creationFinishedc C s^ d| � � }| j�� D ]0\}}|D ]"}|dt�|�p:t|�|f 7 }q"q|d7 }|| j7 }|S )Nz%s z%s: %s z )�_getHeaderLinery rV r r� r4 r� )rK r rW Zvsr r+ r+ r/ r[ s " zMessage.toStringc C s t �d S rr ��NotImplementedErrorrO r+ r+ r/ r� s zMessage._getHeaderLine) r\ r] r^ r_ r� rN r� r� r� r[ 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 rr )r� rN rv � isinstancerq �urir� r� )rK rv r� �versionr+ r+ r/ rN s zRequest.__init__c C s dt | �| j| j�� f S )Nz<SIP Request %d:%s %s>)�idrv r� r[ rO r+ r+ r/ r� s zRequest.__repr__c C s d| j | j�� f S )Nz %s %s SIP/2.0)rv r� r[ rO r+ r+ r/ r� $ s zRequest._getHeaderLineN)r� �r\ r] r^ r_ rN r� r� r+ r+ r+ r/ r� s r� c @ s* e Zd ZdZd dd�Zdd� Zdd � ZdS ) �Responsez% A Response to a URI Request Nr� c C s* t �| � || _|d kr t| }|| _d S rr )r� rN r� r� r� )rK r� r� r� r+ r+ r/ rN . s zResponse.__init__c C s dt | �| jf S )Nz<SIP Response %d:%s>)r� r� rO r+ r+ r/ r� 6 s zResponse.__repr__c C s d| j | jf S )Nz SIP/2.0 %s %s)r� r� rO r+ r+ r/ r� : s zResponse._getHeaderLine)Nr� 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. r� r7 � firstliner c C s || _ | �� d S rr )�messageReceived�reset)rK ZmessageReceivedCallbackr+ r+ r/ rN N s zMessagesParser.__init__re c C s, d| _ d | _d| _d | _d | _| �|� d S )Nr� r )�stater� � bodyReceived�messager� Z setLineMode�rK Z remainingDatar+ r+ r/ r� S s zMessagesParser.resetc C s d| _ | �� d S )N�invalid)r� � setRawModerO 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� � RuntimeErrorrO r+ r+ r/ �dataDonea s zMessagesParser.dataDonec C sH z&t |t�r|�d�}tj�| |� W n t�� | �� Y nX d S )N�utf-8) r� r �encoder � LineReceiver� dataReceivedr �errr� r� r+ r+ r/ r� v s zMessagesParser.dataReceivedc C s t �dS )z2 Expected to create self.message. Nr� �rK �liner+ r+ r/ �handleFirstLine� s z MessagesParser.handleFirstLinec C s | � � d S rr )r� r� r+ r+ r/ �lineLengthExceeded� s z!MessagesParser.lineLengthExceededc C s4 t rt|t�r|�d�}| jdk� r�|�d�s8|�d�rF|dd � }q$|sNd S z|�dd�\}}}W n tk r� | �� Y d S X |dkr�| j r�z t |�}W n tk r� | �� Y 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 | _z|�d d�\}}W n tk �r� | �� Y d S X ||�� f| _|�� d k�r0zt |�� �| _W n tk �r� | �� Y d S X nBd | _| j�r| j j| j� d | _| jdk�r(| �� d S | �� d S )Nr� r� � � r7 rb r: r� ry � rd r r� r )r r� �bytes�decoder� r� r2 rh r� �acceptResponsesri r� r� �acceptRequestsr� �AssertionErrorr� �lstripr� r� r� r� r� )rK r� �a�br r� ro rX r+ r+ r/ � lineReceived� sj zMessagesParser.lineReceivedc C s2 | j dkst�| j�� | �| j� | �|� d S )Nr� )r� r� r� r� r� r� 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� r� r� ) r� r� r r� r� r� r� r� r� rj r� r� )rK r� ZdataLenZ expectedLenr+ r+ r/ �rawDataReceived� s" z MessagesParser.rawDataReceivedN)re )re )r\ r] r^ r_ r� r� r� r� �debugrN 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 rr )�messagesr� � addMessage�parserrO r+ r+ r/ rN � s z Base.__init__c C s | j �|� d S rr )r� r� )rK �msgr+ r+ r/ r� � s zBase.addMessagec C s� | j �|� | j �� | jD ]P}| �||� | jrHt�d|�� |f � t |t �r`| � ||� q | � ||� q g | jd d �< d S )NzReceived %r from %r) r� r� r� r� � _fixupNATr� r r� r[ r� r� �handle_request�handle_response)rK 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)rp ry r= rA r> rL r[ )rK 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 r� N� rp ry rA r= rL r> r` rq � sendMessage�rK ZresponseMessageZdestViar= r> ZdestAddrr+ r+ r/ �deliverResponse s zBase.deliverResponsec C s2 t |�}dD ] }|j�|g �dd� |j|< q |S �z9 Create a response to a request message. )r r r r r( N�r� ry r� �rK r� Zrequest�responsero 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 %rr� N) r< r� r� r r� r[ r� r r� �writer= r> r` )rK 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} Nr� �rK 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} Nr� r� r+ r+ r/ r� O s zBase.handle_responseN)r\ r] r^ r_ r` r� rN 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 Nr� r+ r+ r+ r/ r� Z s r� c @ s e Zd Zdd� ZdS )� Registrationc C s || _ || _d S rr )�secondsToExpiry� contactURL)rK r� r� r+ r+ r/ rN b s zRegistration.__init__Nr� 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/ �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)r\ r] r^ r_ 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)r\ r] r^ r_ 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)�socketZgetfqdnr= r> r� rN )rK r= r> r+ r+ r/ rN � s zProxy.__init__c C s t | j| jd�S )z< Return value of Via header for this proxy. r� )r9 r= r> rO r+ r+ r/ �getVia� s z Proxy.getViac s� t �d� j d �}|d kr �j}z|� |�}W nZ tk rd } z����|j� �� W 5 d }~X Y nB t�� ����d� �� Y n X |d k r�|� � �fdd�� d S )Nzhandle_%s_requestr& c s �� ��| j� ��S rr )r� r� r� )r �r� rK r+ r/ �<lambda>� s z&Proxy.handle_request.<locals>.<lambda>) rT rv �handle_request_defaultr� r� r� r� r r� � addErrback)rK r� r� r r� r r+ r 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 )r[ ry r� )r� r� rO r+ r/ �_mungContactHeader� s z8Proxy.handle_request_default.<locals>._mungContactHeaderr zDropping looped message.Nr r r7 �r� ) r r[ ry r r� �insertr� �locatorr� � addCallbackr� r �_cantForwardRequest) rK r� r� r� r� r Z viaHeaderro r� Ztagsr� r+ rO r/ r � s z Proxy.handle_request_defaultc C s, |� t� |jd d= | �| �d|�� d S )Nr r r% )�trap� LookupErrorry r� r� �rK �errorr� r+ r+ r/ r � s zProxy._cantForwardRequestc C sL t |jd d �}|jp |j}|jp.|jp.| j}t||d�}| �||� dS r� r� r� r+ r+ r/ r� � s zProxy.deliverResponsec C s2 t |�}dD ] }|j�|g �dd� |j|< q |S r� r� r� 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)rp ry r= r> r r� � gotResponser� )rK 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+ r� r+ r+ r/ r s zProxy.gotResponse)r\ r] r^ r_ r` r rN 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)r\ r] r^ 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 rr )r rN ZliveChallenges)rK �argsrM r+ r+ r/ rN @ s zRegisterProxy.__init__c C s |\}}d S rr r+ �rK r� Z host_portr= r> 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�registerry � unauthorized�loginr r+ r+ r/ �handle_REGISTER_requestO s z%RegisterProxy.handle_REGISTER_requestc C s� | � d|�}t| j�D ]Z\}}|�||f�}|d krHd|�� | jf }nd|�� || jf }|j�dg ��|� q| � |� d S )Nr$ z %s realm="%s"z%s %s,realm="%s"r) ) r� r � authorizersr �titler= ry r� r� r� ) rK r� r= r> r ZschemeZauthZchalrX r+ r+ r/ r a s zRegisterProxy.unauthorizedc C s� |j d d �d d�}| j�|d �� �}|r�z|�|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 r7 r& r{ r' )ry r2 r r� r� r� r� r r� r� r� rs r= r r r� r �_cbLoginr �_ebLogin)rK r� r= r> rk r� r r+ r+ r/ r m s2 � ��zRegisterProxy.loginc C s |\}}}| � |||� d S rr )r )rK Zi_a_lr� r= r> r r� r r+ r+ r/ r � s zRegisterProxy._cbLoginc C s |� tjj� | �|||� d S rr )r r r ZUnauthorizedLoginr )rK Zfailurer� r= r> 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 r7 r Nr �expires�0r� )Z callbackArgsZ errbackArgs) r� ry r� � unregister�registryr� r� r� Z addCallbacks� _cbRegister� _ebRegister) rK r� r= r> ro �toURLrl 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%dr r# )r� r� r� r[ r� r� )rK � registrationr� r� r+ r+ r/ r& � s zRegisterProxy._cbRegisterc C s |� tt� d S rr )r r� r r r+ r+ r/ r' � s zRegisterProxy._ebRegisterc C s� zt |jd d �}W n& tk r< | �| �d|�� Y nPX |dkr�|dkrTd}nt|�\}}}| j�|j||�}|� | j |�� | j |� d S )Nr" r r# �*) ri ry rh r� r� r� r% r� r� r � _cbUnregisterr � _ebUnregister) rK r� r( r r" r� ro rl r� r+ r+ r/ r$ � s �zRegisterProxy.unregisterc C s@ | � d|�}|j�dg ��|j�� � |�dd� | �|� d S )Nr" r r" r# )r� ry r� r� r� r[ r� r� )rK r) r� r� r+ r+ r/ r+ � s zRegisterProxy._cbUnregisterc C s d S rr r+ )rK r) r� r+ r+ r/ r, � s zRegisterProxy._ebUnregister)r\ r] r^ r_ r r% r rN 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 rr )�domain�users)rK r. r+ r+ r/ rN � s zInMemoryRegistry.__init__c C sR |j | jkrt�td��S |j| jkr@| j|j \}}t�|�S t�td��S d S �Nzunknown domain� no such user)r= r. r �failr rs r/ �succeed�rK ZuserURI�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 r0 ) r= r. r r2 r rs r/ r3 r� ri �getTime�timer4 r+ r+ r/ r� � s z$InMemoryRegistry.getRegistrationInfoc C sV z| j | \}}W n" tk r4 t�td�� Y S X |�� | j |= t�td|��S )Nr1 r )r/ �KeyErrorr r2 r Zcancelr3 r� )rK rs r5 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.r% i zRegistered %s at %s)r= r. r r� r r2 r� rs r/ r� r Z callLaterr9 r[ r3 r� ri r6 r7 )rK r� r� r� r5 �oldr+ r+ r/ r� � s z InMemoryRegistry.registerAddressc C s | � |j�S rr )r9 rs )rK r� r� r� r+ r+ r/ r� s z"InMemoryRegistry.unregisterAddressN) r\ r] r^ r_ rN r� r� r9 r� r� r+ r+ r+ r/ r- � s r- )NN)NNr );r_ r r7 rB Zzope.interfacer r � collectionsr Ztwistedr Ztwisted.internetr r r Ztwisted.protocolsr Ztwisted.pythonr Ztwisted.python.compatr r r r` Z shortHeadersr� rV rZ r r� r r4 r8 �objectrG r9 rp rq 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 !
Save File
Cancel