File: //opt/alt/python37/lib/python3.7/site-packages/dns/__pycache__/win32util.cpython-37.pyc
B
��fa# � @ s U d dl Z e jdk�rd dlmZ d dlZdZd dlZy edk W n e k
rZ e
ZY nX y d dl Z d dl Z d dl
Z
dZW n e
k
r� dZY nX dd� ZG dd � d �Zer�G d
d � d e j�ZnG d d � d �ZG d
d� d�Zeed< er�er�eZneZdd� ZdS )� N�win32)�AnyTFc C s" | � d�r| dd � } tj�| �S )N�.� )�
startswith�dns�nameZ from_text)�domain� r
�>/opt/alt/python37/lib/python3.7/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 )N)�superr r
�info)r )� __class__r
r r - s
z_WMIGetter.__init__c C sz t �� zbt�� }xT|�� D ]H}|jr |jr t|j�| j_ t
|j �| j_ |j
rbdd� |j
D �| j_P q W W d t �� X d S )Nc S s g | ] }t |��qS r
)r )�.0�xr
r
r �
<listcomp>; s z"_WMIGetter.run.<locals>.<listcomp>)� pythoncomZ CoInitialize�wmiZWMIZ!Win32_NetworkAdapterConfigurationZ IPEnabledZ DNSDomainr r r �listZDNSServerSearchOrderr ZDNSDomainSuffixSearchOrderr ZCoUninitialize)r �systemZ interfacer
r
r �run1 s
z_WMIGetter.runc C s | � � | �� | jS )N)�start�joinr )r r
r
r �getB 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 K s c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�_RegistryGetterc C s t � | _d S )N)r
r )r r
r
r r O 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_charR s z%_RegistryGetter._determine_split_charc C s@ | � |�}|�|�}x&|D ] }|| jjkr| jj�|� qW d S )N)r+ �splitr r �append)r r r* Zns_list�nsr
r
r �_config_nameserversa s
z#_RegistryGetter._config_nameserversc C sH | � |�}|�|�}x.|D ]&}t|�}|| jjkr| jj�|� qW d S )N)r+ r, r r r r- )r r r* Z search_list�sr
r
r �_config_searchh s
z _RegistryGetter._config_searchc C st yt �|d�\}}W n tk
r, d }Y nX |r<| �|� |sD|r�y$t �|d�\}}|rft|�| j_W q� tk
r| Y q�X nvyt �|d�\}}W n tk
r� d }Y nX |r�| �|� y$t �|d�\}}|r�t|�| j_W n tk
r� Y nX yt �|d�\}}W n tk
�r$ d }Y nX |d k�r`yt �|d�\}}W n tk
�r^ d }Y nX |�rp| �|� d S )NZ
NameServer�DomainZDhcpNameServerZ
DhcpDomainZ
SearchListZDhcpSearchList)�winreg� QueryValueEx� WindowsErrorr/ r r r r1 )r �keyZalways_try_domainZservers�_�domr r
r
r �_config_fromkeyp sH
z _RegistryGetter._config_fromkeyc C s� y�t �|d| �}zft �|d�\}}|t jkr2t�t �|d| �}z(t �|d�\}}|t jkrbt�|d@ S |�� X W d |�� X W n tk
r� dS X d S )Nz]SYSTEM\CurrentControlSet\Control\Network\{4D36E972-E325-11CE-BFC1-08002BE10318}\%s\ConnectionZ
PnpInstanceIDz SYSTEM\CurrentControlSet\Enum\%sZ ConfigFlagsr F)r3 �OpenKeyr4 �REG_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 d|�� X t �|d�}ztd}xjyNt �||�}|d7 }t �||�}z | �||�s~wL| �|d� W d|�� X W qL tk
r� P Y qLX qLW W d|�� X W d|�� X | j S )z9Extract resolver configuration from the Windows registry.Nz2SYSTEM\CurrentControlSet\Services\Tcpip\ParametersTz=SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfacesr r F)
r3 ZConnectRegistry�HKEY_LOCAL_MACHINEr: r9 r= �EnumKeyrB �EnvironmentErrorr )r r? Z
tcp_paramsZ
interfaces�ir@ r6 r
r
r r# � s6
z_RegistryGetter.getN)
r r r r r+ r/ r1 r9 rB r# r
r
r
r r% N s '4r% �
_getter_classc C s t � } | �� S )z Extract resolver configuration.)rG r# )�getterr
r
r � get_dns_info� s rI )�sys�platform�typingr Zdns.namer Z _prefer_wmir3 r5 �KeyErrorr> � threadingr r Z _have_wmir r
�Threadr r% �__annotations__rG rI r
r
r
r �<module> s:
%