From 23bb16d1728049320ba49a4c3efc11dbb2cfa81f Mon Sep 17 00:00:00 2001
From: Dobin <dobin.rutishauser@gmail.com>
Date: Mon, 29 Apr 2024 18:38:33 +0100
Subject: [PATCH] ui: make a bit nicer, highlight missing dlls

---
 app/templates/project.html | 54 ++++++++++++++++++++------------------
 app/views_project.py       |  7 +++++
 2 files changed, 35 insertions(+), 26 deletions(-)

diff --git a/app/templates/project.html b/app/templates/project.html
index d3eb4b0..2c0642b 100644
--- a/app/templates/project.html
+++ b/app/templates/project.html
@@ -12,7 +12,7 @@
 
         <div class="row">
             <!-- Row 1: Buttons -->
-            <div class="col-3">
+            <div class="col-1">
                 <div class="row">
                     <div class="col-6">
                     <form method="POST" enctype="multipart/form-data" action="/project/{{project_name}}/build">
@@ -23,7 +23,6 @@
                 
                 {% if is_built %}
                 <div class="row">
-                    <div class="col-6">
                         <form method="POST" enctype="multipart/form-data" action="/project/{{project_name}}/exec">
                             <button class="btn btn-primary" type="submit" value="start">Start Local</button>
                         </form>
@@ -31,22 +30,21 @@
                         <form method="POST" enctype="multipart/form-data" action="/project/{{project_name}}/exec?no_exec=true">
                             <button class="btn btn-primary" type="submit" value="start">File Local</button>
                         </form>
-                    </div>
-                    <div class="col-6">
+
+                        {% if has_remote %}
                         <form method="POST" enctype="multipart/form-data" action="/project/{{project_name}}/exec?remote=true">
                             <button class="btn btn-primary" type="submit" value="start">Start Remote</button>
                         </form>
-
                         <form method="POST" enctype="multipart/form-data" action="/project/{{project_name}}/exec?remote=true&no_exec=true">
                             <button class="btn btn-primary" type="submit" value="start">File Remote</button>
                         </form>
-                    </div>
+                        {% endif %}
                 </div>
                 {% endif %}
             </div>
 
             <!-- Row 2: Input files -->
-            <div class="col-3">
+            <div class="col-2">
                 <!-- leave this here or it will fuck up layout -->
                 <form method="POST" enctype="multipart/form-data" action="/project_add">
                 <input type="hidden" name="project_name" value="{{project_name}}">
@@ -85,11 +83,29 @@
                     {% endfor %}
                 </select>
                 {% endif %}
-
+                <a href="/exes/{{project.settings.inject_exe_in | basename}}">EXE INFO</a>
             </div>
 
-            <!-- Row 3: settings -->
-            <div class="col-3">
+            <!-- Row 3: exe and shellcode info -->
+            <div class="col-2">
+                x64: {{ is_64 }} / Dotnet: {{ is_dotnet}} <br>
+                .text: {{ code_sect_size}}  <br>
+                .rodata: {{ data_sect_size}} 
+                (max: {{ data_sect_largest_gap_size}}) <br>
+
+                {% if unresolved_dlls|length > 0 %}
+                <br>
+                <span class="text-danger">Unresolved dlls:</span>
+                <ul>
+                {% for dll in unresolved_dlls %}
+                    <li>{{ dll }} </li>
+                {% endfor %}
+                </ul>
+                {% endif %}
+            </div>
+
+            <!-- Row 4: leet settings -->
+            <div class="col-2">
                 <select class="form-select" name="source_style" aria-label="SOURCESTYLE" onchange="this.form.submit()">
                     {% for name, value in function_invoke_styles %}
                     <option value="{{name}}"
@@ -115,27 +131,13 @@
                 </select>
             </div>
 
-            <!-- Row 4: more settings -->
-            <div class="col-3">
-                EXE/DLL Is: <br>
-                Is x64: {{ is_64}} <br>
-                Is Dotnet: {{ is_dotnet}} <br>
-                Code Section size: {{ code_sect_size}} <br>
-                Data Section size: {{ data_sect_size}} <br>
-                Data Section largest: {{ data_sect_largest_gap_size}} <br>
-                -> Payload len: {{ payload_len}} <br>
-                {{ project_dir }} <br>
-                Unresolved dlls: <br>
-                {% for dll in unresolved_dlls %}
-                    {{ dll }} <br>
-                {% endfor %}
-                <a href="/exes/{{project.settings.inject_exe_in | basename}}">EXE INFO</a>
-            </div>
+
         </div>
         </form>
 
         <div class="row">
             <div class="col">
+             {{ project_dir }} <br>
               <div class="custom-line"></div>
             </div>
         </div>
diff --git a/app/views_project.py b/app/views_project.py
index 231cc8e..bdc3757 100644
--- a/app/views_project.py
+++ b/app/views_project.py
@@ -59,6 +59,10 @@ def project(name):
     data_sect_largest_gap_size = 0
     payload_len = 0
     unresolved_dlls = []
+    has_remote = False
+
+    if config.get("avred_server") != "":
+        has_remote = True
 
     # when we select a shellcode
     if project.settings.payload_path != "":
@@ -83,6 +87,7 @@ def project(name):
     log_files = get_logfiles(project.settings.main_dir)
 
     exes = list_files_and_sizes(PATH_EXES, prepend=PATH_EXES)
+    exes += list_files_and_sizes(PATH_EXES_MORE, prepend=PATH_EXES_MORE)
     #for file in 
     #    exes.append(PATH_EXES + file)
     #for file in os.listdir(PATH_EXES_MORE):
@@ -115,6 +120,8 @@ def project(name):
         data_sect_largest_gap_size=data_sect_largest_gap_size,
         payload_len=payload_len,
         unresolved_dlls=unresolved_dlls,
+
+        has_remote=has_remote,
     )
 
 def list_files_and_sizes(directory, prepend=""):