1 # -*- coding: utf-8 -*-
2 # Description: freeradius netdata python.d module
5 from base import SimpleService
6 from os.path import isfile
8 from subprocess import Popen, PIPE
10 # default module values (can be overridden per job in `config`)
14 directories = ['/bin/', '/usr/bin/', '/sbin/', '/usr/sbin/']
16 # charts order (can be overridden if you want less charts, or different order)
17 ORDER = ['authentication', 'accounting', 'proxy-auth', 'proxy-acct']
21 'options': [None, "Authentication", "packets/s", 'Authentication', 'freerad.auth', 'line'],
23 ['access-accepts', None, 'incremental'], ['access-rejects', None, 'incremental'],
24 ['auth-dropped-requests', None, 'incremental'], ['auth-duplicate-requests', None, 'incremental'],
25 ['auth-invalid-requests', None, 'incremental'], ['auth-malformed-requests', None, 'incremental'],
26 ['auth-unknown-types', None, 'incremental']
29 'options': [None, "Accounting", "packets/s", 'Accounting', 'freerad.acct', 'line'],
31 ['accounting-requests', None, 'incremental'], ['accounting-responses', None, 'incremental'],
32 ['acct-dropped-requests', None, 'incremental'], ['acct-duplicate-requests', None, 'incremental'],
33 ['acct-invalid-requests', None, 'incremental'], ['acct-malformed-requests', None, 'incremental'],
34 ['acct-unknown-types', None, 'incremental']
37 'options': [None, "Proxy Authentication", "packets/s", 'Authentication', 'freerad.proxy.auth', 'line'],
39 ['proxy-access-accepts', None, 'incremental'], ['proxy-access-rejects', None, 'incremental'],
40 ['proxy-auth-dropped-requests', None, 'incremental'], ['proxy-auth-duplicate-requests', None, 'incremental'],
41 ['proxy-auth-invalid-requests', None, 'incremental'], ['proxy-auth-malformed-requests', None, 'incremental'],
42 ['proxy-auth-unknown-types', None, 'incremental']
45 'options': [None, "Proxy Accounting", "packets/s", 'Accounting', 'freerad.proxy.acct', 'line'],
47 ['proxy-accounting-requests', None, 'incremental'], ['proxy-accounting-responses', None, 'incremental'],
48 ['proxy-acct-dropped-requests', None, 'incremental'], ['proxy-acct-duplicate-requests', None, 'incremental'],
49 ['proxy-acct-invalid-requests', None, 'incremental'], ['proxy-acct-malformed-requests', None, 'incremental'],
50 ['proxy-acct-unknown-types', None, 'incremental']
56 class Service(SimpleService):
57 def __init__(self, configuration=None, name=None):
58 SimpleService.__init__(self, configuration=configuration, name=name)
59 self.host = self.configuration.get('host', 'localhost')
60 self.port = self.configuration.get('port', '18121')
61 self.secret = self.configuration.get('secret', 'adminsecret')
62 self.acct = self.configuration.get('acct', False)
63 self.proxy_auth = self.configuration.get('proxy_auth', False)
64 self.proxy_acct = self.configuration.get('proxy_acct', False)
65 self.echo = [''.join([directory, 'echo']) for directory in directories if isfile(''.join([directory, 'echo']))][:1]
66 self.radclient = [''.join([directory, 'radclient']) for directory in directories if isfile(''.join([directory, 'radclient']))][:1]
67 self.sub_echo = [self.echo, 'Message-Authenticator = 0x00, FreeRADIUS-Statistics-Type = 15, Response-Packet-Type = Access-Accept']
68 self.sub_radclient = [self.radclient, '-r', '1', '-t', '1', ':'.join([self.host, self.port]), 'status', self.secret]
71 if not all([self.echo, self.radclient]):
72 self.error('Command radclient not found')
74 if self._get_raw_data():
75 chart_choice = [True, bool(self.acct), bool(self.proxy_auth), bool(self.proxy_acct)]
76 self.order = [chart for chart, choice in zip(ORDER, chart_choice) if choice]
77 self.definitions = {k:v for k, v in CHARTS.items() if k in self.order}
78 self.info('Plugin was started succesfully')
81 self.error('Request returned no data. Is server alive? Used options: host {}, port {}, secret {}'.format(self.host, self.port, self.secret))
87 Format data received from shell command
90 result = self._get_raw_data()
91 return {k.lower():int(v) for k, v in findall(r'((?<=-)[AP][a-zA-Z-]+) = (\d+)', result)}
93 def _get_raw_data(self):
95 The following code is equivalent to
96 'echo "Message-Authenticator = 0x00, FreeRADIUS-Statistics-Type = 15, Response-Packet-Type = Access-Accept" | radclient -t 1 -r 1 host:port status secret'
100 process_echo = Popen(self.sub_echo, stdout=PIPE, shell=False)
101 process_rad = Popen(self.sub_radclient, stdin=process_echo.stdout, stdout=PIPE, shell=False)
102 process_echo.stdout.close()
103 raw_result = process_rad.communicate()[0]
107 if process_rad.returncode is 0:
108 return raw_result.decode()