MozDef/alerts/sqs_queues_deadman.py

36 строки
1.2 KiB
Python

#!/usr/bin/env python
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at https://mozilla.org/MPL/2.0/.
# Copyright (c) 2017 Mozilla Corporation
from lib.deadman_alerttask import DeadmanAlertTask
from mozdef_util.query_models import SearchQuery, TermMatch
class AlertSQSQueuesDeadman(DeadmanAlertTask):
def main(self):
self.parse_config('sqs_queues_deadman.conf', ['sqs_queues'])
for queue_name in self.config.sqs_queues.split(','):
self.sqs_queue_name = queue_name
self.process_alert()
def process_alert(self):
search_query = SearchQuery(hours=1)
search_query.add_must(TermMatch('tags', self.sqs_queue_name))
self.filtersManual(search_query)
self.searchEventsSimple()
self.walkEvents()
def onNoEvent(self):
category = 'deadman'
tags = [self.sqs_queue_name, 'sqs']
severity = 'ERROR'
summary = 'No events found from {} sqs queue the last hour'.format(self.sqs_queue_name)
# Create the alert object based on these properties
return self.createAlertDict(summary, category, tags, [], severity=severity)