Update main.py

main.py

@@ -107,10 +107,11 @@ def index():
 
 @app.route('/generate', methods=['POST'])
 def generate_script():
+    # Step 7: Generate the randomized assembly information using meaningful words
     assembly_info = {
         'title': random.choice(titles),
         'description': random.choice(descriptions),
-        'configuration': '',
+        'configuration': '',  # Can leave empty
         'company': random.choice(companies),
         'product': "MyProduct",
         'copyright': f"Copyright © {random.choice(companies)} 2024",
@@ -120,6 +121,7 @@ def generate_script():
         'informational_version': random_version()
     }
 
+    # Step 8: Replace placeholders in the base template
     modified_cs = base_cs_template.replace('<<title>>', assembly_info['title']) \
                                   .replace('<<description>>', assembly_info['description']) \
                                   .replace('<<configuration>>', assembly_info['configuration']) \
@@ -134,42 +136,30 @@ def generate_script():
                                   .replace('<<additional_obfuscated_code>>', generate_additional_obfuscated_code()) \
                                   .replace('<<obfuscated_methods>>', generate_obfuscated_methods())
 
-    # Save the modified C# script to a temporary project folder
-    project_name = 'PolymorphicProgram'
-    os.makedirs(project_name, exist_ok=True)
-    script_path = os.path.join(project_name, 'Program.cs')
+    # Save the modified C# script to a file
+    script_path = 'polymorphic_program.cs'
     with open(script_path, 'w') as file:
         file.write(modified_cs)
 
-    # Create a project file
-    project_file_content = f"""
-<Project Sdk="Microsoft.NET.Sdk">
-  <PropertyGroup>
-    <OutputType>WinExe</OutputType>
-    <TargetFramework>net7.0</TargetFramework>
-  </PropertyGroup>
-</Project>
-"""
-    project_file_path = os.path.join(project_name, f"{project_name}.csproj")
-    with open(project_file_path, 'w') as file:
-        file.write(project_file_content)
-
-# Step 10: Compile the C# script using dotnet
-    compile_command = f'dotnet build {script_path.replace(".cs", ".csproj")} -c Release -o /app'
-    
+    # Step 10: Locate csc and compile the C# script
     try:
-        result = subprocess.run(compile_command, shell=True, check=True, text=True, capture_output=True)
-        print(f"Compilation succeeded: {result.stdout}")  # Log the successful output
-    except subprocess.CalledProcessError as e:
-        print(f"Compilation failed with return code {e.returncode}")  # Log the return code
-        print(f"stdout: {e.stdout}")  # Log the standard output
-        print(f"stderr: {e.stderr}")  # Log the standard error
-        return f"Compilation failed: {e.stderr.strip()}", 500
+        # Find the path to csc
+        csc_path = subprocess.run(['which', 'csc'], capture_output=True, text=True, check=True).stdout.strip()
+    except subprocess.CalledProcessError:
+        return "csc not found. Make sure the .NET SDK is installed.", 500
+
+    # Compile the C# script
+    compile_command = f'{csc_path} /target:winexe /platform:x64 /out:run.exe {script_path} /win32icon:app.ico /win32manifest:app.manifest'
+    compile_result = subprocess.run(compile_command, shell=True, check=True, capture_output=True)
+
+    if compile_result.returncode != 0:
+        return f"Compilation failed: {compile_result.stderr.decode()}", 500
 
     # Provide a link to download the compiled executable
     return send_file('run.exe', as_attachment=True)
 
 
+
 # Start the Flask app
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)