admin/SuperMega
1
0
Fork 0
mirror of https://github.com/dobin/SuperMega synced 2026-06-02 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: make nicer ui, web, log

Browse Source
This commit is contained in:
Dobin
2024-03-08 11:09:31 +00:00
parent c990a6699d
commit 215e24ffe0
11 changed files with 60 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/static/styles.css
+13 -1
View File
@@ -63,4 +63,16 @@
max-width: 40em;
width: 40em;
text-align: left;
}
}
.nav-item .btn {
margin-bottom: 5px; /* Add some space between the buttons */
}
.custom-line {
border-top: 1px solid #7e7e7e; /* Create a solid line */
width: 100%; /* Span the full width of the container */
margin: 20px 0; /* Add some vertical spacing */
}
app/storage.py
-6
View File
@@ -25,18 +25,12 @@ class Storage():
self.save_data()
def get_data(self):
print("Read data")
with open("app/data.pickle", "rb") as f:
data = f.read()
data = pickle.loads(data)
for project in data:
print(" {}".format(project.name))
return data
def save_data(self):
print("Save data")
with open("app/data.pickle", "wb") as f:
f.write(pickle.dumps(self.data))
app/templates/files.html
+4 -4
View File
@@ -1,12 +1,12 @@
<div class="container-fluid">
<div class="row">
<div class="col-md-3">
<ul class="nav nav-tabs flex-column" id="myTab" role="tablist">
<div class="col-md-2">
<ul class="nav nav-pills flex-column" id="myTab" role="tablist">
{% for log_file in log_files %}
<li class="nav-item" role="presentation">
<button
class="nav-link {% if loop.last %}active{% endif %}"
class="nav-link btn btn-light {% if loop.last %}active{% endif %}"
id="project-{{log_file['id']}}-tab"
data-bs-toggle="tab"
data-bs-target="#project-{{log_file['id']}}"
@@ -20,7 +20,7 @@
</ul>
</div>
<div class="col-md-9">
<div class="col-md-10">
<div class="tab-content" id="myTabContent">
{% for log_file in log_files %}
<div
app/templates/project.html
+25 -10
View File
@@ -7,12 +7,14 @@
{% include 'navigation.html' %}
<div class="indent">
<h1> Project {{project_name}} </h1>
<h2> {{project_name}} </h2>
<form method="POST" enctype="multipart/form-data" action="/add_project">
<input type="hidden" name="project_name" value="{{project_name}}">
<div class="row">
<div class="col-4">
<input type="text" name="project_name" class="hidden form-control" value="{{project_name}}"
placeholder="Projekt" aria-label="PROJECTNAME" aria-describedby="basic-addon1">
<div class="col-3">
<input type="text" name="comment" class="hidden form-control" value="{{project.comment}}"
placeholder="" aria-label="PROJECTNAME" aria-describedby="basic-addon1">
<select class="form-select" name="shellcode" aria-label="SHELLCODE">
{% for shellcode in shellcodes %}
@@ -33,7 +35,7 @@
{% endfor %}
</select>
</div>
<div class="col-4">
<div class="col-3">
<select class="form-select" name="source_style" aria-label="SOURCESTYLE">
{% for name, value in sourcestyles %}
@@ -52,7 +54,7 @@
</select>
</div>
<div class="col-4">
<div class="col-3">
<select class="form-select" name="alloc_style" aria-label="ALLOCSTYLE">
{% for name, value in allocstyles %}
<option value="{{name}}"
@@ -78,20 +80,33 @@
</select>
</div>
</div>
</form>
</form>
<div class="row">
<div class="col-4">
<div class="col-3">
<button class="btn btn-primary" type="submit" value="save">Save</button>
<form method="POST" enctype="multipart/form-data" action="/start_project">
<input type="hidden" name="project_name" value="{{project_name}}">
<div class="form-check">
<input class="form-check-input" name="try_start" type="checkbox" value="checked" id="flexCheckDefault" checked>
<label class="form-check-label" for="flexCheckDefault">
Start Infected Exe
</label>
</div>
<button class="btn btn-primary" type="submit" value="start">Start</button>
</form>
</div>
</div>
<div class="row">
<div class="col">
<div class="custom-line"></div> <!-- Here's the horizontal line -->
</div>
</div>
<div class="row">
<div class="col-md-12">
app/templates/project_add_get.html
-1
View File
@@ -68,7 +68,6 @@
</div>
<button class="btn btn-primary" type="submit" value="save">Save</button>
<button class="btn btn-primary" type="submit" value="Verify">Verify</button>
</div>
</form>
app/views.py
+10 -46
View File
@@ -2,15 +2,12 @@ from flask import Blueprint, current_app, flash, request, redirect, url_for, ren
from werkzeug.utils import secure_filename
import os
import logging
import io
from typing import List, Tuple
from datetime import date
from pygments import highlight
from pygments.lexers import CLexer, NasmLexer, DiffLexer, HexdumpLexer
from pygments.formatters import HtmlFormatter
import difflib
from ansi2html import Ansi2HTMLConverter
import pickle
from config import config
from model.settings import Settings
@@ -66,7 +63,7 @@ def project(name):
@views.route("/add_project", methods=['POST', 'GET'])
def inject():
def add_project():
if request.method == 'POST':
settings = Settings()
@@ -135,47 +132,18 @@ def inject():
@views.route("/start_project", methods=['POST', 'GET'])
def start_project():
#project_name = request.args.get('project_name')
project_name = request.form['project_name']
project_name = request.form.get('project_name')
try_start = request.form.get('try_start')
if try_start != None:
try_start = True
else:
try_start = False
project = storage.get_project(project_name)
project.settings.try_start_final_infected_exe = try_start
start(project.settings)
return redirect("/project/{}".format(project_name), code=302)
@views.route("/build")
def build():
exes = []
for file in os.listdir("app/upload/exe"):
exes.append(file)
shellcodes = []
for file in os.listdir("app/upload/shellcode"):
shellcodes.append(file)
sourcestyles = [(color.name, color.value) for color in SourceStyle]
allocstyles = [(color.name, color.value) for color in AllocStyle]
decoderstyles = [(color.name, color.value) for color in DecoderStyle]
execstyles = [(color.name, color.value) for color in ExecStyle]
injectstyles = [(color.name, color.value) for color in InjectStyle]
return render_template('build.html',
exes=exes,
shellcodes=shellcodes,
sourcestyles=sourcestyles,
allocstyles=allocstyles,
decoderstyles=decoderstyles,
execstyles=execstyles,
injectstyles=injectstyles,
)
@views.route("/files")
def files():
log_files = get_logfiles()
return render_template('files.html',
log_files=log_files
)
def get_logfiles():
log_files = []
id = 0
@@ -184,13 +152,11 @@ def get_logfiles():
for file in os.listdir(f"{logs_dir}/"):
if file.startswith("."):
continue
print("Handle: ", file)
with open(os.path.join(f"{logs_dir}/", file), "r") as f:
if file.endswith(".bin"):
continue
data = f.read()
print("FILE: {}".format(file))
if 'main_c' in file:
data = highlight(data, CLexer(), HtmlFormatter(full=False))
elif '_asm_' in file:
@@ -203,11 +169,9 @@ def get_logfiles():
elif '.ascii' in file:
data = conv.convert(data, full=False)
elif '.txt' in file:
# skip it
continue
continue # skip it
elif '.hex' in file:
print("-> hex")
continue
continue # skip it
#data = escape(data)
#data = highlight(data, HexdumpLexer(), HtmlFormatter(full=False))
elif '.log' in file:
log.py
+4 -1
View File
@@ -65,4 +65,7 @@ def setup_logging():
list_handler.setFormatter(CustomFormatter())
root_logger.addHandler(ch)
root_logger.addHandler(list_handler)
root_logger.addHandler(list_handler)
def clear_log():
log_messages.clear()
model/project.py
+1
View File
@@ -12,6 +12,7 @@ logger = logging.getLogger("Project")
class Project():
def __init__(self, settings: Settings):
self.name: str = ""
self.comment: str = ""
self.settings: Settings = settings
self.payload: Payload = Payload(self.settings.payload_path)
self.exe_host: ExeHost = ExeHost(self.settings.inject_exe_in)
pe/derbackdoorer.py
+1 -3
View File
@@ -216,8 +216,6 @@ Trailing {sect_name} bytes:
for ins in trampoline.split(';'):
logger.info(f'\t{ins.strip()}')
logger.info('')
return (trampoline, addrOffset)
@@ -245,7 +243,7 @@ Trailing {sect_name} bytes:
self.compiledTrampoline = encoding
self.compiledTrampolineCount = count
logger.info('Successfully backdoored entry point with jump/call to shellcode')
logger.debug('Successfully backdoored entry point with jump/call to shellcode')
return instr.address
return 0
pe/superpe.py
-2
View File
@@ -178,8 +178,6 @@ class SuperPe():
if self.arch == 'x64':
imageBaseRelocType = SuperPe.IMAGE_REL_BASED_DIR64
logger.info('Adding new relocations to backdoored PE file...')
relocsSize = self.pe.OPTIONAL_HEADER.DATA_DIRECTORY[SuperPe.IMAGE_DIRECTORY_ENTRY_BASERELOC].Size
relocsIndex = self.getSectionIndexByDataDir(SuperPe.IMAGE_DIRECTORY_ENTRY_BASERELOC)
addr = self.pe.OPTIONAL_HEADER.DATA_DIRECTORY[SuperPe.IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress
web.py
+2
View File
@@ -3,11 +3,13 @@
import os
import argparse
from flask import Flask
import logging
from app.views import views
from log import setup_logging, writelog
if __name__ == "__main__":
logging.getLogger('werkzeug').setLevel(logging.ERROR)
setup_logging()
parser = argparse.ArgumentParser()
parser.add_argument('--listenip', type=str, help='IP to listen on', default="0.0.0.0")