File: //usr/lib/python3/dist-packages/samba/tests/krb5/__pycache__/spn_tests.cpython-310.pyc
o
eF�c� � @ s� d dl Z d dlZe j�d d� dejd< d dlmZ d dlZd dlm Z d dl
m Z d dl m
Z
mZmZmZ d Zd ZeG d
d � d e ��Zed krVd Zd Zd dlZe�� dS dS )
� Nz
bin/python�1�PYTHONUNBUFFERED)�DynamicTestCase)� KDCBaseTest)�KerberosCredentials)�AES256_CTS_HMAC_SHA1_96�ARCFOUR_HMAC_MD5�KDC_ERR_S_PRINCIPAL_UNKNOWN� NT_PRINCIPALFc s� e Zd Zejjejjejjd�Zdddddd�Z e
� fdd ��Z e
d
d � �Z d d
� Z
� fdd�Zddd�Zddd�Zdd� Zdd� Z� ZS )�SpnTests)�computer�server�rodczldap/{{account}}z&ldap/{{account}}/{netbios_domain_name}z"ldap/{{account}}/{dns_domain_name}zldap/{{account}}/testz'ldap/{{account}}:test/{dns_domain_name})�2_part�3_part_our_domain�3_part_our_realm�3_part_not_our_realm�3_part_instancec s t � �� d | _d S �N)�super�
setUpClass�_mock_rodc_creds)�cls�� __class__� �</usr/lib/python3/dist-packages/samba/tests/krb5/spn_tests.pyr : s
zSpnTests.setUpClassc C sV | j �� D ]#\}}| j�� D ]\}}|� d|� �}||f}| jd|g|�R � qqd S )N�_spn_�test_spn)�test_account_types�items� test_spns�generate_dynamic_test)r �account_type_name� account_type�spn_name�spn�tname�targsr r r �setUpDynamicTestCases@ s ��z SpnTests.setUpDynamicTestCasesc C s, | � |�}| �||�}| jt|�d�d�}| �� }| �|�}| �� }|�� }|� � } | �
|j j �}
t
tf} || jju r>t
} nd } | j|| d�}
|�d�dkrl|�|�sX|�| �rl|| jjurl|| jjurlt}| j}d }nd}d }| j}| j|j|j|j||
||| j |||
ddd�
}| j |d |j|| d � d S )
N�/)� name_type�names)�etype� r �0F)
�expected_crealm�expected_cname�expected_srealm�expected_sname�ticket_decryption_key� check_rep_fn�check_error_fn�check_kdc_private_fn�expected_error_mode�tgt�authenticator_subkey� kdc_options� expect_edata)�cname�realm�sname�etypes) �
_get_creds� _format_spn�PrincipalName_creater
�split�get_client_creds�get_tgt� get_samdb�domain_netbios_name�domain_dns_name� RandomKey� session_keyr- r r � AccountType�SERVER� TicketDecryptionKey_from_creds�count�endswith�RODCr �generic_check_kdc_error�generic_check_kdc_rep�tgs_exchange_dict�crealmr= �srealm�generic_check_kdc_private�_generic_kdc_exchange)�selfr$ r& � target_credsr? � client_credsr9 �samdb�netbios_domain_name�dns_domain_name�subkeyr@ � ticket_etype�decryption_keyr8 r6 r5 �kdc_exchange_dictr r r �_test_spn_with_argsH sh
�
��� �
�z SpnTests._test_spn_with_argsc s t � �� t| _t| _d S r )r �setUp�global_asn1_print�
do_asn1_print�global_hexdump�
do_hexdump)rY r r r rd � s
zSpnTests.setUpNc s t � �fdd�|�S )Nc s �� | � �S r )rB )r& ��credsrY r r �<lambda>� s z'SpnTests._format_spns.<locals>.<lambda>)�map)rY �spnsrj r ri r � _format_spns� s zSpnTests._format_spnsc C s> | � � }|j|�� |�� d�}|d ur |�� }|j|d�}|S )N)r] r^ ��account)rG �formatrH rI � get_username)rY r& rj r\ � account_namer r r rB � s
� zSpnTests._format_spnc C sX | � | j�� �}|| jju r!| j}|d u r | �|�}|t| �_|S | j|d|id�}|S )Nr& )r$ �opts) rn r! �valuesrL rQ r �_get_mock_rodc_creds�type�get_cached_creds)rY r$ rm rj r r r rA � s
���zSpnTests._get_credsc C s0 | � � }|D ]}|j|jd�}||jvr|j�|� q| �� }t�||j�}t� |�}t�
|jtj d�|d< |� |� t
� }|�| �� � |�|j�� � |�|j� |�|j� |�|j� |�|j� |�|� |�|j� |j |tj dgd�}t |d j ddd��} |� | � | �!||�}
| �"||
� |S )Nro �servicePrincipalNamezmsDS-KeyVersionNumber)�base�scope�attrsr )�idx)#�get_mock_rodc_ctxrq �myname�SPNs�appendrG �ldb�Dn�acct_dn�Message�MessageElement�FLAG_MOD_REPLACE�modifyr �guess�get_lp� set_realmr> �upper�
set_domain� domain_name� set_password� acct_pass� set_username�set_workstation�samname�set_dn�set_spn�search�
SCOPE_BASE�int�get�set_kvno�get_keys�creds_set_keys) rY rm �rodc_ctxr& r\ �rodc_dn�msgrj �res�kvno�keysr r r rv � sB
�
�
�
z SpnTests._get_mock_rodc_credsr )�__name__�
__module__� __qualname__r rL �COMPUTERrM rQ r r! � classmethodr r) rc rd rn rB rA rv �
__classcell__r r r r r + s* ��
;
r �__main__)�sys�os�path�insert�environ� samba.testsr r� � samba.tests.krb5.kdc_base_testr � samba.tests.krb5.raw_testcaser �"samba.tests.krb5.rfc4120_constantsr r r r
re rg r r� �unittest�mainr r r r �<module> s(
% �