Small CR changes

This commit is contained in:
Shreya 2020-07-26 23:25:30 +05:30
parent 45fdf4b62c
commit e6560ba3d4
3 changed files with 4 additions and 4 deletions

View File

@ -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
return e.output.decode(), False

View File

@ -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():

View File

@ -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():