File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/dns_wildcard.cpython-310.pyc
o
�/a�3 � @ sd d dl Z d dlmZ d dlmZmZ d dlmZ d dlm Z m
Z
d dlm Z m Z d dl
mZ d dlmZ d dlZe�d�Ze�e�Ze�e� ejd d
d d d
� e�e�Ze�e� e e�Ze�e� e� � \Z Z e� � a e�!t �a"e j#a#e$e �dk r�e�%� e �&d� e d Z'e d a(t"�)ej*� dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2G dd� de�Z3e
e4ed� dS )� N)� credentials)�dns� dnsserver)�record_from_string)�SubunitOptions� TestProgram)�werror� WERRORError)�DNSTestz3dns_wildcard.py <server name> <server ip> [options]z --timeout�int�timeoutz Specify timeout for DNS requests)�type�dest�help� � z1.1.1.1z*.wildcardtestz1.1.1.2zexact.wildcardtestz1.1.1.3z*.level2.wildcardtestz1.1.1.4zexact.level2.wildcardtestc s| e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Z� ZS )�TestWildCardQueriesc s� t t| ��� t| _t| _t| _t| _t| _t j
dt | � � f t
ft j
dt| � � f tft j
dt| � � f tft j
dt| � � f tfg| _| �� }| jD ]
\}}}| �||||� qLd S )N�%s.%s)�superr �setUp� server_name�server� server_ip�lp�credsr r � DNS_QTYPE_A�WILDCARD�get_dns_domain� WILDCARD_IP�EXACT�EXACT_IP�LEVEL2_WILDCARD�LEVEL2_WILDCARD_IP� LEVEL2_EXACT�LEVEL2_EXACT_IP� dns_records� dns_connect�
add_record��self�c�typ�name�data�� __class__� �:/usr/lib/python3/dist-packages/samba/tests/dns_wildcard.pyr N sB ��������zTestWildCardQueries.setUpc C s. | � � }| jD ]
\}}}| �||||� qd S )N)r& r% �
delete_recordr( r0 r0 r1 �tearDownm s �z TestWildCardQueries.tearDownc C s d| j }t�|| j| j�S )Nzncacn_ip_tcp:%s[sign])r r r r )r) � binding_strr0 r0 r1 r& r s
z TestWildCardQueries.dns_connectc
C sx t ||�}t�� }||_z|�tjd| j| �� |d |� W d S ty; } z|j d t
j kr0|�W Y d }~d S d }~ww �Nr ) r r �DNS_RPC_RECORD_BUF�rec�DnssrvUpdateRecord2�DNS_CLIENT_VERSION_LONGHORNr r r �argsr �"WERR_DNS_ERROR_NAME_DOES_NOT_EXIST)r) �dns_connr+ r, r- r7 � del_rec_buf�er0 r0 r1 r2 v s$
����z!TestWildCardQueries.delete_recordc
C sZ t ||�}t�� }||_z|�tjd| j| �� ||d � W d S ty, } z|�d }~ww r5 ) r r r6 r7 r8 r9 r r r )r) r< r+ r, r- r7 � add_rec_bufr> r0 r0 r1 r' � s
���z TestWildCardQueries.add_recordc C � | � tj�}g }d| �� }| �|tjtj�}|�|� | �||� | j || j
d�\}}| � |tj � | �
|tj� | �|jd� | �|jd jtj� | �|jd jt� dS )z2Query an A record, should match the wildcard entryzmiss.wildcardtest.%s��hostr r N��make_name_packetr �DNS_OPCODE_QUERYr �make_name_questionr �
DNS_QCLASS_IN�append�finish_name_packet�dns_transaction_udpr �assert_dns_rcode_equals� DNS_RCODE_OK�assert_dns_opcode_equals� assertEqual�ancount�answers�rr_type�rdatar �r) �p� questionsr, �q�response�response_packetr0 r0 r1 � test_one_a_query_match_wildcard� � �
�z3TestWildCardQueries.test_one_a_query_match_wildcardc C r@ )z~ Query an A record, should match the wild card entry
have two labels to the left of the wild card target.
z label2.label1.wildcardtest.%srA r r NrC rS r0 r0 r1 �(test_one_a_query_match_wildcard_2_labels� � �
�z<TestWildCardQueries.test_one_a_query_match_wildcard_2_labelsc C � | � tj�}g }dt| �� f }| �|tjtj�}|�|� | � ||� | j
|| j d�\}}| � |tj
� | �|tj� | �|jd� | �|jd jtj� | �|jd jt� dS )zQuery the wildcard entryr rA r r N)rD r rE r r rF r rG rH rI rJ r rK rL rM rN rO rP rQ rR r rS r0 r0 r1 � test_one_a_query_wildcard_entry� � �
�z3TestWildCardQueries.test_one_a_query_wildcard_entryc C r] �z\Query an entry that matches the wild card but has an exact match as
well.
r rA r r N)rD r rE r r rF r rG rH rI rJ r rK rL rM rN rO rP rQ rR r rS r0 r0 r1 � test_one_a_query_exact_match� � �
�z0TestWildCardQueries.test_one_a_query_exact_matchc C r@ )z:Query an A record, should match the level 2 wildcard entryzmiss.level2.wildcardtest.%srA r r N�rD r rE r rF r rG rH rI rJ r rK rL rM rN rO rP rQ rR r" rS r0 r0 r1 �"test_one_a_query_match_wildcard_l2� rZ z6TestWildCardQueries.test_one_a_query_match_wildcard_l2c C r@ )z�Query an A record, should match the level 2 wild card entry
have two labels to the left of the wild card target
z$label1.label2.level2.wildcardtest.%srA r r Nrc rS r0 r0 r1 �+test_one_a_query_match_wildcard_l2_2_labels
r\ z?TestWildCardQueries.test_one_a_query_match_wildcard_l2_2_labelsc C r] r` )rD r rE r# r rF r rG rH rI rJ r rK rL rM rN rO rP rQ rR r$ rS r0 r0 r1 � test_one_a_query_exact_match_l2" rb z3TestWildCardQueries.test_one_a_query_exact_match_l2c C r] )z Query the level 2 wildcard entryr rA r r N)rD r rE r! r rF r rG rH rI rJ r rK rL rM rN rO rP rQ rR r" rS r0 r0 r1 �"test_one_a_query_wildcard_entry_l29 r_ z6TestWildCardQueries.test_one_a_query_wildcard_entry_l2)�__name__�
__module__� __qualname__r r3 r& r2 r' rY r[ r^ ra rd re rf rg �
__classcell__r0 r0 r. r1 r L s r )�module�opts)5�sys�sambar � samba.dcerpcr r �samba.dnsserverr �samba.tests.subunitrunr r r r �samba.tests.dns_baser
� samba.getopt�getopt�options�optparse� OptionParser�parser� SambaOptions� sambaopts�add_option_group�
add_option�CredentialsOptions�credopts� subunitopts�
parse_argsrm r: � get_loadparmr �get_credentialsr r �len� print_usage�exitr r �set_krb_forwardable�NO_KRB_FORWARDABLEr r r r r" r! r$ r# r rh r0 r0 r0 r1 �<module> sV �
�