atomicshop 3.2.1__py3-none-any.whl → 3.2.3__py3-none-any.whl

atomicshop/__init__.py

@@ -1,4 +1,4 @@
 """Atomic Basic functions and classes to make developer life easier"""
 
 __author__ = "Den Kras"
-__version__ = '3.2.1'
+__version__ = '3.2.3'

atomicshop/mitm/config_static.py

@@ -3,7 +3,7 @@ from dataclasses import dataclass
 from typing import Literal
 
 from . import import_config
-
+from .message import ClientMessage
 
 # CONFIG = None
 LIST_OF_BOOLEANS: list = [
@@ -173,14 +173,19 @@ def load_config(config_toml_file_path: str):
     return result
 
 
-# def get_listening_addresses() -> list[str]:
-#     """
-#     Get the list of listening addresses from the TCPServer configuration.
-#     If no_engines_usage_to_listen_addresses_enable is True, return the no_engines_listening_address_list.
-#     Otherwise, return an empty list.
-#     """
-#
-#     return ENGINES_LIST
+def get_listening_addresses(client_message: ClientMessage) -> dict:
+    """
+    Get the list of listening addresses from the TCPServer configuration.
+    If no_engines_usage_to_listen_addresses_enable is True, return the no_engines_listening_address_list.
+    Otherwise, return an empty list.
+    """
+
+    for engine in ENGINES_LIST:
+        if engine.engine_name == client_message.engine_name:
+            return {
+                'domain_target_dict': engine.domain_target_dict,
+                'port_target_dict': engine.port_target_dict
+            }
 
 
 # ============ Server Tester Specific ===============

atomicshop/mitm/engines/__reference_general/requester___reference_general.py

@@ -2,6 +2,7 @@
 from atomicshop.mitm.engines.__parent.requester___parent import RequesterParent
 from atomicshop.mitm.shared_functions import create_custom_logger
 from atomicshop.mitm.message import ClientMessage
+from atomicshop.mitm import config_static
 
 """
 import time

atomicshop/mitm/engines/__reference_general/responder___reference_general.py

@@ -2,6 +2,7 @@
 from atomicshop.mitm.engines.__parent.responder___parent import ResponderParent
 from atomicshop.mitm.shared_functions import create_custom_logger
 from atomicshop.mitm.message import ClientMessage
+from atomicshop.mitm import config_static
 
 """
 import time

atomicshop/mitm/engines/create_module_template.py

@@ -82,7 +82,7 @@ class CreateModuleTemplate:
         config_lines_list.append(f'domains = [{", ".join(domains_with_quotes)}]')
         config_lines_list.append('localhost = 1\n')
         config_lines_list.append('[on_port_connect]')
-        config_lines_list.append('#5000 = "31.31.31.31"')
+        config_lines_list.append('#5000 = "31.31.31.31:443"')
         config_lines_list.append('#5000 = "C:\\test\\address.txt"\n')
         config_lines_list.append('[mtls]')
         config_lines_list.append('# "subdomain.domain.com" = "file_name_in_current_dir.pem"\n')

atomicshop/wrappers/pywin32w/wmis/wmi_helpers.py

@@ -143,7 +143,7 @@ def call_method(
     """
 
     # Assign the single value to a tuple if it is not already a tuple or dict and not an EmptyValue.
-    if not isinstance(value, EmptyValue) and not (isinstance(value, tuple) and isinstance(value, dict)):
+    if not isinstance(value, (EmptyValue, tuple, dict)):
         value = (value,)
 
     # Get the method instance out of the WMI object.

atomicshop/wrappers/socketw/dns_server.py

@@ -902,7 +902,11 @@ def get_target_ip_from_engine(
     """
     # Iterate through the list of engines.
     for domain, target_ip_port in engine_domain_target_dict.items():
-        if domain in target_domain:
+        # If the domain is exactly the same as the target domain,
+        if domain == target_domain:
+            # Get the target IP address from the engine.
+            return target_ip_port['ip']
+        elif domain in target_domain:
             # Get the target IP address from the engine.
             return target_ip_port['ip']
 

{atomicshop-3.2.1.dist-info → atomicshop-3.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atomicshop
-Version: 3.2.1
+Version: 3.2.3
 Summary: Atomic functions and classes to make developer life easier
 Author: Denis Kras
 License: MIT License

{atomicshop-3.2.1.dist-info → atomicshop-3.2.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=0QDVialFRk2pBIU7G-tJb5_vrVqmvG0Grp6n90s_aPg,122
+atomicshop/__init__.py,sha256=E9biES7HGkFT33WIn-3TddAU3I0nK3vpYHlxkm5MdD8,122
 atomicshop/_basics_temp.py,sha256=6cu2dd6r2dLrd1BRNcVDKTHlsHs_26Gpw8QS6v32lQ0,3699
 atomicshop/_create_pdf_demo.py,sha256=Yi-PGZuMg0RKvQmLqVeLIZYadqEZwUm-4A9JxBl_vYA,3713
 atomicshop/_patch_import.py,sha256=ENp55sKVJ0e6-4lBvZnpz9PQCt3Otbur7F6aXDlyje4,6334
@@ -134,7 +134,7 @@ atomicshop/file_io/tomls.py,sha256=vZ_Wng5alLf8z6HSEZj7PS0XKDA-Iies9ihVWOkTcKo,1
 atomicshop/file_io/xlsxs.py,sha256=v_dyg9GD4LqgWi6wA1QuWRZ8zG4ZwB6Dz52ytdcmmmI,2184
 atomicshop/file_io/xmls.py,sha256=zh3SuK-dNaFq2NDNhx6ivcf4GYCfGM8M10PcEwDSpxk,2104
 atomicshop/mitm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/mitm/config_static.py,sha256=6vAuksHxZhwejwqL0S66mcI95Fe43I5dsjLYA-wDZ-I,8474
+atomicshop/mitm/config_static.py,sha256=9lnSmGuNKUnN0-IskhObA0XJeLvm63vXD2QiNAUTIb8,8751
 atomicshop/mitm/config_toml_editor.py,sha256=2p1CMcktWRR_NW-SmyDwylu63ad5e0-w1QPMa8ZLDBw,1635
 atomicshop/mitm/connection_thread_worker.py,sha256=E_DNvP4lbbcoeFziiG10Aa7bg7Ka09wdwV_uqaRm5rU,32554
 atomicshop/mitm/import_config.py,sha256=_RJPxoVEEsJQwQY-K_8qPw2PbzsDcV23_sEJEWSKqXY,17831
@@ -145,7 +145,7 @@ atomicshop/mitm/recs_files.py,sha256=ZAAD0twun-FtmbSniXe3XQhIlawvANNB_HxwbHj7kwI
 atomicshop/mitm/shared_functions.py,sha256=0lzeyINd44sVEfFbahJxQmz6KAMWbYrW5ou3UYfItvw,1777
 atomicshop/mitm/statistic_analyzer.py,sha256=5_sAYGX2Xunzo_pS2W5WijNCwr_BlGJbbOO462y_wN4,27533
 atomicshop/mitm/engines/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-atomicshop/mitm/engines/create_module_template.py,sha256=_1aoB-ety98szV__2jnn6cohxmrClU5mOnf6u-nq_Zc,5762
+atomicshop/mitm/engines/create_module_template.py,sha256=VXGicn4S136GezFFvB9YuFteDkR2srUyA_O2E3XmogQ,5766
 atomicshop/mitm/engines/create_module_template_main_example.py,sha256=LeQ44Rp2Gi_KbIDY_4OMS0odkSK3zFZWra_oAka5eJY,243
 atomicshop/mitm/engines/__parent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/mitm/engines/__parent/parser___parent.py,sha256=HHaCXhScl3OlPjz6eUxsDpJaZyk6BNuDMc9xCkeo2Ws,661
@@ -155,8 +155,8 @@ atomicshop/mitm/engines/__parent/responder___parent.py,sha256=mtiS_6ej9nxT9UhAQR
 atomicshop/mitm/engines/__reference_general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/mitm/engines/__reference_general/parser___reference_general.py,sha256=57MEPZMAjTO6xBDZ-yt6lgGJyqRrP0Do5Gk_cgCiPns,2998
 atomicshop/mitm/engines/__reference_general/recorder___reference_general.py,sha256=El2_YHLoHUCiKfkAmGlXxtFpmSjsUFdsb8I1MvSAFaM,653
-atomicshop/mitm/engines/__reference_general/requester___reference_general.py,sha256=yXw8cJ5_0XlupnEtEhqyOR4rvQEAg68Ok39vKa8vKZY,1949
-atomicshop/mitm/engines/__reference_general/responder___reference_general.py,sha256=zVMmON1sy6HH-A35uZZuAURBMfKS5X9ivkPVjPl-sCw,9654
+atomicshop/mitm/engines/__reference_general/requester___reference_general.py,sha256=aYLrLZYvMdX9u7NzYc03GviFcyhusKqUatZ8tKrkaLM,1992
+atomicshop/mitm/engines/__reference_general/responder___reference_general.py,sha256=Q0XKe_dP4Xm3BrUgCn-rUr_n7cjZCkol4moz_Ice0HM,9697
 atomicshop/mitm/statistic_analyzer_helper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/mitm/statistic_analyzer_helper/analyzer_helper.py,sha256=pk6L1t1ea1kvlBoR9QEJptOmaX-mumhwLsP2GCKukbk,5920
 atomicshop/mitm/statistic_analyzer_helper/moving_average_helper.py,sha256=UnnY_FSTiXEfZ8SkDKU2s2qpgPYu1oOT99ghmY-zzas,19992
@@ -317,13 +317,13 @@ atomicshop/wrappers/pywin32w/wmis/msft_netipaddress.py,sha256=Kn2gQsyEquM5QQkdF2
 atomicshop/wrappers/pywin32w/wmis/win32_networkadapterconfiguration.py,sha256=CnHFac1FT6cnH1VwHUyD6vZr6FKPxY8fpbj-1s2_Mzg,11327
 atomicshop/wrappers/pywin32w/wmis/win32networkadapter.py,sha256=iao4OK6u097uRacw6vQuya1kOtJzLA1IM4cgo6UCWLM,3941
 atomicshop/wrappers/pywin32w/wmis/win32process.py,sha256=qMzXtJ5hBZ5ydAyqpDbSx0nO2RJQL95HdmV5SzNKMhk,6826
-atomicshop/wrappers/pywin32w/wmis/wmi_helpers.py,sha256=3SA6hNgw-iBWYl3l1t3RBxRYB4BH-FbTs9gzmcg_Q8w,8710
+atomicshop/wrappers/pywin32w/wmis/wmi_helpers.py,sha256=Ng5pbWeQBNcPqfFuxHEIdkIUeSDAi8w-NjclXlQHfo8,8662
 atomicshop/wrappers/socketw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/socketw/accepter.py,sha256=4I9ORugRDvwaqSzm_gWSjZnRwQGY8hDTlCdsYHwH_ZE,2377
 atomicshop/wrappers/socketw/base.py,sha256=EcosGkD8VzgBY3GeIHDSG29ThQfXwg3-GQPmBTAqTdw,3048
 atomicshop/wrappers/socketw/certificator.py,sha256=mtWPJ_ew3OSwt0-1W4jaoco1VIY4NRCrMv3mDUxb_Cc,12418
 atomicshop/wrappers/socketw/creator.py,sha256=LGI4gcgJ47thx6f96rjwjPz3CsTAIv6VxWFY4EyUF2E,13667
-atomicshop/wrappers/socketw/dns_server.py,sha256=2Y0yE1jIwv--wgNG8SnleqHttPjnKRksndVE_e0CSGM,53806
+atomicshop/wrappers/socketw/dns_server.py,sha256=aWrO16-c9Zt5SnVnmTUEv_j7XENYVBz1udIW9qMeC18,54011
 atomicshop/wrappers/socketw/exception_wrapper.py,sha256=qW_1CKyPgGlsIt7_jusKkMV4A4hih4bX324u0PLnoO8,7382
 atomicshop/wrappers/socketw/get_process.py,sha256=aJC-_qFUv3NgWCSUzDI72E4z8_-VTZE9NVZ0CwUoNlM,5698
 atomicshop/wrappers/socketw/receiver.py,sha256=9B3MvcDqr4C3x2fsnjG5SQognd1wRqsBgikxZa0wXG8,8243
@@ -337,8 +337,8 @@ atomicshop/wrappers/socketw/statistics_csv.py,sha256=WcNyaqEZ82S5-f3kzqi1nllNT2N
 atomicshop/wrappers/winregw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 atomicshop/wrappers/winregw/winreg_installed_software.py,sha256=Qzmyktvob1qp6Tjk2DjLfAqr_yXV0sgWzdMW_9kwNjY,2345
 atomicshop/wrappers/winregw/winreg_network.py,sha256=ih0BVNwByLvf9F_Lac4EdmDYYJA3PzMvmG0PieDZrsE,9905
-atomicshop-3.2.1.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-3.2.1.dist-info/METADATA,sha256=qmcM6sJS_q1IivsuhbARaKxfuqkZ5eYfSxpRCfN_oQ0,10670
-atomicshop-3.2.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-atomicshop-3.2.1.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-3.2.1.dist-info/RECORD,,
+atomicshop-3.2.3.dist-info/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-3.2.3.dist-info/METADATA,sha256=Nx2nweBCrNFQkSYKA7MHTIo6UU60pt6PD-UhLLfFvgg,10670
+atomicshop-3.2.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+atomicshop-3.2.3.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-3.2.3.dist-info/RECORD,,