o

�/a��@sbdZd
dl
Zd
dlmZ
d
dlmZ
d
dlmZ
d
dl	Zd
dl
mZmZ
Gdd�dej
jj�ZdS)	z&Tests for the Auth and AuthZ logging.
�N)
�DONT_USE_KERBEROS)
�AS_SYSTEM_MAGIC_PATH_TOKEN)
�samr)�EVT_ID_SUCCESSFUL_LOGON�EVT_LOGON_NETWORKcsHeZ
dZ�f
d
d�Z�f
dd�Zdd�Zdd�Zd	d
�Zdd�Z�Z	S)
�AuthLogTestsNcalrpcc

stt
|���
t|_dS�
N)�superr�setUpr�
remoteAddress�
�self�
�	__class__��>/usr/lib/python3/dist-packages/samba/tests/auth_log_ncalrpc.pyr
"s


zAuthLogTestsNcalrpc.setUpc

stt
|���
dSr)r	r�tearDownrrrrr&s
zAuthLogTestsNcalrpc.tearDowncsJ��
fd
d�}|r
d|}t�d||�
�|�
|�|�
}||��
�
dS)Nc

s@|d
dko|dddko|dd�dko|dd�
kS)N�type�
Authorization�serviceDescription�DCE/RPC�authTyper
�transportProtectionr)
�msg��	authTypes�
protectionrr�isLastExpectedMessage,s
���zEAuthLogTestsNcalrpc._test_rpc_ncaclrpc.<locals>.isLastExpectedMessagez[%s]z
ncalrpc:%s)r�get_loadparm�waitForMessages)r
r�binding�credsr�
checkFunctionr�messagesrrr�_test_rpc_ncaclrpc)s



z&AuthLogTestsNcalrpc._test_rpc_ncaclrpccCs
t|�
}|�
|t|
�
d
�
|
d}|�
d|d�
|�
d|dd�
|�
|d|dd�
|�
d	|dd
�
|�|�|dd�
�

|
d}|�
d|d�
|�
d
|dd�
|�
d|dd�
|�
|d|dd�
|�
t|dd�
|�
t|dd�
dS)Nz/Did not receive the expected number of messagesr
rrrr�
r�NONEr�	sessionId�Authentication�NT_STATUS_OK�status��authDescription�eventId�	logonType)�len�assertEqual�
assertTrue�is_guidrr)r
r#rr�expected_messagesrrrr�rpc_ncacn_np_ntlm_check:s8


�



�






�


�

�

�z+AuthLogTestsNcalrpc.rpc_ncacn_np_ntlm_checkc
C�.|j|�
�td
�}
|�gd�
d|
d|j�
dS)N��template�kerberos_state��NTLMSSP�ncalrpcr:��SIGN��insta_creds�get_credentialsrr$r4�r
r!rrr�test_ncalrpc_ntlm_dns_signW�

�

�z.AuthLogTestsNcalrpc.test_ncalrpc_ntlm_dns_signc
Cr5)Nr6r9�seal�SEALr>rArrr�test_ncalrpc_ntlm_dns_sealarCz.AuthLogTestsNcalrpc.test_ncalrpc_ntlm_dns_seal)
�__name__�
__module__�__qualname__r
rr$r4rBrF�
__classcell__rrrrr s
r)�__doc__�samba.tests�samba�samba.credentialsr�samba.dcerpc.dcerpcr�samba.dcerpcr�samba.tests.auth_log_base�samba.dcerpc.windows_event_idsrr�tests�
auth_log_base�AuthLogTestBaserrrrr�<module>
s