mirror of
https://github.com/dobin/SuperMega
synced 2026-06-02 17:27:10 +00:00
Dobin
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
import logging
|
|
import shutil
|
|
|
|
from model.defs import *
|
|
from model.payload import Payload
|
|
from model.settings import Settings
|
|
from model.carrier import Carrier
|
|
|
|
logger = logging.getLogger("Project")
|
|
|
|
|
|
class WebProject():
|
|
def __init__(self, name: str, settings: Settings):
|
|
self.name = name
|
|
self.settings: Settings = settings
|
|
self.comment: str = ""
|
|
|
|
|
|
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.carrier: Carrier = Carrier(self.settings.inject_exe_in)
|
|
|
|
self.project_dir: str = ""
|
|
self.project_exe: str = ""
|
|
|
|
|
|
def init(self):
|
|
self.payload.init()
|
|
self.carrier.init()
|
|
|
|
|
|
def prepare_project(project_name, settings):
|
|
src = "{}{}/".format(PATH_CARRIER, settings.source_style.value)
|
|
dst = "{}{}/".format(PATH_WEB_PROJECT, project_name)
|
|
|
|
# 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)
|
|
|
|
# copy *.c *.h files from src directory to dst directory
|
|
for file in os.listdir(src):
|
|
if file.endswith(".c") or file.endswith(".h"):
|
|
logger.info("--( Copy {} to {}".format(src + file, dst))
|
|
shutil.copy2(src + file, dst) |