a )&iK@sddlZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Zddl mZdd lmZejdd d ejd <d Zd ZGdddeZedkrd Zd ZddlZedS)N) SCOPE_SUBTREE) NTSTATUSError)security) ndr_unpack) NT_STATUS_NO_IMPERSONATION_TOKEN)libsmb_samba_internal)param) KDCBaseTestz bin/python1ZPYTHONUNBUFFEREDFc@s2eZdZdZddZddZddZd d d Zd S)SmbTestszfTest for SMB authentication using Kerberos credentials stored in a credentials cache file. cCs |dS)N _run_smb_testselfr;/usr/lib/python3/dist-packages/samba/tests/krb5/test_smb.pytest_smb/szSmbTests.test_smbcCs|jdddS)NT)renamer rrrrtest_smb_rename2szSmbTests.test_smb_renamecCs|jddddS)NFT) include_pac expect_errorr rrrrtest_smb_no_pac5szSmbTests.test_smb_no_pacFTc Cs|}|}d}d}|j|jjdd}|} |} |} |j|| |||d\} } |t j | j |j t d| dgd} |d t| ttj| d dd }|r|}t|}t|tjd |d <||t jd d }|t jjd |d| j t jd <t}|| j!|d}||j"d||"dd|d}||j"d||"ddzt#j$|||| d}WnLt%y}z2|s|&|j'\}}|t(|WYd}~dSd}~00|)||*\}}}}}|||d dS)NZcifstmpF)Z account_typeZ use_cache)Zpacz(sAMAccountName=%s)Z objectSid)ZscopeZ expressionattrsrZsAMAccountNameZ KRB5CCNAMEzFILE:zclient min protocolZNT1zclient max protocol)Zlpcreds)+Z get_samdbZ host_dns_nameZget_cached_credsZ AccountTypeZUSERZ get_usernameZ get_dc_credsZcreate_ccache_with_userZ addCleanuposremovenamesearchrZ assertEquallenrrZdom_sidZget_new_usernameldbZMessageZget_dnZMessageElementZFLAG_MOD_REPLACEZmodifyenvironget __setitem__s3paramZ get_contextloadZget_lpZ configfilesetlibsmbZConnrZfailargsrZ assertFalseZ posix_whoami)rrrrZsamdbZ mach_nameZserviceZshareZuser_credentialsZ user_nameZmach_credentialsrZ cachefileZldb_resZsidnew_namemsgZ krb5_ccnameZs3_lpZ min_protocolZ max_protocolZconneenum_ZuidgidZgidsZsidsZguestrrrr 9sl         zSmbTests._run_smb_testN)FTF)__name__ __module__ __qualname____doc__rrrr rrrrr *sr __main__)sysrr"rZsambarZ samba.dcerpcrZ samba.ndrrZsamba.ntstatusrZ samba.samba3rr)rr&Zsamba.tests.krb5.kdc_base_testr pathinsertr#Zglobal_asn1_printZglobal_hexdumpr r1Zunittestmainrrrrs(         k