File: //lib/python3/dist-packages/samba/tests/__pycache__/getdcname.cpython-310.pyc
o
�/aJ � @ sl d Z ddlmZ ddlmZmZ ddlZddlZddlmZ ddl m
Z
ddl m Z G dd � d ej
j�ZdS )
z)
Tests GetDCNameEx calls in NETLOGON
� )�auth)� WERRORError�werrorN)� Credentials)�netlogon)�GUIDc @ s� e Zd Zdd� Z d,dd�Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� ZdS )-� GetDCNameExc C sd t j�� | _t� | _d | _tj� d�| _
tj� d�| _ tj� d�| _ tj� d�| _
tj� d�| _d S )N�SERVER�REALM�DOMAIN� TRUST_REALM� TRUST_DOMAIN)�samba�tests� env_loadparm�lpr �creds�
netlogon_conn�os�environ�get�server�realm�domain� trust_realm� trust_domain)�self� r �7/usr/lib/python3/dist-packages/samba/tests/getdcname.py�setUp! s zGetDCNameEx.setUpNFr c C sP | j d u rt�d| �� �| _ |r | j �| jd d||||�S | j �| j||||�S )Nzncalrpc:[schannel]r )r r � get_loadparm�netr_DsRGetDCNameEx2r �netr_DsRGetDCNameEx)r r � domain_guid� site_name�ex2�flagsr r r �_call_get_dc_name, s$
�
�
�z GetDCNameEx._call_get_dc_namec C s� | j dd�}| �|jdu� | �|j�d�� | �|jdu� | �|j�d�� | �|j�� | j�� kp=|j�� | j�� k� | j dt j
d�}| � |j�� | j�� � | j dt j d�}| � |j�� | j�� � dS )z�Check the most trivial requirements of Ex2 (no domain or site)
a) The paths are prefixed with two backslashes
b) The returned domains conform to the format requested
c) The domain matches our own domain
T)r% N�\\)r% r& )
r' �
assertTrue�dc_unc�
startswith�
dc_address� domain_name�lowerr r r �DS_RETURN_DNS_NAME� assertEqual�DS_RETURN_FLAT_NAME�r �responser r r �test_get_dc_ex2@ s0 ���� �� �zGetDCNameEx.test_get_dc_ex2c C s | j du rdS | j| j dd�}| j| jdd�}| �|jdu� | �|j�d�� | �|jdu� | �|j�d�� | �|j|j� | �|j|j� | �|j� � | j � � kp^|j� � | j
� � k� | j| j t j dd�}| �
|j� � | j � � � | j| j t jdd�}| �
|j� � | j
� � � dS )z�Check what happens to Ex2 requests after being forwarded to winbind
a) The paths must still have the same backslash prefixes
b) The returned domain does not match our own domain
c) The domain matches the format requested
NT)r r% r( �r r& r% )r r' r r) r* r+ r, �assertNotEqualr- r. r r r/ r0 r1 )r �response_trustr3 r r r � test_get_dc_over_winbind_ex2] sN
���� ���� �� �z(GetDCNameEx.test_get_dc_over_winbind_ex2c C s� | j du rdS | j| j tjd�}| �|jdu� | �|j�d�� | �|jdu� | �|j�d�� | �|j �
� | j �
� � dS )ziTest the standard Ex version (not Ex2)
Ex calls Ex2 anyways, from now on, just test Ex.
N�r r& r( ) r r' r r/ r) r* r+ r, r0 r- r. )r r7 r r r �test_get_dc_over_winbind� s
� �z$GetDCNameEx.test_get_dc_over_winbindc C s� | j du rdS d}| j| j |tjd�}| �|jdu� | �|j�d�� | �|jdu� | �|j�d�� | �|j �
� | j �
� � | �|�
� |j �
� � dS )�pTest the standard Ex version (not Ex2)
We assume that there is a Default-First-Site-Name site.
N�Default-First-Site-Name�r r$ r& r( ) r r' r r/ r) r* r+ r, r0 r- r. � dc_site_name)r �siter7 r r r �"test_get_dc_over_winbind_with_site� s
� �z.GetDCNameEx.test_get_dc_over_winbind_with_sitec
C � | j du rdS d}z| j| j |tjdd�}| �d� W dS tyE } z |j\}}|tjkr:| �d� W Y d}~dS W Y d}~dS d}~ww �zqTest the standard Ex version (not Ex2)
We assume that there is no Invalid-First-Site-Name site.
N�Invalid-First-Site-NameF�r r$ r& r% �3Failed to give the correct error for incorrect site�%Failed to detect an invalid site name� r r' r r/ �failr �argsr �WERR_NO_SUCH_DOMAIN�r r? r7 �e�enum�estrr r r �%test_get_dc_over_winbind_invalid_site� �"
�
���z1GetDCNameEx.test_get_dc_over_winbind_invalid_sitec
C rA )z`Test the Ex2 version.
We assume that there is no Invalid-First-Site-Name site.
NrC TrD rE rF rG rK r r r �)test_get_dc_over_winbind_invalid_site_ex2� rP z5GetDCNameEx.test_get_dc_over_winbind_invalid_site_ex2c
C s� | j du rdS d}z | j| j |tjd�}W n ty0 } z| �dt|� � W Y d}~nd}~ww | �|jdu� | �|j� d�� | �|j
du� | �|j
� d�� | � |j �
� | j �
� � | �|jdu� | �d|j� dS )r; N� r= �(Unable to get empty string site result: r( )r r' r r/ r rH �strr) r* r+ r, r0 r- r. r> r6 �r r? r7 rL r r r �*test_get_dc_over_winbind_empty_string_site� s*
� �� �z6GetDCNameEx.test_get_dc_over_winbind_empty_string_sitec
C s� | j du rdS z | j| jtjdd�}W n ty. } z| �dt|� � W Y d}~nd}~ww | �|du� | � |j
� � | j � � � dS )z#Supply a NETBIOS trust domain name.NFr5 � Failed to succeed over winbind: ) r r' r r r/ r rH rT r) r0 r- r. )r r7 rL r r r � test_get_dc_over_winbind_netbios� s
� �� �z,GetDCNameEx.test_get_dc_over_winbind_netbiosc
C s� | j du rdS d}z
| j| j|tjdd�}W n ty3 } z| �d| j||f � W Y d}~nd}~ww | �|du� | �|j �
� | j �
� � | �|�
� |j �
� � dS )z�Supply a NETBIOS trust domain name.
Sporadically fails because NETBIOS queries do not return site name in
winbind. The site check in NETLOGON will trigger and fail the request.
Currently marked in flapping...
Nr< FrD z7get_dc_name (domain=%s,site=%s) over winbind failed: %s) r r' r r r/ r rH r) r0 r- r. r> rU r r r �*test_get_dc_over_winbind_with_site_netbios s(
�
��� �z6GetDCNameEx.test_get_dc_over_winbind_with_site_netbiosc
C s� | j du rdS t� }z | j| j |tjd�}W n ty1 } z| �dt|� � W Y d}~nd}~ww | �|j du� | �|j �
d�� | �|j du� | �|j �
d�� | � |j
�� | j �� � dS )�?Ensure that we do not reject requests supplied with a NULL GUIDN�r r# r& z'Unable to get NULL domain GUID result: r( )r r r' r r/ r rH rT r) r* r+ r, r0 r- r. )r � null_guidr7 rL r r r �$test_get_dc_over_winbind_domain_guid s&
� �� �z0GetDCNameEx.test_get_dc_over_winbind_domain_guidc C s� d}| j | j|tjd�}| �|jdu� | �|j�d�� | �|jdu� | �|j�d�� | �|j �
� | j�
� � | �|�
� |j �
� � dS )r; r< r= Nr( ) r' r r r/ r) r* r+ r, r0 r- r. r> )r r? r3 r r r �test_get_dc_with_site5 s � �z!GetDCNameEx.test_get_dc_with_sitec
C rA rB ) r r' r r/ rH r rI r rJ �r r? r3 rL rM rN r r r �test_get_dc_invalid_siteJ rP z$GetDCNameEx.test_get_dc_invalid_sitec
C s~ d}z| j | j|tjdd�}| �d� W dS ty> } z |j\}}|tjkr3| �d� W Y d}~dS W Y d}~dS d}~ww )z_Test the Ex2 version
We assume that there is no Invalid-First-Site-Name site.
rC TrD rE rF N) r' r r r/ rH r rI r rJ r_ r r r � test_get_dc_invalid_site_ex2^ s �
���z(GetDCNameEx.test_get_dc_invalid_site_ex2c
C s� d}z | j | j|tjd�}W n ty) } z| �dt|� � W Y d}~nd}~ww | �|jdu� | �|j� d�� | �|j
du� | �|j
� d�� | � |j �
� | j�
� � | �|jdu� | �d|j� dS )r; rR r= rS Nr( )r' r r r/ r rH rT r) r* r+ r, r0 r- r. r> r6 �r r? r3 rL r r r � test_get_dc_empty_string_sitep s&
� �� �z)GetDCNameEx.test_get_dc_empty_string_sitec
C sz z | j | jtjdd�}W n ty' } z| �dt|� � W Y d}~nd}~ww | �|du� | �|j �
� | j �
� � dS )� Supply a NETBIOS domain name.Fr5 rW N) r' r r r/ r rH rT r) r0 r- r. r )r r3 rL r r r �test_get_dc_netbios� s
� �� �z GetDCNameEx.test_get_dc_netbiosc
C s� d}z
| j | j|tjdd�}W n ty* } z| �dt|� � W Y d}~nd}~ww | �|du� | �|j �
� | j �
� � | �|�
� |j �
� � dS )rd r< FrD rW N)
r' r r r/ r rH rT r) r0 r- r. r r> rb r r r � test_get_dc_with_site_netbios� s
� �� �z)GetDCNameEx.test_get_dc_with_site_netbiosc C sz t � }| j| j|tjd�}| �|jdu� | �|j�d�� | �|jdu� | �|j�d�� | � |j
� � | j� � � dS )rZ r[ Nr( ) r r' r r r/ r) r* r+ r, r0 r- r. )r r\ r3 r r r � test_get_dc_with_domain_guid� s � �z(GetDCNameEx.test_get_dc_with_domain_guidc C sp | j dtjd�}| �|jdu� | �|j�d�� | �|jdu� | �|j�d�� | �|j� � | j
� � � dS )z1Ensure that empty domain resolve to the DC domainrR r9 Nr( ) r' r r/ r) r* r+ r, r0 r- r. r r2 r r r �$test_get_dc_with_empty_string_domain� s � �z0GetDCNameEx.test_get_dc_with_empty_string_domain)NNNFr )�__name__�
__module__� __qualname__r r' r4 r8 r: r@ rO rQ rV rX rY r] r^ r` ra rc re rf rg rh r r r r r s.
� * r )�__doc__r r r r � samba.testsr �samba.credentialsr � samba.dcerpcr �samba.dcerpc.miscr r �TestCaser r r r r �<module> s