File: //lib/python3/dist-packages/samba/tests/__pycache__/dns_tkey.cpython-310.pyc
o
�/at � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZm Z d dl
m Z e� d�Z
e�e
�Ze
�e� e
jdddd d
� e�e
�Ze
�e� ee
�Ze
�e� e
�� \ZZejZee�d k ree
�� e �d � ed Z ed Z G d
d� de �Z e e ed� dS )� N)�dns)�SubunitOptions� TestProgram)� DNSTKeyTestz*dns.py <server name> <server ip> [options]z --timeout�int�timeoutz Specify timeout for DNS requests)�type�dest�help� � c sL e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )�TestDNSUpdatesc s t | _t| _tt| ��� d S )N)� server_name�server� server_ip�superr
�setUp��self�� __class__� �6/usr/lib/python3/dist-packages/samba/tests/dns_tkey.pyr 5 s zTestDNSUpdates.setUpc C s | � � dS )ztest DNS TKEY handshakeN)�
tkey_transr r r r � test_tkey: s zTestDNSUpdates.test_tkeyc C sF | � � }| �|| j�\}}| �|tj� | �| j�}| �|tj � dS )z#test DNS update without TSIG recordN)
�make_update_request�dns_transaction_udpr �assert_dns_rcode_equalsr �DNS_RCODE_REFUSED�
search_record�
newrecname�assert_rcode_equals�DNS_RCODE_NXDOMAIN)r �p�response�
response_p�rcoder r r �test_update_wo_tsig? s
z"TestDNSUpdates.test_update_wo_tsigc C s� | � � | �� }| �|d� | �|| j�\}}| �|tj� |jd j }| �
|j tj � | �
|j
d� | �| j�}| �|tj� dS )z5test DNS update with a TSIG record with a bad keyname�badkeyr N)r r � sign_packetr r r r �DNS_RCODE_NOTAUTH�
additional�rdata� assertEqual�error�DNS_RCODE_BADKEY�mac_sizer r r! r" �r r# r$ r% � tsig_recordr& r r r � test_update_tsig_bad_keynameI s z+TestDNSUpdates.test_update_tsig_bad_keynamec C s� | � � | �� }| �|| j� | �|| j�\}}| �|tj� |j d j
}| � |j tj
� | � |jd� | �| j�}| �|tj� dS )z1test DNS update with a TSIG record with a bad MACr N)r r �bad_sign_packet�key_namer r r r r* r+ r, r- r. �DNS_RCODE_BADSIGr0 r r r! r" r1 r r r �test_update_tsig_bad_macY s z'TestDNSUpdates.test_update_tsig_bad_macc C s� | � � | �� }| �|| j�}| �|| j�\}}| �|tj� | � |||� | �
| j �}| � |tj� | jdd�}| �|| j�}| �|| j�\}}| �|tj� | � |||� | �
| j �}| � |tj
� dS )z(test DNS update with correct TSIG recordT��deleteN)r r r) r5 r r r r � DNS_RCODE_OK�
verify_packetr r r! r" )r r# �macr$ r% r& r r r �test_update_tsigi s z TestDNSUpdates.test_update_tsigc C s. d| j }tj}d}| �tj�}| �| �� tjtj�}g }|�|� | � ||� g }t�
� }||_ tj |_
tj|_d|_d|_|�|� t�
� }||_ tj|_
tj|_d|_d|_|�|� t�
� }||_ tj |_
||_||_d|_d|_|�|� t|�|_||_g } t�
� }||_ tj|_
tj|_d|_d|_| �|� t| �|_| |_| �|| j �\}
} | � |
tj � | � � | � || j!�} | �|| j �\}
} | � |
tj"� | �#|
| | � | �$|�}
| �%|
tj"� | j&dd�}| � || j!�} | �|| j �\}
} | � |
tj"� | �#|
| | � | �$| j �}
| �%|
tj'� dS ) zAtest DNS update with correct TSIG record (follow Windows pattern)�wini� r i�� z
10.1.45.64Tr8 N)(r r �
DNS_QCLASS_IN�make_name_packet�DNS_OPCODE_UPDATE�make_name_question�get_dns_domain�
DNS_QTYPE_SOA�append�finish_name_packet�res_rec�name� DNS_QTYPE_A�rr_type�DNS_QCLASS_ANY�rr_class�ttl�length�DNS_QTYPE_AAAAr, �len�nscount�nsrecs�DNS_QTYPE_CNAME�DNS_QCLASS_NONE�ancount�answersr r r r r r) r5 r: r; r r! r r" )r r rL rM r# �q� questions�updates�r�prereqsr$ r% r< r& r r r �test_update_tsig_windows� s|
�
z'TestDNSUpdates.test_update_tsig_windows) �__name__�
__module__� __qualname__r r r' r3 r7 r= r\ �
__classcell__r r r r r
4 s
r
)�module�opts) �sys�optparse� samba.getopt�getopt�options� samba.dcerpcr �samba.tests.subunitrunr r �samba.tests.dns_baser � OptionParser�parser� SambaOptions� sambaopts�add_option_group�
add_option�CredentialsOptions�credopts� subunitopts�
parse_argsrb �argsr rP � print_usage�exitr r r
r] r r r r �<module> s4
�