File: //usr/lib/python3/dist-packages/samba/tests/krb5/__pycache__/alias_tests.cpython-310.pyc
o
eF�c
� @ s� d dl Z d dlZe j�d d� dejd< d dlZd dlmZ d dlm m
m Z d dl m
Z
d dlmZmZmZmZ dZdZG d d
� d
e
�Zed krZdZdZd dlZe�� dS dS ) � Nz
bin/python�1�PYTHONUNBUFFERED)� delete_force)� KDCBaseTest)�AES256_CTS_HMAC_SHA1_96�ARCFOUR_HMAC_MD5�KDC_ERR_TGT_REVOKED� NT_PRINCIPALFc @ sJ e Zd Zdd� Zdd� Zddd�Zdd � Zd
d � Zdd d
�Zddd�Z dS )�
AliasTestsc C � | j dd� d S �N�rename��action��
_run_dc_alias��self� r �>/usr/lib/python3/dist-packages/samba/tests/krb5/alias_tests.py�test_dc_alias_rename+ � z AliasTests.test_dc_alias_renamec C r �N�deleter r r r r r �test_dc_alias_delete. r z AliasTests.test_dc_alias_deleteNc s4 | � � � � �� d d� }| j|� fdd�|d� d S )N���c s � S �Nr r �� target_credsr r �<lambda>5 s z*AliasTests._run_dc_alias.<locals>.<lambda>r )� get_dc_creds� get_username�
_run_alias)r r � target_namer r r r 1 s zAliasTests._run_dc_aliasc C r r ��_run_create_aliasr r r r �test_create_alias_rename7 r z#AliasTests.test_create_alias_renamec C r r r$ r r r r �test_create_alias_delete: r z#AliasTests.test_create_alias_deletec s* � � � �� �fdd�}� j�||d� d S )Nc sT � � � } | �� �� }�� d|� �}d|� �}d|i}� j| �� jj||d�\}}|S )N�.zldap/� dNSHostName)� account_type�spn�additional_details)� get_samdb�domain_dns_name�lower�create_account� AccountType�COMPUTER)�samdb�realm�hostnamer+ �details�creds�fn�r r# r r �
create_target@ s
�
�z3AliasTests._run_create_alias.<locals>.create_targetr )�get_new_usernamer" )r r r: r r9 r r% = s z AliasTests._run_create_aliasc sP �� � }��� }�j||�jjd�\}}��t||� ��||�}|�� � |� � } |�
| � t � t �
||��}
t �|t jd�|
d< |�|
� |�|� �j|ddd�����jd �} |� d� �� � �} ��|t| j�� ��|| j� ��|| j� ��| | j� ��� | j � |dkr���� }
t � t �
||��}
t �|
t jd�|
d< |�|
� n|d kr�|� |� n�� |� |� }|� � }|j |t j!d
gd �}t|d j"d
d d
��}�j#t$d|gd�}�j#t$|gd����%|�}��&t'j(j)�}t*t+f}� ���fdd�}t,}�j-� �� ||||�j.�j/�|dddd�}�j0|d � ||d�}��1||� d S )N)r* �sAMAccountName�0T)� kdc_options�freshzauthorization-data�@r
r r) )�scope�attrsr )�idx�ldap)� name_type�namesc s �j �� �jd d�}|g|fS )N)�namer4 �tgt_session_key�ctype)�PA_S4U2Self_create� session_key)�_kdc_exchange_dict�_callback_dict�req_body�padata�r4 r � target_cname�tgtr r �generate_s4u2self_padata� s �
z7AliasTests._run_alias.<locals>.generate_s4u2self_padataF)�expected_crealm�expected_cname�expected_srealm�expected_sname�ticket_decryption_key�generate_padata_fn�expected_error_mode�check_error_fn�check_kdc_private_fnrR �authenticator_subkeyr> �
expect_pac� expect_edata)�cnamer4 �sname�etypes)2r- r; r0 r1 r2 �
addCleanupr �
get_objectSid� get_realm�get_salt�set_forced_salt�ldb�Message�Dn�MessageElement�FLAG_MOD_REPLACE�modify� set_username�get_tgt� get_pac_data�ticket_privater/ � assertEqual�str� account_name� account_sid�
logon_name�upn� domain_namer �fail�get_dn�search�
SCOPE_BASE�get�PrincipalName_creater � TicketDecryptionKey_from_creds� RandomKey�kcrypto�Enctype�AES256r r r �tgs_exchange_dict�generic_check_kdc_error�generic_check_kdc_private�_generic_kdc_exchange�check_error_rep)r r# �target_creds_fnr r3 � mach_name�
mach_creds�mach_dn�mach_sid�old_salt�msg�pac_datarw �
mach_name2r � target_dn�res�target_hostnamera �target_decryption_keyr] rb rS rZ �kdc_exchange_dict�repr rP r r" W s�
�
�
�
���� ��zAliasTests._run_aliasr )
�__name__�
__module__� __qualname__r r r r& r' r% r" r r r r r
* s
r
�__main__)�sys�os�path�insert�environrh � samba.testsr �samba.tests.krb5.kcrypto�tests�krb5r� � samba.tests.krb5.kdc_base_testr �"samba.tests.krb5.rfc4120_constantsr r r r �global_asn1_print�global_hexdumpr
r� �unittest�mainr r r r �<module> s&
�