0x1949 Team - FAZEMRX - MANAGER
Edit File: test_sip.cpython-310.pyc
o �b�c � @ s� d Z ddlmZmZ ddlmZmZ ddlmZ ddl m Z zddl m Z W n e y1 dZ Y nw ddlmZ d d �d d � Zd �d d �Zd�d d �Zd�d d �Zd�d d �Zd�d d �Zd�d d �Zd�d d �ZG dd� d�ZG dd� de j�ZG dd� de�Z G dd� de j�Z G d d � d e j�Z G d d � d e j�Z G d d!� d!e j�Z eej!�G d"d#� d#��Z"eej!�G d$d%� d%��Z#G d&d'� d'e j�Z$G d(d)� d)e j�Z%G d*d+� d+ej&�Z'G d,d-� d-e j�Z(dS ).z( Session Initialization Protocol tests. � )�checkers�portal)�defer�reactor)�sip)�unittest)�AsyncioSelectorReactorN)� implementerz z?INVITE sip:foo SIP/2.0 From: mo To: joe Content-Length: 4 abcd� z z-INVITE sip:foo SIP/2.0 From: mo To: joe 1234zHINVITE sip:foo SIP/2.0 From: mo To: joe Content-Length: 4 1234 lalalalz�INVITE sip:foo SIP/2.0 From: mo To: joe Content-Length: 0 INVITE sip:loop SIP/2.0 From: foo To: bar Content-Length: 4 abcdINVITE sip:loop SIP/2.0 From: foo To: bar Content-Length: 4 1234z4SIP/2.0 200 OK From: foo To:bar Content-Length: 0 z.INVITE sip:foo SIP/2.0 f: mo t: joe l: 4 abcdz=INVITE sip:foo SIP/2.0 Via: SIP/2.0/UDP 10.0.0.1:5060;rport a� SIP/2.0 200 OK Via: SIP/2.0/UDP server10.biloxi.com ;branch=z9hG4bKnashds8;received=192.0.2.3 Via: SIP/2.0/UDP bigbox3.site3.atlanta.com ;branch=z9hG4bK77ef4c2312983.1;received=192.0.2.2 Via: SIP/2.0/UDP pc33.atlanta.com ;branch=z9hG4bK776asdhds ;received=192.0.2.1 To: Bob <sip:bob@biloxi.com>;tag=a6c85cf From: Alice <sip:alice@atlanta.com>;tag=1928301774 Call-ID: a84b4c76e66710@pc33.atlanta.com CSeq: 314159 INVITE Contact: <sip:bob@192.0.2.4> Content-Type: application/sdp Content-Length: 0 c @ � e Zd Zdd� ZdS )� TestRealmc G s t jd dd� fS )Nc S s d S �N� r r r �7/usr/lib/python3/dist-packages/twisted/test/test_sip.py�<lambda>� s z)TestRealm.requestAvatar.<locals>.<lambda>)r �IContact)�self�avatarId�mind� interfacesr r r � requestAvatar s zTestRealm.requestAvatarN)�__name__� __module__� __qualname__r r r r r r ~ � r c @ �\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�MessageParsingTestsc C s g | _ t�| j j�| _d S r )�lr �MessagesParser�append�parser�r r r r �setUp� s zMessageParsingTests.setUpc C s | j �|� | j �� d S r �r � dataReceived�dataDone)r �messager r r � feedMessage� s z MessageParsingTests.feedMessagec C sN | � |j|� | � |j�� |� | � |j|� | � |j|� | � |jd� dS )z$ Validate Requests. � N)� assertEqual�method�uri�toString�headers�body�finished)r �mr* r+ r- r. r r r �validateMessage� s z#MessageParsingTests.validateMessagec C �H | j }| �t� | �t|�d� | �|d dddgdgdgd�d � d S � Nr( r �INVITE�sip:foo�mo�joe�4��from�tozcontent-length�abcd)r r' �request1r) �lenr1 �r r r r r � testSimple� � �z MessageParsingTests.testSimplec C sr | j }| �t� | �t� | �t|�d� | �|d dddgdgdgd�d � | �|d dddgdgd �d � d S ) N� r r4 r5 r6 r7 r8 r9 r< r( )r: r; �1234)r r' r= �request2r) r> r1 r? r r r �testTwoMessages� s ��z#MessageParsingTests.testTwoMessagesc C r2 ) Nr( r r4 r5 r6 r7 r8 r9 rC )r r' �request3r) r> r1 r? r r r � testGarbage� rA z MessageParsingTests.testGarbagec C s� | j }| �t� | �t|�d� | �|d dddgdgdgd�d � | �|d dd d gd gdgd�d� | �|d dd d gd gdgd�d� d S )N� r r4 r5 r6 r7 �0r9 � r( zsip:loop�foo�barr8 r< rB rC )r r' �request4r) r> r1 r? r r r �testThreeInOne� s0 ���z"MessageParsingTests.testThreeInOnec C r2 r3 )r r' � request_shortr) r> r1 r? r r r � testShort� rA z MessageParsingTests.testShortc C s� | j }| �t� | �t|�d� |d }| �|jd� | �|jd� | �|jdgdgdgd�� | �|jd � | �|j d� d S ) Nr( r �� �OKrK rL rI r9 rJ ) r r' � response1r) r> �code�phraser- r. r/ �r r r0 r r r �testSimpleResponse� s �z&MessageParsingTests.testSimpleResponsec C sn | j }| �t� | �t|�d� |d }| �|jd d d� | �|jd d d� | �|jd d d� dS ) zv A header may be split across multiple lines. Subsequent lines begin with C{" "} or C{"\t"}. r( r �viazHSIP/2.0/UDP server10.biloxi.com;branch=z9hG4bKnashds8;received=192.0.2.3zVSIP/2.0/UDP bigbox3.site3.atlanta.com;branch=z9hG4bK77ef4c2312983.1;received=192.0.2.2rB zHSIP/2.0/UDP pc33.atlanta.com;branch=z9hG4bK776asdhds ;received=192.0.2.1N)r r' �response_multiliner) r> r- rV r r r �test_multiLine� s � � �z"MessageParsingTests.test_multiLineN) r r r r" r' r1 r@ rE rG rN rP rW rZ r r r r r � s r c @ s e Zd ZdZdd� ZdS )�%MessageParsingFeedDataCharByCharTestsz9 Same as base class, but feed data char by char. c C s$ |D ]}| j �|� q| j �� d S r r# )r r& �cr r r r' s z1MessageParsingFeedDataCharByCharTests.feedMessageN)r r r �__doc__r' r r r r r[ s r[ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�MakeMessageTestsc C s, t �dd�}|�dd� | �|�� d� d S )Nr4 r5 rK rL z$INVITE sip:foo SIP/2.0 Foo: bar )r �Request� addHeaderr) r, �r �rr r r � testRequest s z MakeMessageTests.testRequestc C sB t �dd�}|�dd� |�dd� |�d� | �|�� d� d S ) NrQ rR rK rL zContent-Lengthr8 rC z3SIP/2.0 200 OK Foo: bar Content-Length: 4 1234)r �Responser` �bodyDataReceivedr) r, ra r r r � testResponse s �z MakeMessageTests.testResponsec C s t �d�}| �|�� d� d S )NrQ zSIP/2.0 200 OK )r rd r) r, ra r r r �testStatusCode% s z MakeMessageTests.testStatusCodeN)r r r rc rf rg r r r r r^ s r^ c @ r )�ViaTestsc C s" |� � }| �|t�|�� � � d S r )r, r) r �parseViaHeader)r �v�sr r r �checkRoundtrip+ s zViaTests.checkRoundtripc C sH t �d�}t �d�}| �|j|j� | �|j|j� | �|j|j� d S )Nz SIP/2.0/UDP 192.168.1.1:5060z SIP/2.0/UDP 192.168.1.1:5060)r ri r) � transport�host�port)r �v1�v2r r r �testExtraWhitespace/ s z ViaTests.testExtraWhitespacec C s� d}t �|�}| �|jd� | �|jd� | �|jd� | �|j� | �|j� | � |j � | �|j d� | �|j d� | �|j d� | �|jd� | �|�� d � | �|� d S ) zC Test parsing a Via header with one of everything. zVSIP/2.0/UDP first.example.com:4000;ttl=16;maddr=224.2.0.1 ;branch=a7c6a8dlze (Example)�UDPzfirst.example.comi� � z 224.2.0.1� a7c6a8dlzer zKSIP/2.0/UDP first.example.com:4000;ttl=16;branch=a7c6a8dlze;maddr=224.2.0.1N)r ri r) rm rn ro � assertIsNone�rport� rportValue� assertFalse�rportRequested�ttl�maddr�branch�hiddenr, rl �r rk rj r r r � test_complex6 s$ � �zViaTests.test_complexc C s� d}t �|�}| �|jd� | �|jd� | �|jd� | �|j� | �|j� | � |j � | �|j � | �|j � | �|j � | �|j� | �|�� d� | �|� dS )z3 Test parsing a simple Via header. z SIP/2.0/UDP example.com;hiddenrs � example.com�� z#SIP/2.0/UDP example.com:5060;hiddenN)r ri r) rm rn ro rv rw rx ry rz r{ r| r} � assertTruer~ r, rl r r r r � test_simpleP s zViaTests.test_simplec C s t �d�}| �|� d S )Nr� )r �Viarl �r rj r r r � testSimplerc s zViaTests.testSimplerc C s� t jddd�}| j| jgd�}| �t|�d� | �|d d d� | �|d d t� | �|�� d � | �|j � | �|j � | � |j � d S ) z| Setting rport to True is deprecated, but still produces a Via header with the expected properties. �foo.barT�rw ��offendingFunctionsr( r r& �+rport=True is deprecated since Twisted 9.0.�category� SIP/2.0/UDP foo.bar:5060;rportN) r r� � flushWarnings�test_deprecatedRPortr) r> �DeprecationWarningr, r� rw rz rv rx )r rj �warningsr r r r� g s � z ViaTests.test_deprecatedRPortc C s: t jddd�}| �|�� d� | �|j� | �|j� dS )zU An rport setting of None should insert the parameter with no value. r� Nr� r� )r r� r) r, r� rz rv rx r� r r r � test_rportz s zViaTests.test_rportc C sJ t jddd�}| �|�� d� | �|j� | �|jd� | �|jd� dS )zk An rport numeric setting should insert the parameter with the number value given. r� r( r� z SIP/2.0/UDP foo.bar:5060;rport=1N)r r� r) r, ry rz rx rw r� r r r �test_rportValue� s zViaTests.test_rportValuec C sn d}t �|�}| �|jd� | �|jd� | �|jd� | �|jd� | �|jd� | �|� � � d�d� d S ) Nz8SIP/2.0/UDP 10.0.0.1:5060;received=22.13.1.5;rport=12345rs �10.0.0.1r� z 22.13.1.5i90 z rport=12345���) r ri r) rm rn ro �receivedrw �assertNotEqualr, �findr r r r �testNAT� s zViaTests.testNATc C s" d}t �|�}| �|�� |� dS )zZ Parsing and serializing Via headers with unknown parameters should work. z?SIP/2.0/UDP example.com:5060;branch=a12345b;bogus;pie=deliciousN)r ri r) r, r r r r �test_unknownParams� s zViaTests.test_unknownParamsN) r r r rl rr r� r� r� r� r� r� r� r� r r r r rh * s rh c @ � e Zd Zdd� Zdd� ZdS )�URLTestsc C s$ dD ] }| � t�|��� |� qd S )N)zsip:j.doe@big.comz&sip:j.doe:secret@big.com;transport=tcpz!sip:j.doe@big.com?subject=projectzsip:example.com)r) r �parseURLr, )r �urlr r r � testRoundtrip� s �zURLTests.testRoundtripc C sZ d}t �|�}dddddddd d d d d gfdddd�ff D ] \}}| �t||�|� q d S )Nzesip:user:pass@hosta:123;transport=udp;user=phone;method=foo;ttl=12;maddr=1.2.3.4;blah;goo=bar?a=b&c=d)�username�user)�password�pass)rn �hosta)ro �{ )rm �udp)�usertype�phone)r* rK )r{ � )r| �1.2.3.4�other�blahzgoo=barr- �b�d)�ar\ )r r� r) �getattr)r rk r� �krj r r r � testComplex� s" � � �zURLTests.testComplexN)r r r r� r� r r r r r� � s r� c @ r )� ParseTestsc C s� dddi fdddi fdddi fdddi fdd dd d d �ffD ]"\}}}}t �|�\}}}| �||� | �|�� |� | �||� q d S ) Nz""A. G. Bell" <sip:foo@example.com>z A. G. Bellzsip:foo@example.comzAnon <sip:foo@example.com>�AnonrJ z<sip:foo@example.com>z)foo <sip:foo@example.com>;tag=bar;foo=bazrK rL �baz)�tagrK )r � parseAddressr) r, )r �address�name�urls�params�gname�gurl�gparamsr r r �testParseAddress� s& � �� �zParseTests.testParseAddressN)r r r r� r r r r r� � r r� c @ r )� DummyLocatorc C s t �tjddd��S )N� server.comr� �ro )r �succeedr �URL�r � logicalURLr r r � getAddress� s zDummyLocator.getAddressN�r r r r� r r r r r� � � r� c @ r )�FailingLocatorc C s t �t� �S r )r �fail� LookupErrorr� r r r r� � s zFailingLocator.getAddressNr� r r r r r� � r� r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� ProxyTestsc s0 t �d�� _t� � j_g � _� fdd�� j_d S )N� 127.0.0.1c � � j �| |f�S r ��sentr ��dest�msgr! r r r � � z"ProxyTests.setUp.<locals>.<lambda>)r �Proxy�proxyr� �locatorr� � sendMessager! r r! r r" � s zProxyTests.setUpc C s� t �dd�}|�dt �d��� � |�dt �d��� � |�dd� |�dd � |�d d � | j�|�� d � | �t| j �d � | j d \}}| �|j d� | �|j d� | �|j �� d� | �|j d� | �|jd g d�� d S )Nr4 r5 rX r� z1.2.3.5rK rL r; �<sip:joe@server.com>�contactz<sip:joe@1.2.3.5>�r� r� r( r r� r� )�SIP/2.0/UDP 127.0.0.1:5060�SIP/2.0/UDP 1.2.3.4:5060zSIP/2.0/UDP 1.2.3.5:5060)r r_ r` r� r, r� �datagramReceivedr) r> r� ro rn r+ r* r- �r rb r� r0 r r r �testRequestForward� s" �z ProxyTests.testRequestForwardc C s� t �dd�}|�dt �d��� � |�dd� |�dd� |�d d � | j�|�� d � | jd \}}| �|j d d dg� d S )Nr4 r5 rX r� rK rL r; r� r� z<sip:joe@1.2.3.4>�z1.1.1.1r� r r� z)SIP/2.0/UDP 1.2.3.4:5060;received=1.1.1.1) r r_ r` r� r, r� r� r� r) r- r� r r r �testReceivedRequestForward s �z%ProxyTests.testReceivedRequestForwardc C sH t �d�}|�dt �d��� � | j�|�� d� | �t| j �d� d S )NrQ rX �foo.comr� r ) r rd r` r� r, r� r� r) r> r� ra r r r �testResponseWrongVia s z ProxyTests.testResponseWrongViac C s� t �d�}|�dt �d��� � |�dt jddd��� � | j�|�� d� | �t| j �d� | j d \}}| �|j |j fd � | �|j d� | �|j d d g� d S ) NrQ rX r� � client.com�� r� r� r( r )r� r� zSIP/2.0/UDP client.com:1234)r rd r` r� r, r� r� r) r> r� rn ro rT r- r� r r r �testResponseForward s z ProxyTests.testResponseForwardc C s� t �d�}|�dt �d��� � |�dt jddd��� � | j�|�� d� | �t| j �d� | j d \}}| �|j |j fd � d S ) NrQ rX r� r� r� )r� r� r( r �r� r� ) r rd r` r� r, r� r� r) r> r� rn ro r� r r r �testReceivedResponseForward$ s z&ProxyTests.testReceivedResponseForwardc s� t �d�}|�dt �d��� � g � � fdd�| j_| j�|�� d� | �t � �d� � d \}}| �t |j � dg ��d� | �|j d� d S ) NrQ rX r� c s � � | �S r )r )r� �r r r r 1 s z-ProxyTests.testResponseToUs.<locals>.<lambda>r� r( r ) r rd r` r� r, r� � gotResponser� r) r> r- �getrT )r rb r0 �addrr r� r �testResponseToUs- s zProxyTests.testResponseToUsc C s\ t �dd�}|�dt �d��� � |�dt �d��� � | j�|�� d� | �| jg � d S )Nr4 r5 rX r� r� r� ) r r_ r` r� r, r� r� r) r� ra r r r �testLoop8 s zProxyTests.testLoopc C s� t �dd�}|�dt �d��� � |�dd� t� | j_| j�|�� d� | � t | j �d� | j d \}}| � |j |j fd� | � |jd � | � |jd d g� d S ) Nr4 r5 rX r� r; r� r� r( r i� r� )r r_ r` r� r, r� r� r� r� r) r> r� rn ro rT r- r� r r r �testCantForwardRequest? s z!ProxyTests.testCantForwardRequestN) r r r r" r� r� r� r� r� r� r� r� r r r r r� � s r� c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )�RegistrationTestsc sF t jdd�� _t �d�� _� j � j_� j_g � _� fdd�� j_d S )Nr� �rn �bell.example.comc r� r r� r� r! r r r R r� z)RegistrationTests.setUp.<locals>.<lambda>)r � RegisterProxyr� �InMemoryRegistry�registryr� r� r� r! r r! r r" M s zRegistrationTests.setUpc C s& | j j�� D ]\}}|�� q| `d S r )r� �users�values�cancelr� )r r� r+ r r r �tearDownT s zRegistrationTests.tearDownc C sP t �dd�}|�dd� |�dd� |�dt �d��� � | j�|�� d � d S ) N�REGISTER�sip:bell.example.comr; �sip:joe@bell.example.comr� �sip:joe@client.com:1234rX r� r� �r r_ r` r� r, r� r� ra r r r �registerY s zRegistrationTests.registerc C s\ t �dd�}|�dd� |�dd� |�dt �d��� � |�d d � | j�|�� d � d S ) Nr� r� r; r� r� �*rX r� �expiresrI r� r ra r r r � unregister` s z RegistrationTests.unregisterc C s | � � | jd \}}| �|j|jfd� | �|jd� | �|jd dg� | �|jd dg� | �|jd d g� tt�t krN| � t |jd d �d v � | �t | j j�d � | j jd \}}| �|�� d � | jj�tjd dd��}|�dd� � |�| jd� |S )Nr r� rQ rX �SIP/2.0/UDP client.com:5060r; r� r� �sip:joe@client.com:5060r )i i i i r( r7 r� �r� rn c S s | j | jfS r �rn ro )�desturlr r r r { s z0RegistrationTests.testRegister.<locals>.<lambda>)r r� r) rn ro rT r- �typer r r� �intr> r� r� r, r� r� r� r r� � addCallback)r r� r0 �dcr+ r� r r r � testRegisterh s$ �z RegistrationTests.testRegisterc C s� | � � | �� | jd \}}| �|j|jfd� | �|jd� | �|jd dg� | �|jd dg� | �|jd d g� | �|jd d g� | �| jj i � d S ) Nr( r� rQ rX r r; r� r� r r rI ) r r r� r) rn ro rT r- r� r� �r r� r0 r r r �testUnregister s z RegistrationTests.testUnregisterc C s: t � }t�|�}t�� }|�dd� |�|� || j_d S )NzuserXname@127.0.0.1� passXword)r r �Portalr �'InMemoryUsernamePasswordDatabaseDontUse�addUser�registerCheckerr� )r rb �pr\ r r r � addPortal� s zRegistrationTests.addPortalc C sV | � � | �� | �t| jj�d� | �t| j�d� | jd \}}| �|jd� d S )Nr r( i� )r r r) r> r� r� r� rT r r r r �testFailedAuthentication� s z*RegistrationTests.testFailedAuthenticationc C �b t �dd�}|�dd� |�dd� |�dt �d��� � | j�|�� d � | �t| j �d � d S ) Nr� z sip:wrong.comr; r� r� r rX r� r� r � r r_ r` r� r, r� r� r) r> r� ra r r r �testWrongDomainRegister� � z)RegistrationTests.testWrongDomainRegisterc C r ) Nr� r� r; zsip:joe@foo.comr� r rX r� r� r r ra r r r �testWrongToDomainRegister� r z+RegistrationTests.testWrongToDomainRegisterc C �4 | � � tjddd�}| jj�|�}| �|t� |S )Nr7 r� r �r r r� r� r� r� � assertFailurer� �r r� r� r r r �testWrongDomainLookup� � z'RegistrationTests.testWrongDomainLookupc C r )N�janer� r r r" r r r �testNoContactLookup� r$ z%RegistrationTests.testNoContactLookupN)r r r r" r� r r r r r r r r r# r& r r r r r� L s r� c @ r� )�Clientc C s t j�| � g | _t�� | _d S r )r �Base�__init__r� r �Deferred�deferredr! r r r r) � s zClient.__init__c C s | j �|� | j�| j � d S r )r� r r+ �callback)r �responser� r r r �handle_response� s zClient.handle_responseN)r r r r) r. r r r r r' � s r' c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � LiveTestsc C sz t jdd�| _t �d�| _| j | j_| j_tjd| jdd�| _t � | _ tjd| j dd�| _ | j� � j | j� � jf| _d S )Nr� r� r� r )� interface)r r� r� r� r� r� r � listenUDP� serverPortr' �client� clientPort�getHostrn ro � serverAddressr! r r r r" � s �zLiveTests.setUpc C sH | j j�� D ]\}}|�� qt�| jj�}t�| jj�}t� ||g�S r ) r� r� r� r� r � maybeDeferredr4 � stopListeningr2 � gatherResults)r r� r+ �d1�d2r r r r� � s zLiveTests.tearDownc s� � j �� j}t�dd�}|�dd� |�dd| � |�dtjd|d ��� � � j� tj d� j d d �|� � jj }� fd d �}|� |� |S )Nr� r� r; r� r� �sip:joe@127.0.0.1:%drX r� r� r( r c �* � � t| �d� | d }� � |jd� d S �Nr( r rQ �r) r> rT �r� rb r! r r �check� � z%LiveTests.testRegister.<locals>.check)r4 r5 ro r r_ r` r� r, r3 r� r� r6 r+ r )r r rb r� rA r r! r r � s � zLiveTests.testRegisterc s� � j �� j}t�dd�}|�dd� |�dd| � |�dtjd|d d ��� � � j� j gd �}� � t |�d � � � |d d d� � � |d d t � � j �tjd� jd d�|� � j j}� fdd�}|�|� |S )ap rport is allowed without a value, apparently because server implementors might be too stupid to check the received port against 5060 and see if they're equal, and because client implementors might be too stupid to bind to port 5060, or set a value on the rport parameter they send if they bind to another port. r� r� r; r� r� r< rX r� T)ro rw r� r( r r& r� r� r c r= r> r? r@ r! r r rA rB z)LiveTests.test_amoralRPort.<locals>.check)r4 r5 ro r r_ r` r� r, r� �test_amoralRPortr) r> r� r3 r� r� r6 r+ r )r r rb r� r� rA r r! r rC � s$ �� zLiveTests.test_amoralRPortN)r r r r" r� r rC r r r r r/ � s r/ ))r] � twisted.credr r �twisted.internetr r �twisted.protocolsr � twisted.trialr � twisted.internet.asyncioreactorr � BaseException�zope.interfacer �replacer= rD rF rM rS rO �request_nattedrY r �TestCaser r[ r^ rh r� r� �ILocatorr� r� r� r� r( r' r/ r r r r �<module> s� � ����� �� ���� �� �� �� x eo