File: //home/arjun/projects/env/lib64/python3.10/site-packages/dns/__pycache__/win32util.cpython-310.pyc
o
!we-# � @ s U d dl Z e jdkr�d dlmZ d dlZdZd dlZzedu W n e y+ e
ZY nw zd dl Z d dl Z d dl
Z
dZW n e
yG dZY nw dd� ZG dd � d �Zer_G d
d � d e j�ZnG d d � d �ZG d
d� d�Zeed< erxerxeZneZdd� ZdS dS )� N�win32)�AnyTFc C s" | � d�r | dd � } tj�| �S )N�.� )�
startswith�dns�name� from_text)�domain� r �F/home/arjun/projects/env/lib/python3.10/site-packages/dns/win32util.py�_config_domain s
r
c @ s e Zd Zdd� ZdS )�DnsInfoc C s d | _ g | _g | _d S �N)r
� nameservers�search��selfr r r �__init__% s
zDnsInfo.__init__N)�__name__�
__module__� __qualname__r r r r r r $ s r c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�
_WMIGetterc s t � �� t� | _d S r )�superr r �infor �� __class__r r r - s
z_WMIGetter.__init__c C s� t �� z<t�� }|�� D ]%}|jr2|jr2t|j�| j_ t
|j �| j_ |j
r0dd� |j
D �| j_ nq
W t �� d S W t �� d S t �� w )Nc S s g | ]}t |��qS r )r
)�.0�xr r r �
<listcomp>: s ��z"_WMIGetter.run.<locals>.<listcomp>)� pythoncom� CoInitialize�wmi�WMI�!Win32_NetworkAdapterConfiguration� IPEnabled� DNSDomainr
r r
�list�DNSServerSearchOrderr �DNSDomainSuffixSearchOrderr �CoUninitialize)r �system� interfacer r r �run1 s"
��� �z_WMIGetter.runc C s | � � | �� | jS r )�start�joinr r r r r �getB s z_WMIGetter.get)r r r r r- r0 �
__classcell__r r r r r , s r c @ s e Zd ZdS )r N)r r r r r r r r K s c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�_RegistryGetterc C s t � | _d S r )r r r r r r r O s z_RegistryGetter.__init__c C s4 |� d�dkr d}|S |� d�dkrd}|S d}|S )N� r �,)�find)r �entry�
split_charr r r �_determine_split_charR s ��z%_RegistryGetter._determine_split_charc C s< | � |�}|�|�}|D ]}|| jjvr| jj�|� q d S r )r8 �splitr r �append)r r r7 �ns_list�nsr r r �_config_nameserversa s
��z#_RegistryGetter._config_nameserversc C sD | � |�}|�|�}|D ]}t|�}|| jjvr | jj�|� q d S r )r8 r9 r
r r r: )r r r7 � search_list�sr r r �_config_searchh s
��z _RegistryGetter._config_searchc C sb z
t �|d�\}}W n ty d }Y nw |r | �|� |s!|r=zt �|d�\}}|r2t|�| j_W nB ty< Y n:w z
t �|d�\}}W n tyR d }Y nw |rv| �|� zt �|d�\}}|rkt|�| j_W n tyu Y nw z
t �|d�\}}W n ty� d }Y nw |d u r�z
t �|d�\}}W n ty� d }Y nw |r�| �|� d S d S )N�
NameServer�Domain�DhcpNameServer�
DhcpDomain�
SearchList�DhcpSearchList)�winreg� QueryValueEx� WindowsErrorr= r
r r
r@ )r �key�always_try_domain�servers�_�domr r r r �_config_fromkeyp sZ �
� � �
� � � ��z _RegistryGetter._config_fromkeyc C s� zJt �|d| �}z<t �|d�\}}|t jkrt�t �|d| �}z t �|d�\}}|t jkr1t�|d@ W |�� W |�� W S |�� w |�� w tyT Y dS w )Nz]SYSTEM\CurrentControlSet\Control\Network\{4D36E972-E325-11CE-BFC1-08002BE10318}\%s\Connection�
PnpInstanceIDz SYSTEM\CurrentControlSet\Enum\%s� ConfigFlagsr F)rG �OpenKeyrH �REG_SZ�
ValueError� REG_DWORD�Close� Exception)r �lm�guid�connection_key�pnp_id�ttype�
device_key�flagsr r r �_is_nic_enabled� s8 ���
�
�
�z _RegistryGetter._is_nic_enabledc C s� t �dt j�}zqt �|d�}z | �|d� W |�� n|�� w t �|d�}zEd} z1t �||�}|d7 }t �||�}z| �||�sIW |�� W q)| �|d� W |�� n|�� w W n tyd Y nw q*W |�� n|�� w W |�� | j S |�� w )z9Extract resolver configuration from the Windows registry.Nz2SYSTEM\CurrentControlSet\Services\Tcpip\ParametersTz=SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfacesr r F)
rG �ConnectRegistry�HKEY_LOCAL_MACHINErR rO rV �EnumKeyr_ �EnvironmentErrorr )r rX �
tcp_params�
interfaces�irY rJ r r r r0 � sD �� � ��
�z_RegistryGetter.getN)
r r r r r8 r= r@ rO r_ r0 r r r r r2 N s ' 4r2 �
_getter_classc C s t � } | �� S )z Extract resolver configuration.)rg r0 )�getterr r r � get_dns_info� s ri )�sys�platform�typingr �dns.namer � _prefer_wmirG rI �KeyErrorrW � threadingr r" � _have_wmir
r �Threadr r2 �__annotations__rg ri r r r r �<module> sB
� � # ��