File: //lib/python3/dist-packages/samba/tests/dcerpc/__pycache__/createtrustrelax.cpython-310.pyc
o
�/a� � @ sh d Z ddlZddlZddlmZ ddlmZmZmZ ddl m
Z
m Z m Z ddl
mZ G dd� de�ZdS ) z.Tests for the CreateTrustedDomainRelax wrapper� N)�TestCase)�lsa�security�drsblobs)� Credentials�SMB_ENCRYPTION_REQUIRED�SMB_ENCRYPTION_OFF)�CreateTrustedDomainRelaxc s> e Zd Z� fdd�Zdd� Zd dd�Zdd � Zd
d � Z� ZS )
� CreateTrustedDomainRelaxTestc s t t| ��� d S )N)�superr
�setUp��self�� __class__� �E/usr/lib/python3/dist-packages/samba/tests/dcerpc/createtrustrelax.pyr s z"CreateTrustedDomainRelaxTest.setUpc C sT t � }|�� tj�d�}tj�d�}tj�d�}|�|� |�|� |�|� |S )N�DOMAIN�USERNAME�PASSWORD)r �guess�samba�tests�env_get_var_value�
set_domain� set_username� set_password)r �c�domain�username�passwordr r r �get_user_creds! s
z+CreateTrustedDomainRelaxTest.get_user_credsTc
C s, | � � }|r |�t� n|�t� | �� }dtj�d� }t� |||�}|r.| �
|� � � n| � |� � � t�
� }t�� |_|�d|tj�}| �|� t�� }d|_z|�||tj�} |�|| jj� W n tym Y nw t� � } |j| j _d| j _t� d�| _tj tj!B | _"tj#| _$tj%| _&t�'d�(d��}
t)�*� } t+|
�| _,|
| _-t)�.� } d | _/tj0| _1| | _2t)�3� }
d
|
_4| g|
_5t)�6� }d
|_4|
|_7d }z t8||| tj9||�}W n( tj:y� } zt;|��d }~w t<y� } z |r�t;|��W Y d }~nd }~ww |�r| �|� |�|| j� d S | �=|� d S ) Nz ncacn_np:%s�SERVER� ztests.samba.example.com�createtrustrelaxz&S-1-5-21-538490383-3740119673-95748416r z utf-16-ler � )>r! �set_smb_encryptionr r � get_loadparmr r r r �lsarpc�
assertTrue�transport_encrypted� assertFalse�ObjectAttribute�QosInfo�sec_qos� OpenPolicy2r �SEC_FLAG_MAXIMUM_ALLOWED�assertIsNotNone�String�string� QueryTrustedDomainInfoByName�!LSA_TRUSTED_DOMAIN_INFO_FULL_INFO�DeleteTrustedDomain�info_ex�sid� RuntimeError�TrustDomainInfoInfoEx� domain_name� netbios_name�dom_sid�LSA_TRUST_DIRECTION_INBOUND� LSA_TRUST_DIRECTION_OUTBOUND�trust_direction�LSA_TRUST_TYPE_UPLEVEL�
trust_type�%LSA_TRUST_ATTRIBUTE_FOREST_TRANSITIVE�trust_attributes�string_to_byte_array�encoder �
AuthInfoClear�len�sizer �AuthenticationInformation�LastUpdateTime�TRUST_AUTH_TYPE_CLEAR�AuthType�AuthInfo� AuthenticationInformationArray�count�array�trustAuthInOutBlob�currentr �SEC_STD_DELETE�
NTSTATUSError�AssertionError�OSError� assertIsNone)r �
smbencrypt�creds�lp�binding_string�lsa_conn�
objectAttr�
pol_handle�name�info�
password_blob� clear_value� clear_authentication_information� authentication_information_array�outgoing�trustdom_handle�nt�er r r �_create_trust_relax, s�
�
� �
�� ���
z0CreateTrustedDomainRelaxTest._create_trust_relaxc C � | � d� d S )NT�rj r
r r r � test_create_trust_relax_encrypt � z<CreateTrustedDomainRelaxTest.test_create_trust_relax_encryptc C rk )NFrl r
r r r � test_create_trust_relax_no_enc� rn z;CreateTrustedDomainRelaxTest.test_create_trust_relax_no_enc)T) �__name__�
__module__� __qualname__r r! rj rm ro �
__classcell__r r r r r
s
Sr
)�__doc__�osr � samba.testsr � samba.dcerpcr r r �samba.credentialsr r r �samba.trust_utilsr r
r r r r �<module> s