MAB-malware/minimizer.py at main · bitsecurerlab/MAB-malware · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from utils import *

class Minimizer:
    def __init__(self, samples_manager):
        self.samples_manager = samples_manager

    def run(self):
        while True:
            count_evasive = self.samples_manager.get_count_with_status(SAMPLE_STATUS_EVASIVE)
            count_minimal = self.samples_manager.get_count_with_status(SAMPLE_STATUS_MINIMAL)

            if count_evasive == 0:
                logger_min.info('No evasive samples, sleep...')
                time.sleep(5)
            else:
                self.samples_manager.minimize_evasive_sample()
                self.samples_manager.update_evasive_list()
                if Utils.is_cuckoo_enable():
                    self.samples_manager.update_minimal_list()

            if ( Utils.is_cuckoo_enable() and count_evasive + count_minimal == 0 ) or (not Utils.is_cuckoo_enable() and count_evasive == 0):
                if os.path.exists(REWRITER_EXIT_SIGN):
                    logger_min.info('%%%%%%%%%%%%%%%%%%%%%%%% Minimizer Finish %%%%%%%%%%%%%%%%%%%%%%%%')
                    exit()

            time.sleep(0.5)