
    tf2              
       8   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j	                     n6e j                            dd          r ej
                     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ej                  Z G d dej                  Z G d d ej                  Z G 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                       e Zd Zd ZdS )TestSigVectorsc                 (   	 t                      }t          D ]		d         r>t          j        	d                   }d}t	          j        d|i          |j        _        n#d }d }t	          j        d           |j        _        t          j	        t          j        	d                             }t	          j
        dt          	d                             }|                    |||          } 	fd	} fd
}t          j        |||           d S )Nr   
tls-uniquer   r   
cryptosignr   r   r   channel_id_typec                                          t          |           t          k                                   dt	          |                                          d         |            d S )N   r   )
assertTruetypestrassertEquallen)signedselftestvecs    f/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/autobahn/wamp/test/test_wamp_cryptosign.pysuccessz,TestSigVectors.test_vectors.<locals>.success   su    V 3444   KK     K(        c                 L                         t          |                      d S Nfailr+   errr/   s    r1   failedz+TestSigVectors.test_vectors.<locals>.failed   s!    		#c((#####r3   )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_signedr2   r:   r0   s
   `        @r1   test_vectorszTestSigVectors.test_vectorso   sR   &&% )	; )	;G|$ _%-gl.CDD
".7<7MZfhrYs7t7t7t"44!
"&7<7MY]7^7^7^"4 (283CGMDZ3[3[\\K dW[EY6Z6Z6Z[[I #11)=GBQ 2 S SH      $ $ $ $ $ '6::::S)	; )	;r3   N)__name__
__module____qualname__rI    r3   r1   r   r   m   s#        ,; ,; ,; ,; ,;r3   r   c                   &    e Zd Zd Zd Zd Zd ZdS )TestAuthc                 j   t          j        t                    | _        | j        j                            t                    | _        t          j	                    }|
                    d                                           |                                }t          j        d|i          | _        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   s      r1   setUpzTestAuth.setUp   s     /888=//
/CC N	#**,,--- XXZZ
!&!7LR\C]!^!^!^r3   c                 d    |                      | j                            d          d           d S )NFbinary@1adfc8bfe1d35616e64dffbd900096f23b066f914c8c2ffbb66f6075b96e116d)r,   rT   
public_keyr/   s    r1   test_public_keyzTestAuth.test_public_key   s2    ,,E,::<~r3   c                 &    t                      } j        |j        _        t          j        dt          d                    } j                            | j        j        d         d          } fd} fd}t          j
        |||           d S )Nr#   r   r$   r!   r%   c                 z                         dt          |                                           d|            d S )Nr(   r   )r,   r-   )r.   r/   s    r1   r2   z$TestAuth.test_valid.<locals>.success   sT    F    S    r3   c                 L                         t          |                      d S r5   r6   r8   s    r1   r:   z#TestAuth.test_valid.<locals>.failed   !    IIc#hhr3   )r   r@   r?   r   rB   rC   rT   rD   r   rE   rF   )r/   rG   r   rH   r2   r:   s   `     r1   
test_validzTestAuth.test_valid   s    &&/3/E,OL$2K2K2KLL	8**96:6L6WXd6e;G + I I	 	 	 	 		  	  	  	  	  	Hgv66666r3   c                 ,    t          ddddi j                  }t                      } j        |j        _        t          j        dt          d                    }|                    ||          } fd} fd	}t          j
        |||           d S )
Nr#   someonechannel_bindingr!   )authid	authextraprivkeyr   r$   c                 4                         | d           d S )Nr   )r,   )replyr/   s    r1   r2   z,TestAuth.test_authenticator.<locals>.success   s.     S    r3   c                 L                         t          |                      d S r5   r6   r8   s    r1   r:   z+TestAuth.test_authenticator.<locals>.failed   rh   r3   )r   rW   r   r@   r?   r   rB   rC   on_challengerE   rF   )r/   authenticatorrG   r   f_replyr2   r:   s   `      r1   test_authenticatorzTestAuth.test_authenticator   s    ,(,7$	
 
 
 &&/3/E,OL$2K2K2KLL	,,Wi@@	 	 	 	 		  	  	  	  	  	GWf55555r3   N)rJ   rK   rL   r]   rd   ri   rv   rM   r3   r1   rO   rO      sV        _ _ _@ @ @7 7 706 6 6 6 6r3   rO   c                        e Zd Zd Zd Zd ZdS )TestKeyc                    |                      t          d          d           |                      t          d          d           |                      t          d          d           |                      t          d          d           |                      t          j        t          d                                                    d	           d S )
Nr   r3      s      s      s   	
<0102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e)r,   r   r<   b2a_hexdecoderc   s    r1   test_padzTestKey.test_pad   s    !c***!k222!o666"  (]  	^  	^  	^)(2,,77>>@@  CA  	B  	B  	B  	B  	Br3   c                 @   t          j        d          5 }|                    t                     |                    d           t          j        |j                  }|                     |	                                d           d d d            d S # 1 swxY w Y   d S )Nw+tr   ra   )
tempfileNamedTemporaryFilewriterS   seekr   from_ssh_filenamer,   rb   r/   fprT   s      r1   test_keyzTestKey.test_key   s    (// 	s2HHWGGAJJJ-bg66CS^^--/qrrr	s 	s 	s 	s 	s 	s 	s 	s 	s 	s 	s 	s 	s 	s 	s 	s 	s 	ss   A1BBBc                 z   t          j        d          5 }|                    t                     |                    d           t          j        |j                  }|                     |	                    d          d           |                     |j
        d           d d d            d S # 1 swxY w Y   d S )Nr   r   Fr_   @9569de18c7c0843212569dcddf2615c7f46125dc9b2292dea30b07b56a4d02a6zsomeuser@example.com)r   r   r   pubkeyr   r   r   r   r,   rb   commentr   s      r1   test_pubkeyzTestKey.test_pubkey   s    (// 	B2HHVGGAJJJ-bg66CS^^5^99;}~~~S[*@AAA	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	B 	Bs   BB00B47B4N)rJ   rK   rL   r   r   r   rM   r3   r1   rx   rx      sL        B B Bs s sB B B B Br3   rx   c                        e Zd Zd Zd Zd ZdS )TestAuthExtrac                 r    t                      }|                     |                                i            d S r5   r
   r,   marshal)r/   aes     r1   test_default_ctorzTestAuthExtra.test_default_ctor   s1     ""r*****r3   c                     t          d          }|                     |                                ddi           t          dd          }|                     |                                dddid           d S )	Ns    )r   r   r      )r   	bandwidthr   )r   reservationr   )r/   ae1s     r1   	test_ctorzTestAuthExtra.test_ctor   s    !666i)
 	 	 	 "EEES)
 )
 	 	 	 	 	r3   c                 :   ddddddddd	d
dddd}t          j        |          }|                                }|d                                         |d<   dD ]+}|d         |                                         |d         |<   ,|                     ||           d S )N@9019a424b040859c108edee02e64c1dcb32b253686d7b5db56c306e9bdb2fe7e@fe81c84e94a75a357c259d6b37361e43966a45f57dff181bb61b2f91a0f4ac88r!   @2e642bf991f48ece9133a0a32d15550921dda12bfebfbc941571d4b2960540bc*0xe78ea2fE1533D4beD9A10d91934e109A130D0ad8i  i[*0x163D58cE482560B7826b4612f40aa2A7d53310C4*0x72b3486d38E9f49215b487CeAaDF27D6acf22115*0x52d66f36A7927cF9612e1b40bD6549d08E0513Ffr   )chain_idblock_norealmdelegateseederr   ڂ747763c69394270603f64af5be3f8256a14b41ff51027e583ee81db9f1f15a01cc8e55218a76139f26dbaaa78d8a537d80d248b3fc6245ecf4602cc5fbb0f6452e)r   r   rl   r   	trustrootr   r   r   )r   r   r   r   )r
   parser   lowerr,   )r/   data_originalr   data_marshalledks        r1   
test_parsezTestAuthExtra.test_parse  s    X[+\E%EHF   ^
 
  "'66++-- &3;%?%E%E%G%Gk"0 	V 	VA.;M.J1.M.S.S.U.UM-(++-88888r3   N)rJ   rK   rL   r   r   r   rM   r3   r1   r   r      sA        + + +  9 9 9 9 9r3   r   ) osrX   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   r   rS   r   r;   TestCaser   rO   rx   r   rM   r3   r1   <module>r      s  6 
			          :>>-&& EEZ^^M4(( EE
,C
D
DD       3 3 3 3 3 3 R R R R R R R R R R )666666(((((( %
 YW X	  YW X	  YW X	  YYW X	  YYW X	  YYW X	 E(V.; .; .; .; .;X& .; .; .;b?6 ?6 ?6 ?6 ?6x  ?6 ?6 ?6DB B B B Bh B B B6,9 ,9 ,9 ,9 ,9H% ,9 ,9 ,9 ,9 ,9r3   