python.d/fail2ban.chart.py

   1 # -*- coding: utf-8 -*-
   2 # Description: fail2ban log netdata python.d module
   3 # Author: l2isbad
   4
   5 from base import LogService
   6 from re import compile
   7 try:
   8     from itertools import filterfalse
   9 except ImportError:
  10     from itertools import ifilterfalse as filterfalse
  11 from os import access as is_accessible, R_OK
  12
  13 priority = 60000
  14 retries = 60
  15 regex = compile(r'([A-Za-z-]+\]) enabled = ([a-z]+)')
  16
  17 ORDER = ['jails_group']
  18
  19
  20 class Service(LogService):
  21     def __init__(self, configuration=None, name=None):
  22         LogService.__init__(self, configuration=configuration, name=name)
  23         self.order = ORDER
  24         self.log_path = self.configuration.get('log_path', '/var/log/fail2ban.log')
  25         self.conf_path = self.configuration.get('conf_path', '/etc/fail2ban/jail.local')
  26         self.default_jails = ['ssh']
  27         try:
  28             self.exclude = self.configuration['exclude'].split()
  29         except (KeyError, AttributeError):
  30             self.exclude = []
  31
  32
  33     def _get_data(self):
  34         """
  35         Parse new log lines
  36         :return: dict
  37         """
  38
  39         # If _get_raw_data returns empty list (no new lines in log file) we will send to Netdata this
  40         self.data = {jail: 0 for jail in self.jails_list}
  41
  42         try:
  43             raw = self._get_raw_data()
  44             if raw is None:
  45                 return None
  46             elif not raw:
  47                 return self.data
  48         except (ValueError, AttributeError):
  49             return None
  50
  51         # Fail2ban logs looks like
  52         # 2016-12-25 12:36:04,711 fail2ban.actions[2455]: WARNING [ssh] Ban 178.156.32.231
  53         self.data = dict(
  54             zip(
  55                 self.jails_list,
  56                 [len(list(filterfalse(lambda line: (jail + '] Ban') not in line, raw))) for jail in self.jails_list]
  57             ))
  58
  59         return self.data
  60
  61     def check(self):
  62
  63         # Check "log_path" is accessible.
  64         # If NOT STOP plugin
  65         if not is_accessible(self.log_path, R_OK):
  66             self.error('Cannot access file %s' % (self.log_path))
  67             return False
  68
  69         # Check "conf_path" is accessible.
  70         # If "conf_path" is accesible try to parse it to find enabled jails
  71         if is_accessible(self.conf_path, R_OK):
  72             with open(self.conf_path, 'rt') as jails_conf:
  73                 jails_list = regex.findall(' '.join(jails_conf.read().split()))
  74             self.jails_list = [jail[:-1] for jail, status in jails_list if status == 'true']
  75         else:
  76             self.jails_list = []
  77             self.error('Cannot access jail.local file %s.' % (self.conf_path))
  78
  79         # If for some reason parse failed we still can START with default jails_list.
  80         self.jails_list = [jail for jail in self.jails_list if jail not in self.exclude]\
  81                                               if self.jails_list else self.default_jails
  82         self.create_dimensions()
  83         self.info('Plugin succefully started. Jails: %s' % (self.jails_list))
  84         return True
  85
  86     def create_dimensions(self):
  87         self.definitions = {'jails_group':
  88                                 {'options':
  89                                      [None, "Jails ban statistics", "bans/s", 'Jails', 'jail.ban', 'line'], 'lines': []}}
  90         for jail in self.jails_list:
  91             self.definitions['jails_group']['lines'].append([jail, jail, 'absolute'])