From d55ed342565f851a8701e3758e94acddeba56939 Mon Sep 17 00:00:00 2001
From: Ivan Maslov <Ivan.Maslov@UnicodeLabs.ru>
Date: Mon, 2 Mar 2020 21:59:38 +0300
Subject: [PATCH] # Created + tested SafeSource!!! You can Create and run
 crypted .py code

---
 Sources/pyOpenRPA/Tools/SafeSource/Crypter.py | 15 +++++------
 .../pyOpenRPA/Tools/SafeSource/DistrCreate.py | 12 +++++----
 .../pyOpenRPA/Tools/SafeSource/DistrRun.py    | 26 +++++++++++++++++--
 v1.0.41 => v1.0.42                            |  0
 4 files changed, 37 insertions(+), 16 deletions(-)
 rename v1.0.41 => v1.0.42 (100%)

diff --git a/Sources/pyOpenRPA/Tools/SafeSource/Crypter.py b/Sources/pyOpenRPA/Tools/SafeSource/Crypter.py
index 6a44c717..3921c79b 100644
--- a/Sources/pyOpenRPA/Tools/SafeSource/Crypter.py
+++ b/Sources/pyOpenRPA/Tools/SafeSource/Crypter.py
@@ -66,17 +66,14 @@ def decrypt_file_bytes(key, in_filename, chunksize=24*1024):
         заданным ключом.
     """
     lResult = b''
-    if not out_filename:
-        out_filename = os.path.splitext(in_filename)[0]
     with open(in_filename, 'rb') as infile:
         origsize = struct.unpack('<Q', infile.read(struct.calcsize('Q')))[0]
         iv = infile.read(16)
         decryptor = AES.new(key, AES.MODE_CBC, iv)
-        with open(out_filename, 'wb') as outfile:
-            while True:
-                chunk = infile.read(chunksize)
-                if len(chunk) == 0:
-                    break
-                lResult = lResult+decryptor.decrypt(chunk)
-            lResult=lResult[0:origsize-1]
+        while True:
+            chunk = infile.read(chunksize)
+            if len(chunk) == 0:
+                break
+            lResult = lResult+decryptor.decrypt(chunk)
+        lResult=lResult[0:origsize]
     return lResult
\ No newline at end of file
diff --git a/Sources/pyOpenRPA/Tools/SafeSource/DistrCreate.py b/Sources/pyOpenRPA/Tools/SafeSource/DistrCreate.py
index d8d4f322..94a758db 100644
--- a/Sources/pyOpenRPA/Tools/SafeSource/DistrCreate.py
+++ b/Sources/pyOpenRPA/Tools/SafeSource/DistrCreate.py
@@ -43,11 +43,13 @@ def Run():
     lKeyHashStr_2 = hashlib.sha256(pyautogui.password('Please repeat the key to protect source code').encode("utf-8")).digest()
     if lKeyHashStr_1 == lKeyHashStr_2:
         for lFileItem in lPyFileList:
-            #Create right \\ splashes
-            lFileItem = os.path.abspath(lFileItem)
-            Crypter.encrypt_file(lKeyHashStr_1, lFileItem, f"{lFileItem[0:-2]}{gOutCryptedExtension}")
-            #Remove old file
-            os.remove(lFileItem)
+            #Hotfix - dont encrypt __main__.py to __main__.cry - python dont see it
+            if True:#"__main__.py" not in lFileItem:
+                #Create right \\ splashes
+                lFileItem = os.path.abspath(lFileItem)
+                Crypter.encrypt_file(lKeyHashStr_1, lFileItem, f"{lFileItem[0:-2]}{gOutCryptedExtension}")
+                #Remove old file
+                os.remove(lFileItem)
     else:
         raise Exception("User set different secret key 1 and key 2")
     ############ Step 6 - Final stage
diff --git a/Sources/pyOpenRPA/Tools/SafeSource/DistrRun.py b/Sources/pyOpenRPA/Tools/SafeSource/DistrRun.py
index 3e7d2b40..999ef827 100644
--- a/Sources/pyOpenRPA/Tools/SafeSource/DistrRun.py
+++ b/Sources/pyOpenRPA/Tools/SafeSource/DistrRun.py
@@ -5,6 +5,9 @@ import sys
 import imp
 import base64
 from . import Crypter # Crypto functions
+import datetime #Datetime
+import hashlib #Get hash of the word
+import pyautogui
 EXT = '.cry'
 
 # How to run
@@ -87,7 +90,9 @@ class Base64Importer(object):
             src = ''
 
         return src
-
+    # for __main__
+    def get_code(self, inModName):
+        return self.get_source(inModName)
     #---------------------------------------------------------------------------
     def __collect_modules_info(self, root_package_path):
         """Собирает информацию о модулях из указанного пакета
@@ -125,14 +130,31 @@ class Base64Importer(object):
 gInUncryptedExtension = "py" # cry for filename.cry
 gOutCryptedExtension = "cry" # cry for filename.cry
 gFileMaskToDelete = "pyc" #Remove all .pyc files
+
+#Settings to run crypted module
+gSettings = {
+    #runpy
+    "run_module": {
+        "mod_name": "TestPackage", #"TestPackage",
+        "init_globals": None,
+        "run_name": None,
+        "alter_sys": True
+    }
+}
+
 # Create process run
 def Run():
+    global gSettings
     print(f"{str(datetime.datetime.now())}: Run decryptography")
     ############# Step 5 - Ask and confirm the secret word
     lKeyHashStr_1 = hashlib.sha256(pyautogui.password('Please enter the key to protect source code').encode("utf-8")).digest()
     lKeyHashStr_2 = hashlib.sha256(pyautogui.password('Please repeat the key to protect source code').encode("utf-8")).digest()
     if lKeyHashStr_1 == lKeyHashStr_2:
-        CryptographerInit("", inKey = lKeyHashStr_1)
+        sys.meta_path.append(Base64Importer("TestPackage",inKey = lKeyHashStr_1))
+        #CryptographerInit("", inKey = lKeyHashStr_1)
+        #Init the functions
+        import runpy
+        runpy.run_module(**gSettings["run_module"])
     else:
         raise Exception("User set different secret key 1 and key 2")
     ############ Step 6 - Final stage
diff --git a/v1.0.41 b/v1.0.42
similarity index 100%
rename from v1.0.41
rename to v1.0.42