o
    >h2                  
   @   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r&e	  ne jddr2e
  nedd dlmZ d dlmZ d dlmZmZmZ erZd 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)MockUSE_TWISTED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
@4d57d97a68f555696620a6d849c0ce582568518d729eb753dc7c732de2804510@ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffb32675b221f08593213737bef8240e7c15228b07028e19595294678c90d11c0cae80a357331bfc5cc9fb71081464e6e75013517c2cf067ad566a6b7b728e5d03ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff)
channel_idprivate_key	challenge	signature@d511fe78e23934b3dadb52fcd022974b80bd92bccc7c5cf404e46cc0a8a2f5cd@b26c1f87c13fc1da14997f1b5a71995dff8fbe0a62fae8473c7bdbd05bfb607dd4209ad10d5aff6bfbc009d7e924795de138a63515efc7afc6b01b7fe5201372190374886a70207b042294af5bd64ce725cd8dceb344e6d11c09d1aaaf4d660fb26c1f87c13fc1da14997f1b5a71995dff8fbe0a62fae8473c7bdbd05bfb607d@6e1fde9cf9e2359a87420b65a87dc0c66136e66945196ba2475990d8a0c3a25b@b05e6b8ad4d69abf74aa3be3c0ee40ae07d66e1895b9ab09285a2f1192d562d27beb282184baadd08f166f16dd683b39cab53816ed81e6955def951cb2ddad1ec184e206746fd82bda075af03711d3d5658fc84a76196b0fa8d1ebc92ef9f30bb05e6b8ad4d69abf74aa3be3c0ee40ae07d66e1895b9ab09285a2f1192d562d2@62e935ae755f3d48f80d4d59f6121358c435722a67e859cc0caa8b539027f2ff9b6f41540c9b95b4b7b281c3042fa9c54cef43c842d62ea3fd6030fcb66e70b3e80d49d44c29d1635da9348d02ec93f3ed1ef227dfb59a07b580095c2b82f80f9d16ca518aa0c2b707f2b2a609edeca73bca8dd59817a633f35574ac6fd80d00305aaa3ac25e98f651427688b3fc43fe7d8a68a7ec1d7d61c61517c519bd4a427c3015599d83ca28b4c652333920223844ef0725eb5dc2febfd6af7677b73f01d0852a29b460fc92ec943242ac638a053bbacc200512b18b30d15083cbdc9282ee3c7644fd8070532bc1fde3d70d742267da545d8c8f03e63bda63f1ad4214f4d2c4bfdb4eb9526def42deeb7e31602a6ff99eba893e0a4ad4d45892ca75e608d2b75e24a189a7f78ca776ba36fc53f6c3e31c32f251f2c524f0a44202f2902dc                   @   s   e Zd Zdd ZdS )TestSigVectorsc           	         s   t  }tD ]W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selftestvec /var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/autobahn/wamp/test/test_wamp_cryptosign.pysuccess   s   z,TestSigVectors.test_vectors.<locals>.successc                         t|  d S Nfailr'   errr-   r/   r0   failed      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_signedr1   r9   r/   r,   r0   test_vectorso   s&   zTestSigVectors.test_vectorsN)__name__
__module____qualname__rI   r/   r/   r/   r0   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
   from_ssh_byteskeybodykey_keyencoder   privkey_hexhashlibsha256updatedigestr   r>   r@   )r-   mr   r/   r/   r0   setUp   s   zTestAuth.setUpc                 C   s   |  | jjddd d S )NFbinary@1adfc8bfe1d35616e64dffbd900096f23b066f914c8c2ffbb66f6075b96e116d)r(   rQ   
public_keyr8   r/   r/   r0   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*   r8   r/   r0   r1      s   z$TestAuth.test_valid.<locals>.successc                    r2   r3   r4   r6   r8   r/   r0   r9      r:   z#TestAuth.test_valid.<locals>.failed)r   r@   r?   r   rB   rC   rQ   rD   r   rE   rF   )r-   rG   r   rH   r1   r9   r/   r8   r0   
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    someonechannel_bindingr   )authid	authextraprivkeyr   r!   c                    s     | d d S )Nr   )r(   )replyr8   r/   r0   r1      s   z,TestAuth.test_authenticator.<locals>.successc                    r2   r3   r4   r6   r8   r/   r0   r9      r:   z+TestAuth.test_authenticator.<locals>.failed)r   rT   r   r@   r?   r   rB   rC   on_challengerE   rF   )r-   authenticatorrG   r   f_replyr1   r9   r/   r8   r0   test_authenticator   s   
zTestAuth.test_authenticatorN)rJ   rK   rL   rZ   r_   r`   rj   r/   r/   r/   r0   rM      s
    rM   c                   @   $   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   	
<0102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e)r(   r   r<   b2a_hexdecoder8   r/   r/   r0   test_pad   s
   zTestKey.test_padc                 C   s`   t d!}|t |d t|j}| |	 d W d    d S 1 s)w   Y  d S )Nw+tr   r]   )
tempfileNamedTemporaryFilewriterP   seekr
   from_ssh_filenamer(   r^   r-   fprQ   r/   r/   r0   test_key   s   

"zTestKey.test_keyc                 C   sr   t d*}|t |d t|j}| |j	ddd | |j
d W d    d S 1 s2w   Y  d S )Nru   r   Fr[   @9569de18c7c0843212569dcddf2615c7f46125dc9b2292dea30b07b56a4d02a6zsomeuser@example.com)rv   rw   rx   pubkeyry   r
   rz   r{   r(   r^   commentr|   r/   r/   r0   test_pubkey   s   

"zTestKey.test_pubkeyN)rJ   rK   rL   rt   r~   r   r/   r/   r/   r0   rl      s    rl   c                   @   rk   )TestAuthExtrac                 C   s   t  }| | i  d S r3   r	   r(   marshal)r-   aer/   r/   r0   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    )r   r   r      )r   	bandwidthr   )r   reservationr   )r-   ae1r/   r/   r0   	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 |< q$| || d S )N@9019a424b040859c108edee02e64c1dcb32b253686d7b5db56c306e9bdb2fe7e@fe81c84e94a75a357c259d6b37361e43966a45f57dff181bb61b2f91a0f4ac88r   @2e642bf991f48ece9133a0a32d15550921dda12bfebfbc941571d4b2960540bc*0xe78ea2fE1533D4beD9A10d91934e109A130D0ad8i  i[*0x163D58cE482560B7826b4612f40aa2A7d53310C4*0x72b3486d38E9f49215b487CeAaDF27D6acf22115*0x52d66f36A7927cF9612e1b40bD6549d08E0513Ffr   )chain_idblock_norealmdelegateseederr   ڂ747763c69394270603f64af5be3f8256a14b41ff51027e583ee81db9f1f15a01cc8e55218a76139f26dbaaa78d8a537d80d248b3fc6245ecf4602cc5fbb0f6452e)r   r   rb   r   	trustrootr   r   r   )r   r   r   r   )r	   parser   lowerr(   )r-   data_originalr   data_marshalledkr/   r/   r0   
test_parse  s(   
zTestAuthExtra.test_parseN)rJ   rK   rL   r   r   r   r/   r/   r/   r0   r      s    r   ) osrU   r<   unittestunittest.mockr   rE   environgetuse_twisteduse_asyncioRuntimeErrorautobahn.wampr   autobahn.wamp.authr   autobahn.wamp.cryptosignr   r   r	   r
   nacl.encodingr   rv   rP   r   r;   TestCaser   rM   rl   r   r/   r/   r/   r0   <module>   sn   

	+1B