NetExec/nxc/first_run.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from os import mkdir
from os.path import exists
from os.path import join as path_join
import shutil
from nxc.paths import NXC_PATH, CONFIG_PATH, TMP_PATH, DATA_PATH
from nxc.nxcdb import initialize_db
from nxc.logger import nxc_logger


def first_run_setup(logger=nxc_logger):
    if not exists(TMP_PATH):
        mkdir(TMP_PATH)

    if not exists(NXC_PATH):
        logger.display("First time use detected")
        logger.display("Creating home directory structure")
        mkdir(NXC_PATH)

    folders = (
        "logs",
        "modules",
        "protocols",
        "workspaces",
        "obfuscated_scripts",
        "screenshots",
    )
    for folder in folders:
        if not exists(path_join(NXC_PATH, folder)):
            logger.display(f"Creating missing folder {folder}")
            mkdir(path_join(NXC_PATH, folder))

    initialize_db(logger)

    if not exists(CONFIG_PATH):
        logger.display("Copying default configuration file")
        default_path = path_join(DATA_PATH, "nxc.conf")
        shutil.copy(default_path, NXC_PATH)

    # if not exists(CERT_PATH):
    #     logger.display('Generating SSL certificate')
    #     try:
    #         check_output(['openssl', 'help'], stderr=PIPE)
    #         if os.name != 'nt':
    #             os.system('openssl req -new -x509 -keyout {path} -out {path} -days 365 -nodes -subj "/C=US" > /dev/null 2>&1'.format(path=CERT_PATH))
    #         else:
    #             os.system('openssl req -new -x509 -keyout {path} -out {path} -days 365 -nodes -subj "/C=US"'.format(path=CERT_PATH))
    #     except OSError as e:
    #         if e.errno == errno.ENOENT:
    #             logger.error('OpenSSL command line utility is not installed, could not generate certificate, using default certificate')
    #             default_path = path_join(DATA_PATH, 'default.pem')
    #             shutil.copy(default_path, CERT_PATH)
    #         else:
    #             logger.error('Error while generating SSL certificate: {}'.format(e))
    #             sys.exit(1)