U
    ¹êW[¢“  ã                6   @   s’  d Z ddlZddlZddlZddlmZmZ ddlmZ ddl	m
Z
 ddlmZmZmZ ddlmZ ddlmZ dd	lmZmZmZ d
Zddddddddddœ	Zi Ze ¡ D ]\ZZeee< q¬[[dddddddd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„ diej/ƒ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ƒZ9ee5e6ƒG dzd{„ d{ƒƒZ:dS )~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Úcontactzcontent-encodingúcontent-lengthzcontent-typeÚfromZsubjectÚtoÚviaZTryingZRingingzCall Is Being ForwardedZQueuedzSession ProgressZOKzMultiple ChoiceszMoved PermanentlyzMoved Temporarilyz	See Otherz	Use ProxyzAlternative ServicezBad RequestZUnauthorizedzPayment RequiredZ	Forbiddenz	Not FoundzMethod Not AllowedzNot AcceptablezProxy Authentication RequiredzRequest TimeoutZConflictZGonezLength RequiredzRequest Entity Too LargezRequest-URI Too LargezUnsupported Media TypezUnsupported URI SchemezBad ExtensionzExtension RequiredzInterval Too BriefzTemporarily UnavailablezCall/Transaction Does Not ExistzLoop DetectedzToo Many HopszAddress IncompleteZ	Ambiguousz	Busy HerezRequest TerminatedzNot Acceptable HerezRequest PendingZUndecipherablezInternal Server ErrorzNot ImplementedzBad GatewayzService UnavailablezServer Time-outzSIP Version not supportedzMessage Too LargezBusy EverywhereZDeclinezDoes 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^  ZCSeqzCall-IDzWWW-Authenticate)Úcseqr   úwww-authenticatec                 C   s   d  dd„ |  d¡D ƒ¡S )zK
    Capitalize a string, making sure to treat '-' as a word separator
    ú-c                 S   s   g | ]}|  ¡ ‘qS © )Ú
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   krdkr,n n| dd… S | S )Nr   éÿÿÿÿú"é   r+   r3   r+   r+   r/   Úunqx   s     r8   c                   @   sL   e Zd ZdZeddddeddfdd„Zdd„ Zd	d
„ Ze	eeƒZ
dd„ ZdS )Ú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}
    ZUDPNFc
                 K   s˜   || _ || _|| _|| _|| _|| _|dkrJtjdtdd d| _	d| _
n8|dkr`d| _	d| _
n"|tkrvd| _	d| _
n|| _	d| _
|| _|	| _|
| _dS )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 dkrdS | jdk	r| jS dS dS )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| _dS )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   ZnewRPortr+   r+   r/   Ú	_setrportÚ   s    zVia._setrportc                 C   s¾   d| j | j| jf }| jr"|d7 }dD ]&}t| |ƒ}|dk	r&|d||f 7 }q&| jr^|d7 }n| jdk	rx|d| jf 7 }t| j	 
¡ ƒ}|D ].\}}|dkr¨|d| 7 }qŠ|d||f 7 }qŠ|S )	zI
        Serialize this header for use in a request or response.
        zSIP/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 |dd…  }}|  dd¡\}}| ¡ }i }|  d¡\}}}	|dks\|dkrjtd	| f ƒ‚|	|d
< d|krž|  d¡\}
}t|ƒ|d< |
|d< n||d< |D ]–}| ¡   dd¡}t|ƒdkrÚ|d d }}n|\}}|dkrôd|d< qª|  dd¡}t|ƒdkr|d d }} n|\}} |dkr8t| ƒ} | ||< qªtf |ŽS )za
    Parse a Via header.

    @return: The parsed version of this header.
    @rtype: L{Via}
    rS   r   r7   Nú ú/ZSIPz2.0zwrong protocol or version: %rr<   ú:r>   r=   Ú r@   Tú=)rL   r?   )r2   ÚstripÚ
ValueErrorÚintÚlenr9   )rX   ÚpartsZsentÚparamsZprotocolinfoZbyÚresultZpnameZpversionr<   r=   r>   ÚpZcommentÚnamer+   r+   r/   ÚparseViaHeader  s<    



rp   c                   @   s2   e Zd ZdZddd„Zdd„ Zdd„ Zd	d
„ ZdS )ÚURLz
    A SIP URL.
    Nc                 C   sl   || _ || _|| _|| _|| _|| _|| _|
| _|| _|	| _	|d krLg | _
n|| _
|d krbi | _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ƒ | jd krF|| jƒ | jd kr>|d| j ƒ |dƒ || jƒ | jd krh|d| j ƒ | jd kr€|d| j ƒ dD ]&}t| |ƒ}|d kr„|d||f ƒ q„| jD ]}|d| ƒ q²| jrð|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 ‘qS )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| jf 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"td| dd…  ƒ‚| dd…  d¡}|d |dd…  }}| dd¡}t|ƒd	krª|\}}	| d
d¡}
t|
ƒdkr|
d |d< q²|
d |d< |
d |d< n|d }	|	 d
d¡}t|ƒdkrØ|d |d< n|d |d< t|d ƒ|d< |dkr||d< |dkr||d< |D ]æ}||d kr|d|kr|i  |d< }| dd¡\}}| d¡D ]}| d¡\}}|||< q^| dd¡}t|ƒdkr¬| dg ¡ |¡ q|\}}|dkrÈ||d< n8|dkrî|dkrät|ƒ}|||< n| dg ¡ |¡ qtf |ŽS )zV
    Return string into URL object.

    URIs are of form 'sip:user@example.com'.
    rz   zunsupported 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   ZudpartsZuserpassZhostportZuppartsZhppartsrn   r   ry   ÚheaderrZ   r   Znvro   rX   r+   r+   r/   ÚparseURLk  sV    







r   c                 C   s   d| _ d| _d| _i | _dS )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&dt| ||di fS i }|  dd¡\}}|  ¡ }| d¡rX|dd… }| d¡rn|dd… }| d	d¡\}}t|||d}|  ¡ }|rÄ| d
¡D ] }|s¬q¢| d¡\}	}
|
||	< q¢|ràd|_i |_d|_d|_|||fS )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   )Zaddressr=   r>   Úcleanrl   ro   rŠ   Zparamstringr   rZ   r   r+   r+   r/   ÚparseAddress®  s2    



r”   c                   @   s   e Zd Zddd„ZdS )ÚSIPErrorNc                 C   s4   |d krt | }t | d||f ¡ || _|| _d S )NzSIP error (%d): %s)ÚstatusCodesÚ	ExceptionrN   ÚcodeÚphrase)rK   r˜   r™   r+   r+   r/   rN   Ó  s
    zSIPError.__init__)N©r\   r]   r^   rN   r+   r+   r+   r/   r•   Ò  s   r•   c                   @   s   e Zd ZdZdS )ÚRegistrationErrorz(
    Registration was not possible.
    N©r\   r]   r^   r_   r+   r+   r+   r/   r›   Ü  s   r›   c                   @   sD   e Zd ZdZdZdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dd„ Z
dS )ÚMessagez
    A SIP message.
    Nc                 C   s   t ƒ | _d| _d| _d S )Nre   r   )r   ry   ÚbodyÚfinishedrO   r+   r+   r/   rN   ê  s    zMessage.__init__c                 C   s>   |  ¡ }t ||¡}|dkr&t|ƒ| _| j |g ¡ |¡ d S )Nr   )ÚlowerÚlongHeadersr€   ri   Úlengthry   r‰   r‚   )rK   ro   rX   r+   r+   r/   Ú	addHeaderð  s
    
zMessage.addHeaderc                 C   s   |  j |7  _ d S rr   )rž   ©rK   Údatar+   r+   r/   ÚbodyDataReceivedø  s    zMessage.bodyDataReceivedc                 C   s,   | j d kr"| j t| jƒkr"tdƒ‚d| _d S )Nzwrong body lengthr7   )r¢   rj   rž   rh   rŸ   rO   r+   r+   r/   ÚcreationFinishedü  s    zMessage.creationFinishedc                 C   s^   d|   ¡  }| j ¡ D ]0\}}|D ]"}|dt |¡p:t|ƒ|f 7 }q"q|d7 }|| j7 }|S )Nz%s
z%s: %s
z
)Ú_getHeaderLinery   rV   r   r€   r4   rž   )rK   r   rW   Zvsr   r+   r+   r/   r[     s    "
zMessage.toStringc                 C   s   t ‚d S rr   ©ÚNotImplementedErrorrO   r+   r+   r/   r¨     s    zMessage._getHeaderLine)r\   r]   r^   r_   r¢   rN   r£   r¦   r§   r[   r¨   r+   r+   r+   r/   r   ã  s   
r   c                   @   s*   e Zd ZdZd
dd„Zdd„ Zdd„ Zd	S )ÚRequestz
    A Request for a URI
    úSIP/2.0c                 C   s:   t  | ¡ || _t|tƒr"|| _nt|ƒ| _t| jƒ d S rr   )r   rN   rv   Ú
isinstancerq   Úurir   rŽ   )rK   rv   r®   Úversionr+   r+   r/   rN     s    


zRequest.__init__c                 C   s   dt | ƒ| j| j ¡ f S )Nz<SIP Request %d:%s %s>)Úidrv   r®   r[   rO   r+   r+   r/   r…      s    zRequest.__repr__c                 C   s   d| j | j ¡ f S )Nz%s %s SIP/2.0)rv   r®   r[   rO   r+   r+   r/   r¨   $  s    zRequest._getHeaderLineN)r¬   ©r\   r]   r^   r_   rN   r…   r¨   r+   r+   r+   r/   r«     s   

r«   c                   @   s*   e Zd ZdZd
dd„Zdd„ Zdd	„ ZdS )ÚResponsez%
    A Response to a URI Request
    Nr¬   c                 C   s*   t  | ¡ || _|d kr t| }|| _d S rr   )r   rN   r˜   r–   r™   )rK   r˜   r™   r¯   r+   r+   r/   rN   .  s
    
zResponse.__init__c                 C   s   dt | ƒ| jf S )Nz<SIP Response %d:%s>)r°   r˜   rO   r+   r+   r/   r…   6  s    zResponse.__repr__c                 C   s   d| j | jf S )NzSIP/2.0 %s %s)r˜   r™   rO   r+   r+   r/   r¨   :  s    zResponse._getHeaderLine)Nr¬   r±   r+   r+   r+   r/   r²   )  s   
r²   c                   @   sx   e Zd ZdZdZdZdZdZdZdd„ Z	dd	d
„Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zddd„Zdd„ ZdS )Ú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   ZmessageReceivedCallbackr+   r+   r/   rN   N  s    zMessagesParser.__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    zMessagesParser.resetc                 C   s   d| _ |  ¡  d S )NÚinvalid)r·   Ú
setRawModerO   r+   r+   r/   ÚinvalidMessage\  s    zMessagesParser.invalidMessagec                 C   sb   |   ¡  | jdkrdS | jdkr,|  ¡  dS | jdkr@|  ¡  n| j| jk rV|  ¡  ntdƒ‚dS )zI
        Clear out any buffered data that may be hanging around.
        r´   Nrž   zthis should never happen)ZclearLineBufferr·   r¶   r¢   ÚmessageDoner¸   ÚRuntimeErrorrO   r+   r+   r/   ÚdataDonea  s    




zMessagesParser.dataDonec                 C   sH   z&t |tƒr| d¡}tj | |¡ W n   t ¡  |  ¡  Y nX d S )Núutf-8)	r­   r   Úencoder	   ÚLineReceiverÚdataReceivedr
   Úerrr½   r¤   r+   r+   r/   rÄ   v  s    

zMessagesParser.dataReceivedc                 C   s   t ‚dS )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 rt|tƒr| d¡}| jdk rþ| d¡s8| d¡rF|dd … }q$|sNd S z| dd¡\}}}W n tk
r‚   |  ¡  Y d S X |dkrÌ| j	rÌzt
|ƒ}W n tk
r¼   |  ¡  Y d S X t||ƒ| _n(|dkrè| jrèt||ƒ| _n|  ¡  d S d	| _d S | jd	kst‚|rî| d¡s,| d
¡rJ| j\}}||| ¡  f| _n¢| jrf| jj| jŽ  d | _z| dd¡\}}W n  tk
rš   |  ¡  Y d S X || ¡ f| _| ¡ dkr0zt
| ¡ ƒ| _W n  tk
rê   |  ¡  Y d S X nBd| _| jr| jj| jŽ  d | _| jdk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    


zMessagesParser.lineReceivedc                 C   s2   | j dkst‚| j ¡  |  | j¡ |  |¡ d S )Nrž   )r·   rÑ   r¹   r§   rµ   r¶   rº   r+   r+   r/   r¾   Ï  s    
zMessagesParser.messageDonec                 C   sÄ   | j dkst‚tr&t|tƒr&| d¡}| j dkr4d S | jd krL| j |¡ ntt	|ƒ}| j| j
 }||kr’| j |d |… ¡ |  ||d … ¡ d S |  j
|7  _
| j |¡ | j
| jkrÀ|  ¡  d S )N)rž   r»   rÁ   r»   )r·   rÑ   r   r­   rÍ   rÎ   r¢   r¹   r¦   rj   r¸   r¾   )rK   r¥   ZdataLenZ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 Zd ZdZeZdZdd„ Zdd„ Zdd„ Zd	d
„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ ZdS )Ú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    zBase.addMessagec                 C   s€   | j  |¡ | j  ¡  | jD ]P}|  ||¡ | jrHt d| ¡ |f ¡ t	|t
ƒr`|  ||¡ q|  ||¡ qg | jd d …< d S )NzReceived %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    


zBase.datagramReceivedc                 C   sz   |\}}t |jd d ƒ}|j|krN||_|j|kr:||_| ¡ |jd d< n(|jdkrv||_||_| ¡ |jd d< d S )Nr   r   T)rp   ry   r=   rA   r>   rL   r[   )rK   r¹   Ú
sourcePeerÚsrcHostÚsrcPortZ	senderViar+   r+   r/   rÝ     s    


zBase._fixupNATc                 C   sL   t |jd d ƒ}|jp|j}|jp.|jp.| j}t||d}|  ||¡ dS ©zX
        Deliver response.

        Destination is based on topmost Via header.
        r   r   r   N©	rp   ry   rA   r=   rL   r>   r`   rq   ÚsendMessage©rK   ZresponseMessageZdestViar=   r>   ZdestAddrr+   r+   r/   ÚdeliverResponse  s
    zBase.deliverResponsec                 C   s2   t |ƒ}dD ] }|j |g ¡dd… |j|< q|S ©z9
        Create a response to a request message.
        )r   r   r   r   r(   N©r²   ry   r€   ©rK   r˜   ZrequestÚresponsero   r+   r+   r/   ÚresponseFromRequest)  s    zBase.responseFromRequestc                 C   sj   |j dkrtdƒ‚| jr.t d| ¡ |f ¡ | ¡ }t|tƒrJ| d¡}| j  	||j
|jp`| jf¡ dS )z£
        Send a message.

        @param destURL: C{URL}. This should be a *physical* URL, not a logical one.
        @param message: The message to send.
        )ZudpNzonly UDP currently supportedzSending %r to %rrÁ   N)r<   r¿   r×   r
   rÜ   r[   r­   r   rÂ   Úwriter=   r>   r`   )rK   ZdestURLr¹   r¥   r+   r+   r/   rç   4  s    


zBase.sendMessagec                 C   s   t ‚dS )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    zBase.handle_requestc                 C   s   t ‚dS )z…
        Override to define behavior for responses received.

        @type message: C{Message}
        @type addr: C{tuple}
        Nr©   rð   r+   r+   r/   rß   O  s    zBase.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 Zd ZdZdS )ÚIContactz(
    A user of a registrar or proxy
    Nrœ   r+   r+   r+   r/   rñ   Z  s   rñ   c                   @   s   e Zd Zdd„ ZdS )ÚRegistrationc                 C   s   || _ || _d S rr   )ÚsecondsToExpiryÚ
contactURL)rK   ró   rô   r+   r+   r/   rN   b  s    zRegistration.__init__Nrš   r+   r+   r+   r/   rò   a  s   rò   c                   @   s(   e Zd ZdZdd„ Zdd„ Zdd„ ZdS )	Ú	IRegistryz?
    Allows registration of logical->physical URL mapping.
    c                 C   s   dS )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    zIRegistry.registerAddressc                 C   s   dS )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    zIRegistry.unregisterAddressc                 C   s   dS )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 Zd ZdZdd„ ZdS )ÚILocatorz<
    Allow looking up physical address for logical URL.
    c                 C   s   dS )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    zILocator.getAddressN)r\   r]   r^   r_   rÿ   r+   r+   r+   r/   rþ   †  s   rþ   c                   @   sf   e Zd ZdZeZdZdefdd„Zdd„ Zdd„ Zd	d
„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ ZdS )ÚProxyz
    SIP proxy.
    Nc                 C   s"   |p
t  ¡ | _|| _t | ¡ dS )z˜
        Create new instance.

        @param host: our hostname/IP as set in Via headers.
        @param port: our port as set in Via headers.
        N)ÚsocketZgetfqdnr=   r>   rØ   rN   )rK   r=   r>   r+   r+   r/   rN   ž  s    zProxy.__init__c                 C   s   t | j| jd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 kr ˆj}z|ˆ |ƒ}W nZ tk
rd } zˆ ˆ |jˆ ¡¡ W 5 d }~X Y nB   t ¡  ˆ ˆ dˆ ¡¡ Y nX |d k	r¦| 	‡ ‡fdd„¡ d S )Nzhandle_%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    $zProxy.handle_requestc                    sš   |\}}‡ fdd„}ˆ   ¡ }| ¡ |jd kr<t d¡ dS |jd  d| ¡ ¡ t|jd d dd	\}}}	ˆ j |¡}
|
 	ˆ j
|¡ |
 ˆ j|¡ dS )
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   |   ¡ |jd d< ˆ  | |¡S )Nr   r   )r[   ry   rç   )r®   r¹   rO   r+   r/   Ú_mungContactHeaderÐ  s    z8Proxy.handle_request_default.<locals>._mungContactHeaderr   zDropping 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®   Ztagsr‹   r+   rO   r/   r  Å  s    

zProxy.handle_request_defaultc                 C   s,   |  t¡ |jd d= |  |  d|¡¡ d S )Nr   r   r%   )ÚtrapÚLookupErrorry   ré   rî   ©rK   Úerrorr¹   r+   r+   r/   r  ã  s    
zProxy._cantForwardRequestc                 C   sL   t |jd d ƒ}|jp|j}|jp.|jp.| j}t||d}|  ||¡ dS rå   ræ   rè   r+   r+   r/   ré   é  s
    zProxy.deliverResponsec                 C   s2   t |ƒ}dD ] }|j |g ¡dd… |j|< q|S rê   rë   rì   r+   r+   r/   rî   ø  s    zProxy.responseFromRequestc                 C   sl   t |jd d ƒ}|j|jf| j| jfkr8t d¡ dS |jd d= |jd s^|  ||¡ dS |  |¡ dS )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    

zProxy.handle_responsec                 C   s   dS )zJ
        Called with responses that are addressed at this server.
        Nr+   rð   r+   r+   r/   r    s    zProxy.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 Zd Zdd„ Zdd„ ZdS )ÚIAuthorizerc                 C   s   dS )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+   )Zpeerr+   r+   r/   ÚgetChallenge  s    zIAuthorizer.getChallengec                 C   s   dS )zf
        Create a credentials object from the given response.

        @type response: C{str}
        Nr+   )rí   r+   r+   r/   rÎ   *  s    zIAuthorizer.decodeN)r\   r]   r^   r  rÎ   r+   r+   r+   r/   r    s   r  c                   @   s„   e Zd ZdZdZdZi Zdd„ Zdd„ Zdd„ Z	d	d
„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )Ú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   ZliveChallenges)rK   ÚargsrM   r+   r+   r/   rN   @  s    zRegisterProxy.__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 dkr"|  |||¡ n&d|jkr:|  |||¡S |  |||¡S dS )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 krHd| ¡ | jf }nd| ¡ || jf }|j dg ¡ |¡ 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   ZschemeZauthZchalrX   r+   r+   r/   r  a  s    zRegisterProxy.unauthorizedc              	   C   sÜ   |j d d  d d¡}| j |d  ¡ ¡}|rÆz| |d ¡}W n: tk
rV   ‚ Y qØ   t ¡  |  	|  
d|¡¡ Y qØX | jd| j 7  _| j |d t¡ | j|||¡ | j|||¡ tj¡ 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       ÿ   þýzRegisterProxy.loginc                 C   s   |\}}}|   |||¡ d S rr   )r  )rK   Zi_a_lr¹   r=   r>   r   rÓ   r   r+   r+   r/   r   ƒ  s    
zRegisterProxy._cbLoginc                 C   s    |  tjj¡ |  |||¡ d S rr   )r  r   r  ZUnauthorizedLoginr  )rK   Zfailurer¹   r=   r>   r+   r+   r/   r!  ‰  s    zRegisterProxy._ebLoginc           
      C   sº   t |jd d dd\}}}d}d|jkr8|jd d }|j ddg¡d dkr`|  |||¡ nV|dk	rt |||d	\}}}| j |j||¡}	n| j |¡}	|	j| j	| j
|f|fd
 dS )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    
þzRegisterProxy.registerc                 C   sT   |   d|¡}|jd kr:| d|j ¡ ¡ | dd|j ¡ | dd¡ |  |¡ d S )Nr"   r   r"  z%dr   r#  )rî   rô   r£   r[   ró   ré   )rK   Úregistrationr¹   rí   r+   r+   r/   r&  §  s    
zRegisterProxy._cbRegisterc                 C   s   |  tt¡ d S rr   )r  r›   r  r  r+   r+   r/   r'  °  s    zRegisterProxy._ebRegisterc           	   	   C   s   zt |jd d ƒ}W n& tk
r<   |  |  d|¡¡ Y nPX |dkrŒ|dkrTd}nt|ƒ\}}}| 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     ÿzRegisterProxy.unregisterc                 C   s@   |   d|¡}|j dg ¡ |j ¡ ¡ | dd¡ |  |¡ d S )Nr"   r   r"  r#  )rî   ry   r‰   r‚   rô   r[   r£   ré   )rK   r)  r¹   rÜ   r+   r+   r/   r+  Ç  s    zRegisterProxy._cbUnregisterc                 C   s   d S rr   r+   )rK   r)  r¹   r+   r+   r/   r,  Î  s    zRegisterProxy._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 Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dS )ÚInMemoryRegistryz6
    A simplistic registry for a specific domain.
    c                 C   s   || _ i | _d S rr   )ÚdomainÚusers)rK   r.  r+   r+   r/   rN   Ø  s    zInMemoryRegistry.__init__c                 C   sR   |j | jkrt tdƒ¡S |j| jkr@| j|j \}}t |¡S t tdƒ¡S d S ©Nzunknown domainúno such user)r=   r.  r   Úfailr  rs   r/  Úsucceed©rK   ZuserURIÚdcrŠ   r+   r+   r/   rÿ   Ý  s    
zInMemoryRegistry.getAddressc                 C   sh   |j | jkrt tdƒ¡S |j| jkrV| j|j \}}t tt	| 
¡ t ¡  ƒ|ƒ¡S t tdƒ¡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" tk
r4   t tdƒ¡ Y S X | ¡  | j |= t td|ƒ¡S )Nr1  r   )r/  ÚKeyErrorr   r2  r  Zcancelr3  rò   )rK   rs   r5  rŠ   r+   r+   r/   Ú_expireRegistrationñ  s    z$InMemoryRegistry._expireRegistrationc                 C   sÌ   |j | jkr$t d¡ t tdƒ¡S |j | jkrHt d¡ t tdƒ¡S |j| jkrp| j|j \}}| 	d¡ nt
 d| j|j¡}t d| ¡ | ¡ f ¡ ||f| j|j< t tt| ¡ t ¡  ƒ|ƒ¡S )Nz(Registration for domain we don't handle.r%   i  zRegistered %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   Zzope.interfacer   r   Úcollectionsr   Ztwistedr   Ztwisted.internetr   r   r   Ztwisted.protocolsr	   Ztwisted.pythonr
   Ztwisted.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³   ZDatagramProtocolrØ   rñ   rò   rõ   rþ   r   r  r  r-  r+   r+   r+   r/   Ú<module>   sà   ø
Æ>ý ,>
8
$
. /m 	 !