mirror of
https://github.com/dobin/SuperMega
synced 2026-06-02 17:27:10 +00:00
Dobin Rutishauser
67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
import logging
|
|
import shutil
|
|
|
|
from model.defs import *
|
|
from model.payload import Payload
|
|
from model.settings import Settings
|
|
from model.injectable import Injectable
|
|
|
|
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.injectable: Injectable = Injectable(self.settings.inject_exe_in)
|
|
|
|
self.project_dir: str = ""
|
|
self.project_exe: str = ""
|
|
|
|
|
|
def init(self) -> bool:
|
|
if not self.payload.init():
|
|
return False
|
|
if not self.injectable.init():
|
|
return False
|
|
return True
|
|
|
|
|
|
def prepare_project(project_name, settings):
|
|
src = "{}{}/".format(PATH_CARRIER, settings.carrier_name)
|
|
dst = "{}{}/".format(PATH_WEB_PROJECT, 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)
|
|
|
|
# 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) |