File: //lib/python3/dist-packages/samba/tests/__pycache__/ntlmdisabled.cpython-310.pyc
o
�/a � @ sr d dl mZ d dlZd dlZd dlmZmZmZ d dlmZm Z d dl
Z
d dl m Z m
Z
mZ G dd� de�ZdS )� )�TestCaseN)� Credentials�DONT_USE_KERBEROS�MUST_USE_KERBEROS)�
NTSTATUSError�ntstatus)�srvsvc�samr�lsac s8 e Zd Z� fdd�Z� fdd�Zdd� Zdd� Z� ZS ) �NtlmDisabledTestsc s| t t| ��� | �� | _t�d�| _t� | _ | j �
| j� | j � t�d�� | j � | j� | j �
t�d�� | j �t� d S )N�SERVER�USERNAME�PASSWORD)�superr �setUp� get_loadparm�lp�os�getenv�serverr �creds�guess� set_username�
set_domain� set_password�set_kerberos_stater ��self�� __class__� �:/usr/lib/python3/dist-packages/samba/tests/ntlmdisabled.pyr $ s
zNtlmDisabledTests.setUpc s t t| ��� d S )N)r r �tearDownr r r r! r" 1 s zNtlmDisabledTests.tearDownc
C sz zt � d| j | j| j�}| �|� W d S ty< } z t�|jd �j }|t
j kr0| � d� n� W Y d }~d S d }~ww )Nzncacn_np:%s[smb2,ntlm]r � NTLM is disabled on this server)
r r r r �assertIsNotNoner �ctypes�c_uint32�args�valuer �NT_STATUS_NTLM_BLOCKED�fail)r �conn�e�enumr r r! �test_ntlm_connection4 s
���z&NtlmDisabledTests.test_ntlm_connectionc
C s� | j �t� t�dt�d� �}t�� }| j|_ t�� }t�d�|_ z|�
||d d dd d � W d S t ye } z)t �
|jd �j}|tjkrL| �d� n|tjkrRn� W Y d }~d S W Y d }~d S d }~ww )Nzncacn_np:%s[krb5,seal,smb2]r r
Tr r# )r r r r r r r
�Stringr �string�ChangePasswordUser2r r% r& r' r( r r) r* �NT_STATUS_WRONG_PASSWORD)r r+ r �usernamer, r- r r r! �test_samr_change_passwordA s&
���z+NtlmDisabledTests.test_samr_change_password)�__name__�
__module__� __qualname__r r" r. r4 �
__classcell__r r r r! r " s
r )� samba.testsr r �samba�samba.credentialsr r r r r r% � samba.dcerpcr r r
r r r r r! �<module> s