From e6560ba3d410014b87554688313ec3271c9583ef Mon Sep 17 00:00:00 2001 From: Shreya Date: Sun, 26 Jul 2020 23:25:30 +0530 Subject: [PATCH] Small CR changes --- .../post_breach/actions/modify_shell_startup_files.py | 2 +- .../cc/services/attack/technique_reports/T1156.py | 4 ++-- .../cc/services/attack/technique_reports/T1504.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/monkey/infection_monkey/post_breach/actions/modify_shell_startup_files.py b/monkey/infection_monkey/post_breach/actions/modify_shell_startup_files.py index 5e8208c96..f7bd43a6e 100644 --- a/monkey/infection_monkey/post_breach/actions/modify_shell_startup_files.py +++ b/monkey/infection_monkey/post_breach/actions/modify_shell_startup_files.py @@ -59,4 +59,4 @@ class ModifyShellStartupFiles(PBA): return output, True except subprocess.CalledProcessError as e: # Return error output of the command - return e.output.decode(), False \ No newline at end of file + return e.output.decode(), False diff --git a/monkey/monkey_island/cc/services/attack/technique_reports/T1156.py b/monkey/monkey_island/cc/services/attack/technique_reports/T1156.py index a9c0ea338..64c401e8f 100644 --- a/monkey/monkey_island/cc/services/attack/technique_reports/T1156.py +++ b/monkey/monkey_island/cc/services/attack/technique_reports/T1156.py @@ -20,8 +20,8 @@ class T1156(AttackTechnique): 'ips': [{'$arrayElemAt': ['$data.ip', 0]}]}, 'result': '$data.result'}}, {'$unwind': '$result'}, - {'$match': {'$or': [{'result': {'$regex': '\.bash'}}, # noqa: W605 - {'result': {'$regex': '\.profile'}}]}}] # noqa: W605 + {'$match': {'$or': [{'result': {'$regex': r'\.bash'}}, + {'result': {'$regex': r'\.profile'}}]}}] @staticmethod def get_report_data(): diff --git a/monkey/monkey_island/cc/services/attack/technique_reports/T1504.py b/monkey/monkey_island/cc/services/attack/technique_reports/T1504.py index 7bfda1d8e..05568fcb8 100644 --- a/monkey/monkey_island/cc/services/attack/technique_reports/T1504.py +++ b/monkey/monkey_island/cc/services/attack/technique_reports/T1504.py @@ -20,7 +20,7 @@ class T1504(AttackTechnique): 'ips': [{'$arrayElemAt': ['$data.ip', 0]}]}, 'result': '$data.result'}}, {'$unwind': '$result'}, - {'$match': {'result': {'$regex': 'profile\.ps1'}}}] # noqa: W605 + {'$match': {'result': {'$regex': r'profile\.ps1'}}}] @staticmethod def get_report_data():