File: //lib/python3/dist-packages/samba/tests/dcerpc/__pycache__/samr_change_password.cpython-310.pyc
o
�/an � @ s� d Z ddlZddlZddlZddlmZmZmZmZ ddl m
Z
ddl m Z ddl m
Z
ddlmZmZmZ ddlmZ dd lmZ G d
d � d e�ZdS ) z$Tests for samba.dcerpc.samr.password� N)�crypto�generate_random_password�generate_random_bytes�ntstatus)�system_session)� Credentials)�SMB_ENCRYPTION_REQUIRED)�samr�security�lsa)�SamDB)�RpcInterfaceTestCasec sd e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z ddd�Z
ddd�Z � Z S )�SamrPasswordTestsc s� t t| ��� | �� | �d� tj�d�| _tj�d�| _ tj�d�| _
tj�d�| _ d| j | _ t
� | _| j�| j� | j�| j
� | j�| j � d S )Ni' �SERVER�DOMAIN�USERNAME�PASSWORDzncacn_np:%s[krb5])�superr �setUp�
open_samdb�create_user_account�samba�tests�env_get_var_value�
remote_server�
remote_domain� remote_user�remote_password�remote_binding_stringr � remote_creds�guess�lp� set_username� set_password��self�� __class__� �I/usr/lib/python3/dist-packages/samba/tests/dcerpc/samr_change_password.pyr # s
zSamrPasswordTests.setUpc sH t t| ��� t�| j� t�| j� t�| j� tj �
| j | j � d S �N)
r r �tearDownr �Close� user_handle�
domain_handle�handler r � delete_force�samdb�user_dnr$ r& r( r) r+ 4 s
zSamrPasswordTests.tearDownc C sD t j�� | _t� | _| j�| j� t� | _t | j| j| jd�| _
d S )N)� session_info� credentialsr! ) r r � env_loadparmr! r � local_credsr r �sessionr r1 r$ r( r( r) r @ s �z SamrPasswordTests.open_samdbc C sH | j �d tj�| _| j �| jt�| j��| _ | j �
| jtj| j �| _ d S r* ) �conn�Connect2r
�SEC_FLAG_MAXIMUM_ALLOWEDr/ � LookupDomainr �Stringr �
domain_sid�
OpenDomainr. r$ r( r( r) �open_domain_handleM s �
�
�z$SamrPasswordTests.open_domain_handlec C sB t �| j�}| j�| j|g�}| j�| jtj|d j d �| _
d S )Nr ) r r< � user_namer8 � LookupNamesr. �OpenUserr
r: �idsr- )r% �name�ridsr( r( r) �open_user_handleX s
�z"SamrPasswordTests.open_user_handlec C sp d| | _ tdd�| _d| j | j�� f | _tj�| j| j� | jj | j | jd| j d | j | j | j d� d S )Nz SAMR_USER_%d� zcn=%s,cn=users,%sz
Password for z is )� description� givenname�surname)
r@ r � user_passr1 � domain_dnr2 r r r0 �newuser)r% �user_idr( r( r) r c s
�z%SamrPasswordTests.create_user_accountc C sN dd� }zt �� t�|||��}W t �� nt �� w t�� }t|�|_|S )Nc S st t dgd �}t�| �d��}t|�}td| �}d| }||d|�< ||||| �< |jddd�|dd�< t|�S )Nr i z utf-16-lei � �little)� byteorder)� bytearrayr �string_to_byte_array�encode�lenr �to_bytes�bytes)�password�data�p�plen�b�ir( r( r) �encode_pw_buffers s zCSamrPasswordTests.init_samr_CryptPassword.<locals>.encode_pw_buffer) r �set_relax_moder �arcfour_encrypt�set_strict_moder �
CryptPassword�listrY )r% rX � session_keyr^ �encrypted_blob�out_blobr( r( r) �init_samr_CryptPasswordq s
z)SamrPasswordTests.init_samr_CryptPassword�P@ssw0rdc
C s� t � | j| �� | j�| _| �� | �� d}d}t �� }d|_| � || jj
�|_ z | j� | j
||� W d S tjyb } z!t�|jd �j}t|� |tjkrV| jjdkrVn� W Y d }~d S d }~ww )Nrh � r �
disallowed)r r � get_loadparmr r8 r? rF �
UserInfo24�password_expiredrg rd rX � SetUserInfo2r- r �
NTSTATUSError�ctypes�c_uint32�args�value�printr �NT_STATUS_ACCESS_DENIEDr! � weak_crypto)r% rX �level�info�e�coder( r( r) �test_setUserInfo2_Password� s. �
���z,SamrPasswordTests.test_setUserInfo2_Passwordc C sr | j �t� t�| j| �� | j �| _| �� | �� d}d}t� � }d|_
| � || jj �|_
| j�| j||� d S )Nrh ri r )r �set_smb_encryptionr r r rk r8 r? rF rl rm rg rd rX rn r- )r% rX rw rx r( r( r) �$test_setUserInfo2_Password_Encrypted� s �z6SamrPasswordTests.test_setUserInfo2_Password_Encrypted)rh )
�__name__�
__module__� __qualname__r r+ r r? rF r rg r{ r} �
__classcell__r( r( r& r) r " s
r )�__doc__�osrp � samba.testsr r r r r �
samba.authr �samba.credentialsr r � samba.dcerpcr r
r � samba.samdbr r
r r( r( r( r) �<module> s