File: //lib/python3/dist-packages/samba/tests/__pycache__/gensec.cpython-310.pyc
o
�/a�# � @ s@ d Z ddlmZ ddlmZmZ ddlZG dd� dejj�Z dS )z�Tests for GENSEC.
Note that this just tests the bindings work. It does not intend to test
the functionality, that's already done in other tests.
� )� Credentials)�gensec�authNc s� e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
d �Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Z� ZS )!� GensecTestsc s` t t| ��� i | _tj�� | jd< | _| j�d�| jd< | j� dd� t
j � | j�| _
d S )N�lp_ctxz netbios name�target_hostname�spnego:simulate_w2k�no)
�superr �setUp�settings�samba�tests� env_loadparmr �get�setr �Security� start_client��self�� __class__� �4/usr/lib/python3/dist-packages/samba/tests/gensec.pyr s zGensecTests.setUpc C s | � t| jjd� d S )N�foo)� assertRaises� RuntimeErrorr �start_mech_by_namer r r r � test_start_mech_by_unknown_name) s z+GensecTests.test_start_mech_by_unknown_namec C s | j �d� d S )N�spnego)r r r r r r �test_start_mech_by_name, � z#GensecTests.test_start_mech_by_namec C s | � t| jj� d S )N)r r r � session_infor r r r �test_info_uninitialized/ s z#GensecTests.test_info_uninitializedNc
C s� |r| j �|�}|s
d}| j �|d� tj�| j�| _| j�| � � � | j�
tj � |dur6| j� |� n| j�
|� |rE| j �|d� tjj| jtj| j d�d�| _t� }|�| j � |�| j � | j�|� | j�
tj � | j�
|� d}d}d}d} |s�|r�| j �|d� td
� z
| j�|�\}} W n tjy� }
zt|
��d}
~
ww |r�| j �|d� |s�td � z
| j�| �\}}W n tjy� }
zt|
��d}
~
ww |r�|r�nq�|r�| j �||� | �|� | �|� | j�� } d } z| j� | �}
| j� |
�}W n tj�y } z
| � t |�� W Y d}~nd}~ww | � | |� d
} | j� | �}
| j� |
�}| � | |� | j�!� }| j�!� }| � ||� dS )�JTest GENSEC by doing an exchange with ourselves using GSSAPI against a KDC� �yesNr �r �r � auth_contextF� Tz running client gensec_updatez running server gensec_update� Hello Server� Hello Client)"r r r r r r r �
gensec_client�set_credentials�get_credentials� want_feature� FEATURE_SEALr �start_mech_by_sasl_name� start_serverr � AuthContext�
gensec_serverr �guess�set_machine_account�print�updater
�
NTSTATUSError�AssertionError�
assertTruer" �wrap�unwrap�fail�str� assertEqual� session_key)r �mech� client_mech�client_only_opt�orig_client_opt�creds�client_finished�server_finished�server_to_client�client_to_server�ntr" �
test_bytes� test_wrapped�test_unwrapped�e�client_session_key�server_session_keyr r r � _test_update2 s�
� �����
��
zGensecTests._test_updatec C � | � d� d S )N�GSSAPI�rS r r r r � test_update� � zGensecTests.test_updatec C rT )N�
GSS-SPNEGOrV r r r r �test_update_spnego� rX z GensecTests.test_update_spnegoc C � | � ddd� d S )NrY r zgensec:gssapi_krb5rV r r r r � test_update_spnego_downgrade� � z(GensecTests.test_update_spnego_downgradec C r[ )NrY r zspnego:client_no_optimisticrV r r r r � test_update_no_optimistic_spnego� r] z,GensecTests.test_update_no_optimistic_spnegoc C s: | j �dd� tj�| j�| _| j �dd� | �d� d S )Nr r& r rY �r r r r r r rS r r r r � test_update_w2k_spnego_client� s z)GensecTests.test_update_w2k_spnego_clientc C s, t j�| j�| _ | j�dd� | �d� d S )Nr r& rY )r r r r r r rS r r r r � test_update_w2k_spnego_server� s z)GensecTests.test_update_w2k_spnego_serverc C s, | j �dd� tj�| j�| _| �d� d S )Nr r rY r_ r r r r �test_update_w2k_spnego� s z"GensecTests.test_update_w2k_spnegoc C � | � dd� d S )NrY � gssapi_krb5rV r r r r � test_update_gss_krb5_to_spnego� r! z*GensecTests.test_update_gss_krb5_to_spnegoc C rc )NrY �ntlmssprV r r r r � test_update_ntlmssp_to_spnego� r! z)GensecTests.test_update_ntlmssp_to_spnegoc
C s� t j�| j�| _| j�| �� � | j�t j� | j� d� | j�
d� t jj | jt j
| jd�d�| _t� }|�| j� |�| j� | j�|� | j�t j� | j� d� | j�
d� d}d}d} d}|rh|s�|d7 }|s�td t|�|f � | j�|�\}}|s�td
t|�|f � | j�|�\}}|rh|rh | �|d k� | j�� }d }| j�|�} | j�| �}
| �||
� d
}| j�|�} | j�| �}
| �||
� | j�� } | j�� } | �| | � dS )r$ � r r' r( Fr* r � z$running client gensec_update: %d: %rz$running server gensec_update: %d: %r�
r+ r, N) r r r r r- r. r/ r0 � FEATURE_SIGN�set_max_update_sizer r3 r r4 r r5 r r6 r7 r8 �lenr9 r< r" r= r> rA rB )
r rG rH rI rJ �irK r" rM rN rO rQ rR r r r �test_max_update_size� sX
� �
z GensecTests.test_max_update_size)NN)�__name__�
__module__� __qualname__r r r r# rS rW rZ r\ r^ r` ra rb re rg ro �
__classcell__r r r r r s
T r )
�__doc__�samba.credentialsr r
r r � samba.testsr �TestCaser r r r r �<module> s