File: //usr/local/CyberPanel/lib64/python3.10/site-packages/DNS/__pycache__/Base.cpython-310.pyc
o
��h}? �
@ s4 d Z ddlZddlZddlZddlZddlZddlZddlmZm Z m
Z
z ddl m Z e � Z W n ddl Z Y G dd� de
�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de�ZddlmZ dde
jejddddg d� ad$dd�Zdd� Zd d � Zd d � ZG d d!� d!�Z d"d#� Z dS )%u<
$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-14 Scott Kitterman <scott@kitterman.com>
Base functionality. Request and Response classes, that sort of thing.
� N� )�Type�Class�Opcode)� SystemRandomc @ � e Zd ZdS )�DNSErrorN��__name__�
__module__� __qualname__� r
r
�>/usr/local/CyberPanel/lib/python3.10/site-packages/DNS/Base.pyr � r c @ r )�
ArgumentErrorNr r
r
r
r r r r c @ r )� SocketErrorNr r
r
r
r r r r c @ r )� TimeoutErrorNr r
r
r
r r ! r r c @ s e Zd Zdd� ZdS )� ServerErrorc C s t �| ||� || _|| _d S �N)r �__init__�message�rcode)�selfr r r
r
r r $ s
zServerError.__init__N)r
r r r r
r
r
r r # s r c @ r )�IncompleteReplyErrorNr r
r
r
r r ) r r )�Lib�udp�5 � ) �protocol�port�opcode�qtype�rd�timing�timeout�
server_rotate�server�/etc/resolv.confc C s\ z t | d��
}t|�W d � W S 1 sw Y W dS ty- td �d� Y S w )zCparses the /etc/resolv.conf file and sets defaults for name servers�rNr& z 127.0.0.1)�open�ParseResolvConfFromIterable�FileNotFoundError�defaults�append)� resolv_path�streamr
r
r �ParseResolvConf2 s (� �r0 c C s� | D ]V}|� � }|r|d dks|d dkrq|�� }t|�dk r"q|d dkr4t|�dkr4|d td< |d dkr; |d dkrB |d d krI |d d
krXtd �|d � qd S )
zHparses a resolv.conf formatted stream and sets defaults for name serversr �;�#� �domainr �search�options�sortlist�
nameserverr& N)�strip�split�lenr, r- )�lines�line�fieldsr
r
r r* ; s& ��r* c C sH dd l } | jdv rddlm} |�� td< d S | jdkr!t� d S t� S )Nr )�win32�ntr )�win32dnsr& �darwin)�sys�platform� rA �RegistryResolver, �ParseOSXSysConfigr0 )rC rA r
r
r �_DiscoverNameServersP s
rH c C s dS )zbDon't call, only here for backward compatability. We do discovery for
you automatically.
Nr
r
r
r
r �DiscoverNameServersZ s rI c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) �
DnsRequestz high level Request object c O sJ d | _ i | _| �||� | j| _d| _d| _t| jd �dkr#td��d S )Nr rE r& z"No working name servers discovered)�donefuncr, �argparse�args�tid�
resulttyper; r )r �namerM r
r
r r b s �zDnsRequest.__init__c C s |sd| j v r| j d |d< t|�tu st|�tu r ||d< nt|�dkr/|d r/|d |d< t d rNtt d �tjkrNt d dd � t d d d� t d< tt �� �D ]}||vrm|| j v rg| j | ||< qTt | ||< qTt|d �tks~t|d �tkr�|d g|d< || _ d S )NrP r r r% r& )
r, �type�bytes�strr; �types�ListType�list�keysrM )r rP rM �ir
r
r rL l s&
$
�
zDnsRequest.argparsec C s t � ||�| _d S r )�socket�s)r �a�br
r
r �
socketInit� s zDnsRequest.socketInitc C sh | j dkrt�| jgg g | j �\}}}t|�std��| j�d�\| _| _t�� | _ | j
| j d< | � � S )Nr �Timeout� r& )
r$ �selectrZ r; r �recvfrom�reply� from_address�time� time_finish�nsrM � processReply)r r( �w�er
r
r �processUDPReply� s
zDnsRequest.processUDPReplyc C s� |� |�}t|�|k rG| jdkr(| j| j t�� }|dkr"td��| j�|� |� |t|� �}|s=tdt|�|f ��||7 }t|�|k s |S )Nr r^ z incomplete reply - %d of %d read)�readr; r$ �
time_startrd r rZ �
settimeout)r �f�count�res�rem�bufr
r
r �_readall� s
�
zDnsRequest._readallc C s� | j dkr
| j�| j � n| j�d � | j�d�}z| �|d�}t�|�}| �||�| _W |�� n|�� w t � � | _
| j | j d< | �
� S )Nr �rbr3 r& )r$ rZ rm �makefilers r � unpack16bitrb �closerd re rf rM rg )r rn �headerro r
r
r �processTCPReply� s
zDnsRequest.processTCPReplyc C s� | j | j d | jd< | jst�| j�}n7| jdkr!t�| j�}n+| jdkr-t�| j�}n | jdkr9t� | j�}n| jdkrEt�
| j�}nt d| j ��t� || j�}| j|_|S )Ni� �elapsed�default�binary�text�integerzUnknown resulttype: )
re rl rM rO r � Munpackerrb �MunpackerDefault�MunpackerBinary�
MunpackerText�MunpackerInteger� SyntaxError� DnsResult)r �ur( r
r
r rg � s
zDnsRequest.processReplyc
C s\ zt �dd�}| j�d|f� W dS tjy, } z
|jtjkr"� W Y d}~nd}~ww q)z<Pick random source port to avoid DNS cache poisoning attack.Ti r_ rE N)�random�randintrZ �bindrY �error�errno�
EADDRINUSE)r � source_port�msgr
r
r � getSource� s ���zDnsRequest.getSourcec C s | � � | j�| j| jf� d S r )r� rZ �connectrf r )r r
r
r �conn� s zDnsRequest.connc O s� | � ||� | jd }| jd | _t�dd�| _| jd | _| jd }| jd }| jd }d | jv r:| jd | _nd
| _t| jd �t ksOt| jd �t
krizt t t
| jd �
� ��}W n tyh td ��w | jd }d
| jvr|t| j� td��| jd
}|t jkr�|dkr�td� d}t�� } | �| jd|dd|ddddddd�
| �||tj� | �� | _z|dkr�| �|� W | j S | �|� W | j S t j y� }
zt |
��d}
~
ww )aW
Request function for the DnsRequest class. In addition to standard
DNS args, the special pydns arg 'resulttype' can optionally be passed.
Valid resulttypes are 'default', 'text', 'decimal', and 'binary'.
Defaults are configured to be compatible with pydns:
AAAA: decimal
Others: text
r r r r_ r$ r r" r& rO r{ r! �unknown query typerP �nothing to lookup�tcp�'Query type AXFR, protocol forced to TCPr r N) rL rM r r� r� rN r$ rO rQ rR rS �getattrr �upper�AttributeErrorr �print�AXFRr �Mpacker� addHeader� addQuestionr �IN�getbuf�request�sendUDPRequest�sendTCPRequestrY r� r �response� r rP rM r r r" r& r! �qname�m�reasonr
r
r �qry� sV
$ �
�
����zDnsRequest.qryc O s� | � ||� z| jd rtd�| jd ���W n Y | jd }| jd | _t�dd�| _| jd | _| jd }| jd }| jd
}t | jd �t
ksUt | jd �t krozt t
t | jd �� ��}W n tyn td ��w | jd }d
| jvr�t| j� td��| jd
}|t
jkr�|dkr�td� d}t�� } | �| jd|dd|ddddddd�
| �||tj� | �� | _z|dkr�| �|� W | j S | �|� W | j S t j y� }
zt |
��d}
~
ww )z needs a refactoring rO zERestulttype {0} set with DNS.req, use DNS.qry to specify result type.r r r r_ r$ r r" r& r! r� rP r� r� r� r r N) rL rM r �formatr r� r� rN r$ rQ rR rS r� r r� r� r� r� r r� r� r� r r� r� r� r� r� rY r� r r� r� r
r
r �req sZ
�
$ �
�
����zDnsRequest.reqc C sh d}d| _ |D ]�| _zl| j�d�r$ttd�r"tjr"| �tjtj� n
W q| �tj tj� zBt
�
� | _ | � � | j
�| j� | �� }|jd | jksQ| jd | jkre| �� }|jd | jksQ| jd | jksQ|| _ W | j
�� n| j
�� w W n, tjy� } z
|p�|}W Y d}~qd}~w ty� } z
|p�|}W Y d}~qd}~ww | j r� nq| j s�|r�|�dS dS )z refactor meN�:�has_ipv6�idr )r� rf ro �hasattrrY r� r] �AF_INET6�
SOCK_DGRAM�AF_INETrd rl r� rZ �sendr� rj rx rN rc r rw r� r )r r& �first_socket_errorr( ri �tr
r
r r� J sH
� � ���
�zDnsRequest.sendUDPRequestc C sb d}d| _ |D ]�| _zi| j�d�r$ttd�r"tjr"| �tjtj� n
W q| �tj tj� z?t
�
� | _ | � � t
�t| j��| j }| j�d� | j�|� | �� }|jd | jkre|| _ W | j�� W nAW | j�� n| j�� w W n, tjy� } z
|p|}W Y d}~qd}~w ty� } z
|p�|}W Y d}~qd}~ww | j r� nq| j s�|r�|�dS dS )z& do the work of sending a TCP request Nr� r� r r� )r� rf ro r� rY r� r] r� � SOCK_STREAMr� rd rl r� r � pack16bitr; r� rZ � setblocking�sendallry rx rN rw r� r )r r& r� rr r( ri r� r
r
r r� t sJ
� � ���
�zDnsRequest.sendTCPRequestN)r
r r �__doc__r rL r] rj rs ry rg r� r� r� r� r� r� r
r
r
r rJ ` s
% 81 *rJ c C s� ddl } ddl}| �dd�}|�d�}g }d} |�� }|s n/|�� }t|�dk s0|d tjvr3d}q|� |�}|rM|du rEg }|�
|� |�
|� d�� q|� � |D ] }t
d �|� qTdS ) zMRetrieves the current Mac OS X resolver settings using the scutil(8) command.r Nz/usr/sbin/scutil --dnsr( z$^\s+nameserver[]0-9[]*\s*\:\s*(\S+)$Tr r& )�os�re�popen�compile�readline�rstripr; �string�
whitespace�matchr- �grouprw r, �extend)r� r� �scutil�res_re�sets�
currentset�lr� r
r
r rG � s0
��rG )r' ) r� rY r� rT rd r` r� rE r r r r� r � Exceptionr r r r r r r �QUERY�Ar, r0 r* rH rI rJ rG r
r
r
r �<module> s8 (
�
@