a
�M�e�# � @ s� U d dl Z d dlZe jdkr�d dlmZ d dlZdZd dlZz e du W n e
y^ e Z Y n0 ej �
d�r�d dlZd dlZd dlZdZndZdd� ZG d d
� d
�Zer�G d d � d ej�ZnG d
d � d �ZG dd� d�Zeed< er�er�eZneZdd� ZdS )� N�win32)�AnyT�wmiFc C s" | � d�r| dd � } tj�| �S )N�.� )�
startswith�dns�nameZ from_text)�domain� r �1/usr/lib/python3.9/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 �� znt�� }|�� D ]P}|jr|jrt|j�| j_ |j
rLt |j
�| j_ |j
rfdd� |j
D �| j_ qlqW t �� n
t �� 0 d S )Nc S s g | ] }t |��qS r )r
)�.0�xr r r �
<listcomp>= s �z"_WMIGetter.run.<locals>.<listcomp>)� pythoncomZ CoInitializer ZWMIZ!Win32_NetworkAdapterConfigurationZ IPEnabledZDNSServerSearchOrder�listr r Z DNSDomainr
r
ZDNSDomainSuffixSearchOrderr ZCoUninitialize)r �systemZ interfacer r r �run3 s �
z_WMIGetter.runc C s | � � | �� | jS r )�start�joinr r r r r �getE s z_WMIGetter.get)r r r r r# r& �
__classcell__r r r r r . s r c @ s e Zd ZdS )r N)r r r r r r r r N 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 R s z_RegistryGetter.__init__c C s0 |� d�dkrd}n|� d�dkr(d}nd}|S )N� r �,)�find)r �entry�
split_charr r r �_determine_split_charU s z%_RegistryGetter._determine_split_charc C s< | � |�}|�|�}|D ] }|| jjvr| jj�|� qd S r )r. �splitr r �append)r r r- Zns_list�nsr r r �_config_nameserversd s
z#_RegistryGetter._config_nameserversc C sD | � |�}|�|�}|D ]&}t|�}|| jjvr| jj�|� qd S r )r. r/ r
r r r0 )r r r- Z search_list�sr r r �_config_searchk s
z _RegistryGetter._config_searchc C sh zt �|d�\}}W n ty* d }Y n0 |r:| �|� |sB|r|z$t �|d�\}}|rdt|�| j_W q� tyx Y q�0 nrzt �|d�\}}W n ty� d }Y n0 |r�| �|� z$t �|d�\}}|r�t|�| j_W n ty� Y n0 zt �|d�\}}W n t�y d }Y n0 |d u �rTzt �|d�\}}W n t�yR d }Y n0 |�rd| �|� d S )NZ
NameServerZDomainZDhcpNameServerZ
DhcpDomainZ
SearchListZDhcpSearchList)�winreg� QueryValueEx� WindowsErrorr2 r
r r
r4 )r �keyZalways_try_domainZservers�_Zdomr r r r �_config_fromkeys sH
z _RegistryGetter._config_fromkeyc C s� z�t �|d| �}z�t �|d�\}}|t jkr2t�t �|d| �}z>t �|d�\}}|t jkrbt�|d@ W |�� W |�� W S |�� 0 W |�� n
|�� 0 W n ty� Y dS 0 d S )Nz]SYSTEM\CurrentControlSet\Control\Network\{4D36E972-E325-11CE-BFC1-08002BE10318}\%s\ConnectionZ
PnpInstanceIDz SYSTEM\CurrentControlSet\Enum\%sZ ConfigFlagsr F)r5 �OpenKeyr6 ZREG_SZ�
ValueErrorZ REG_DWORD�Close� Exception)r �lm�guidZconnection_keyZpnp_idZttypeZ
device_key�flagsr r r �_is_nic_enabled� s: ���
�
�� z _RegistryGetter._is_nic_enabledc C s t �dt j�}z�t �|d�}z| �|d� W |�� n
|�� 0 t �|d�}z�d}zbt �||�}|d7 }t �||�}z2| �||�s�W |�� W qR| �|d� W |�� n
|�� 0 W qR ty� Y q�Y qR0 qRW |�� n
|�� 0 W |�� n
|�� 0 | j S )z9Extract resolver configuration from the Windows registry.Nz2SYSTEM\CurrentControlSet\Services\Tcpip\ParametersTz=SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfacesr r F)
r5 ZConnectRegistry�HKEY_LOCAL_MACHINEr; r: r= ZEnumKeyrB �EnvironmentErrorr )r r? Z
tcp_paramsZ
interfaces�ir@ r8 r r r r& � s< �� � z_RegistryGetter.getN)
r r r r r. r2 r4 r: rB r& r r r r r( Q s '4r( �
_getter_classc C s t � } | �� S )z Extract resolver configuration.)rF r& )�getterr r r � get_dns_info� s rH )�sysZ
dns._featuresr �platform�typingr Zdns.nameZ _prefer_wmir5 r7 �KeyErrorr> Z _featuresZhave� threadingr r Z _have_wmir
r �Threadr r( �__annotations__rF rH r r r r �<module> s8
#