U
    •uÄcª2  ã                
   @   s\  d dl Z d dlZd dlZd dlZd dlmZ d dlZe j dd¡rLe 	¡  n e j dd¡rde 
¡  nedƒ‚d dlmZ d dlmZ d dlmZmZmZ er´d d	lmZ d d
lmZ d dlZdZdZ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dddœgZG dd„ dejƒZG dd„ dejƒZG dd „ d ejƒZG d!d"„ d"ejƒZdS )#é    N)ÚMockZUSE_TWISTEDZUSE_ASYNCIOz*need either USE_TWISTED=1 or USE_ASYNCIO=1)Útypes)Úcreate_authenticator)Ú_makepadÚHAS_CRYPTOSIGNÚCryptosignAuthextra)ÚCryptosignKey)Ú
HexEncoderaš  -----BEGIN OPENSSH PRIVATE KEY-----
b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW
QyNTUxOQAAACAa38i/4dNWFuZN/72QAJbyOwZvkUyML/u2b2B1uW4RbQAAAJj4FLyB+BS8
gQAAAAtzc2gtZWQyNTUxOQAAACAa38i/4dNWFuZN/72QAJbyOwZvkUyML/u2b2B1uW4RbQ
AAAEBNV9l6aPVVaWYgpthJwM5YJWhRjXKet1PcfHMt4oBFEBrfyL/h01YW5k3/vZAAlvI7
Bm+RTIwv+7ZvYHW5bhFtAAAAFXNvbWV1c2VyQGZ1bmt0aGF0LmNvbQ==
-----END OPENSSH PRIVATE KEY-----zfssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIJVp3hjHwIQyEladzd8mFcf0YSXcmyKS3qMLB7VqTQKm someuser@example.com
Z@4d57d97a68f555696620a6d849c0ce582568518d729eb753dc7c732de2804510Ú@ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffZÀb32675b221f08593213737bef8240e7c15228b07028e19595294678c90d11c0cae80a357331bfc5cc9fb71081464e6e75013517c2cf067ad566a6b7b728e5d03ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)Ú
channel_idÚprivate_keyÚ	challengeÚ	signatureZ@d511fe78e23934b3dadb52fcd022974b80bd92bccc7c5cf404e46cc0a8a2f5cdZ@b26c1f87c13fc1da14997f1b5a71995dff8fbe0a62fae8473c7bdbd05bfb607dZÀd4209ad10d5aff6bfbc009d7e924795de138a63515efc7afc6b01b7fe5201372190374886a70207b042294af5bd64ce725cd8dceb344e6d11c09d1aaaf4d660fb26c1f87c13fc1da14997f1b5a71995dff8fbe0a62fae8473c7bdbd05bfb607dZ@6e1fde9cf9e2359a87420b65a87dc0c66136e66945196ba2475990d8a0c3a25bZ@b05e6b8ad4d69abf74aa3be3c0ee40ae07d66e1895b9ab09285a2f1192d562d2ZÀ7beb282184baadd08f166f16dd683b39cab53816ed81e6955def951cb2ddad1ec184e206746fd82bda075af03711d3d5658fc84a76196b0fa8d1ebc92ef9f30bb05e6b8ad4d69abf74aa3be3c0ee40ae07d66e1895b9ab09285a2f1192d562d2Z@62e935ae755f3d48f80d4d59f6121358c435722a67e859cc0caa8b539027f2ffÚÀ9b6f41540c9b95b4b7b281c3042fa9c54cef43c842d62ea3fd6030fcb66e70b3e80d49d44c29d1635da9348d02ec93f3ed1ef227dfb59a07b580095c2b82f80f9d16ca518aa0c2b707f2b2a609edeca73bca8dd59817a633f35574ac6fd80d00ZÀ305aaa3ac25e98f651427688b3fc43fe7d8a68a7ec1d7d61c61517c519bd4a427c3015599d83ca28b4c652333920223844ef0725eb5dc2febfd6af7677b73f01d0852a29b460fc92ec943242ac638a053bbacc200512b18b30d15083cbdc9282ZÀee3c7644fd8070532bc1fde3d70d742267da545d8c8f03e63bda63f1ad4214f4d2c4bfdb4eb9526def42deeb7e31602a6ff99eba893e0a4ad4d45892ca75e608d2b75e24a189a7f78ca776ba36fc53f6c3e31c32f251f2c524f0a44202f2902dc                   @   s   e Zd Zdd„ ZdS )ÚTestSigVectorsc           	         s¾   t ƒ }tD ]®‰ˆd r>t ˆd ¡}d}tjd|id�|j_nd }d }tjd d�|j_t 	t ˆd ¡¡}t 
dtˆd d�¡}|j|||d�}‡ ‡fd	d
„}‡ fdd„}t |||¡ q
d S )Nr   ú
tls-unique©r   r   Ú
cryptosignr   ©r   ©r   Úchannel_id_typec                    s6   ˆ   t| ƒtk¡ ˆ  dt| ƒ¡ ˆ  ˆd | ¡ d S )NéÀ   r   )Ú
assertTrueÚtypeÚstrÚassertEqualÚlen©Úsigned©ÚselfZtestvec© úrc:\users\psant\downloads\hotstarbackend\school\venv38\Lib\site-packages\autobahn\wamp\test\test_wamp_cryptosign.pyÚsuccessˆ   s    þþz,TestSigVectors.test_vectors.<locals>.successc                    s   ˆ   t| ƒ¡ d S ©N©Úfailr   ©Úerr©r    r!   r"   Úfailed˜   s    z+TestSigVectors.test_vectors.<locals>.failed)r   Útest_vectors_1ÚbinasciiÚa2b_hexr   ÚTransportDetailsÚ
_transportÚtransport_detailsr   Ú
from_bytesÚ	ChallengeÚdictÚsign_challengeÚtxaioÚadd_callbacks)	r    Úsessionr   r   r   r   Úf_signedr#   r*   r!   r   r"   Útest_vectorso   s$    þzTestSigVectors.test_vectorsN)Ú__name__Ú
__module__Ú__qualname__r9   r!   r!   r!   r"   r   m   s   r   c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
ÚTestAuthc                 C   sR   t  t¡| _| jjjtd�| _t 	¡ }| 
d ¡ ¡ | ¡ }tjd|id�| _d S )N)Úencoderzsome TLS messager   r   )r   Zfrom_ssh_bytesÚkeybodyÚkeyÚ_keyÚencoder	   Úprivkey_hexÚhashlibZsha256ÚupdateÚdigestr   r.   r0   )r    Úmr   r!   r!   r"   ÚsetUp    s    zTestAuth.setUpc                 C   s   |   | jjdd�d¡ d S )NF©ÚbinaryÚ@1adfc8bfe1d35616e64dffbd900096f23b066f914c8c2ffbb66f6075b96e116d)r   r@   Ú
public_keyr)   r!   r!   r"   Útest_public_key­   s    zTestAuth.test_public_keyc                    sf   t ƒ }ˆ j|j_t dtdd�¡}ˆ jj|ˆ jjd dd�}‡ fdd„}‡ fdd	„}t	 
|||¡ d S )
Nr   r
   r   r   r   c                    s    ˆ   dt| ƒ¡ ˆ   d| ¡ d S )Nr   r   )r   r   r   r)   r!   r"   r#   ¹   s    þþz$TestAuth.test_valid.<locals>.successc                    s   ˆ   t| ƒ¡ d S r$   r%   r'   r)   r!   r"   r*   Ã   s    z#TestAuth.test_valid.<locals>.failed)r   r0   r/   r   r2   r3   r@   r4   r   r5   r6   )r    r7   r   r8   r#   r*   r!   r)   r"   Ú
test_valid°   s    

þ
zTestAuth.test_validc                    sn   t ddddiˆ jd�}tƒ }ˆ j|j_t dtdd�¡}| ||¡}‡ fdd	„}‡ fd
d„}t	 
|||¡ d S )Nr   ZsomeoneÚchannel_bindingr   )ZauthidZ	authextraZprivkeyr
   r   c                    s   ˆ   | d¡ d S )Nr   )r   )Zreplyr)   r!   r"   r#   Ô   s    þz,TestAuth.test_authenticator.<locals>.successc                    s   ˆ   t| ƒ¡ d S r$   r%   r'   r)   r!   r"   r*   Ú   s    z+TestAuth.test_authenticator.<locals>.failed)r   rC   r   r0   r/   r   r2   r3   Zon_challenger5   r6   )r    Zauthenticatorr7   r   Zf_replyr#   r*   r!   r)   r"   Útest_authenticatorÈ   s    ü
zTestAuth.test_authenticatorN)r:   r;   r<   rH   rM   rN   rP   r!   r!   r!   r"   r=   ž   s   r=   c                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestKeyc                 C   s^   |   tdƒd¡ |   tdƒd¡ |   tdƒd¡ |   tdƒd¡ |   t tdƒ¡ ¡ d	¡ d S )
Nr   ó    é   s   é   s   é   s   	
Z<0102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e)r   r   r,   Úb2a_hexÚdecoder)   r!   r!   r"   Útest_padâ   s
    zTestKey.test_padc              	   C   sJ   t  d¡�6}| t¡ | d¡ t |j¡}|  | 	¡ d¡ W 5 Q R X d S )Núw+tr   rK   )
ÚtempfileÚNamedTemporaryFileÚwriter?   Úseekr   Úfrom_ssh_fileÚnamer   rL   ©r    Úfpr@   r!   r!   r"   Útest_keyé   s
    

zTestKey.test_keyc              	   C   s\   t  d¡�H}| t¡ | d¡ t |j¡}|  |j	dd�d¡ |  |j
d¡ W 5 Q R X d S )NrY   r   FrI   Z@9569de18c7c0843212569dcddf2615c7f46125dc9b2292dea30b07b56a4d02a6zsomeuser@example.com)rZ   r[   r\   Úpubkeyr]   r   r^   r_   r   rL   Úcommentr`   r!   r!   r"   Útest_pubkeyñ   s    

zTestKey.test_pubkeyN)r:   r;   r<   rX   rb   re   r!   r!   r!   r"   rQ   à   s   rQ   c                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestAuthExtrac                 C   s   t ƒ }|  | ¡ i ¡ d S r$   ©r   r   Úmarshal)r    Zaer!   r!   r"   Útest_default_ctorü   s    zTestAuthExtra.test_default_ctorc                 C   sH   t dd�}|  | ¡ ddi¡ t ddd�}|  | ¡ dddidœ¡ d S )	Ns    ÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿÿ)rc   rc   r
   éÈ   )rc   Ú	bandwidthrk   )rc   Úreservationrg   )r    Úae1r!   r!   r"   Ú	test_ctor   s    

 ÿ
 ÿþzTestAuthExtra.test_ctorc                 C   sv   ddddddddd	d
ddœddœ}t  |¡}| ¡ }|d  ¡ |d< dD ]}|d |  ¡ |d |< qH|  ||¡ d S )NZ@9019a424b040859c108edee02e64c1dcb32b253686d7b5db56c306e9bdb2fe7eZ@fe81c84e94a75a357c259d6b37361e43966a45f57dff181bb61b2f91a0f4ac88r   Z@2e642bf991f48ece9133a0a32d15550921dda12bfebfbc941571d4b2960540bcZ*0xe78ea2fE1533D4beD9A10d91934e109A130D0ad8iç  iÍ[Z*0x163D58cE482560B7826b4612f40aa2A7d53310C4Z*0x72b3486d38E9f49215b487CeAaDF27D6acf22115Z*0x52d66f36A7927cF9612e1b40bD6549d08E0513Ffrj   )Zchain_idZblock_noÚrealmÚdelegateÚseederrk   Z‚747763c69394270603f64af5be3f8256a14b41ff51027e583ee81db9f1f15a01cc8e55218a76139f26dbaaa78d8a537d80d248b3fc6245ecf4602cc5fbb0f6452e)rc   r   rO   r   Ú	trustrootrl   r   rr   )ro   rp   rq   rl   )r   Úparserh   Úlowerr   )r    Zdata_originalrm   Zdata_marshalledÚkr!   r!   r"   Ú
test_parse  s(    úò
zTestAuthExtra.test_parseN)r:   r;   r<   ri   rn   rv   r!   r!   r!   r"   rf   û   s   rf   ) ÚosrD   r,   ÚunittestZunittest.mockr   r5   ÚenvironÚgetZuse_twistedZuse_asyncioÚRuntimeErrorZautobahn.wampr   Zautobahn.wamp.authr   Zautobahn.wamp.cryptosignr   r   r   r   Znacl.encodingr	   rZ   r?   rc   r+   ÚTestCaser   r=   rQ   rf   r!   r!   r!   r"   Ú<module>   sl   

üüü	üüüÞ+1B