File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/auth_log_samlogon.cpython-310.pyc
o
�/a� � @ s� d Z ddlZddlZddlmZ ddlZddlmZm Z m
Z
ddl m Z m
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZmZ dd lmZ dd lmZm Z G d
d� dej j j �Z dS )z5
Tests auth logging tests that exercise SamLogon
� N)�SamDB)� Credentials�DONT_USE_KERBEROS�CLI_CRED_NTLMv2_AUTH)�ntlmssp�netlogon)�AS_SYSTEM_MAGIC_PATH_TOKEN)�ndr_pack)�system_session)� delete_force)� UF_WORKSTATION_TRUST_ACCOUNT�UF_PASSWD_NOTREQD)�SEC_CHAN_WKSTA��EVT_ID_SUCCESSFUL_LOGON�EVT_LOGON_NETWORKc s@ e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Z� ZS ) �AuthLogTestsSamLogonc st t t| ��� tj�� | _t� | _t | j| jd�| _
t j d | _
d| _d| _t| _| j
�� | _d| j| jf | _d S )N)� session_info�lp�DOMAIN� SamLogonTest�
abcdefghijzcn=%s,cn=users,%s)�superr �setUp�samba�tests� env_loadparmr r
�sessionr �ldb�os�environ�domain� netbios_name� machinepassr �
remoteAddress� domain_dn�base_dn� samlogon_dn��self�� __class__� �?/usr/lib/python3/dist-packages/samba/tests/auth_log_samlogon.pyr . s �
�zAuthLogTestsSamLogon.setUpc s t t| ��� t| j| j� d S )N)r r �tearDownr r r' r( r* r, r- r. > s z AuthLogTestsSamLogon.tearDownc C s� dd� }|r d| }nd}d| j d �d�}| j�| jdd| j tttB �|d �� t � }|�
| � � � |� t
� |�| j � |�| jd
� t�d | | � � |�}d }t�� } d
| _t�� }
tj|
_| j|
_t�� } tj| _| j| _t�� } tj| _|
| | g| _t | �}
|j t ||
d�}d}tj }t j!j�"� }dd� |D �|_#t�$� |_%t&|d �|j%_'dd� |d D �|j%_(t j!j�)� |_*|�+� \}}||j*j,_-||j*j._-|�/� |j*j0_-t j!jj1}|�2t3j4d |�/� ||||�}|\}}}| �5||�}||� d S )Nc S s\ | d dko-| d d dko-| d d dko-| d d dko-| d d t ko-| d d
tkS ) N�type�Authentication�serviceDescription�SamLogon�authDescription�network� passwordType�NTLMv2�eventId� logonTyper )�msgr, r, r- �isLastExpectedMessageD s ���
���zBAuthLogTestsSamLogon._test_samlogon.<locals>.isLastExpectedMessagez
[schannel,%s]z
[schannel]�"z utf-16-le�computerz%s$)�dn� objectclass�sAMAccountName�userAccountControl�
unicodePwd�$z
ncalrpc:%ss abcdefgh� )�flags� challenge� target_infor c S �" g | ]
}t |t�r |nt|��qS r, ��
isinstance�int�ord��.0�xr, r, r- �
<listcomp>� s �z7AuthLogTestsSamLogon._test_samlogon.<locals>.<listcomp>� nt_responsec S rG r, rH rL r, r, r- rO � s ���SERVER)6r# �encoder �addr' r" �strr r
r �guess� get_loadparm�set_secure_channel_typer � set_password� set_usernamer r � AV_PAIR_LIST�count�AV_PAIR�MsvAvNbDomainName�AvIdr! �Value�MsvAvNbComputerName�MsvAvEOL�pairr �get_ntlm_responser �$NetlogonNetworkTransitiveInformationr �dcerpc�netr_NetworkInforE �netr_ChallengeResponse�nt�len�length�data�netr_IdentityInfo�
identity_info�get_ntlm_username_domain� domain_name�string� account_name�get_workstation� workstation�NetlogonValidationSamInfo4�netr_LogonSamLogonExr r �waitForMessages)r) �binding�creds�
checkFunctionr: �utf16pw�
machine_creds�
netlogon_connrE rF �
domainname� computername�eol�target_info_blob�response�
netr_flags� logon_level�logon�usernamer! �validation_level�result�
validation�
authoritative�netr_flags_out�messagesr, r, r- �_test_samlogonB s�
�
� � �
�
�
z#AuthLogTestsSamLogon._test_samlogonc C s� | � |�}d}| �|t|�d� |d }| �d|d � | �d|d d � | �d|d d � | �d
|d d � | �| �|d d �� d S )
N� z/Did not receive the expected number of messagesr �
Authorizationr/ zDCE/RPCr1 �ncalrpc�authType�NONE�transportProtection� sessionId)�remove_netlogon_messages� assertEqualri �
assertTrue�is_guid)r) r� �expected_messagesr9 r, r, r- �samlogon_check� s
�
� z#AuthLogTestsSamLogon.samlogon_checkc
C sb | j | �� td�}z | �d|| j� W d S ty0 } z| �dt|� � W Y d }~d S d }~ww )N)�template�kerberos_state�SEALzUnexpected exception: )� insta_creds�get_credentialsr r� r� � Exception�failrT )r) rx �er, r, r- �test_ncalrpc_samlogon� s
� ��z*AuthLogTestsSamLogon.test_ncalrpc_samlogon) �__name__�
__module__� __qualname__r r. r� r� r� �
__classcell__r, r, r* r- r , s [r )!�__doc__� samba.testsr r � samba.samdbr �samba.tests.auth_log_base�samba.credentialsr r r � samba.dcerpcr r �samba.dcerpc.dcerpcr � samba.ndrr �
samba.authr
r �
samba.dsdbr r
�samba.dcerpc.miscr � samba.dcerpc.windows_event_idsr r r �
auth_log_base�AuthLogTestBaser r, r, r, r- �<module> s