File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/auth_log_base.cpython-310.pyc
o
�/a� � @ sx d Z ddlZddlmZ ddlmZmZ ddlm Z ddl
Z
ddl Z ddl Z ddl
Z
ddlmZ G dd� dejj�ZdS ) z&Tests for the Auth and AuthZ logging.
� N)� Messaging)� MSG_AUTH_LOG�AUTH_EVENT_NAME)�LoadParm)�paramc s^ e Zd Zedd� �Zedd� �Z� fdd�Zddd �Zed
d � �Zd d
� Z dZ
dd� Z � Z S )�AuthLogTestBasec C s� t �d�}|r
t|d�}ntj�� td|d�| _| j�t � tj�� }|�
d�d }|� d�d }|| _ dd � }d
g i| _
|| j
f| _| jj| jtd � d | _ t jd | _d | _d S )
N�SERVERCONFFILE)�filename_for_non_global_lp)� )�lp_ctx�
interfacesr �/c S s$ t |� t�|�}| d �|� d S )N�messages)�print�json�loads�append)�context�msgType�src�message�jsonMsg� r �;/usr/lib/python3/dist-packages/samba/tests/auth_log_base.py�messageHandler: s
z2AuthLogTestBase.setUpClass.<locals>.messageHandlerr ��msg_type�SERVER)�os�getenvr �samba�tests� env_loadparmr �msg_ctx�
irpc_add_namer �get�split�
remoteAddressr �msg_handler_and_context�registerr �environ�server�
connection)�self� server_confr �client_ip_and_mask� client_ipr r r r �
setUpClass" s&
�
zAuthLogTestBase.setUpClassc C s, | j r| jj| j td� | j�t� d S d S )Nr )r( r# �
deregisterr �irpc_remove_namer �r- r r r �
tearDownClassJ s
��z AuthLogTestBase.tearDownClassc s t t| ��� | �� d S �N)�superr �setUp�discardMessagesr4 �� __class__r r r8 Q s zAuthLogTestBase.setUpNc s� � �fdd�}�fdd��|�_ t�� }|�jd �s5�j�d� t�� | dkr.d�_ g S |�jd �rd�_ tt��jd ��S ) z�Wait for all the expected messages to arrive
The connection is passed through to keep the connection alive
until all the logging messages have been received.
c s$ | D ]
}�|�r� |�r dS qdS )NTFr )r r )�isLastExpectedMessage�isRemoter r � completed[ s
�z2AuthLogTestBase.waitForMessages.<locals>.completedc sz � j d u rdS d }| d dkr| d d }n| d dkr#| d d }ndS z
|�d�}|d � j kW S ty< Y dS w ) NT�type�
Authorizationr' �AuthenticationF�:r
)r' r& �
IndexError)r �remote�addrr4 r r r= a s
�z1AuthLogTestBase.waitForMessages.<locals>.isRemoter g�������?r
N)r, �timer r# � loop_once�list�filter)r- r< r, r> �
start_timer )r< r= r- r �waitForMessagesU s �z AuthLogTestBase.waitForMessagesc C sF | j �d� t| jd �r!g | jd< | j �d� t| jd �s
d S d S )Ng����MbP?r )r# rG �lenr r4 r r r r9 � s
�z AuthLogTestBase.discardMessagesc C s dd� }t t||��S )Nc S s d| vrdS | d d }|dkS )NrA T�serviceDescription�NETLOGONr )�msg�sdr r r �is_not_netlogon� s zAAuthLogTestBase.remove_netlogon_messages.<locals>.is_not_netlogon)rH rI )r- r rQ r r r �remove_netlogon_messages� s z(AuthLogTestBase.remove_netlogon_messagesz<[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}c C s t �| j|�S r6 )�re�match�GUID_RE)r- �guidr r r �is_guid� s zAuthLogTestBase.is_guidr6 )
�__name__�
__module__� __qualname__� classmethodr1 r5 r8 rK r9 rR rU rW �
__classcell__r r r: r r s
'
+
r )�__doc__� samba.testsr �samba.messagingr �samba.dcerpc.messagingr r � samba.paramr rF r r rS r r! �TestCaser r r r r �<module> s