79 lines
3.2 KiB
Python
79 lines
3.2 KiB
Python
from servicebase import ServiceBase
|
|
import paramiko
|
|
import datetime
|
|
import re
|
|
|
|
class ServiceDelegate(ServiceBase) :
|
|
|
|
_connections = None
|
|
|
|
def get_arguments(cls) :
|
|
"""Returns an array of information used to construct an argumentparser argument."""
|
|
return ['-r', '--vpn','store_true','Return VPN information about the subject (openvpn)']
|
|
|
|
def startup(self) :
|
|
for requirement in ['hosts','username','key'] :
|
|
if requirement not in self._config or (requirement in self._config and self._config[requirement] == ''):
|
|
self._error.append('Missing required config option ' + requirement)
|
|
rself._error
|
|
self.hosts = self._config['hosts']
|
|
self.debug('Logging into OpenVPN servers...',1)
|
|
connections = {}
|
|
for host in self._config['hosts'] :
|
|
try :
|
|
sshclient = paramiko.SSHClient()
|
|
sshclient.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
sshclient.connect(host,22,username=self._config['username'],password=self._config['key'])
|
|
connections.update({host:sshclient})
|
|
except Exception as exception :
|
|
self._error.append('Unable to ssh into ' + host + ': ' + str(exception))
|
|
self._connections = connections
|
|
|
|
|
|
def shutdown(self) :
|
|
for connection in self._connections.values() :
|
|
connection.close()
|
|
|
|
def lookup(self,subject) :
|
|
search_command = 'sudo -S cat /var/log/openvpn/openvpn.log | grep \'primary virtual IP for\' | grep -i \'' + subject + '\' | tail -n 1'
|
|
final_dictionary = {}
|
|
for host,connection in self._connections.items() :
|
|
try:
|
|
stdin,stdout,stderr=connection.exec_command(search_command)
|
|
stdin.write(self._config['key'] + '\n')
|
|
stdin.flush()
|
|
result = stdout.readlines()
|
|
if len(result) > 0 :
|
|
self.debug('Retrieved line from ssh session: \n' + result[0].strip(),2)
|
|
result = result[0]
|
|
else :
|
|
result = ''
|
|
items = {'nat_address': r'.*\ ([0-9a-fA-F\.\:]*)$', 'source_address': r'.*/([0-9a-fA-F\.]*)\:.*', 'user_name': r'.*primary\ virtual\ IP\ for\ ([a-zA-Z0-9\.]*)/.*', 'timestamp': r'^([a-zA-Z]{3}\ +[0-9]{1,2}\ [0-9:]*)\ .*'}
|
|
return_dictionary = {}
|
|
for item in items.keys() :
|
|
matches = re.match(items[item],result)
|
|
if matches and len(matches.groups()) and matches[1] :
|
|
return_dictionary.update({item: matches[1]})
|
|
if len(return_dictionary) > 0 :
|
|
return_dictionary.update({'host':host})
|
|
|
|
epoch = None
|
|
if 'timestamp' in final_dictionary.keys() and 'timestamp' in return_dictionary.keys() :
|
|
epoch = int(datetime.datetime.strptime(return_dictionary['timestamp'] + ' ' + str(datetime.datetime.now().year), '%b %d %H:%M:%S %Y').strftime('%s'))
|
|
if len(return_dictionary) > 0 :
|
|
if epoch :
|
|
if epoch > int(datetime.datetime.strptime(final_dictionary['timestamp'] + ' ' + str(datetime.datetime.now().year), '%b %d %H:%M:%S %Y').strftime('%s')) :
|
|
final_dictionary = {}
|
|
final_dictionary.update(return_dictionary)
|
|
else :
|
|
final_dictionary = return_dictionary
|
|
else :
|
|
self.debug('No results for ' + subject + ' on host ' + host,2)
|
|
except Exception as exception :
|
|
raise exception
|
|
self._error.append('Unable to get results from ssh: ' + str(exception))
|
|
if len(final_dictionary) > 0 :
|
|
return final_dictionary
|
|
else :
|
|
return None
|