o l>j'@sddlZddlZddlZddlZddlZddlZejeje Z ej eje dZ ej e ssr z#Unexpected build result... aborting2AC6BDB41324A9A9BF166DE5E1389582FAF72B6651987EE07FC3192943DB56050A37329CBB4A099ED8193E0757767A13DD52312AB4B03310DCD7F48A9DA04FD50E8083969EDB767B0CF6095179A163AB3661A05FBD5FAAAE82918A9962F0B93B855F97993EC975EEAA80D740ADBF4FF747359D041D5C33EA71D281E446B14773BCA97B43A23FB801676BD207A436C6481F1D2B9078717461A5B9D32E688F87748544523B524B0D57D5EA77A2775D2ECFA032CFBDBF52FB3786160279004E57AE6AF874E7303CE53299CCC041C7BC308D82A5698F3A8D0C38271AE35F8E9DBFBB694B5C803D89F7AE435DE236D525F54759B65E372FCD68EF20FA7111F9E4AFF73asciic@seZdZejejdddfddZddZddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZddZdd Zd!d"Zd#d$ZdS)%SRPTestsNc " Cs|j} |j} |j} d} d} | | | ||||\}}| | | ||||dd| }|}|\}}| | | |||||d| }||||\}}||||||| ||||||||d| }|}|\}}|||}|||}|||||}| |||||||||| }|||||}|||| |||d|||||| } ||| | ||}!|||!||||| | | | | | | | | | dS)Ntestuser testpassword) UserVerifiercreate_salted_verification_keyget_ephemeral_secretstart_authentication assertEqual get_challengeprocess_challengeverify_session assertTrue authenticated)"selfu_modv_modg_modhash_algng_typen_hexg_hexk_hexrrrusernamepassword_s_vusrbytes_aunameAusr2uname2A2svrbytes_bsBMM2HAMKsvr2HAMK2svr3HAMK3rrr doit/sH               z SRPTests.doitcC|tttdSN)r<_pysrprrrr test_pure_python_defaultshz"SRPTests.test_pure_python_defaultscCr=r>)r<_ctsrpr@rrr test_ctypes_defaultskrBzSRPTests.test_ctypes_defaultscC|tttdSr>r<r?rCr@rrr test_mix1nrBzSRPTests.test_mix1cC|tttdSr>rFr@rrr test_mix2qrBzSRPTests.test_mix2cCrEr>r<rCr?r@rrr test_mix3trBzSRPTests.test_mix3cCrHr>rJr@rrr test_mix4wrBzSRPTests.test_mix4cC|jttttjddS)N)r!)r<rCsrpSHA512r@rrr test_hash_SHA512{zSRPTests.test_hash_SHA512cCrM)N)r")r<rCrNNG_8192r@rrr test_NG_8192~rQzSRPTests.test_NG_8192cCs|jttttjttddS)N)r"r#r$)r<rCrN NG_CUSTOM test_n_hex test_g_hexr@rrr test_NG_CUSTOMzSRPTests.test_NG_CUSTOMc Cs |jttttjtjttddS)N)r!r"r#r$)r<rCr?rNSHA256rTrUrVr@rrr test_all1s zSRPTests.test_all1cCs|jttttjtjddS)N)r!r")r<rCr?rNSHA224NG_4096r@rrr test_all2rXzSRPTests.test_all2cCs.tdD]}td}||d?dkqdS)z{ Verify that the Python implementation guarantees byte length by setting most significant bit to 1 rN)ranger?get_random_of_lengthr)rxvalrrr test_random_of_lengths  zSRPTests.test_random_of_lengthcsXttdttd}fdd}ttfD]}|fD]}|||q!qdS)zW Verify that all implementations require 32 bytes for ephemeral values !cst}|jdd|dWdn1swYdt|jt}|jd|dWdn1s@wYdt|jdS)Nr,pwd)r+r+)r2r2) assertRaises ValueErrorrassertInstr exceptionr)modrdctxrandom31rrr verf_lens  z0SRPTests.test_ephemeral_length..verf_lenN)r? long_to_bytesrbrC)rrandom33rrrnrdrrpr test_ephemeral_lengths  zSRPTests.test_ephemeral_lengthcCs<tdd}|| tdd}|| dS)Ntest)r?rrrrC)rr*rrr test_authenticated_on_inits  z#SRPTests.test_authenticated_on_initcCs|jtttdddS)N5)r%rJr@rrr test_custom_kszSRPTests.test_custom_kc Csvttd}ttfD],}|d||}|t }||Wdn1s*wY|dt |j q dS)Nr_r,bytes_A) r?rsrbrCrrirjrrkrlrm)r randbytesrnr1rorrr test_verifier_requires_As   z!SRPTests.test_verifier_requires_A)__name__ __module__ __qualname__rNSHA1NG_2048r<rArDrGrIrKrLrPrSrWrZr]rerurwryr|rrrr r-s$9  rzSHA1 r[rYSHA384rO)rriiii )rrrrrrc Csx|tt||}|\}}|||||||}|\} } || | } || } || |r6|s:t ddS)NzAuthentication failed!) rr&r'rrrrrr Exception) rnr!r"r(r)r*r,r-r1r3r4r5r7rrr do_auths     rr^csnttt\|afdd}t}|dkr,t|d|d8}|dks|t|}|S)Ncs0tdkrttd8atdksdSdS)Nrr)NLEFTrrr(r)r!rnr"rr test_threadsz%performance_test..test_threadrr)rNrr&r'rtime_threadstart_new_thread)rnr!r"niternthreadsrstartdurationrrr performance_tests  rcCs*ddd}d||jt|t|f}|S)NPythonctypes)z srp._pysrpz srp._ctsrpz %s, %s, %d:)r}hash_map prime_map)rnr!r"mcfgrrr get_param_strs rcCs<t||||}t|||}td|dd||||S)Nz z%.6f)rrprintljust)rnr!r"rrrrrr param_tests rcCsRtdtdtttjtj}tttjtj}tdtdtd||dS)N<************************************************************zDefault Parameter Timings:rzPerformance increases: z ctypes-module : )rrr?rNrrrC)py_timect_timerrr print_default_timingssrc Cstjtjtjtjg}tjtjtjtjtj g}t dt dt dt ddd|D]}t dt | dddq*t dt d |D]$}t d t |dd|D]}t d tt||d ddqRt dqDdS) Nrz0Hash Algorithm vs Prime Number performance tablerz | )endzNG_%d z<------------------------------------------------------------z%s |z{0:>12f}r^)rNNG_1024rr\rRrr[rYrrOrrrjustrformatrrC)ng_types hash_typesngr!rrr print_performance_tables    rc CsLtdtdd}tddD]}td|tttjtj|||qdS)NrzThread Performance Test:dr z Thread Count {0:>2}: {1:8f})rrarrrCrNrr)rrrrr print_thread_performance#s &rr*z* Testing Implementation) verbosityz* Performance Testingg?)r^r)r^)1unittestos.pathossysrrpathdirnameabspath__file__this_dirjoin build_direxistsrexitlistdir plat_dirslenplat_dirinsertrN srp._pysrpr? srp._ctsrprCrVencoderUTestCaserrrr&r'rrrrrrrr TestLoaderloadTestsFromTestCasesuiteTextTestRunnerrunsleeprrrr sf