File: //usr/local/CyberCP/lib/python3.10/site-packages/DNS/tests/__pycache__/testPackers.cpython-310.pyc
o
�hI � @ s� d dl Z e j�d d� d dlZd dlZd dlZdZG dd� dej�ZG dd� dej�Z G dd � d ej�Z
G d
d � d ej�Z G d d
� d
ej�Z G dd� dej�Z
G dd� de
�ZG dd� de
�ZG dd� de
�ZG dd� de
�ZG dd� de
�ZG dd� de
�ZG d d � d e
�ZG d d � d e
�ZG d d!� d!e
�ZG d"d#� d#e
�ZG d$d%� d%e
�ZG d&d'� d'e
�ZG d(d)� d)e
�ZG d*d+� d+e
�ZG d,d-� d-e
�Z G d.d/� d/e
�Z G d0d1� d1e
�Z G d2d3� d3e
�Z d4d5� Z d6d7� Z!e"d8kr�e�#� dS dS )9� Nz..�
TestCompletedc @ � e Zd ZdZdd� Zdd� ZdS )� Int16Packing))�
s
)�� s �)�� s �)�b� s �b)� s ��c C �, | j D ]\}}tj�|�}| �||� qdS )z4 pack16bit should give known output for known input N)� knownValues�DNS�Lib� pack16bit� assertEqual��self�i�s�result� r �H/usr/local/CyberCP/lib/python3.10/site-packages/DNS/tests/testPackers.py�test16bitPacking � �z Int16Packing.test16bitPackingc C �, | j D ]\}}tj�|�}| �||� qdS )z6 unpack16bit should give known output for known input N)r r r
� unpack16bitr r r r r �test16bitUnpacking" r z Int16Packing.test16bitUnpackingN)�__name__�
__module__� __qualname__r r r r r r r r � r c @ r )� Int32Packing))r s
)r s �)r s �)r s �b)r s ��)i��s ��)i�k�s �k�)i���s ���c C r
)z4 pack32bit should give known output for known input N)r r r
� pack32bitr r r r r �test32bitPacking2 r z Int32Packing.test32bitPackingc C r )z6 unpack32bit should give known output for known input N)r r r
� unpack32bitr r r r r �test32bitUnpacking8 r z Int32Packing.test32bitUnpackingN)r r r r r" r$ r r r r r ( s r c @ r )�
IPaddrPacking))z 127.0.0.1i )z 10.99.23.13i
c
)z 192.35.59.45l -;F )z255.255.255.255l �� c C r
)z3 addr2bin should give known output for known input N)r r r
�addr2binr r r r r �testIPaddrPackingG r z IPaddrPacking.testIPaddrPackingc C r )z3 bin2addr should give known output for known input N)r r r
�bin2addrr r r r r �testIPaddrUnpackingM r z!IPaddrPacking.testIPaddrUnpackingN)r r r r r' r) r r r r r% ? r r% c @ s\ e Zd Zdgdfg d�dfg d�dfgZdgdfg d�dfg d�dfgZdd � Zd
d � Zd S )
�PackerClassPacking� www.ekit.coms wwwekitcom )�
ns1.ekorp.comz
ns2.ekorp.comz
ns3.ekorp.coms ns1ekorpcom ns2�ns3�)za.root-servers.net.zb.root-servers.net.zc.root-servers.net.zd.root-servers.net.ze.root-servers.net.zf.root-servers.net.s( a root-serversnet b�c�d�e�f�)za.root-servers.netzb.root-servers.netzc.root-servers.netzd.root-servers.netze.root-servers.netzf.root-servers.netc C sJ ddl m} | jD ]\}}|� }|D ]}|�|� q| �|�� |� q d S )Nr )�Packer)�DNS.Libr- �knownPackValues�addnamer �getbuf)r r- �namelistr �p�nr r r �
testPackNamesi s �z PackerClassPacking.testPackNamesc C s\ ddl m} | jD ]"\}}||�}g }tt|��D ] }|�� }|�|� q| �||� q d S )Nr )�Unpacker)r. r6 �knownUnpackValues�range�len�getname�appendr )r r6 r2 r �u�namesr r4 r r r �testUnpackNamesq s �z"PackerClassPacking.testUnpackNamesN)r r r r/ r7 r5 r> r r r r r* S s& ��� ��� r* c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
�testUnpackingMangled�%addA(self, name, klass, ttl, address)� www02ekitcom Q� ��
c C s t j�| j�}|�� d S �N)r r
�
RRunpacker�
packerCorrect�getAdata�r r< r r r �
testWithoutRR� s z"testUnpackingMangled.testWithoutRRc C s, t j�| j�}|�� | �t jj|j� d S rB )r r
rC rD � getRRheader� assertRaises� UnpackErrorrF r r r �testWithTwoRRs� � z#testUnpackingMangled.testWithTwoRRsc C s, t j�| j�}|�� | �t jj|j� d S rB )r r
rC rD rH rI rJ �endRRrF r r r �testWithNoGetData� rL z&testUnpackingMangled.testWithNoGetDataN)r r r �__doc__rD rG rK rN r r r r r? � s r? c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�PackerTestCasezD base class for tests of Packing code. Laziness on my part, I know. c C s t jj| _t jj| _d S rB )r r
�RRpackerrC �r r r r �setUp� �
zPackerTestCase.setUpc C s4 | � � }| �|�}|d ur|tur| �|�S d S d S rB )rQ �doPackr �checkPackResult)r r3 �checkr r r �
testPacker� s
�zPackerTestCase.testPackerc C sD t | d�s| jjdkrtd| t|�� �� d S d S | �|�� | j�S )N�packerExpectedResultrP zP***)�hasattr� __class__r �print�reprr1 r rY )r �bufr r r rV � �
�
�z PackerTestCase.checkPackResultc C sD t | d�s| jjdkrtd| t||f�� d S d S | �||f| j�S )N�unpackerExpectedResultrP zU***)rZ r[ r r\ r] r r` )r �rrbits�specbitsr r r �checkUnpackResult� r_ z PackerTestCase.checkUnpackResultc C s� | j d urSt| jd�rAt| d�rA| �| j�}|�� d d� }| � |�}z|�� W n tjj y: | �
dd� Y nw | � ||�S | jj }|dkrU| �
| jj d� d S d S d S )N�doUnpackrY � r zNot at end of RR!rP zUnpack NotImplemented)rd rZ r[ rC rY rH rM r r
rJ r rc r � assertEquals)r r< ra rb �mer r r � testUnpacker� s(
�
�
��
zPackerTestCase.testUnpackerc C � dS �z! stub. don't test the base class Nr �r r3 r r r rU � � zPackerTestCase.doPackc C ri rj r rk r r r rd � rl zPackerTestCase.doUnpackN) r r r rO rS rX rV rc rh rU rd r r r r rP � s rP c @ �< e Zd ZdZdd� Zdd� Zdejjej j
dfdfZ d Z d
S ) �testPackingOfCNAME�'addCNAME(self, name, klass, ttl, cname)c C � |� dtjjdd� d S )N�www.sub.domain� �realhost.sub.domain��addCNAMEr �Class�INrk r r r rU � � ztestPackingOfCNAME.doPackc C � |� � S rB �� getCNAMEdatarF r r r rd � � ztestPackingOfCNAME.doUnpackrq rr rs s% wwwsubdomain realhost�N�
r r r rO rU rd r �Type�CNAMErv rw r` rY r r r r rn � s �rn c @ rm ) �testPackingOfCNAME2ro c C rp )N� www.cust.com�� �www023.big.isp.comrt rk r r r rU � rx ztestPackingOfCNAME2.doPackc C ry rB rz rF r r r rd � r| z testPackingOfCNAME2.doUnpackr� r� r� s) wwwcustcom � www023bigisp� Nr} r r r r r� � � �r� c @ rm ) �testPackingOfCNAME3ro c C rp )N� www.fred.com�Q �webhost.loa.comrt rk r r r rU � rx ztestPackingOfCNAME3.doPackc C ry rB rz rF r r r rd � r| z testPackingOfCNAME3.doUnpackr� r� r� s& wwwfredcom Q� webhostloa� Nr} r r r r r� � r� r� c @ �( e Zd ZdZdd� Zdd� ZdZdZdS ) �testPackingOfHINFO�)addHINFO(self, name, klass, ttl, cpu, os)c C � |� dtjjddd� d S )N�www.sub.domain.comrr �i686�linux��addHINFOr rv rw rk r r r rU � � ztestPackingOfHINFO.doPackc C ry rB �� getHINFOdatarF r r r rd � r| ztestPackingOfHINFO.doUnpack))r� �
� rr )r� r� s) wwwsubdomaincom
i686linuxN�r r r rO rU rd r` rY r r r r r� � � �r� c @ r� ) �testPackingOfHINFO2r� c C r� )N�core1.lax.foo.comrr �cisco�iosr� rk r r r rU � r� ztestPackingOfHINFO2.doPackc C ry rB r� rF r r r rd � r| z testPackingOfHINFO2.doUnpack))r� r� r� rr )r� r� s' core1laxfoocom
ciscoiosNr� r r r r r� � r� r� c @ r� ) �testPackingOfMX�3addMX(self, name, klass, ttl, preference, exchange)c C r� )N�sub.domain.comr� r �mailhost1.isp.com��addMXr rv rw rk r r r rU r� ztestPackingOfMX.doPackc C ry rB )� getMXdatarF r r r rd r| ztestPackingOfMX.doUnpacks, subdomaincom Q�
mailhost1isp� ))r� � r� r� )r r� N�r r r rO rU rd rY r` r r r r r� s �r� c @ s0 e Zd ZdZdd� Zdd� Zdg d�fZdZd S )
�testPackingOfMX2r� c C sF |� dtjjddd� |� dtjjddd� |� dtjjddd� d S ) Nz
ekit-inc.com.r� r �
mx1.ekorp.com� �
mx2.ekorp.com� �
mx3.ekorp.comr� rk r r r rU s ztestPackingOfMX2.doPackc C sF |� � g}|�� d d� }||� � 7 }|�� d d� }||� � 7 }|S )Nre )r� rH )r r< �res�dummyr r r rd s
ztestPackingOfMX2.doUnpack)� ekit-inc.comr� r� r� ))r r� r� r� r� r� sL ekit-inccom Q�
mx1ekorp� Q� mx2� Q� mx3� Nr� r r r r r�
s �r� c @ r� ) �testPackingOfNSz&addNS(self, name, klass, ttl, nsdname)c C rp )Nr� r� r, )�addNSr rv rw rk r r r rU # rx ztestPackingOfNS.doPackc C ry rB )� getNSdatarF r r r rd % r| ztestPackingOfNS.doUnpack))r� � r� r� r, s$ ekit-inccom Q� ns1ekorp� Nr� r r r r r� ! � r� c @ r� ) �testPackingOfPTRz(addPTR(self, name, klass, ttl, ptrdname)c C rp )N�www.ekit-inc.comrr �www-real01.ekorp.com)�addPTRr rv rw rk r r r rU , rx ztestPackingOfPTR.doPackc C ry rB )�
getPTRdatarF r r r rd . r| ztestPackingOfPTR.doUnpack))r� � r� rr r� s/ wwwekit-inccom
www-real01ekorp�
Nr� r r r r r� * r� r� c @ r� ) �testPackingOfSOAz`addSOA(self, name, klass, ttl, mname,
rname, serial, refresh, retry, expire, minimum)c C s$ |� dtjjdddddddd �
d S )
Nr� rr r, zhostmaster.ekit-inc.com��gTw�d r� �, � )�addSOAr rv rw rk r r r rU 6 s �ztestPackingOfSOA.doPackc C ry rB )�
getSOAdatarF r r r rd 9 r| ztestPackingOfSOA.doUnpack))r� � r� rr )r, �
hostmaster)�serialr� )zrefresh r� z 1 minutes)�retryr� z 3 minutes)�expirer� z 5 minutes)�minimumr� z 6 minutessD ekit-inccom ,ns1ekorp�
hostmaster wTg� d � , �Nr� r r r r r� 3 s r� c @ r� ) �testPackingOfAr@ c C rp )N�www02.ekit.comr� � 192.168.10.2��addAr rv rw rk r r r rU A rx ztestPackingOfA.doPackc C ry rB �rE rF r r r rd C r| ztestPackingOfA.doUnpack))r� r� r� r� r� rA Nr� r r r r r� ? r� r� c @ r� ) �testPackingOfA2� addA(self, name, ttl, address)c C rp )Nr+ r� � 10.98.1.0r� rk r r r rU J rx ztestPackingOfA2.doPackc C ry rB r� rF r r r rd L r| ztestPackingOfA2.doUnpack))r+ r� r� r� r� s wwwekitcom Q�
b Nr� r r r r r� H r� r� c @ r� ) �testPackingOfA3r� c C sT |� dtjjdd� |� dtjjdd� |� dtjjdd� |� dtjjdd� d S )N� www.zol.comr� � 192.168.10.4� 192.168.10.3r� � 192.168.10.1r� rk r r r rU S s ztestPackingOfA3.doPackc C sH |� � |�� |� � |�� |� � |�� |� � f\}}}}}}}||||fS rB )rE rH ) r r< �u1�d1�u2�d2�u3�d3�u4r r r rd X s < ztestPackingOfA3.doUnpack))r� r� r� r� )r� r� r� r� sH wwwzolcom Q� ��
Q� ��
Q� ��
Q� ��
Nr� r r r r r� Q s r� c @ s. e Zd ZdZdd� Zdd� ZdZddgfZd S )
�testPackingOfTXT�$addTXT(self, name, klass, ttl, list)c C rp )Nr� rr zthis is a text record��addTXTr rv rw rk r r r rU ` rx ztestPackingOfTXT.doPackc C ry rB )�
getTXTdatarF r r r rd b r| ztestPackingOfTXT.doUnpacks. ekit-inccom this is a text record)r� � r� rr s this is a text recordNr� r r r r r� ^ s r� c @ s e Zd ZdZdd� ZdZdS )�testPackingOfTXT2r� c C s |fdd�}| � t|� tS )Nc S s | � dtjjdd�S )Nr� rr a� the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
the quick brown fox jumped over the lazy brown dog
r� )r3 r r r �<lambda>k s z*testPackingOfTXT2.doPack.<locals>.<lambda>)rI �
ValueErrorr )r r3 �fr r r rU j s ztestPackingOfTXT2.doPackN)r r r rO rU rd r r r r r� h s r� c @ �D e Zd ZdZdd� Zdd� Zdd� ZdZd ej j
ej j d
fd fZ
d S )
�testPackingOfAAAAText�(addAAAA(self, name, klass, ttl, address)c C � t jj| _t jj| _d S rB )r r
rQ �RRunpackerTextrC rR r r r rS r rT ztestPackingOfAAAAText.setUpc C � t |dtjjdd� d S �N�
google.comre �2607:f8b0:4005:802::1005��addAAAAr rv rw rk r r r rU v rx z testPackingOfAAAAText.doPackc C � |� � }|S rB �� getAAAAdata�r r< �rr r r rd x � z testPackingOfAAAAText.doUnpack�& googlecom &��@ r� re r� N�r r r rO rS rU rd rY r r~ �AAAArv rw r` r r r r r� p � r� c @ r� )
�testPackingOfAAAABinaryr� c C r� rB )r r
rQ �RRunpackerBinaryrC rR r r r rS � rT z testPackingOfAAAABinary.setUpc C r� r� r� rk r r r rU � rx z testPackingOfAAAABinary.doPackc C s | � t|d�� |�d�}|S )Nr� r� )� assertFalserZ �getbytesr� r r r rd � s
z testPackingOfAAAABinary.doUnpackr� r� re s &��@ Nr� r r r r r� ~ s r� c @ r� )
�testPackingOfAAAAIntegerr� c C r� rB )r r
rQ �RRunpackerIntegerrC rR r r r rS � rT z testPackingOfAAAAInteger.setUpc C r� r� r� rk r r r rU � rx z testPackingOfAAAAInteger.doPackc C r� rB r� r� r r r rd � r� z!testPackingOfAAAAInteger.doUnpackr� r� re l � ,�& Nr� r r r r r� � r� r� c C s: t �t j|�}| �|tjj||� | j| | _| �� | S )z!Add AAAA record to a packer.
) �socket� inet_pton�AF_INET6� addRRheaderr r~ r� r^ rM )r3 �name�klass�ttl�address�addr_bufr r r r� � s
r� c C s ddl m} | � �t�S )Nr ��
TestLoader)�unittestr �loadTestsFromNamer r
r r r �
test_suite� s r �__main__)$�sys�path�insertr r r r �TestCaser r r% r* r? rP rn r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r �mainr r r r �<module> sF ( 7
�