File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/prefork_restart.cpython-310.pyc
o
�/a@ � @ s� d Z ddlZddlZddlZddlZddlmZmZ ddlm Z m
Z
ddl m Z ddl
mZ ddlmZmZ ddlmZ dd lmZmZ dd
lmZ dd lmZ d Zd
Z G dd� de�Z dS )z�Tests process restarting in the pre-fork process model.
NOTE: As this test kills samba processes it won't play nicely with other
tests, so needs to be run in it's own environment.
� N)�TestCase� delete_force)�echo�netlogon)� Messaging)�SamDB)� Credentials�DONT_USE_KERBEROS)�
get_string)� UF_WORKSTATION_TRUST_ACCOUNT�UF_PASSWD_NOTREQD)�SEC_CHAN_WKSTA)�system_session� �PFRSc s� e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z� ZS )'�PreforkProcessRestartTestsc s& t t| ��� | �� }t|d�| _d S )N)�lp_ctx)�superr �setUp� get_loadparmr �msg_ctx)�selfr �� __class__� �=/usr/lib/python3/dist-packages/samba/tests/prefork_restart.pyr / s z PreforkProcessRestartTests.setUpc s t t| ��� d S �N)r r �tearDown)r r r r r 4 s z#PreforkProcessRestartTests.tearDownc C s: | j �� }g }|D ]}|jD ] }|�|j|jf� qq |S r )r �irpc_all_servers�ids�append�name�pid)r �services� processes�service�idr r r �get_process_data7 s
�z+PreforkProcessRestartTests.get_process_datac C s* | � � }|D ] \}}||kr| S qd S r )r' )r r! r$ �pnamer" r r r � get_process@ s �z&PreforkProcessRestartTests.get_processc C s@ g }t |�D ]}d�||�}|�| �|�� | �|| � q|S )N�prefork-worker-{0}-{1})�range�formatr r) �assertIsNotNone)r r! �workers�pids�x� process_namer r r �get_worker_pidsG s z*PreforkProcessRestartTests.get_worker_pidsc C s< t |�}t|�D ]}d�||�}| �||| ddd� qd S )Nr* r � � )�lenr+ r, �wait_for_process)r r! r. � num_workersr0 r1 r r r �wait_for_workersO s
�z+PreforkProcessRestartTests.wait_for_workersc C sd t �|� |}||k r'| �|�}|d ur||krd S t �|� ||7 }||k s | �d�||�� d S )Nz6Times out after {0} seconds waiting for {1} to restart)�time�sleepr) �failr, )r r! r" �
initial_delay�wait�timeout�delay�pr r r r6 U s
��z+PreforkProcessRestartTests.wait_for_processc C sh | � � }i }|D ])\}}|�d�s|�d�s|�d�r1||v r1||| kr1| �d�|||| �� qd S )Nzprefork-�_server�srvz+Duplicate process for {0}, pids {1} and {2})r' �
startswith�endswithr; r, )r r$ � process_mapr! r@ r r r � check_for_duplicate_processesb s
�� ���z8PreforkProcessRestartTests.check_for_duplicate_processesc C sL | j | �� d�}|�d|�� |�� f � tdtjd | �� |d�| _ d S )N)�templatez%s\%sz
ldaps://%s�SERVER)�url�lp� credentials)
� insta_creds�get_credentials� set_bind_dn�
get_domain� get_usernamer �os�environr �samdb)r �credsr r r � simple_bindp s � �z&PreforkProcessRestartTests.simple_bindc C s. t �d| �� �}| �g d�|�g d��� d S )Nzncalrpc:)r3 � � )r �rpcechor � assertEqual�EchoData)r �connr r r �rpc_echoy s z#PreforkProcessRestartTests.rpc_echoc C s t jd }t jd }| �� }| �� }t� }td| |||d�}t�dd�}t}d||� � f } t
|| � dt |� � d�}
|�
| d d
| tttB �|
d �� t� } | �|� | �t� | �t� | �|� | �|d � | �|� t�d
| || � t
|| � d S )NrH � SERVER_IPz ldap://%s)rI � session_inforK rJ � zcn=%s,%sz"%s"z utf-16-le�computerz%s$)�dn� objectclass�sAMAccountName�userAccountControl�
unicodePwd�$z ncacn_ip_tcp:%s[schannel,seal])rQ rR r rM r r �samba�generate_random_password� MACHINE_NAME� domain_dnr r
�encode�add�strr r r �guess�set_secure_channel_typer
�set_kerberos_stater � set_password� set_username�set_workstationr ) r �server�hostrJ rK �session�ldb� machine_pass� machine_name�
machine_dn�utf16pw�
machine_credsr r r r } sF
�
�
�z#PreforkProcessRestartTests.netlogonc C � | � � | �d�}| �|� | �dt�}t�|tj� | � d|ddd� | �
d|� | �d�}| �|� | � ||� | �dt�}t t�D ] }| � || || � qD| �
� | � � d S )N�prefork-master-ldap�ldapr3 r4 )rU r) r- r2 � NUM_WORKERSrQ �kill�signal�SIGTERMr6 r8 �assertNotEqualsr+ rF �r r" r. �new_pid� new_workersr0 r r r �test_ldap_master_restart� s
z3PreforkProcessRestartTests.test_ldap_master_restartc C s� | � � | �d�}| �|� | �dt�}t�|d tj� | � d|ddd� | �d�}| �|� | �
||� | �dt�}| � |d |d � | �
|d |d � | �
|d |d � | �
|d |d � | � � | � � d S ) Nr~ r r zprefork-worker-ldap-0r3 r4 rV rW )
rU r) r- r2 r� rQ r� r� r� r6 rY r� rF �r r" r. r� r� r r r �test_ldap_worker_restart� s
z3PreforkProcessRestartTests.test_ldap_worker_restartc C s� | � � | �d�}| �|� | �dt�}|D ] }t�|tj� q| � d|� | �d�}| �|� | �
||� | �dt�}t t�D ] }| � || || � q@| �
� | � � d S )Nr~ r )rU r) r- r2 r� rQ r� r� r� r8 rY r+ r� rF �r r" r. r0 r� r� r r r � test_ldap_all_workers_restart� s
z8PreforkProcessRestartTests.test_ldap_all_workers_restartc C r} )N�prefork-master-rpc�rpcr3 r4 )r\ r) r- r2 r� rQ r� r� r� r6 r8 r� r+ rF r� r r r �test_rpc_master_restart s
z2PreforkProcessRestartTests.test_rpc_master_restartc C s� | � � | �� | �d�}| �|� | �dt�}t�|d tj � | �
d|d ddd� | �d�}| �|� | � ||� | �dt�}| � |d |d � | � |d |d � | � |d |d � | � |d |d � | �
� | � � | �� d S ) Nr� r� r zprefork-worker-rpc-0r3 r4 rV rW )r\ r r) r- r2 r� rQ r� r� r� r6 rY r� rF r� r r r � test_rpc_worker_zero_restartB s$
z7PreforkProcessRestartTests.test_rpc_worker_zero_restartc C s� | � � | �� | �d�}| �|� | �dt�}|D ] }t�|tj � qt
t�D ]}| � d� |�|| ddd� q(| �d�}| �|� | �
||� | �dt�}t
t�D ] }| �|| || � qS| �� | � � | �� d S )Nr� r� zprefork-worker-rpc-{0}r r3 r4 )r\ r r) r- r2 r� rQ r� r� r� r+ r6 r, rY r� rF r� r r r � test_rpc_all_workers_restartj s(
�
z7PreforkProcessRestartTests.test_rpc_all_workers_restartc C s� | � d�}| �|� d}dD ]=}| �dt�}| � d�}t�|tj� t�� }| � d|ddd� | �
d|� t�� }|| }| � ||| � | � ||� q| �dt�} t
t�D ] }
| �||
| |
� qV| �� d S ) Nzprefork-master-echo� �r r� �
r� �kdcr r3 r4 r )r) r- r2 r� rQ r� r� r� r9 r6 r8 �
assertLess�assertGreaterEqualr+ r� rF ) r r" �backoff_increment�expectedr. �process�start�end�durationr� r0 r r r �test_master_restart_backoff� s$
z6PreforkProcessRestartTests.test_master_restart_backoffc C s� d}dD ]6}| � d�}| �|� t�|tj� t�� }| �d|ddd� t�� }|| }| �||| � | � ||� q| �
� d S )Nr� r� zprefork-worker-echo-2r r3 r4 ) r) r- rQ r� r� r� r9 r6 r� r� rF )r r� r� r� r� r� r� r r r �test_worker_restart_backoff� s
z6PreforkProcessRestartTests.test_worker_restart_backoff)�__name__�
__module__� __qualname__r r r' r) r2 r8 r6 rF rU r\ r r� r� r� r� r� r� r� r� �
__classcell__r r r r r - s(
*'*$&(((r ) �__doc__rQ r� r9 rg � samba.testsr r � samba.dcerpcr r �samba.messagingr � samba.samdbr �samba.credentialsr r � samba.commonr
�
samba.dsdbr r �samba.dcerpc.miscr
�
samba.authr r� ri r r r r r �<module> s"