atomicshop 3.3.23__py3-none-any.whl → 3.3.25__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.3.23'
+__version__ = '3.3.25'

atomicshop/wrappers/githubw.py

@@ -21,7 +21,7 @@ class GitHubWrapper:
             user_name: str = None,
             repo_name: str = None,
             repo_url: str = None,
-            branch: str = 'master',
+            branch: str = 'main',
             path: str = None,
             pat: str = None,
             branch_file_extension: Literal[

atomicshop/wrappers/socketw/socket_wrapper.py

@@ -1,5 +1,7 @@
 import multiprocessing
 import threading
+import time
+
 import select
 from typing import Literal, Union, Callable, Any
 from pathlib import Path
@@ -14,7 +16,7 @@ from ..loggingw import loggingw
 from ...script_as_string_processor import ScriptAsStringProcessor
 from ...permissions import permissions
 from ... import filesystem, certificates
-from ...basics import booleans
+from ...basics import booleans, tracebacks
 from ...print_api import print_api
 
 from . import base, creator, get_process, accepter, statistics_csv, ssl_base, sni
@@ -498,6 +500,13 @@ class SocketWrapper:
         listening_sockets: list = [listening_socket_object]
 
         while True:
+            engine_name: str = ''
+            source_ip: str = ''
+            source_hostname: str = ''
+            dest_port: int = 0
+            process_name: str = ''
+            domain_from_engine: str = ''
+
             try:
                 # Using "select.select" which is currently the only API function that works on all
                 # operating system types: Windows / Linux / BSD.
@@ -508,7 +517,6 @@ class SocketWrapper:
 
                 listening_ip, listening_port = listening_socket_object.getsockname()
 
-                domain_from_engine = None
                 # Get the domain to connect on this process in case on no SNI provided.
                 for domain, ip_port_dict in self.engine.domain_target_dict.items():
                     if ip_port_dict['ip'] == listening_ip:
@@ -534,16 +542,20 @@ class SocketWrapper:
 
                 self.logger.info(f"Requested domain setting: {domain_from_engine}")
 
+                engine_name = get_engine_name(domain_from_engine, [self.engine])
+
                 # Wait from any connection on "accept()".
                 # 'client_socket' is socket or ssl socket, 'client_address' is a tuple (ip_address, port).
                 client_socket, client_address, accept_error_message = accepter.accept_connection_with_error(
                     listening_socket_object, domain_from_dns_server=domain_from_engine,
                     print_kwargs={'logger': self.logger})
 
+                source_ip: str = client_address[0]
+                dest_port: int = listening_socket_object.getsockname()[1]
+
                 # This is the earliest stage to ask for process name.
                 # SSH Remote / LOCALHOST script execution to identify process section.
                 # If 'get_process_name' was set to True, then this will be executed.
-                process_name = None
                 if self.get_process_name:
                     # Get the process name from the socket.
                     get_command_instance = get_process.GetCommandLine(
@@ -554,21 +566,20 @@ class SocketWrapper:
                         logger=self.logger)
                     process_name = get_command_instance.get_process_name(print_kwargs={'logger': self.logger})
 
-                source_ip: str = client_address[0]
-                engine_name: str = get_engine_name(domain_from_engine, [self.engine])
-                dest_port: int = listening_socket_object.getsockname()[1]
-
                 # Not always there will be a hostname resolved by the IP address, so we will leave it empty if it fails.
                 try:
-                    source_hostname: str = socket.gethostbyaddr(source_ip)[0]
+                    source_hostname = socket.gethostbyaddr(source_ip)[0]
                 except socket.herror:
-                    source_hostname = ''
+                    pass
 
                 # If 'accept()' function worked well, SSL worked well, then 'client_socket' won't be empty.
                 if client_socket:
                     # Get the protocol type from the socket.
                     is_tls: bool = False
+
+                    time.sleep(1)  # Wait a second to gather some data.
                     tls_properties = ssl_base.is_tls(client_socket)
+
                     if tls_properties:
                         is_tls = True
                         tls_type, tls_version = tls_properties
@@ -681,8 +692,22 @@ class SocketWrapper:
                         dest_port=str(dest_port),
                         host=domain_from_engine,
                         process_name=process_name)
+            except ConnectionResetError as e:
+                exception_string: str = tracebacks.get_as_string()
+                full_string: str = f"{str(e)} | {exception_string}"
+                self.statistics_writer.write_accept_error(
+                    engine=engine_name,
+                    source_host=source_hostname,
+                    source_ip=source_ip,
+                    error_message=full_string,
+                    dest_port=str(dest_port),
+                    host=domain_from_engine,
+                    process_name=process_name)
             except Exception as e:
-                self.exceptions_logger.write(e)
+                _ = e
+                exception_string: str = tracebacks.get_as_string()
+                full_string: str = f"Engine: {engine_name} | {exception_string}"
+                self.exceptions_logger.write(full_string)
 
 
 def before_socket_thread_worker(

{atomicshop-3.3.23.dist-info → atomicshop-3.3.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomicshop
-Version: 3.3.23
+Version: 3.3.25
 Summary: Atomic functions and classes to make developer life easier
 Author: Denis Kras
 License-Expression: MIT
@@ -8,7 +8,7 @@ Project-URL: Homepage, https://github.com/BugSec-Official/atomicshop
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.10
+Requires-Python: <3.13,>=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
 Requires-Dist: wheel

{atomicshop-3.3.23.dist-info → atomicshop-3.3.25.dist-info}/RECORD

@@ -1,4 +1,4 @@
-atomicshop/__init__.py,sha256=SVEyQmBJFUE371lRYyRuD9Bi1k7XEA5UIZulvlGbD8k,123
+atomicshop/__init__.py,sha256=peqNB9IL7p86gZd32BF4w1uBeAZvZo0dOfoMRX3xB5U,123
 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
@@ -187,7 +187,7 @@ atomicshop/wrappers/astw.py,sha256=VkYfkfyc_PJLIOxByT6L7B8uUmKY6-I8XGZl4t_z828,4
 atomicshop/wrappers/configparserw.py,sha256=JwDTPjZoSrv44YKwIRcjyUnpN-FjgXVfMqMK_tJuSgU,22800
 atomicshop/wrappers/cryptographyw.py,sha256=QEUpDn8vUvMg3ADz6-4oC2kbDNC_woDlw7C0zU7qFVM,14233
 atomicshop/wrappers/ffmpegw.py,sha256=wcq0ZnAe0yajBOuTKZCCaKI7CDBjkq7FAgdW5IsKcVE,6031
-atomicshop/wrappers/githubw.py,sha256=PpG0U1MRoDY2LLYpLxnoUbSb8Pz9d47HtwBnOwxMaRY,27550
+atomicshop/wrappers/githubw.py,sha256=BEEJuWd1eBfiv1FSgb76D4nEBRpFU9j37lztn-KbYa4,27548
 atomicshop/wrappers/netshw.py,sha256=8WE_576XiiHykwFuE-VkCx5CydMpFlztX4frlEteCtI,6350
 atomicshop/wrappers/numpyw.py,sha256=sBV4gSKyr23kXTalqAb1oqttzE_2XxBooCui66jbAqc,1025
 atomicshop/wrappers/olefilew.py,sha256=biD5m58rogifCYmYhJBrAFb9O_Bn_spLek_9HofLeYE,2051
@@ -308,14 +308,14 @@ atomicshop/wrappers/socketw/sender.py,sha256=aX_K8l_rHjd5AWb8bi5mt8-YTkMYVRDB6Dn
 atomicshop/wrappers/socketw/sni.py,sha256=uj6KKYKmSrzXcKBhVLaHQhYn1wNfIUpdnmcvn21V9iE,18176
 atomicshop/wrappers/socketw/socket_client.py,sha256=WWIiCxUX9irN9aWzJ6-1xrXNB_iv_diq3ha1yrWsNGU,22671
 atomicshop/wrappers/socketw/socket_server_tester.py,sha256=Qobmh4XV8ZxLUaw-eW4ESKAbeSLecCKn2OWFzMhadk0,6420
-atomicshop/wrappers/socketw/socket_wrapper.py,sha256=8tRoccWvnDNXaK2iYsT68RyxHHjILdPjw2R7SxPC6go,41621
+atomicshop/wrappers/socketw/socket_wrapper.py,sha256=bSFRfmzlfOsqb0jBLGsdCxDkZl6L0Si5QJZkShjOxEM,42589
 atomicshop/wrappers/socketw/ssl_base.py,sha256=62-hPm7zla1rh3m_WvDnXqKH-sDUTdiRptD8STCkgdk,2313
 atomicshop/wrappers/socketw/statistics_csv.py,sha256=_gA8bMX6Sw_UCXKi2y9wNAwlqifgExgDGfQIa9pFxQA,5543
 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.3.23.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
-atomicshop-3.3.23.dist-info/METADATA,sha256=xbmLVSZnBgpxy0Q-fUDt--RFuw8RANnPINcEeZZwg3U,9312
-atomicshop-3.3.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-atomicshop-3.3.23.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
-atomicshop-3.3.23.dist-info/RECORD,,
+atomicshop-3.3.25.dist-info/licenses/LICENSE.txt,sha256=lLU7EYycfYcK2NR_1gfnhnRC8b8ccOTElACYplgZN88,1094
+atomicshop-3.3.25.dist-info/METADATA,sha256=VAymfm9H0qFDtFWy28IxXTtb6CGaePBQVC4QkN7nGv8,9318
+atomicshop-3.3.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+atomicshop-3.3.25.dist-info/top_level.txt,sha256=EgKJB-7xcrAPeqTRF2laD_Np2gNGYkJkd4OyXqpJphA,11
+atomicshop-3.3.25.dist-info/RECORD,,