File: //usr/local/CyberPanel/lib64/python3.10/site-packages/DNS/__pycache__/Lib.cpython-310.pyc
o
��h�c � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlZddl m
Z
zddl Z W n
e y? ddl
Z Y nw d Zd
ZG d d � d e
�ZG d
d� de
�ZddlmZ ddlmZ ddlmZmZmZmZ dd� Zdd� Z dd� Z dd� Z dd� Z d d � Z d d � Z!d d!� Z"G d"d#� d#�Z#G d$d%� d%�Z$d&d'� Z%G d(d)� d)e#�Z&d*d+� Z'G d,d-� d-e$�Z(G d.d/� d/e(�Z)G d0d1� d1e)�Z*G d2d3� d3e)�Z+G d4d5� d5e$�Z,G d6d7� d7e#�Z-G d8d9� d9e$�Z.G d:d;� d;e#�Z/G d<d=� d=e$�Z0G d>d?� d?e&e/e-�Z1G d@dA� dAe(e0e.�Z2G dBdC� dCe)e0e.�Z3G dDdE� dEe*e0e.�Z4G dFdG� dGe,e0e.�Z5G dHdI� dIe+e0e.�Z6dJdK� Z7G dLdM� dM�Z8dNdO� Z9dPdQ� Z:e;dRk�rKe%� dS dS )Su>
$Id$
This file is part of the py3dns project.
Homepage: https://launchpad.net/py3dns
This code is covered by the standard Python License. See LICENSE for details.
Changes for Python3 port © 2011-13 Scott Kitterman <scott@kitterman.com>
Library code. Largely this is packers and unpackers for various types.
� N� )�Type)�Class)�Opcode)�Status)�DNSErrorF�idnac @ � e Zd ZdS )� UnpackErrorN��__name__�
__module__� __qualname__� r r �=/usr/local/CyberPanel/lib/python3.10/site-packages/DNS/Lib.pyr
/ � r
c @ r )� PackErrorNr r r r r r 0 r r )�pack)�unpack)� inet_ntoa� inet_aton� inet_ntop�AF_INET6c C �
t d| �S )N�!H�� struct_pack��nr r r � pack16bit8 �
r c C r �N�!Lr r r r r � pack32bit; r r# c C � t d| �d S )Nr r ��
struct_unpack��sr r r � unpack16bit> � r) c C r$ )Nr" r r% r' r r r � unpack32bitA r* r+ c C s t dt| ��d S )aD Convert a string IPv4 address into an unsigned integer.
Examples::
>>> addr2bin('127.0.0.1')
2130706433
>>> addr2bin('127.0.0.1') == socket.INADDR_LOOPBACK
1
>>> addr2bin('255.255.255.254')
4294967294L
>>> addr2bin('192.168.0.1')
3232235521L
Unlike old DNS.addr2bin, the n, n.n, and n.n.n forms for IP addresses
are handled as well::
>>> addr2bin('10.65536')
167837696
>>> 10 * (2 ** 24) + 65536
167837696
>>> addr2bin('10.93.512')
173867520
>>> 10 * (2 ** 24) + 93 * (2 ** 16) + 512
173867520
r" r )r& r )�addrr r r �addr2binD s r- c C s t td| ��S r! )r r r r r r �bin2addrc r* r. c C s
t t| �S �N)r r r r r r � bin2addr6f r r0 c C s t d| �\}}|d> |B S )Nz!QQ�@ r% )�str�h�lr r r � bin2long6i s r5 c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�PackerzE packer base class. supports basic byte/16bit/32bit/addr/string/name c C s( t jrd}nt j}td|�| _i | _d S )N�utf8� )�DNS�
LABEL_UTF8�LABEL_ENCODING�bytes�buf�index��self�encr r r �__init__r s
zPacker.__init__c C s | j S r/ )r= �r@ r r r �getbufy s z
Packer.getbufc C s< t |�dkr
td��tjrd}ntj}| jt||� | _d S )Nr zone character expectedr7 )�len� TypeErrorr9 r: r; r= r< )r@ �crA r r r �addbyte{ s
zPacker.addbytec C s( t jrd}nt j}| jt||� | _d S �Nr7 )r9 r: r; r= r< )r@ �abytesrA r r r �addbytes� s zPacker.addbytesc C � | j tt|�� | _ d S r/ )r= r< r �r@ r r r r �add16bit� � zPacker.add16bitc C rL r/ )r= r< r# rM r r r �add32bit� rO zPacker.add32bitc C s t |�}| jtt|�� | _d S r/ )r- r= r< r# )r@ r, r r r r �addaddr� s zPacker.addaddrc C s@ t |�dkrtddt |� ��| �tt |��� | �|� d S )N� z Can't encode string of length z
%s (> 255))rE �
ValueErrorrH �chrrK )r@ r( r r r � addstring� s
�zPacker.addstringc C s� g }|� d�D ]
}|s q|�|� qg }tt|��D ] }d�||d � �}|�� }|�|� || jv r:| j| } nqt|�}d }t| j�}g } tj rNd}
ntj
}
t d|
�} t|�D ]\} || }z|� |
�}W n t
y� tj sr� |�d�s{d| }|� |
�}Y nw t|�}
|
dkr�td��|t| � dk r�| �|| |t| � f� ntd� td � | t |
g� | } qZ|r�| t|d
B � } n| t d |
� } | j| | _| D ] \}}|| j|< q�d S ) N�.r7 r8 u �? zlabel too longi�? zDNS.Lib.Packer.addname:zwarning: pointer too bigi � � )�split�append�rangerE �join�upperr> r= r9 r: r; r< �encode�UnicodeEncodeError�
startswithr �printr )r@ �name�nlist�label�keys�i�key�pointer�offsetr> rA r= �jr �valuer r r �addname� s^
�
� �zPacker.addnamec C s� t | j�� �}|�� td� |D ]
}td|| j| f � qtd� d}tdt| j�d d�D ]G}| j||d � dkrE|sBt� d}q0d}td| � | j||d � D ] }d| k rbd k rkn ntd
| � qVtd t|� � qVt� q0td� d S ) Nz(----------------------------------------z%20s %3dr r � z**z%4d� �z %cz%2d) �listr> re �sortra r[ rE r= �ord)r@ re rg �spacerf rG r r r �dump� s(
z Packer.dumpN)r r
r �__doc__rB rD rH rK rN rP rQ rU rl rt r r r r r6 p s 8r6 c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�Unpackerc C s || _ d| _d S )Nr )r= ri �r@ r= r r r rB � s
zUnpacker.__init__c C s4 | j t| j�kr td��| j| j }| j d | _ |S )NzRan off end of datar )ri rE r= r
)r@ rG r r r �getbyte� s
zUnpacker.getbytec C s: | j | j| j| � }t|�|krtd��| j| | _|S )Nznot enough data left)r= ri rE r
)r@ r r( r r r �getbytes� s zUnpacker.getbytesc C � t | �d��S )Nrm )r) ry rC r r r �get16bit� r* zUnpacker.get16bitc C rz )N� )r+ ry rC r r r �get32bit� r* zUnpacker.get32bitc C s$ t jrd}nt j}tt| �� �|�S rI )r9 r: r; r< r. r} r? r r r �getaddr� s zUnpacker.getaddrc C s
| � d�S )N� )ry rC r r r �getaddr6 r zUnpacker.getaddr6c C s | � | �� �S r/ )ry rx rC r r r � getstring r* zUnpacker.getstringc C s� | � � }|r/|d@ dkr/| � � }|}|d> |B d@ }| j}z
|| _| �� }W || _|S || _w |dkr5dS tjr;d}ntj}t| �|�|�}| �� }|sN|S |d | S )N� � i�?��r r8 r7 rV )rx ri �getnamer9 r: r; r2 ry ) r@ rf �drj rh � save_offset�domainrA �remainsr r r r� s,
� zUnpacker.getnameN) r r
r rB rx ry r{ r} r~ r� r� r� r r r r rv � s rv c C s� d} t t| ��}dd l}|�� |D ]2}t� }|�d� |�d� |�d� |�d� |�d� |�d� |�d � |�d
� |�d � q|�� t |�
� d � t t |�
� | d
�d� t |j
�}|�� |�d� |�� |�d� |�� |�d� |�� |�d� |�� |�� |D ](}t |j
�}|�� |�d�|�� |�d�|�� |�d�|�� |�d�|�� f }q�|�� t |�
� d� t t |�
� | d
�d� d S )Ni� r z 192.168.0.1z********************z
f.ISI.ARPAz********zFoo.F.isi.arpaz******************�arpaz**************************r8 zms total for packingr| zms per packing� r� � � zms total for unpackingzms per unpacking)rp r[ �timing�startr6 rQ rK rl �finishra �milli�roundrv r= r~ ry r� )�N�Rr� rf �p�u�resr r r �
testpacker$ sX
� r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�RRpackerc C s t �| � d | _d S r/ )r6 rB �rdstartrC r r r rB Y s
zRRpacker.__init__c G sh | � |� | �|� | �|� | �|� |r%|dd � r td��|d }nd}| �|� t| j�| _d S )Nr z
too many argsr )rl rN rP rF rE r= r� )r@ rb �RRtype�klass�ttl�rest�rdlengthr r r � addRRheader\ s
zRRpacker.addRRheaderc C s� t | j| jd | j� �}|t| j�| j krd S | j| jd � }| j}d}z$| jd | jd � | _| �t|�� | j| | _d}W |sI|| _d S d S |sP|| _w )Nrm r r )r) r= r� rE rN )r@ r� �rdata�save_buf�okr r r �
patchrdlengthh s zRRpacker.patchrdlengthc C s | j d ur | �� d | _ d S r/ )r� r� rC r r r �endRRv s
zRRpacker.endRRc C s | j d ur | �� t�| �S r/ )r� r� r6 rD rC r r r rD z s
zRRpacker.getbufc C �( | � |tj||� | �|� | �� d S r/ )r� r �CNAMErl r� )r@ rb r� r� �cnamer r r �addCNAME~ �
zRRpacker.addCNAMEc C s2 | � |tj||� | �|� | �|� | �� d S r/ )r� r �HINFOrU r� )r@ rb r� r� �cpu�osr r r �addHINFO� �
zRRpacker.addHINFOc C s2 | � |tj||� | �|� | �|� | �� d S r/ )r� r �MXrN rl r� )r@ rb r� r� �
preference�exchanger r r �addMX� r� zRRpacker.addMXc C r� r/ )r� r �NSrl r� )r@ rb r� r� �nsdnamer r r �addNS� r� zRRpacker.addNSc C r� r/ )r� r �PTRrl r� )r@ rb r� r� �ptrdnamer r r �addPTR� r� zRRpacker.addPTRc C sd | � |tj||� | �|� | �|� | �|� | �|� | �|� | �| � | �|
� | �� d S r/ )r� r �SOArl rP r� ) r@ rb r� r� �mname�rname�serial�refresh�retry�expire�minimumr r r �addSOA� s
zRRpacker.addSOAc C �P | � |tj||� t|�tu st|�tu r|g}|D ]}| �|� q| �� d S r/ �r� r �TXT�typer< r2 rU r� �r@ rb r� r� �tlist�txtdatar r r �addTXT� � zRRpacker.addTXTc C r� r/ r� r� r r r �addSPF� r� zRRpacker.addSPFc C r� r/ )r� r �ArQ r� )r@ rb r� r� �addressr r r �addA� r� z
RRpacker.addAc C sB | � |tjtj|� | �|� | �t|�� | �|� | � � d S r/ )
r� r �WKSr �INrQ rH rT rK r� )r@ rb r� r� �protocol�bitmapr r r �addWKS� s
zRRpacker.addWKSc C s t �r/ )�NotImplementedErrorrC r r r �addSRV� s zRRpacker.addSRVN)r r
r rB r� r� r� rD r� r� r� r� r� r� r� r� r� r� r� r r r r r� X s" r� c C sl | dk r
| d| fS | dk r| d| d fS | dk r"| d| d fS | dk r.| d| d fS | d | d fS )
N�< z
%d secondsi z
%d minutesi�Q z%d hoursi�: z%d daysz%d weeksr )�secondsr r r �
prettyTime� s r� c @ sx e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z e Z
dd� Zdd� Zdd� ZdS ) �
RRunpackerc C � t �| |� d | _d S r/ �rv rB �rdendrw r r r rB � �
zRRunpacker.__init__c C sB | � � }| �� }| �� }| �� }| �� }| j| | _|||||fS r/ )r� r{ r} ri r� �r@ rb �rrtyper� r� r� r r r � getRRheader� s zRRunpacker.getRRheaderc C � | j | jkr
td��d S �Nzend of RR not reached�ri r� r
rC r r r r� � � �zRRunpacker.endRRc C � | � � S r/ �r� rC r r r � getCNAMEdata� � zRRunpacker.getCNAMEdatac C s. t jrd}nt j}t| �� |�t| �� |�fS rI )r9 r: r; r2 r� r? r r r � getHINFOdata� s zRRunpacker.getHINFOdatac C s | � � | �� fS r/ �r{ r� rC r r r � getMXdata� � zRRunpacker.getMXdatac C r� r/ r� rC r r r � getNSdata� r� zRRunpacker.getNSdatac C r� r/ r� rC r r r �
getPTRdata� r� zRRunpacker.getPTRdatac
C sT | � � | � � d| �� f dt| �� � dt| �� � dt| �� � dt| �� � fS )N)r� )zrefresh )r� )r� )r� )r� r} r� rC r r r �
getSOAdata� s �zRRunpacker.getSOAdatac C s2 g }| j | jkr|�t| �� �� | j | jks|S r/ )ri r� rZ r< r� �r@ r� r r r �
getTXTdata� �
�zRRunpacker.getTXTdatac C � t jrd}nt j}| �� �|�S rI �r9 r: r; r~ �decoder? r r r �getAdata� � zRRunpacker.getAdatac C s0 | � � }t| �� �}| �| j| j �}|||fS r/ )r~ rr rx ry r� ri )r@ r� r� r� r r r �
getWKSdata� s
zRRunpacker.getWKSdatac C s, | � � }| � � }| � � }| �� }||||fS )zP
_Service._Proto.Name TTL Class SRV Priority Weight Port Target
r� )r@ �priority�weight�port�targetr r r �
getSRVdata� s
zRRunpacker.getSRVdataN)r r
r rB r� r� r� r� r� r� r� r� r� �
getSPFdatar� r� r� r r r r r� � s r� c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�RRunpackerDefaultc C r� r/ )r� rB r� rw r r r rB
r� zRRunpackerDefault.__init__c C s: t jrd}nt j}t�| �� �|��}t�t d|�d �S �Nr7 z!Ir )
r9 r: r; �socketr r~ r� � ipaddress� IPv4Addressr& �r@ rA �xr r r r�
s
zRRunpackerDefault.getAdatac C s t �t| �� ��S r/ )r � IPv6Addressr0 r� rC r r r � getAAAAdata s z RRunpackerDefault.getAAAAdataN�r r
r rB r� r r r r r r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�RRunpackerTextc C � t �| |� d S r/ �r rB rw r r r rB r� zRRunpackerText.__init__c C r� rI r� r? r r r r� r� zRRunpackerText.getAdatac C � t | �� �S r/ )r0 r� rC r r r r � zRRunpackerText.getAAAAdatac C sF t jrd}nt j}g }| j| jkr!|�t| �� |�� | j| jks|S rI )r9 r: r; ri r� rZ r2 r� )r@ rA r� r r r r� " s �zRRunpackerText.getTXTdataN)r r
r rB r� r r� r r r r r
s
r
c @ r� )�RRunpackerIntegerc C r r/ r rw r r r rB - r� zRRunpackerInteger.__init__c C s4 t jrd}nt j}t�| �� �|��}td|�d S r )r9 r: r; r r r~ r� r& r r r r r� / s
zRRunpackerInteger.getAdatac C r
r/ )r5 r� rC r r r r 6 r z RRunpackerInteger.getAAAAdataNr r r r r r , s r c @ s0 e Zd Zdd� Zdd� Zdd� Zdd� ZeZd S )
�RRunpackerBinaryc C r� r/ r� rw r r r rB : r� zRRunpackerBinary.__init__c C sH | � � }| �� }| �� }| �� }| �� }|| _| j| | _|||||fS r/ )r� r{ r} r� ri r� r� r r r r� = s z RRunpackerBinary.getRRheaderc C r� r� r� rC r r r r� F r� zRRunpackerBinary.endRRc C s2 g }| j | jkr|�| �| j�� | j | jks|S r/ )ri r� rZ ry r� r� r r r r� I r� zRRunpackerBinary.getTXTdataN)r r
r rB r� r� r� r� r r r r r 9 s r c @ � e Zd Zdd� ZdS )�Hpackerc C s� | � |� | � |d@ d> |d@ d> B |d@ d> B |d@ d> B |d@ d> B |d@ d> B |d@ d> B | d@ B � | � |
� | � | � | � | � | � |
� d S ) Nr � � �
� r� � r| )rN )r@ �id�qr�opcode�aa�tc�rd�ra�z�rcode�qdcount�ancount�nscount�arcountr r r � addHeaderS s
&
�
�
�
��
zHpacker.addHeaderN)r r
r r% r r r r r R � r c @ r )� Hunpackerc
C s� | � � }| � � }|d? d@ |d? d@ |d? d@ |d? d@ |d? d@ |d? d@ |d? d@ |d ? d@ f\}}}}}}} }
| � � } | � � } | � � }
| � � }|||||||| |
| | |
|f
S )
Nr r r r r r� r r| r )r{ )r@ r �flagsr r r r r r r r r! r" r# r$ r r r � getHeader_ s$
� �zHunpacker.getHeaderN)r r
r r) r r r r r' ^ r& r' c @ r )�Qpackerc C s" | � |� | �|� | �|� d S r/ )rl rN )r@ �qname�qtype�qclassr r r � addQuestionv s
zQpacker.addQuestionN)r r
r r. r r r r r* u r& r* c @ r )� Qunpackerc C s | � � | �� | �� fS r/ )r� r{ rC r r r � getQuestion| s zQunpacker.getQuestionN)r r
r r0 r r r r r/ { r& r/ c @ r )�MpackerNr r r r r r1 � � r1 c @ r )� MunpackerNr r r r r r3 � � r3 c @ r )�MunpackerDefaultNr r r r r r5 � r4 r5 c @ r )�
MunpackerTextNr r r r r r6 � r2 r6 c @ r )�MunpackerBinaryNr r r r r r7 � r2 r7 c @ r )�MunpackerIntegerNr r r r r r8 � r2 r8 c
C s t d� | �� \
}}}}}}}}} }
} } }
t d| � t d|||||||| f � |r/t d� | r7t d| � t d|
| | |
f � t|
�D ] }t d| � t| � qEt| �D ] }t d| � t| � qVt| �D ] }t d | � t| � qgt|
�D ] }t d
| � t| � qxd S ) NzHEADER:zid=%d,z=qr=%d, opcode=%d, aa=%d, tc=%d, rd=%d, ra=%d, z=%d, rcode=%d,z*** response truncated! ***z *** nonzero error code! (%d) ***z0 qdcount=%d, ancount=%d, nscount=%d, arcount=%dz QUESTION %d:z
ANSWER %d:zAUTHORITY RECORD %d:zADDITIONAL RECORD %d:)ra r) r[ �dumpQ�dumpRR)r� r r r r r r r r r r! r" r# r$ rf r r r �dumpM� s4 � �
�
�r; c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � DnsResultc C s2 i | _ g | _g | _g | _g | _|| _| �|� d S r/ )�header� questions�answers� authority�
additional�args�storeM)r@ r� rB r r r rB � s zDnsResult.__init__c C s� dd l }td| jd | jd f � d}| jd r |d }| j}td| � td � td
|d |d |d
f � tt|fdd�d��}tdd�|�|d |d |d |d f � td� | jD ]}td|d |d |d f � qat� td � | jD ]}td |d t |d �|d |d f � q}t� td!� | j
D ]}td |d t |d �|d |d f � q�t� td"� | j D ]}td |d t |d �|d |d f � q�t� d#| jv r�td$| jd# � td%| jd& � td'|� |� � � � d S )(Nr z ; <<>> PDG.py 1.0 <<>> %s %srb r, r8 r zrecurs z ;; options: z;; got answer:z+;; ->>HEADER<<- opcode %s, status %s, id %dr �statusr c S s || S r/ r )r r3 r r r �<lambda>� s z DnsResult.show.<locals>.<lambda>)r r r r r z4;; flags: %s; Ques: %d, Ans: %d, Auth: %d, Addit: %drn r! r" r# r$ z
;; QUESTIONS:z!;; %s, type = %s, class = %sr+ �qtypestr� qclassstrz ;; ANSWERS:z%-20s %-6s %-6s %sr� �typename�dataz;; AUTHORITY RECORDS:z;; ADDITIONAL RECORDS:�elapsedz ;; Total query time: %d msecz;; To SERVER: %s�serverz ;; WHEN: %s)
�timera rB r= rp �filterr\ r> r? �reprr@ rA �ctime)r@ rL �optr3 r( �q�ar r r �show� s\ �
��
�
�
�
�
zDnsResult.showc C s8 |� � \
| jd< | jd< | jd< | jd< | jd< | jd< | jd< | jd< | jd < | jd
< | jd < | jd < | jd
< t�| jd �| jd< t�| jd �| jd< t| jd
�D ] }| j�| � |�� qUt| jd �D ] }| j
�| � |�� qht| jd �D ] }| j �| � |�� q{t| jd
�D ] }| j
�| � |�� q�d S )Nr r r r r r r r r r! r" r# r$ � opcodestrrD )r) r= r rT r � statusstrr[ r> rZ �storeQr? �storeRRr@ rA )r@ r� rf r r r rC � s" ��zDnsResult.storeMc C sF i }|� � \|d< |d< |d< t�|d �|d< t�|d �|d< |S )Nr+ r, r- rF rG )r0 r �typestrr �classstr)r@ r� rQ r r r rV � s
zDnsResult.storeQc C s� i }|� � \|d< |d< |d< |d< |d< t�|d �|d< t�|d �|d< d|d }t||�r<t||�� |d < |S |�|d �|d < |S )
Nrb r� �classr� r� rH rY � get%sdatarI )r� r rX r rY �hasattr�getattrry )r@ r� �rr� r r r rW s &
�zDnsResult.storeRRN)r r
r rB rS rC rV rW r r r r r< � s ) r< c C s4 | � � \}}}td||t�|�|t�|�f � d S )Nz%qname=%s, qtype=%d(%s), qclass=%d(%s))r0 ra r rX r rY )r� r+ r, r- r r r r9 s
�
�r9 c C sv | � � \}}}}}t�|�}td||||t�|�|f � d| }t| |�r1tdt| |�� � d S td| �|�� d S )Nz*name=%s, type=%d(%s), class=%d(%s), ttl=%dr[ z formatted rdata:z binary rdata:) r� r rX ra r rY r\ r] ry )r� rb r� r� r� r� rH r� r r r r: s
��
r: �__main__)<ru �typesr r8 r r r r r9 �Baser r � ImportError�ipaddrr: r; r
r �structr r r r& r r r r r r# r) r+ r- r. r0 r5 r6 rv r� r� r� r� r r
r r r r' r* r/ r1 r3 r5 r6 r7 r8 r; r< r9 r: r r r r r �<module> sp � x<4c A
a
�