File: //lib/python3/dist-packages/samba/tests/dcerpc/__pycache__/binding.cpython-310.pyc
o
�/a� � @ sn d Z ddlZddlmZmZ ddlmZ ddlm m Z ddl
m Z m Z m
Z
ddlmZ G dd� de�ZdS ) z8Tests for samba.dcerpc., credentials and binding strings� N)�RpcInterfaceTestCase�TestCase)�lsa)� Credentials�SMB_ENCRYPTION_REQUIRED�SMB_ENCRYPTION_OFF)�
NTSTATUSErrorc sD e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Z� Z S )
�RpcBindingTestsc s t t| ��� d S )N)�superr �setUp)�self�� __class__� �</usr/lib/python3/dist-packages/samba/tests/dcerpc/binding.pyr s zRpcBindingTests.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 RpcBindingTests.get_user_credsc C � | � � }|�t� | �� }|�dd� |�dd� dtj�d� }t� |||�}| �
|� � � t� � }t�
� |_|�d|tj�}| �|� d S �N�client ipc max protocol�SMB3�client ipc min protocol� ncacn_np:%s�SERVER� )r �set_smb_encryptionr � get_loadparm�setr r r r �lsarpc� assertFalse�transport_encrypted�ObjectAttribute�QosInfo�sec_qos� OpenPolicy2�security�SEC_FLAG_MAXIMUM_ALLOWED�assertIsNotNone�r �creds�lp�binding_string�lsa_conn�
objectAttr�
pol_handler r r � test_smb3_dcerpc_no_encryption+ �
�z.RpcBindingTests.test_smb3_dcerpc_no_encryptionc C r r! )r r( r r) r* r r r r r+ �
assertTruer- r. r/ r0 r1 r2 r3 r4 r5 r r r �test_smb3_dcerpc_encryption? r= z+RpcBindingTests.test_smb3_dcerpc_encryptionc C �Z | � � }|�t� | �� }|�dd� |�dd� dtj�d� }| �t t
j |||� d S )Nr" �SMB2r$ r% r& � r r( r r) r* r r r � assertRaisesr r r+ �r r6 r7 r8 r r r �test_smb2_dcerpc_encryptionS �
z+RpcBindingTests.test_smb2_dcerpc_encryptionc C r@ )Nr" �NT1r$ r% r& rB rD r r r �test_smb1_dcerpc_encryption^ rF z+RpcBindingTests.test_smb1_dcerpc_encryption)
�__name__�
__module__� __qualname__r r r< r? rE rH �
__classcell__r r r
r r s r )�__doc__� samba.testsr r r � samba.dcerpcr �samba.dcerpc.security�dcerpcr2 �samba.credentialsr r r r r r r r r �<module> s