a )&i@sddlZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z ej dddejd<d Zd ZGd d d e Zed krd Zd ZddlZedS) N) NTSTATUSError credentials)lsa)NT_STATUS_ACCESS_DENIED NT_STATUS_NO_IMPERSONATION_TOKEN) KDCBaseTestz bin/python1ZPYTHONUNBUFFEREDFc@s:eZdZdZddZddZddZdd d Zd d ZdS)RpcTestszfTest for RPC authentication using Kerberos credentials stored in a credentials cache file. cCs |dS)N _run_rpc_testselfr;/usr/lib/python3/dist-packages/samba/tests/krb5/test_rpc.pytest_rpc.szRpcTests.test_rpccCs|jdddS)NT)renamer r rrrtest_rpc_rename1szRpcTests.test_rpc_renamecCs|jdddddS)NFT) include_pac expect_anon allow_errorr r rrrtest_rpc_no_pac4szRpcTests.test_rpc_no_pacFTc Cs<|}|j}d}|j|jjdd}|} |} |j|| |||d\} } |t j | j |r| } t |}t | t jd|d<||d|}zt||| }WnNty}z4|s||j\}}||tthWYd}~dSd}~00|ddd\}}|r*|| |jn|| |jdS)NZcifsF)Z account_typeZ use_cache)ZpacZsAMAccountNamencacn_np:%s[\pipe\lsarpc]) Z get_samdbhostZget_cached_credsZ AccountTypeZUSERZ get_usernameZget_server_credsZcreate_ccache_with_userZ addCleanuposremovenameZget_new_usernameldbZMessageZget_dnZMessageElementZFLAG_MOD_REPLACEZmodifyrlsarpcget_lprZfailargsZassertInrr GetUserNameZassertNotEqualstring assertEqual)r rrrrZsamdb mach_nameZserviceZuser_credentialsZ user_nameZmach_credentialsZcredsZ cachefilenew_namemsg binding_strconneenum_ account_namerrrr 8sN   zRpcTests._run_rpc_testcCsX|j}t}|d|}t|||}|ddd\}}|d|j dS)NrzANONYMOUS LOGON) rrZ CredentialsZ set_anonymousrrrr r"r!upper)r r#Z anon_credsr&r'r+r*rrrtest_rpc_anonymousxszRpcTests.test_rpc_anonymousN)FTFF) __name__ __module__ __qualname____doc__rrrr r-rrrrr )s @r __main__)sysrrZsambarrZ samba.dcerpcrZsamba.ntstatusrrZsamba.tests.krb5.kdc_base_testrpathinsertenvironZglobal_asn1_printZglobal_hexdumpr r.Zunittestmainrrrrs    ]