Moved attack scan status enum to common

This commit is contained in:
VakarisZ 2019-04-02 11:08:56 +03:00
parent 2e2b77226d
commit 77b14177c5
3 changed files with 11 additions and 11 deletions

View File

@ -0,0 +1,10 @@
from enum import Enum
class ScanStatus(Enum):
# Technique wasn't scanned
UNSCANNED = 0
# Technique was attempted/scanned
SCANNED = 1
# Technique was attempted and succeeded
USED = 2

View File

@ -17,7 +17,7 @@ from infection_monkey.system_info import SystemInfoCollector
from infection_monkey.system_singleton import SystemSingleton from infection_monkey.system_singleton import SystemSingleton
from infection_monkey.windows_upgrader import WindowsUpgrader from infection_monkey.windows_upgrader import WindowsUpgrader
from infection_monkey.post_breach.post_breach_handler import PostBreach from infection_monkey.post_breach.post_breach_handler import PostBreach
from infection_monkey.transport.attack_telems.base_telem import ScanStatus from common.utils.attack_status_enum import ScanStatus
from infection_monkey.transport.attack_telems.victim_host_telem import VictimHostTelem from infection_monkey.transport.attack_telems.victim_host_telem import VictimHostTelem
__author__ = 'itamar' __author__ = 'itamar'

View File

@ -1,4 +1,3 @@
from enum import Enum
from infection_monkey.config import WormConfiguration, GUID from infection_monkey.config import WormConfiguration, GUID
import requests import requests
import json import json
@ -10,15 +9,6 @@ __author__ = "VakarisZ"
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
class ScanStatus(Enum):
# Technique wasn't scanned
UNSCANNED = 0
# Technique was attempted/scanned
SCANNED = 1
# Technique was attempted and succeeded
USED = 2
class AttackTelem(object): class AttackTelem(object):
def __init__(self, technique, status, data=None): def __init__(self, technique, status, data=None):