File: //lib/python3/dist-packages/samba/tests/__pycache__/join.cpython-310.pyc
o
�/a� � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z m
Z
m Z d dl m
Z
d dlmZ d dd �ZG d
d � d e�ZdS )
� N)� DNSTKeyTest)�
DCJoinContext)�drsuapi�misc�dns)� Credentials)�interface_ips_v4�subunitc C s( ddl }|�| �}|�|�tj�� |S )zGet a logger object.r N)�logging� getLogger�
addHandler�
StreamHandler�sys�stderr)�namer
�logger� r �2/usr/lib/python3/dist-packages/samba/tests/join.py�
get_logger s
r c s8 e Zd Z� fdd�Z� fdd�Zdd� Zdd� Z� ZS ) � JoinTestCasec
s� t j�d�| _t j�d�| _tt| ��� t j�� | _ | �
� | _ d| _ t
� }t| j| j | �� | j | jd |dd�| _t jjt jjB | j_| j jtjtjB O _| jj| j_tj | j_ | j� � d| j_ | j� � d S )N�SERVER� SERVER_IP� jointest1�SAMBA_INTERNAL)�server�creds�lp� netbios_name� targetdir�domainr � dns_backendT)!�samba�tests�env_get_var_valuer � server_ip�superr �setUp� env_loadparmr �get_credentialsr r r r � get_loadparm�tempdir�join_ctx�dsdb�UF_SERVER_TRUST_ACCOUNT�UF_TRUSTED_FOR_DELEGATION�userAccountControl�
replica_flagsr �DRSUAPI_DRS_WRIT_REP�!DRSUAPI_DRS_FULL_SYNC_IN_PROGRESS�domain_replica_flagsr � SEC_CHAN_BDC�secure_channel_type�cleanup_old_join�
force_all_ips�do_join)�selfr �� __class__r r r& ' s2
�� �
zJoinTestCase.setUpc s� z| j j}W n ty d }Y nw |d urNt�|j� t�|j� t�tj� | j
d�� t�tj� | j
d�� t� tj� | j
d�� t�tj� | j
d�� | j j dd� t
t| ��� d S )N�etczmsg.lockz names.tdbzbind-dnsT)�force)r+ �paths�AttributeError�shutil�rmtree� private_dir� state_dir�os�path�joinr* �unlinkr6 r% r �tearDown)r9 r> r: r r rH D s � zJoinTestCase.tearDownc C s^ | � tj�}g }| jj}| �|tjtj�}|�|� t | j
dd�}| � ||� | j || j
d�\}}| �|tj� | �|tj� | �|jt|�� g }d| jj| jjf }| �|tjtj�}|�|� | � ||� | j || j
d�\}}| �|tj� | �|tj� | �|jdt|� � | �|jd jtj� | �|jd j| jj� | �|jd jtj� dS )z9create a query packet containing one query record via TCPT��all_interfaces��hostz %s._msdcs.%s� r N)�make_name_packetr �DNS_OPCODE_QUERYr+ � dnshostname�make_name_question� DNS_QTYPE_A�
DNS_QCLASS_IN�appendr r �finish_name_packet�dns_transaction_tcpr$ �assert_dns_rcode_equals� DNS_RCODE_OK�assert_dns_opcode_equals� assertEqual�ancount�len� ntds_guid� dnsforest�answers�rr_type�DNS_QTYPE_CNAME�rdata)r9 �p� questionsr �q�IPs�response�response_packetr r r �test_join_makes_recordsV s.
z$JoinTestCase.test_join_makes_recordsc C s� t � }|�| jj� |�| jj� | j|d� | �tj�}| � | jj
tj tj �}g }|�
|� | �||� g }t| jdd�}|dd � D ]A}d|v rdt�� }| jj|_tj|_tj|_d|_d|_|} nt�� }| jj|_tj|_tj|_d|_d|_|} | |_|�
|� qDt|�|_ ||_ | � || j �}
| � || j!�\} } | �"| tj#� | �$| | |
� | �tj%�}g }| jj}
| � |
tjtj �}|�
|� | �||� | j&|| j!d�\} }| �"| tj#� | �'| tj%� | �(| j)d� d S ) N)r TrI rM �:r i�� rK )*r �guessr+ r �set_machine_account�
tkey_transrN r �DNS_OPCODE_UPDATErQ � dnsdomain�
DNS_QTYPE_SOArS rT rU r �res_recrP r �DNS_QTYPE_AAAAr` �DNS_QCLASS_NONE�rr_class�ttl�lengthrR rb r\ �nscount�nsrecs� sign_packet�key_name�dns_transaction_udpr$ rW rX �
verify_packetrO rV rY rZ r[ )r9 �dc_credsrc re rd �updatesrf �IP�rrb �macrg �
response_pr rh r r r � test_join_records_can_updatew s`
�
z)JoinTestCase.test_join_records_can_update)�__name__�
__module__� __qualname__r& rH ri r� �
__classcell__r r r: r r & s
!r )r )r! r r@ rD �samba.tests.dns_baser �
samba.joinr � samba.dcerpcr r r �samba.credentialsr �samba.provisionr r r r r r r �<module> s