python-obfuscation-framework 1.4.1__py3-none-any.whl

pof/utils/generator/base.py

@@ -0,0 +1,178 @@
+# TODO (deoktr): store BUILTINS in another file to share it
+BUILTINS = (
+    "__file__",
+    "__name__",
+    "__doc__",
+    "__package__",
+    "__loader__",
+    "__spec__",
+    "__build_class__",
+    "__import__",
+    "abs",
+    "all",
+    "any",
+    "ascii",
+    "bin",
+    "breakpoint",
+    "callable",
+    "chr",
+    "compile",
+    "delattr",
+    "dir",
+    "divmod",
+    "eval",
+    "exec",
+    "format",
+    "getattr",
+    "globals",
+    "hasattr",
+    "hash",
+    "hex",
+    "id",
+    "input",
+    "isinstance",
+    "issubclass",
+    "iter",
+    "aiter",
+    "len",
+    "locals",
+    "max",
+    "min",
+    "next",
+    "anext",
+    "oct",
+    "ord",
+    "pow",
+    "print",
+    "repr",
+    "round",
+    "setattr",
+    "sorted",
+    "sum",
+    "vars",
+    "None",
+    "Ellipsis",
+    "NotImplemented",
+    "False",
+    "True",
+    "bool",
+    "memoryview",
+    "bytearray",
+    "bytes",
+    "classmethod",
+    "complex",
+    "dict",
+    "enumerate",
+    "filter",
+    "float",
+    "frozenset",
+    "property",
+    "int",
+    "list",
+    "map",
+    "object",
+    "range",
+    "reversed",
+    "set",
+    "slice",
+    "staticmethod",
+    "str",
+    "super",
+    "tuple",
+    "type",
+    "zip",
+    "__debug__",
+    "BaseException",
+    "Exception",
+    "TypeError",
+    "StopAsyncIteration",
+    "StopIteration",
+    "GeneratorExit",
+    "SystemExit",
+    "KeyboardInterrupt",
+    "ImportError",
+    "ModuleNotFoundError",
+    "OSError",
+    "EnvironmentError",
+    "IOError",
+    "EOFError",
+    "RuntimeError",
+    "RecursionError",
+    "NotImplementedError",
+    "NameError",
+    "UnboundLocalError",
+    "AttributeError",
+    "SyntaxError",
+    "IndentationError",
+    "TabError",
+    "LookupError",
+    "IndexError",
+    "KeyError",
+    "ValueError",
+    "UnicodeError",
+    "UnicodeEncodeError",
+    "UnicodeDecodeError",
+    "UnicodeTranslateError",
+    "AssertionError",
+    "ArithmeticError",
+    "FloatingPointError",
+    "OverflowError",
+    "ZeroDivisionError",
+    "SystemError",
+    "ReferenceError",
+    "MemoryError",
+    "BufferError",
+    "Warning",
+    "UserWarning",
+    "EncodingWarning",
+    "DeprecationWarning",
+    "PendingDeprecationWarning",
+    "SyntaxWarning",
+    "RuntimeWarning",
+    "FutureWarning",
+    "ImportWarning",
+    "UnicodeWarning",
+    "BytesWarning",
+    "ResourceWarning",
+    "ConnectionError",
+    "BlockingIOError",
+    "BrokenPipeError",
+    "ChildProcessError",
+    "ConnectionAbortedError",
+    "ConnectionRefusedError",
+    "ConnectionResetError",
+    "FileExistsError",
+    "FileNotFoundError",
+    "IsADirectoryError",
+    "NotADirectoryError",
+    "InterruptedError",
+    "PermissionError",
+    "ProcessLookupError",
+    "TimeoutError",
+    "open",
+    "quit",
+    "exit",
+    "copyright",
+    "credits",
+    "license",
+    "help",
+)
+
+RESERVED_WORDS = (
+    "__init__",
+    "__eq__",
+    "__lt__",
+    "append",  # on list
+    "update",  # on dict
+    "copy",  # copy dict or list
+    "join",  # on string "".join()
+    # TODO (deoktr): add all the others
+)
+
+
+class BaseGenerator:
+    RESERVED: list[str] = []  # noqa: RUF012
+
+    @classmethod
+    def extend_reserved(cls, extension):
+        cls.RESERVED.extend(extension)

pof/utils/generator/basic.py

@@ -0,0 +1,107 @@
+"""Basic random names generators."""
+
+import random
+
+from .base import BaseGenerator
+
+
+class BasicGenerator(BaseGenerator):
+    @classmethod
+    def alphabet_generator(
+        cls,
+        min_length: int = 3,
+        max_length: int = 10,
+        chars: str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890__",
+        first_chars: str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ",
+    ):
+        """Random words that can be variables."""
+        previous = []
+        if first_chars is None:
+            first_chars = chars
+        while True:
+            if min_length != max_length:
+                length = random.randint(min_length, max_length)
+            else:
+                length = max_length
+            name = random.choice(first_chars)
+            for _ in range(length - 1):
+                name += random.choice(chars)
+            if name in previous or name in cls.RESERVED:
+                continue
+            previous.append(name)
+            yield name
+
+    @classmethod
+    def number_name_generator(cls, length=1, prefix="_"):
+        """Generator for endless number of variable name."""
+        numbers = list(range(10**length, 10 ** (length + 1)))
+        while True:
+            if len(numbers) == 0:
+                length += 1
+                numbers = list(range(10**length, 10 ** (length + 1)))
+            n = random.choice(numbers)
+            numbers.remove(n)
+            yield prefix + str(n)
+
+    @classmethod
+    def single_symbol_generator(cls, symbol="_"):
+        # TODO (deoktr): generate a couple in a list, randomize the order and take them
+        # instead of increasing it by one in order
+        length = 1
+        while True:
+            length += 1
+            yield symbol * length
+
+    @classmethod
+    def pointer_generator(
+        cls,
+        name_format="_0x{}",
+        length=12,
+        chars="0123456789abcdef",
+    ):
+        """Generate var names in the format of hexadecimal."""
+        ag = cls.alphabet_generator(
+            min_length=length,
+            max_length=length,
+            chars=chars,
+            first_chars=chars,
+        )
+        while True:
+            name = next(ag)
+            yield name_format.format(name)
+
+    @classmethod
+    def function_generator(
+        cls,
+        *args,
+        **kwargs,
+    ):
+        """Generate class names."""
+        chars: str = "abcdefghijklmnopqrstuvwxyz__"
+        first_chars: str = "abcdefghijklmnopqrstuvwxyz"
+        ag = cls.alphabet_generator(
+            *args,
+            chars=chars,
+            first_chars=first_chars,
+            **kwargs,
+        )
+        while True:
+            yield next(ag)
+
+    @classmethod
+    def class_generator(
+        cls,
+        *args,
+        **kwargs,
+    ):
+        """Generate class names."""
+        chars: str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+        first_chars: str = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+        ag = cls.alphabet_generator(
+            *args,
+            chars=chars,
+            first_chars=first_chars,
+            **kwargs,
+        )
+        while True:
+            yield next(ag)