File: //lib/python3/dist-packages/samba/tests/__pycache__/dns_packet.cpython-310.pyc
o
�/a� � @ s� d Z ddlZddlZddlZddlZddlmZmZ ddlm Z dd� Z
ej d Z ej d � d ej d
� �Z
d Zd d
� ZG dd� de �ZG dd� de�ZG dd� de�ZdS )ztSanity tests for DNS and NBT server parsing.
We don't use a proper client library so we can make improper packets.
� N)�dns�nbt)�TestCasec c s � t dd�D ]} | V qq)NT� i�� )�range)�i� r �8/usr/lib/python3/dist-packages/samba/tests/dns_packet.py�_msg_id! s
��r
� SERVER_IP�SERVER�.�REALMg �?c C sJ g }| d dd� } | D ]}|� |d? d � |� |d@ d � q t|�S )zEEven RFC 1002 uses distancing quotes when calling this "compression".s N� � �A � )�append�bytes)�chars�out�cr r r �encode_netbios_bytes, s r c @ sD e Zd Ze� Zdd� Zdd� Zdd� Zdd d
�Zd d � Z d
d� Z
dS )�TestDnsPacketBasec C s8 t d�D ] }| �� }|r d S qtddt � d�� d S )N�( z'the server is STILL unresponsive after z seconds)r �_known_good_query�print�TIMEOUT)�selfr �okr r r �tearDown9 s �zTestDnsPacketBase.tearDownc C s0 |d d� }t �d|�\}}}}}}d|d@ iS )N� �!6H�rcoder )�struct�unpack) r �data�header�id�flags�n_q�n_a�n_rec�n_extar r r � decode_replyB s ��z TestDnsPacketBase.decode_replyc C s� t �dt| j�dt|�ddd�}t �dd| jd�}g }|D ]'}t|t�r-|�d�� d�}n|}|D ]
}|�
dt|�|f � q1|�
|� q |d � |� S )
a� Create a query packet containing one query record.
*names* is either a single string name in the usual dotted
form, or a list of names. In the latter case, each name can
be a dotted string or a list of byte components, which allows
dots in components. Where I say list, I mean non-string
iterable.
Examples:
# these 3 are all the same
"example.com"
["example.com"]
[[b"example", b"com"]]
# this is three names in the same request
["example.com",
[b"example", b"com", b"..!"],
(b"first component", b" 2nd component")]
r" � r z!BHHr �utf8� .s %c%s� ) r$ �pack�next�msg_id�len�qtype�
isinstance�str�encode�splitr �join)r �namesr' �tail� encoded_bits�name�bits�br r r �construct_queryJ s, ��
z!TestDnsPacketBase.construct_queryr Nc C s~ t |t�r|g}| �|�}t�tjtj�}|�|| j� t�|gg g t �\}}}|�
� ||gkr=| � � }| � |d� d S d S )N�the server is unresponsive)
r8 r9 rC �socket�AF_INET�
SOCK_DGRAM�sendto�server�selectr �closer �
assertTrue)r r= �expected_rcode�packet�s�r�_r r r r � _test_queryy s
�z TestDnsPacketBase._test_queryc
C s� | j d dkr
t}tj}n td�dg}tj}| �|g�}t�tj tj
�}|� || j � t � |gg g t
�\}}}||gkrB|�� dS |�d�\}}|�� | �|�d } || kS )Nr �5 s nxdomainFi r# )rI � SERVER_NAMEr � DNS_RCODE_OKr r �
NBT_RCODE_NAMrC rE rF rG rH rJ r rK �recvfromr. )
r r@ rM rN rO rP rQ r& �addrr# r r r r � s
z#TestDnsPacketBase._known_good_queryc C sB d}t � t jt j�}|�|| j� |�� | �� }| �|d� d S )Nr2 rD )rE rF rG rH rI rK r rL )r rN rO r r r r �_test_empty_packet� s z$TestDnsPacketBase._test_empty_packet)r N) �__name__�
__module__� __qualname__r
r5 r r. rC rR r rY r r r r r 6 s
/ r c @ s: e Zd ZedfZdZd dd�Zdd� Zdd � Zd
d � Z dS )
�TestDnsPacketsrS r Nc C s |g| }| j |g|d� d S )N�rM )rR �r �label�nrM r@ r r r � _test_many_repeated_components� s
�z-TestDnsPackets._test_many_repeated_componentsc C � d}| � |d� d S �Ns? ...............................................................� �rb �r r` r r r � test_127_very_dotty_components� � z-TestDnsPackets.test_127_very_dotty_componentsc C rc �Ns? x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.xre rf rg r r r � test_127_half_dotty_components� ri z-TestDnsPackets.test_127_half_dotty_componentsc C � | � � d S �N�rY �r r r r �test_empty_packet� � z TestDnsPackets.test_empty_packetrm )
rZ r[ r\ r rI r7 rb rh rk rp r r r r r] � s
r] c @ sB e Zd ZedfZdZdd� Zddd�Zdd � Zd
d � Z d d
� Z
dS )�TestNbtPackets� � c O s� t |t�r|g}g }|D ]'}t |t�r |�d��d�}n|}t|d �g}|�|dd � � |�|� q | j|g|�R i |�� d S )Nr0 r1 r r )r8 r9 r: r; r �extendr rR )r r= �args�kwargs� nbt_namesr@ rA �encodedr r r �_test_nbt_encode_query� s
z%TestNbtPackets._test_nbt_encode_queryNc C s* |g| }t |�|d< | j|g|d� d S )Nr r^ )r rR r_ r r r rb � s
�z-TestNbtPackets._test_many_repeated_componentsc C rc rd rf rg r r r rh � ri z-TestNbtPackets.test_127_very_dotty_componentsc C rc rj rf rg r r r rk � ri z-TestNbtPackets.test_127_half_dotty_componentsc C rl rm rn ro r r r rp � rq z TestNbtPackets.test_empty_packetrm ) rZ r[ r\ r rI r7 rz rb rh rk rp r r r r rr � s
rr )�__doc__�osr$ rE rJ � samba.dcerpcr r � samba.testsr r
�environr rT r r r r] rr r r r r �<module> s
w