From 2fa45ef5bc89eacbcfe5f33b8d6d72b3ca4be075 Mon Sep 17 00:00:00 2001 From: VakarisZ Date: Mon, 8 Jul 2019 14:51:57 +0300 Subject: [PATCH] small fixes --- .../cc/services/attack/technique_reports/__init__.py | 4 ++-- .../cc/ui/src/components/attack/techniques/T1107.js | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/monkey/monkey_island/cc/services/attack/technique_reports/__init__.py b/monkey/monkey_island/cc/services/attack/technique_reports/__init__.py index 15ce5ac19..d765b5f09 100644 --- a/monkey/monkey_island/cc/services/attack/technique_reports/__init__.py +++ b/monkey/monkey_island/cc/services/attack/technique_reports/__init__.py @@ -50,7 +50,7 @@ class AttackTechnique(object): def technique_status(cls): """ Gets the status of a certain attack technique. - :return: ScanStatus Enum object + :return: ScanStatus numeric value """ if mongo.db.telemetry.find_one({'telem_category': 'attack', 'data.status': ScanStatus.USED.value, @@ -97,7 +97,7 @@ class AttackTechnique(object): def get_tech_base_data(cls): """ Gathers basic attack technique data into a dict. - :return: dict E.g. {'message': 'Brute force used', 'status': 'Used', 'title': 'T1110 Brute force'} + :return: dict E.g. {'message': 'Brute force used', 'status': 2, 'title': 'T1110 Brute force'} """ data = {} status = cls.technique_status() diff --git a/monkey/monkey_island/cc/ui/src/components/attack/techniques/T1107.js b/monkey/monkey_island/cc/ui/src/components/attack/techniques/T1107.js index 73b512706..56edd685a 100644 --- a/monkey/monkey_island/cc/ui/src/components/attack/techniques/T1107.js +++ b/monkey/monkey_island/cc/ui/src/components/attack/techniques/T1107.js @@ -11,7 +11,7 @@ class T1107 extends React.Component { } static renderDelete(status){ - if(status === 2){ + if(status === scanStatus.USED){ return Yes } else { return No