Draft

davidteather · davidteather · commit 463a6d892f28 · 2020-12-04T00:55:32.000-06:00
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,4 @@
+test.py
+__pycache__
+dist
+build
diff --git a/README.md b/README.md
diff --git a/python_obfuscator.egg-info/PKG-INFO b/python_obfuscator.egg-info/PKG-INFO
@@ -0,0 +1,24 @@
+Metadata-Version: 2.1
+Name: python-obfuscator
+Version: 0.0.1
+Summary: It's a python obfuscator.
+Home-page: https://github.com/davidteather/tiktok-api
+Author: David Teather
+Author-email: contact.davidteather@gmail.com
+License: MIT
+Download-URL: https://github.com/davidteather/TikTok-Api/tarball/master
+Description: UNKNOWN
+Keywords: obfuscator
+Platform: UNKNOWN
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Build Tools
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.3
+Classifier: Programming Language :: Python :: 3.4
+Classifier: Programming Language :: Python :: 3.5
+Classifier: Programming Language :: Python :: 3.6
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Description-Content-Type: text/markdown
diff --git a/python_obfuscator.egg-info/SOURCES.txt b/python_obfuscator.egg-info/SOURCES.txt
@@ -0,0 +1,11 @@
+README.md
+setup.py
+python_obfuscator/__init__.py
+python_obfuscator/cli.py
+python_obfuscator/obfuscator.py
+python_obfuscator/utilities.py
+python_obfuscator.egg-info/PKG-INFO
+python_obfuscator.egg-info/SOURCES.txt
+python_obfuscator.egg-info/dependency_links.txt
+python_obfuscator.egg-info/entry_points.txt
+python_obfuscator.egg-info/top_level.txt
diff --git a/python_obfuscator.egg-info/dependency_links.txt b/python_obfuscator.egg-info/dependency_links.txt
@@ -0,0 +1 @@
+
diff --git a/python_obfuscator.egg-info/entry_points.txt b/python_obfuscator.egg-info/entry_points.txt
@@ -0,0 +1,3 @@
+[console_scripts]
+python-obfuscator = python_obfuscator.cli:main
+
diff --git a/python_obfuscator.egg-info/top_level.txt b/python_obfuscator.egg-info/top_level.txt
@@ -0,0 +1 @@
+python_obfuscator
diff --git a/python_obfuscator/__init__.py b/python_obfuscator/__init__.py
@@ -0,0 +1 @@
+from .obfuscator import obfuscator
diff --git a/python_obfuscator/cli.py b/python_obfuscator/cli.py
@@ -0,0 +1,17 @@
+import sys
+from .obfuscator import obfuscator
+from .utilities import one_liner
+
+def convert_file(file_path):
+    obfuscate = obfuscator()
+
+    with open(file_path, 'r') as f:
+        data = f.read()
+        obfuscated_data = obfuscate.obfuscate(data)
+
+    with open(file_path, 'w+') as f:
+        f.write(obfuscated_data)
+
+if __name__ == '__main__':
+    python_file = sys.argv[1]
+    convert_file(python_file)
diff --git a/python_obfuscator/helpers/__init__.py b/python_obfuscator/helpers/__init__.py
@@ -0,0 +1,2 @@
+from .variable_name_generator import VariableNameGenerator
+from .random_datatype import RandomDataTypeGenerator
diff --git a/python_obfuscator/helpers/random_datatype.py b/python_obfuscator/helpers/random_datatype.py
@@ -0,0 +1,19 @@
+import random
+import string
+import time
+
+class RandomDataTypeGenerator:
+    def __init__(self):
+        self.generator_options = [self.random_string, self.random_int]
+    
+    def get_random(self):
+        return random.choice(self.generator_options)()
+
+    def random_string(self, length=79):
+        # Why is it 79 by default? 
+        # See: https://stackoverflow.com/a/16920876/11472374
+        # As kirelagin commented readability is very important
+        return ''.join(random.choice(string.ascii_lowercase + string.ascii_uppercase) for i in range(length))
+
+    def random_int(self):
+        return random.randint(random.randint(0,300), random.randint(300,999))
diff --git a/python_obfuscator/helpers/variable_name_generator.py b/python_obfuscator/helpers/variable_name_generator.py
@@ -0,0 +1,32 @@
+import random
+import string
+import time
+
+class VariableNameGenerator:
+    def __init__(self):
+        self.generator_options = [self.random_string, self.l_and_i, self.time_based, self.just_id, self.scream, self.single_letter_a_lot]
+    
+    def get_random(self, id):
+        return random.choice(self.generator_options)(id)
+
+    def random_string(self, id, length=79):
+        # Why is it 79 by default? 
+        # See: https://stackoverflow.com/a/16920876/11472374
+        # As kirelagin commented readability is very important
+        return ''.join(random.choice(string.ascii_lowercase) for i in range(length)) + str(id)
+
+    def l_and_i(self, id):
+        return ''.join(random.choice('Il') for i in range(id))
+
+    def time_based(self, id):
+        return random.choice(string.ascii_letters) + str(time.time()).replace(".", "") + str(id)
+
+    def just_id(self, id):
+        # python doesn't work with numbers for variable names
+        return random.choice(string.ascii_letters) + str(id)
+
+    def scream(self, id):
+        return ''.join(random.choice("Aa") for i in range(id))
+
+    def single_letter_a_lot(self, id):
+        return random.choice(string.ascii_letters) * id
diff --git a/python_obfuscator/obfuscator.py b/python_obfuscator/obfuscator.py
@@ -0,0 +1,9 @@
+import logging
+from .utilities import obfuscate
+
+class obfuscator:
+    def __init__(self, logging_level=logging.error):
+        pass
+
+    def obfuscate(self, code, remove_techniques=[]):
+        return obfuscate(code, remove_techniques)
diff --git a/python_obfuscator/utilities.py b/python_obfuscator/utilities.py
@@ -0,0 +1,72 @@
+import re
+import ast
+import random
+import time
+from .helpers import VariableNameGenerator, RandomDataTypeGenerator
+import regex
+
+def one_liner(code):
+    return """exec(\"\"\"{}\"\"\")""".format(code.replace("\n", ";").replace('"""', '\\"\\"\\"'))
+
+def variable_renamer(code):
+    variable_names = re.findall(r"(\w+)(?=( |)=( |))", code)
+    name_generator = VariableNameGenerator()
+    for i in range(len(variable_names)):
+        obfuscated_name = name_generator.get_random(i+1)
+        code = re.sub(r"\b(" + variable_names[i][0] + r")\b", obfuscated_name, code)
+
+    return code
+
+def add_random_variables(code):
+    useless_variables_to_add = random.randint(100,400)
+    name_generator = VariableNameGenerator()
+    data_generator = RandomDataTypeGenerator()
+    for v in range(useless_variables_to_add):
+        rand_data = data_generator.get_random()
+        if type(rand_data) == str:
+            rand_data = '"{}"'.format(rand_data)
+        if v % 2 == 0:
+            code = "{} = {}\n".format(name_generator.get_random(v), rand_data) + code
+        else:
+            code = code + "\n{} = {}".format(name_generator.get_random(v), rand_data)
+    return code
+
+def str_to_bytes(code):
+    # (?<=(( |	|)\w+( |)=( |))("""|"|'))[\W\w]*?(?=("""|"|'))
+    python_string_decoraters = ['"""', '"', "'"]
+
+    variable_names = []
+    for s in python_string_decoraters:
+        t = regex.findall(r"(?<=(( |	|\n)\w+( |)=( |))(\"\"\"|\"|'))[\W\w]*?(?=(\"\"\"|\"|'))", code)
+        variable_names += t
+        print(t)
+
+    print(variable_names)
+    raise Exception()
+    for name in variable_names:
+        var_value = code.split(name + " = ").split("\n")[0]
+        if var_value[0] == '"':
+            # need to check if """ string
+            if var_value[:3] == '"""':
+                # is a """ string
+                pass
+            else:
+                var_value
+        elif var_value[0] == "'":
+            pass
+    return code
+
+def obfuscate(code, remove_techniques=[]):
+    if len(remove_techniques) == 0:
+        methods = all_methods
+    else:
+        methods = all_methods.copy()
+        for technique in remove_techniques:
+            methods.remove(technique)
+
+    for technique in methods:
+        code = technique(code)
+
+    return code
+
+all_methods = [str_to_bytes, variable_renamer, add_random_variables, one_liner]
diff --git a/setup.py b/setup.py
@@ -0,0 +1,40 @@
+from distutils.core import setup
+import os.path
+import setuptools
+
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+setuptools.setup(
+    name="python_obfuscator",
+    packages=["python_obfuscator"],
+    version="0.0.1",
+    license="MIT",
+    description="It's a python obfuscator.",
+    author="David Teather",
+    author_email="contact.davidteather@gmail.com",
+    url="https://github.com/davidteather/tiktok-api",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    download_url="https://github.com/davidteather/TikTok-Api/tarball/master",
+    keywords=["obfuscator"],
+    install_requires=[],
+    classifiers=[
+        "Development Status :: 3 - Alpha",
+        "Intended Audience :: Developers",
+        "Topic :: Software Development :: Build Tools",
+        "License :: OSI Approved :: MIT License",
+        "Programming Language :: Python :: 3.3",
+        "Programming Language :: Python :: 3.4",
+        "Programming Language :: Python :: 3.5",
+        "Programming Language :: Python :: 3.6",
+        "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+    ],
+    entry_points = {
+        "console_scripts": [
+            "python-obfuscator=python_obfuscator.cli:main"
+        ]
+    }
+)

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +test.py
 +__pycache__
 +dist
 +build
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+[console_scripts]`
	`2`	`+python-obfuscator = python_obfuscator.cli:main`
	`3`	`+`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+from .variable_name_generator import VariableNameGenerator`
	`2`	`+from .random_datatype import RandomDataTypeGenerator`