mirror of
https://github.com/dobin/SuperMega
synced 2026-06-02 17:27:10 +00:00
Dobin Rutishauser
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
import logging
|
|
|
|
from model.defs import *
|
|
from model.payload import Payload
|
|
from model.settings import Settings
|
|
from model.injectable import Injectable
|
|
|
|
logger = logging.getLogger("Project")
|
|
|
|
|
|
class Project():
|
|
def __init__(self, settings: Settings):
|
|
self.settings: Settings = settings
|
|
|
|
# Set by init()
|
|
self.payload: Payload
|
|
self.injectable: Injectable
|
|
|
|
|
|
def init(self) -> bool:
|
|
self.payload: Payload = Payload(self.settings.get_payload_path())
|
|
self.injectable: Injectable = Injectable(self.settings.get_inject_exe_in())
|
|
|
|
if not self.payload.init():
|
|
return False
|
|
if not self.injectable.init():
|
|
return False
|
|
return True
|
|
|
|
|
|
def print(self):
|
|
logger.info("Project Name: {}".format(self.settings.project_name))
|
|
logger.info("Comment: {}".format(self.settings.project_comment))
|
|
logger.info("Settings: {}".format(self.settings.__dict__))
|
|
logger.info("Payload Path: {}".format(self.payload.payload_path))
|
|
logger.info("Injectable Path: {}".format(self.injectable.exe_filepath))
|
|
|
|
|
|
def prepare_project(project_name):
|
|
dst = "{}{}/".format(PATH_WEB_PROJECT, project_name)
|
|
|
|
logger.info("-[ Prepare and cleanup project: {}".format(project_name))
|
|
|
|
if not os.path.exists(dst):
|
|
os.makedirs(dst)
|
|
|
|
# delete all files in dst directory
|
|
for file in os.listdir(dst):
|
|
if file == "project.pickle":
|
|
continue
|
|
if file.startswith("."):
|
|
continue
|
|
if file.endswith(".exe"):
|
|
# keep all exes except:
|
|
if file != "main.exe" and not file.endswith(".infected.exe"):
|
|
continue
|
|
if file.endswith(".dll"):
|
|
# keep all dlls except:
|
|
if not file.endswith(".infected.dll"):
|
|
continue
|
|
|
|
os.remove(dst + file)
|