File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/audit_log_base.cpython-310.pyc
o
�/a� � @ st d Z ddlZddlmZ ddlmZmZ ddlm Z ddl
Z
ddl Z ddl Z ddl
Z
dd� ZG dd � d ejj�ZdS )
z Tests for DSDB audit logging.
� N)� Messaging)� MSG_AUTH_LOG�AUTH_EVENT_NAME)�LoadParmc C s d| vrd S | d }| | }|S )N�type� )�messager �auditr r �</usr/lib/python3/dist-packages/samba/tests/audit_log_base.py�getAudit s
r c sZ e Zd Z� fdd�Zdd� Zdd� Zddd �Zd
d � Zd Zd
d� Z dd� Z
dd� Z � Z S )�AuditLogTestBasec s t t���� t�d�}|rt|d�}n��� }td|d��_�j� �j
� ��� }|� d�d }|� d�d }|�_
�fdd �� � fd
d �}g d d ��_|�jf�_�jj�j�jd
� �j� t� � fdd�}ddd��_|�jf�_�jj�jtd
� ��� tjd �_d �_d S )N�SERVERCONFFILE)�filename_for_non_global_lp)� )�lp_ctx�
interfacesr �/c sX t | �}|d u r
dS |d }|d u rdS z
|�d�}|d � jkW S ty+ Y dS w )NF�
remoteAddress�:r )r �splitr �
IndexError)r r �remote�addr��selfr r
�isRemoteF s
�z(AuditLogTestBase.setUp.<locals>.isRemotec sl t |� t�|�}|d dks|d dks|d dkr(� |�r(| d �|� d S |d dkr4|| d< d S d S )Nr �passwordChange�
dsdbChange� groupChange�messages�dsdbTransaction�
txnMessage)�print�json�loads�append��context�msgType�srcr �jsonMsg�r r r
�messageHandlerU s
� �z.AuditLogTestBase.setUp.<locals>.messageHandler)r r! ��msg_typec sR t �|�}|d dkr%� |�r't|� |d d | d< |d d | d< d S d S d S )Nr �
Authorization� sessionId�serviceDescription)r# r$ r" r& r+ r r
� authHandleri s
��z+AuditLogTestBase.setUp.<locals>.authHandler� )r0 r1 �SERVER)�superr �setUp�os�getenvr � get_loadparmr �msg_ctx�
irpc_add_name�
event_type�getr r r' �msg_handler_and_context�register� message_typer � auth_context�auth_handler_and_contextr �discardMessages�environ�server�
connection)r � server_confr �client_ip_and_mask� client_ipr, r2 �� __class__)r r r
r6 ) s8
�
�
zAuditLogTestBase.setUpc C s\ | � � | j�| j� | j�t� | jr | jj| j| jd� | jr,| jj| jt d� d S d S )Nr- )
rC r: �irpc_remove_namer< r r> �
deregisterr@ rB r r r r r
�tearDown} s
�
��zAuditLogTestBase.tearDownc C sd |d u r
t | jd �|kS d}| jd D ]}t|�}|d �� |�� kr/|d7 }||kr/ dS qdS )Nr r �dnr TF)�lenr' r �lower)r �expectedrO �received�msgr r r r
� haveExpected� s �z AuditLogTestBase.haveExpectedNc C s� || _ t�� }| �||�s*| j�d� t�� | dkr$d| _ td� g S | �||�r
d| _ |du r6| jd S g }| jd D ]}t|�}|d �� |�� krR|� |� q=|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.
g�������?r Nz Timed outr rO )
rF �timerU r: � loop_oncer" r' r rQ r% )r �numberrF rO �
start_timer rT r r r r
�waitForMessages� s& �
�z AuditLogTestBase.waitForMessagesc C sl | j �d� t| jd �s| jd d ur4g | jd< d | jd< | j �d� t| jd �s| jd d usd S d S )Ng����MbP?r r! )r: rW rP r' r r r r
rC � s
�z AuditLogTestBase.discardMessagesz<[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 )N)�re�match�GUID_RE)r �guidr r r
�is_guid� s zAuditLogTestBase.is_guidc C �
| j d S )Nr0 �rA r r r r
� get_session� �
z AuditLogTestBase.get_sessionc C r` )Nr1 ra r r r r
�get_service_description� rc z(AuditLogTestBase.get_service_description)NN)
�__name__�
__module__� __qualname__r6 rN rU rZ rC r] r_ rb rd �
__classcell__r r rJ r
r ' s T
r )�__doc__� samba.tests�samba�samba.messagingr �samba.dcerpc.messagingr r � samba.paramr rV r# r7 r[ r �tests�TestCaser r r r r
�<module> s