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

from IPacker import IPacker
from lib.utils import *


class PackerHyperion(IPacker):
    # gives: 5^6 ~ 262k brute-forces to undertake during launch
    default_hyperion_args = '-s 4 -k 6'
    hyperion_cmdline_template = '<command> <options> <infile> <outfile>'

    metadata = {
        'author': 'nullsecurity team',
        'url': 'https://nullsecurity.net/tools/binary.html',
        'licensing': 'open-source',
        'description': 'Robust PE EXE runtime AES encrypter for x86/x64 with own-key brute-forcing logic',
        'type': PackerType.PEProtector,
        'input': ['PE', ],
        'output': ['PE', ],
    }

    def __init__(self, logger, options):
        self.hyperion_args = PackerHyperion.default_hyperion_args
        self.logger = logger
        self.options = options

    @staticmethod
    def get_name():
        return 'Hyperion'

    def help(self, parser):
        if parser != None:
            parser.add_argument('--hyperion-path', metavar='PATH', dest='hyperion_path',
                                help='(required) Path to hyperion binary capable of compressing x86/x64 executables.')

            parser.add_argument('--hyperion-args', metavar='ARGS', dest='hyperion_args',
                                help='Optional hyperion-specific arguments to pass during compression.')

        else:
            if not self.options['config']:
                self.logger.fatal('Config file not specified!')

            self.options['hyperion_path'] = configPath( self.options['hyperion_path'])

            if not os.path.isfile(self.options['hyperion_path']):
                self.logger.fatal('--hyperion-path option must be specified!')

            if 'hyperion_args' in self.options.keys() and self.options['hyperion_args'] != None \
                    and len(self.options['hyperion_args']) > 0:
                self.hyperion_args += ' ' + self.options['hyperion_args']

    @ensureInputFileIsPE
    def process(self, arch, infile, outfile):
        out = shell(self.logger, self.options['hyperion_path'] + ' --help')
        ver = ''
        cwd = ''

        m = re.search(r'Version\s+([\d\.]+)', out, re.I)
        if m:
            ver = m.group(1)
            self.logger.info(f'Working with Hyperion v{ver}')

        try:
            cwd = os.getcwd()
            base = os.path.dirname(self.options['hyperion_path'])

            self.logger.dbg('changed working directory to "{}"'.format(base))
            os.chdir(base)

            out = shell(self.logger, IPacker.build_cmdline(
                PackerHyperion.hyperion_cmdline_template,
                os.path.basename(self.options['hyperion_path']),
                self.hyperion_args,
                infile,
                outfile
            ), output=self.options['verbose'] or self.options['debug'], timeout=self.options['timeout'])

        except Exception as e:
            raise

        finally:
            if len(cwd) > 0:
                self.logger.dbg(
                    'reverted to original working directory "{}"'.format(cwd))
                os.chdir(cwd)

        status = os.path.isfile(outfile)

        if not status:
            self.logger.err('Something went wrong: there is no output artefact ({})!\n'.format(
                outfile
            ))

            if len(out) > 0 and not (self.options['verbose'] or self.options['debug']):
                self.logger.info(f'''{PackerHyperion.get_name()} returned:
----------------------------------------
{out}
----------------------------------------
''', forced=True, noprefix=True)

        else:
            print(f'Now that {PackerHyperion.get_name()} produced binary, remember that it\'s taking a longer while before it executes due to decryption key Brute-Force involved. :-)')

        return status