File: //proc/1233/root/usr/lib/python3/dist-packages/samba/__pycache__/dnsserver.cpython-310.pyc
o
�/a�7 � @ s d dl Z d dlZd dlmZmZ G dd� de�ZG dd� dej�ZG dd� dej�Z G d d
� d
ej�Z
G d d � d ej�Z G d
d� dej�Z G dd� dej�Z
G dd� dej�ZG dd� dej�ZG dd� dej�Zi Zdd� Ze� [d%dd�Zdd � Zd d � Zd d � Zd!d"� Zd#d$� ZdS )&� N)� dnsserver�dnspc @ s e Zd ZdS )�
DNSParseErrorN)�__name__�
__module__� __qualname__� r r �1/usr/lib/python3/dist-packages/samba/dnsserver.pyr , s r c �6 e Zd Zddejdf� fdd� Zed dd��Z� ZS )
�ARecord� � r c �6 t t| ��� tj| _||B | _|| _|| _|| _ d S �N)
�superr �__init__r �
DNS_TYPE_A�wType�dwFlags�dwSerial� dwTtlSeconds�data)�self�ip_addr�serial�ttl�rank� node_flag�� __class__r r r 1 �
zARecord.__init__Nc K � | |fi |��S r r ��clsr �sep�kwargsr r r � from_string: � zARecord.from_stringr � r r r r �
DNS_RANK_ZONEr � classmethodr& �
__classcell__r r r r r 0 s � r c r
)
�
AAAARecordr r
r c r r )
r r, r r �
DNS_TYPE_AAAAr r r r r )r �ip6_addrr r r r r r r r A r zAAAARecord.__init__Nc K r! r r r" r r r r& J r' zAAAARecord.from_stringr r( r r r r r, ? s � r, c r
)
� PTRRecordr r
r c �N t t| ��� tj| _||B | _|| _|| _t �
� }||_ t |�|_ || _
d S r )r r/ r r � DNS_TYPE_PTRr r r r r � DNS_RPC_NAME�str�lenr )r �ptrr r r r �ptr_namer r r r Q �
zPTRRecord.__init__Nc K r! r r r" r r r r& ] r' zPTRRecord.from_stringr r( r r r r r/ O � � r/ c r
)
� CNAMERecordr r
r c sJ t � �� tj| _||B | _|| _|| _t� � }||_
t |�|_ || _ d S r )
r r r �DNS_TYPE_CNAMEr r r r r r2 r3 r4 r )r �cnamer r r r �
cname_namer r r r d s
zCNAMERecord.__init__Nc K r! r r r" r r r r& p r' zCNAMERecord.from_stringr r( r r r r r9 b r8 r9 c r
)
�NSRecordr r
r c r0 r )r r= r r � DNS_TYPE_NSr r r r r r2 r3 r4 r )r �
dns_serverr r r r �nsr r r r w r7 zNSRecord.__init__Nc K r! r r r" r r r r& � r' zNSRecord.from_stringr r( r r r r r= u r8 r= c r
)
�MXRecordr r
r c sX t t| ��� tj| _||B | _|| _|| _t �
� }||_ ||j _
t|�|j _|| _d S r )r rA r r � DNS_TYPE_MXr r r r r � DNS_RPC_RECORD_NAME_PREFERENCE� wPreference� nameExchanger3 r4 r )r � mail_server�
preferencer r r r �mxr r r r � s
zMXRecord.__init__Nc
K sT z
|� |�\}}t|�}W n ty } ztd| �|�d }~ww | ||fi |��S )Nz?MX data must have server and priority (space separated), not %r)�split�int�
ValueErrorr )r# r r$ r% �server�priority�er r r r& � s ����zMXRecord.from_stringr r( r r r r rA � s �
rA c s@ e Zd Zddddddejejf� fdd� Zed d d
��Z� Z S ) � SOARecordr r
iX i�Q i c s� t t| ��� tj| _| |
B | _|| _|| _t �
� } || _ || _ || _
|| _|| _|| j_t|�| j_|| j_t|�| j_| | _d S r )r rO r r � DNS_TYPE_SOAr r r r r �DNS_RPC_RECORD_SOA�
dwSerialNo� dwRefresh�dwRetry�dwExpire� dwMinimumTtl�NamePrimaryServerr3 r4 �ZoneAdministratorEmailr ) r �mname�rnamer �refresh�retry�expire�minimumr r r �soar r r r � s
zSOARecord.__init__Nc
K st |� |�}t|�dkrtd��ztdd�D ]
}t|| �||< qW n ty2 } ztd�|�d }~ww | |i |��S )N� zhData requires 7 space separated elements - nameserver, email, serial, refresh, retry, expire, minimumttl� zBSOA serial, refresh, retry, expire, minimumttl' should be integers)rI r4 r �rangerJ rK )r# r r$ r% �args�irN r r r r& � s
����zSOARecord.from_stringr )
r r r r r) �DNS_RPC_FLAG_AUTH_ZONE_ROOTr r* r& r+ r r r r rO � s
�rO c s: e Zd Zddddejdf� fdd� Zed
dd ��Z� ZS ) � SRVRecordr �d r r
c
sd t t| ��� tj| _||B | _|| _|| _t �
� } || _ || _ || _
|| j_t|�| j_| | _d S r )r rf r r � DNS_TYPE_SRVr r r r r �DNS_RPC_RECORD_SRV� wPriority�wWeight�wPort�
nameTargetr3 r4 r )
r �target�portrM �weightr r r r �srvr r r r � s
zSRVRecord.__init__Nc
K s� z |� |�\}}}}W n ty } ztd| �|�d }~ww z|t|�t|�t|�f} W n ty> } ztd�|�d }~ww | | i |��S )NzXSRV data must have four space separated elements: server, port, priority, weight; not %rz/SRV port, priority, and weight must be integers)rI rK r rJ )
r# r r$ r% rn ro rM rp rN rc r r r r& � s& ���� ���zSRVRecord.from_stringr r( r r r r rf � s �rf c r
)
� TXTRecordr r
r c
s� t t| ��� tj| _||B | _|| _|| _t |t
�r |g}g }|D ]}t � � }||_
t
|�|_
|�|� q"t �� } t
|�| _|| _
| | _d S r )r rr r r � DNS_TYPE_TXTr r r r �
isinstancer3 r r2 r4 �append�DNS_RPC_RECORD_STRING�countr )
r �slistr r r r �names�s�name�txtr r r r � s"
zTXTRecord.__init__Nc K s t �|�}| |fi |��S r )�shlexrI )r# r r$ r% rx r r r r& s
zTXTRecord.from_stringr r( r r r r rr � s �rr c C sR t � �� D ]!\} }| dd � dkr&| d d� } ttd| �}|t| < |t|< qd S )Ni�����Record� DNS_TYPE_)�globals�items�getattrr �_RECORD_TYPE_LUT)�k�v�flagr r r �_setup_record_type_lut s ��r� c
K sX t | t�r | �� } zt| }W n ty } ztd�|�d}~ww |j|fd|i|��S )z�Get a DNS record of type t based on the data string.
Additional keywords (ttl, rank, etc) can be passed in.
t can be a dnsp.DNS_TYPE_* integer or a string like "A", "TXT", etc.
zUnsupported record typeNr$ )rt r3 �upperr� �KeyErrorr r& )�tr r$ r% r~ rN r r r �record_from_string s
��r� c C s6 | � � }zttd| �W S ty td| �t�w )Nr z Unknown type of DNS record %s)r� r� r �AttributeErrorr rN )�rec_type�rtyper r r �flag_from_string' s �r� c O s t | i |��}t�� }||_|S r )r� r �DNS_RPC_RECORD_BUF�rec)rc r% r� �bufr r r �recbuf_from_string/ s r� c C s | j �d��� |j �d��� kS )z%Match dns name (of type DNS_RPC_NAME)�.)r3 �rstrip�lower)�n1�n2r r r �dns_name_equal6 s r� c C s t �t j| �}t �t j|�S )z.Convert an AAAA adresss into a canonical form.)�socket� inet_pton�AF_INET6� inet_ntop)�addr�packedr r r �ipv6_normalise; s r� c C s� t ||�}tj}z| �tjd|||d||dd�
\}} W n ty8 }
z|
jd tjkr2W Y d}
~
dS |
�d}
~
ww | r@| j dkrBdS | j
d j D �]} | j |krQqHd} |t
jkra| j|jkr`d} n�|t
jkrst| j�t|j�krrd} n�|t
jkr�t| j|j�r�d} n�|t
jkr�t| j|j�r�d} n�|t
jkr�t| j|j�r�d} n�|t
jkr�t| jj|jj�r�| jj|jjkr�d} n�|t
jkr�| jj|jjkr�| jj|jjkr�| jj |jj kr�t| jj |jj �r�d} nv|t
j k�r,| jj |jj k�r+| jj |jj k�r+| jj!|jj!k�r+| jj"|jj"k�r+| jj#|jj#k�r+t| jj$|jj$��r+t| jj%|jj%��r+d} n,|t
j&k�rX| jj |jj k�rXd} t'| jj �D ]}
| �oU| jj(|
j(|jj(|
j(k} �qC| �r_| S qHdS )z1Find a dns record that matches the specified datar NFT))r� r �DNS_RPC_VIEW_AUTHORITY_DATA�DnssrvEnumRecords2�DNS_CLIENT_VERSION_LONGHORN� WERRORErrorrc �werror�"WERR_DNS_ERROR_NAME_DOES_NOT_EXISTrw r� �recordsr r r r r- r� r1 r� r: r> rB rE rD rh rj rk rl rm rP rR rS rT rU rV rW rX rs rb r3 )�dns_connrL �zoner{ � record_typer �urec� select_flags�buflen�resrN r� �foundrd r r r �dns_record_matchA s�
���
�
�
�
�
�
�
�� ����� ��r� r )r} r� � samba.dcerpcr r rK r �DNS_RPC_RECORDr r, r/ r9 r= rA rO rf rr r� r� r� r� r� r� r� r� r r r r �<module> s. &#