o

�/a�
�@s
dd
lZddl
mZ
ddlmZ
ddlmZmZ
ddlm	Z	
dd
l
mZ
dd
l
Z
e
�d�
Ze�e�
Ze�e�

ejddd	d
d�
e�e�
Ze�e�

ee�
Ze�e�

e��\ZZe��ae�t�
aejaee�
dkrte� �
e�!d�

eda"t�#ej$�

Gd
d�de	�Z%ee&ed�
d
S)�N)
�credentials)
�dns)�SubunitOptions�TestProgram)
�DNSTestz$dns_invalid.py <server ip> [options]z	--timeout�int�timeoutz Specify timeout for DNS requests)�type�dest�help�
cs$eZ
dZ�f
d
d�Zdd�Z�ZS)�TestBrokenQueriesc

s*tt
|���
t|_t|_t|_t|_dS)
N)�superr
�setUp�	server_ip�lp�credsr)
�self�
�	__class__��9/usr/lib/python3/dist-packages/samba/tests/dns_invalid.pyr9s





zTestBrokenQueries.setUpc
Csv|�t
j�
}
g}d
|��}|�|t
jt
j�}td|j�

|�	|�

|�
|
|�
|j|
td�\}}|�
|t
j�
dS)z=Check the server refuses invalid characters in the query nameu¨.%sz
asking for %s)
�hostN)�make_name_packetr�DNS_OPCODE_QUERY�get_dns_domain�make_name_question�DNS_QTYPE_A�
DNS_QCLASS_IN�print�name�append�finish_name_packet�dns_transaction_udpr�assert_dns_rcode_equals�DNS_RCODE_NXDOMAIN)r�
p�	questionsr �
q�response�response_packetrrr�test_invalid_chars_in_nameAs






z,TestBrokenQueries.test_invalid_chars_in_name)�__name__�
__module__�__qualname__rr+�
__classcell__rrrrr
8s
r
)�module�opts)'�sys�sambar�samba.dcerpcr�samba.tests.subunitrunrr�samba.tests.dns_baser�samba.getopt�getopt�options�optparse�OptionParser�parser�SambaOptions�	sambaopts�add_option_group�
add_option�CredentialsOptions�credopts�subunitopts�
parse_argsr1�args�get_loadparmr�get_credentialsrr�len�print_usage�exitr�set_krb_forwardable�NO_KRB_FORWARDABLEr
r,rrrr�<module>
s8












�