atomicshop 2.15.11__py3-none-any.whl → 3.10.5__py3-none-any.whl

atomicshop/wrappers/winregw/winreg_network.py

@@ -0,0 +1,232 @@
+import winreg
+import socket
+
+
+def get_network_interfaces_settings(
+        interface_guid: str = None
+) -> dict:
+    """
+    Get network interface settings from the Windows registry.
+
+    :param interface_guid: str, GUID of the network interface to retrieve settings for.
+        If None, settings for all interfaces will be retrieved.
+    :return: dict, network interface settings.
+    """
+    registry_path = r"SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfaces"
+    network_info = {}
+
+    with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, registry_path) as interfaces_key:
+        interface_count = winreg.QueryInfoKey(interfaces_key)[0]
+
+        for i in range(interface_count):
+            current_interface_guid = winreg.EnumKey(interfaces_key, i)
+
+            # If an interface GUID is provided, and it doesn't match the current one, skip it
+            if interface_guid and interface_guid != current_interface_guid:
+                continue
+
+            interface_path = f"{registry_path}\\{current_interface_guid}"
+            interface_data = {}
+
+            with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, interface_path) as key:
+                value_count = winreg.QueryInfoKey(key)[1]
+
+                for j in range(value_count):
+                    value_name, value_data, _ = winreg.EnumValue(key, j)
+                    interface_data[value_name] = value_data
+
+                # Populate the dictionary for the current interface
+                network_info[current_interface_guid] = interface_data
+
+    return network_info
+
+
+def get_network_connections_to_guids() -> dict:
+    """
+    Get a dictionary mapping network connection names to their corresponding GUIDs.
+
+    :return: dict, GUIDs to connection names.
+    """
+    adapters = {}
+    registry_path = r"SYSTEM\CurrentControlSet\Control\Network\{4D36E972-E325-11CE-BFC1-08002BE10318}"
+
+    with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, registry_path) as network_key:
+        adapter_count = winreg.QueryInfoKey(network_key)[0]
+
+        for i in range(adapter_count):
+            adapter_guid = winreg.EnumKey(network_key, i)
+            adapter_path = f"{registry_path}\\{adapter_guid}\\Connection"
+
+            try:
+                with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, adapter_path) as connection_key:
+                    adapter_name, _ = winreg.QueryValueEx(connection_key, "Name")
+                    adapters[adapter_guid] = adapter_name
+            except FileNotFoundError:
+                # Some GUIDs might not have a corresponding 'Connection' key, so we skip them
+                continue
+
+    return adapters
+
+
+def get_enum_info_by_pnpinstanceid(
+        pnp_instance_id: str
+) -> dict:
+    """
+    Get all information from the Enum registry key for a device with a specific PnPInstanceId.
+
+    :param pnp_instance_id: str, PnPInstanceId of the device.
+    :return: dict, device information.
+    """
+    enum_registry_path = r"SYSTEM\CurrentControlSet\Enum"
+    device_info = {}
+
+    # Construct the full path to the device in the Enum registry
+    hardware_path = f"{enum_registry_path}\\{pnp_instance_id}"
+
+    # Open the registry key corresponding to the device
+    with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, hardware_path) as hardware_key:
+        num_values = winreg.QueryInfoKey(hardware_key)[1]
+
+        # Retrieve all values under this key
+        for i in range(num_values):
+            value_name, value_data, _ = winreg.EnumValue(hardware_key, i)
+            device_info[value_name] = value_data
+
+    return device_info
+
+
+def get_network_connections_details(get_enum_info: bool = True) -> dict:
+    """
+    Get network adapter details from the Windows registry.
+
+    :param get_enum_info: bool, if True, retrieve all information from the corresponding Enum key.
+        This is useful for getting additional information about the network adapter, like make, model, diver details.
+    :return: dict, network adapter details.
+    """
+    adapter_details = {}
+    network_registry_path = r"SYSTEM\CurrentControlSet\Control\Network\{4D36E972-E325-11CE-BFC1-08002BE10318}"
+
+    with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, network_registry_path) as network_key:
+        adapter_count = winreg.QueryInfoKey(network_key)[0]
+
+        for i in range(adapter_count):
+            adapter_guid = winreg.EnumKey(network_key, i)
+            adapter_path = f"{network_registry_path}\\{adapter_guid}\\Connection"
+
+            try:
+                with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, adapter_path) as connection_key:
+                    adapter_name, _ = winreg.QueryValueEx(connection_key, "Name")
+                    pnp_instance_id, _ = winreg.QueryValueEx(connection_key, "PnPInstanceId")
+
+                    # Get all information from the corresponding Enum key
+                    if get_enum_info:
+                        enum_info: dict = get_enum_info_by_pnpinstanceid(pnp_instance_id)
+                    else:
+                        enum_info: dict = {}
+
+                    # Store the retrieved information
+                    adapter_details[adapter_guid] = {
+                        "Name": adapter_name,
+                        "PnPInstanceId": pnp_instance_id,
+                        "EnumInfo": enum_info
+                    }
+
+            except FileNotFoundError:
+                continue
+
+    return adapter_details
+
+
+def _get_default_dns_gateway() -> tuple[bool, list[str]]:
+    """
+    NOTICE: This stopped working from the last Windows update on 11.06.2025.
+    They moved it to 'ProfileNameServer', anyway Since Windows 8 the recommended API has been the WMI
+    NetTCPIP CIM provider (MSFT_DNSClientServerAddress) - didn't test it though.
+    Just moved to netsh wrapping for now.
+
+    Get the default DNS gateway from the Windows registry.
+
+    :return: tuple(is dynamic boolean, list of DNS server IPv4s).
+        If nothing found will return (None, None).
+    """
+
+    def get_current_interface_status(current_interface_settings: dict) -> tuple[bool, list[str]]:
+        if current_interface_settings['NameServer']:
+            result = (False, current_interface_settings['NameServer'].split(','))
+        else:
+            result = (True, current_interface_settings['DhcpNameServer'].split(','))
+
+        return result
+
+
+    # Get current default IPv4 address of the interface that is being used for internet.
+    default_ipv4_address: str = socket.gethostbyname(socket.gethostname())
+    # If the default IPv4 address is localhost, then it could mean that the system is not connected to the internet.
+    # Or there is no network adapter at all.
+    default_dns_gateway_list: list[str] = []
+    if default_ipv4_address == '127.0.0.1':
+        from ... import dns
+        default_dns_gateway_list = dns.get_default_dns_gateway_with_dns_resolver()
+
+    # Get all network interface settings from the registry.
+    all_interfaces_configurations = get_network_interfaces_settings()
+
+    # Find the interface that has this IPv4 assigned.
+    function_result: tuple = tuple()
+    for interface_guid, interface_settings in all_interfaces_configurations.items():
+        if not interface_settings:
+            continue
+
+        if ' ' in interface_settings['NameServer']:
+            interface_settings['NameServer'] = interface_settings['NameServer'].replace(' ', ',')
+        if 'DhcpNameServer' in interface_settings and ' ' in interface_settings['DhcpNameServer']:
+            interface_settings['DhcpNameServer'] = interface_settings['DhcpNameServer'].replace(' ', ',')
+
+        if not default_dns_gateway_list:
+            current_interface_static_ipv4_address: list = interface_settings.get('IPAddress', None)
+            current_interface_dynamic_ipv4_address: str = interface_settings.get('DhcpIPAddress', None)
+
+            static_and_ip_match: bool = (
+                    current_interface_static_ipv4_address and
+                    current_interface_static_ipv4_address[0] == default_ipv4_address)
+            dynamic_and_ip_match: bool = (
+                    current_interface_dynamic_ipv4_address and
+                    current_interface_dynamic_ipv4_address == default_ipv4_address)
+            if static_and_ip_match or dynamic_and_ip_match:
+                function_result = get_current_interface_status(interface_settings)
+
+                break
+        else:
+            current_interface_name_server_list: list[str] = interface_settings['NameServer'].split(',')
+            current_interface_dhcp_name_server_list: list[str] = interface_settings['DhcpNameServer'].split(',')
+            if (current_interface_name_server_list == default_dns_gateway_list or
+                    current_interface_dhcp_name_server_list == default_dns_gateway_list):
+                function_result = get_current_interface_status(interface_settings)
+
+                break
+
+    if not function_result:
+        function_result = (None, None)
+
+    # noinspection PyTypeChecker
+    return function_result
+
+
+def change_metric_of_network_adapter(
+        adapter_guid: str,
+        metric: int
+) -> None:
+    """
+    Change the metric of a network adapter.
+    After you set new metric you need to restart the network adapter for the changes to take effect.
+
+    :param adapter_guid: str, GUID of the network adapter.
+    :param metric: int, new metric value.
+    :return: None
+    """
+
+    reg_path = fr"SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfaces\{adapter_guid}"
+
+    with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, reg_path, 0, winreg.KEY_SET_VALUE) as key:
+        winreg.SetValueEx(key, "UseAutomaticMetric", 0, winreg.REG_DWORD, 0)
+        winreg.SetValueEx(key, "InterfaceMetric", 0, winreg.REG_DWORD, metric)

{atomicshop-2.15.11.dist-info → atomicshop-3.10.5.dist-info}/METADATA

@@ -1,67 +1,47 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: atomicshop
-Version: 2.15.11
+Version: 3.10.5
 Summary: Atomic functions and classes to make developer life easier
 Author: Denis Kras
-License: MIT License
-        
-        Copyright (c) 2023 Bugsec, Denis Kras
-        
-        Permission is hereby granted, free of charge, to any person obtaining a copy
-        of this software and associated documentation files (the "Software"), to deal
-        in the Software without restriction, including without limitation the rights
-        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-        copies of the Software, and to permit persons to whom the Software is
-        furnished to do so, subject to the following conditions:
-        
-        The above copyright notice and this permission notice shall be included in all
-        copies or substantial portions of the Software.
-        
-        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-        SOFTWARE.
+License-Expression: MIT
 Project-URL: Homepage, https://github.com/BugSec-Official/atomicshop
 Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.11
+Requires-Python: <3.14,>=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
 Requires-Dist: wheel
-Requires-Dist: beautifulsoup4
-Requires-Dist: cryptography
-Requires-Dist: dnslib
-Requires-Dist: dnspython
-Requires-Dist: docker
-Requires-Dist: elasticsearch
+Requires-Dist: setuptools
+Requires-Dist: beautifulsoup4==4.14.3
+Requires-Dist: cryptography==46.0.3
+Requires-Dist: dkarchiver==1.0.1
+Requires-Dist: dkinst
+Requires-Dist: dnslib==0.9.26
+Requires-Dist: dnspython==2.8.0
+Requires-Dist: docker==7.1.0
+Requires-Dist: flask_socketio==5.5.1
+Requires-Dist: google-api-python-client==2.187.0
+Requires-Dist: google-genai==1.53.0
+Requires-Dist: icmplib==3.0.4
 Requires-Dist: numpy
-Requires-Dist: olefile
-Requires-Dist: openpyxl
-Requires-Dist: pandas
-Requires-Dist: paramiko
-Requires-Dist: playwright
-Requires-Dist: playwright-stealth
-Requires-Dist: protobuf
-Requires-Dist: psutil
-Requires-Dist: py7zr
-Requires-Dist: pyautogui
-Requires-Dist: pymongo
-Requires-Dist: pyopenssl
-Requires-Dist: python-bidi
+Requires-Dist: olefile==0.47
+Requires-Dist: openpyxl==3.1.5
+Requires-Dist: pandas==2.3.3
+Requires-Dist: paramiko==4.0.0
+Requires-Dist: pefile==2024.8.26
+Requires-Dist: playwright==1.56.0
+Requires-Dist: psutil==7.1.3
+Requires-Dist: pymongo==4.15.5
+Requires-Dist: pyopenssl==25.3.0
 Requires-Dist: python-docx
-Requires-Dist: python-magic
-Requires-Dist: reportlab
-Requires-Dist: setuptools
+Requires-Dist: pywin32==311; platform_system == "Windows"
 Requires-Dist: SoundCard
 Requires-Dist: soundfile
 Requires-Dist: SpeechRecognition
-Requires-Dist: tldextract
-Requires-Dist: pywin32 ; platform_system == "Windows"
+Requires-Dist: tldextract==5.3.0
+Requires-Dist: websockets==15.0.1
+Dynamic: license-file
 
 <h1 align="center">Atomic Workshop</h1>
 
@@ -153,9 +133,9 @@ To get a local copy up and running follow these simple steps.
    ```
 
    The latest version on PyPI is 0.2, so you will need to install from GitHub.
-   Alternatively, you can use a cmd file in the addon folder after 'atomicshop' installation:
+   Alternatively, you can use a command anywhere in CMD after 'atomicshop' installation:
    ```sh
-   "\Lib\site-packages\atomicshop\addons\a_setup_scripts\install_pywintrace_0.3.cmd"
+   pywintrace install
    ```
 
 4. If you get an exception while installing the 'psycopg2' package on ubuntu, install this binary: