File Manager

X7ROOT File Manager

Current Path: /opt/hc_python/lib/python3.12/site-packages/dns

opt / hc_python / lib / python3.12 / site-packages / dns /

?? ..
?? __init__.py(1.62 KB)
?? __pycache__
?? _asyncbackend.py(2.34 KB)
?? _asyncio_backend.py(8.84 KB)
?? _ddr.py(5.12 KB)
?? _features.py(2.43 KB)
?? _immutable_ctx.py(2.4 KB)
?? _trio_backend.py(8.27 KB)
?? asyncbackend.py(2.73 KB)
?? asyncquery.py(30.1 KB)
?? asyncresolver.py(17.43 KB)
?? dnssec.py(40.74 KB)
?? dnssecalgs
?? dnssectypes.py(1.76 KB)
?? e164.py(3.88 KB)
?? edns.py(16.69 KB)
?? entropy.py(4.14 KB)
?? enum.py(3.6 KB)
?? exception.py(5.81 KB)
?? flags.py(2.69 KB)
?? grange.py(2.09 KB)
?? immutable.py(1.97 KB)
?? inet.py(5.64 KB)
?? ipv4.py(2.49 KB)
?? ipv6.py(6.4 KB)
?? message.py(66.59 KB)
?? name.py(41.78 KB)
?? namedict.py(3.91 KB)
?? nameserver.py(9.88 KB)
?? node.py(12.37 KB)
?? opcode.py(2.67 KB)
?? py.typed(0 B)
?? query.py(54.98 KB)
?? quic
?? rcode.py(4.06 KB)
?? rdata.py(30.29 KB)
?? rdataclass.py(2.91 KB)
?? rdataset.py(16.27 KB)
?? rdatatype.py(7.27 KB)
?? rdtypes
?? renderer.py(10.99 KB)
?? resolver.py(72 KB)
?? reversename.py(3.74 KB)
?? rrset.py(8.96 KB)
?? serial.py(3.52 KB)
?? set.py(9 KB)
?? tokenizer.py(23.03 KB)
?? transaction.py(22.06 KB)
?? tsig.py(11.15 KB)
?? tsigkeyring.py(2.57 KB)
?? ttl.py(2.91 KB)
?? update.py(11.96 KB)
?? version.py(1.88 KB)
?? versioned.py(11.49 KB)
?? win32util.py(8.67 KB)
?? wire.py(2.76 KB)
?? xfr.py(12.96 KB)
?? zone.py(50.87 KB)
?? zonefile.py(27.27 KB)
?? zonetypes.py(690 B)

Editing: win32util.py

import sys

import dns._features

if sys.platform == "win32":
    from typing import Any

import dns.name

_prefer_wmi = True

import winreg  # pylint: disable=import-error

# Keep pylint quiet on non-windows.
    try:
        _ = WindowsError  # pylint: disable=used-before-assignment
    except NameError:
        WindowsError = Exception

if dns._features.have("wmi"):
        import threading

import pythoncom  # pylint: disable=import-error
        import wmi  # pylint: disable=import-error

_have_wmi = True
    else:
        _have_wmi = False

def _config_domain(domain):
        # Sometimes DHCP servers add a '.' prefix to the default domain, and
        # Windows just stores such values in the registry (see #687).
        # Check for this and fix it.
        if domain.startswith("."):
            domain = domain[1:]
        return dns.name.from_text(domain)

class DnsInfo:
        def __init__(self):
            self.domain = None
            self.nameservers = []
            self.search = []

if _have_wmi:

class _WMIGetter(threading.Thread):
            # pylint: disable=possibly-used-before-assignment
            def __init__(self):
                super().__init__()
                self.info = DnsInfo()

def run(self):
                pythoncom.CoInitialize()
                try:
                    system = wmi.WMI()
                    for interface in system.Win32_NetworkAdapterConfiguration():
                        if interface.IPEnabled and interface.DNSServerSearchOrder:
                            self.info.nameservers = list(interface.DNSServerSearchOrder)
                            if interface.DNSDomain:
                                self.info.domain = _config_domain(interface.DNSDomain)
                            if interface.DNSDomainSuffixSearchOrder:
                                self.info.search = [
                                    _config_domain(x)
                                    for x in interface.DNSDomainSuffixSearchOrder
                                ]
                            break
                finally:
                    pythoncom.CoUninitialize()

def get(self):
                # We always run in a separate thread to avoid any issues with
                # the COM threading model.
                self.start()
                self.join()
                return self.info

else:

class _WMIGetter:  # type: ignore
            pass

class _RegistryGetter:
        def __init__(self):
            self.info = DnsInfo()

def _split(self, text):
            # The windows registry has used both " " and "," as a delimiter, and while
            # it is currently using "," in Windows 10 and later, updates can seemingly
            # leave a space in too, e.g. "a, b".  So we just convert all commas to
            # spaces, and use split() in its default configuration, which splits on
            # all whitespace and ignores empty strings.
            return text.replace(",", " ").split()

def _config_nameservers(self, nameservers):
            for ns in self._split(nameservers):
                if ns not in self.info.nameservers:
                    self.info.nameservers.append(ns)

def _config_search(self, search):
            for s in self._split(search):
                s = _config_domain(s)
                if s not in self.info.search:
                    self.info.search.append(s)

def _config_fromkey(self, key, always_try_domain):
            try:
                servers, _ = winreg.QueryValueEx(key, "NameServer")
            except WindowsError:
                servers = None
            if servers:
                self._config_nameservers(servers)
            if servers or always_try_domain:
                try:
                    dom, _ = winreg.QueryValueEx(key, "Domain")
                    if dom:
                        self.info.domain = _config_domain(dom)
                except WindowsError:
                    pass
            else:
                try:
                    servers, _ = winreg.QueryValueEx(key, "DhcpNameServer")
                except WindowsError:
                    servers = None
                if servers:
                    self._config_nameservers(servers)
                    try:
                        dom, _ = winreg.QueryValueEx(key, "DhcpDomain")
                        if dom:
                            self.info.domain = _config_domain(dom)
                    except WindowsError:
                        pass
            try:
                search, _ = winreg.QueryValueEx(key, "SearchList")
            except WindowsError:
                search = None
            if search is None:
                try:
                    search, _ = winreg.QueryValueEx(key, "DhcpSearchList")
                except WindowsError:
                    search = None
            if search:
                self._config_search(search)

def _is_nic_enabled(self, lm, guid):
            # Look in the Windows Registry to determine whether the network
            # interface corresponding to the given guid is enabled.
            #
            # (Code contributed by Paul Marks, thanks!)
            #
            try:
                # This hard-coded location seems to be consistent, at least
                # from Windows 2000 through Vista.
                connection_key = winreg.OpenKey(
                    lm,
                    r"SYSTEM\CurrentControlSet\Control\Network"
                    r"\{4D36E972-E325-11CE-BFC1-08002BE10318}"
                    rf"\{guid}\Connection",
                )

try:
                    # The PnpInstanceID points to a key inside Enum
                    (pnp_id, ttype) = winreg.QueryValueEx(
                        connection_key, "PnpInstanceID"
                    )

if ttype != winreg.REG_SZ:
                        raise ValueError  # pragma: no cover

device_key = winreg.OpenKey(
                        lm, rf"SYSTEM\CurrentControlSet\Enum\{pnp_id}"
                    )

try:
                        # Get ConfigFlags for this device
                        (flags, ttype) = winreg.QueryValueEx(device_key, "ConfigFlags")

if ttype != winreg.REG_DWORD:
                            raise ValueError  # pragma: no cover

# Based on experimentation, bit 0x1 indicates that the
                        # device is disabled.
                        #
                        # XXXRTH I suspect we really want to & with 0x03 so
                        # that CONFIGFLAGS_REMOVED devices are also ignored,
                        # but we're shifting to WMI as ConfigFlags is not
                        # supposed to be used.
                        return not flags & 0x1

finally:
                        device_key.Close()
                finally:
                    connection_key.Close()
            except Exception:  # pragma: no cover
                return False

def get(self):
            """Extract resolver configuration from the Windows registry."""

lm = winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE)
            try:
                tcp_params = winreg.OpenKey(
                    lm, r"SYSTEM\CurrentControlSet\Services\Tcpip\Parameters"
                )
                try:
                    self._config_fromkey(tcp_params, True)
                finally:
                    tcp_params.Close()
                interfaces = winreg.OpenKey(
                    lm,
                    r"SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfaces",
                )
                try:
                    i = 0
                    while True:
                        try:
                            guid = winreg.EnumKey(interfaces, i)
                            i += 1
                            key = winreg.OpenKey(interfaces, guid)
                            try:
                                if not self._is_nic_enabled(lm, guid):
                                    continue
                                self._config_fromkey(key, False)
                            finally:
                                key.Close()
                        except OSError:
                            break
                finally:
                    interfaces.Close()
            finally:
                lm.Close()
            return self.info

_getter_class: Any
    if _have_wmi and _prefer_wmi:
        _getter_class = _WMIGetter
    else:
        _getter_class = _RegistryGetter

def get_dns_info():
        """Extract resolver configuration."""
        getter = _getter_class()
        return getter.get()

X7ROOT File Manager

Editing: win32util.py

Upload File

Create Folder