Ahenk/opt/ahenk/base/registration/Registration.py

131 lines
5.3 KiB
Python
Raw Normal View History

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
import datetime
import json
import netifaces
import socket
import uuid
from uuid import getnode as get_mac
from base.Scope import Scope
from base.messaging.AnonymousMessager import AnonymousMessager
class Registration():
def __init__(self):
scope = Scope().getInstance()
self.logger = scope.getLogger()
self.message_manager = scope.getMessageManager()
self.event_manager = scope.getEventManager()
self.messager = scope.getMessager()
self.db_service = scope.getDbService()
if self.is_registered():
self.logger.debug('[Registration] Ahenk already registered')
else:
2016-03-07 12:55:14 +02:00
self.register(True)
2016-03-07 12:55:14 +02:00
def registration_request(self):
self.logger.debug('[Registration] Requesting registration')
anon_messager = AnonymousMessager(self.message_manager.registration_msg(), None)
anon_messager.connect_to_server()
2016-03-10 09:53:51 +02:00
def ldap_registration_request(self):
self.logger.debug('[Registration] Requesting LDAP registration')
self.messager.send_Direct_message(self.message_manager.ldap_registration_msg())
2016-03-10 09:53:51 +02:00
def confirm_registration(self, reg_reply):
self.logger.debug('[Registration] Reading registration reply')
j = json.loads(reg_reply)
self.logger.debug('[Registration]' + j['message'])
status = str(j['status']).lower()
dn = str(j['agentDn']).lower()
self.logger.debug('[Registration] Registration status: ' + str(status))
if 'registered' == str(status) or 'registered_without_ldap' == str(status):
self.logger.debug('dn:' + dn)
self.update_conf_file(dn)
elif 'registration_error' == str(status):
self.logger.info('[Registration] Registration is failed. New registration request will send')
self.re_register()
self.registration_request()
elif 'already_exists' == str(status):
self.update_conf_file(dn)
self.logger.info('[Registration] Ahenk already registered')
def update_conf_file(self, dn=None):
self.logger.debug('[Registration] Registration configuration is updating...')
self.db_service.update('registration', ['dn', 'registered'], [dn, 1], ' registered = 0')
2016-03-10 09:53:51 +02:00
2016-03-23 17:34:12 +02:00
if self.conf_manager.has_section('CONNECTION'):
self.conf_manager.set('CONNECTION', 'uid',self.conf_manager.get('REGISTRATION', 'from'))
self.conf_manager.set('CONNECTION', 'password',self.conf_manager.get('REGISTRATION', 'password'))
#TODO get file path?
with open('/etc/ahenk/ahenk.conf', 'w') as configfile:
self.conf_manager.write(configfile)
self.logger.debug('[Registration] Registration configuration file is updated')
def is_registered(self):
registered = self.db_service.select_one_result('registration', 'registered', 'registered = 1')
if registered == 1:
return True
else:
return False
2016-03-10 09:53:51 +02:00
def is_ldap_registered(self):
dn = self.db_service.select_one_result('registration', 'dn', 'registered = 1')
if dn is not None and dn != '':
2016-03-10 09:53:51 +02:00
return True
else:
return False
def register(self, uuid_depend_mac=False):
cols = ['jid', 'password', 'registered', 'params', 'timestamp']
vals = [str(self.generate_uuid(uuid_depend_mac)), str(self.generate_password()), 0, str(self.get_registration_params()), str(datetime.datetime.now().strftime("%d-%m-%Y %I:%M"))]
self.db_service.delete('registration', ' 1==1 ')
self.db_service.update('registration', cols, vals)
self.logger.debug('[Registration] Registration parameters were created')
def get_registration_params(self):
params = {
'ipAddresses': str(self.get_ipAddresses()),
'macAddresses': str(':'.join(("%012X" % get_mac())[i:i + 2] for i in range(0, 12, 2))),
'hostname': str(socket.gethostname())
}
return json.dumps(params)
def unregister(self):
self.logger.debug('[Registration] Ahenk is unregistering...')
self.db_service.delete('registration', ' 1==1 ')
self.logger.debug('[Registration] Ahenk is unregistered')
2016-03-07 12:55:14 +02:00
def re_register(self):
self.logger.debug('[Registration] Reregistrating...')
2016-03-07 12:55:14 +02:00
self.unregister()
self.register(False)
def generate_uuid(self, depend_mac=True):
if depend_mac is False:
self.logger.debug('[Registration] uuid creating randomly')
return uuid.uuid4() # make a random UUID
else:
self.logger.debug('[Registration] uuid creating according to mac address')
return uuid.uuid3(uuid.NAMESPACE_DNS, str(get_mac())) # make a UUID using an MD5 hash of a namespace UUID and a mac address
def generate_password(self):
return uuid.uuid4()
def get_ipAddresses(self):
self.logger.debug('[Registration] looking for network interfaces')
ip_address = ""
for interface in netifaces.interfaces():
if (str(interface) != "lo"):
ip_address += str(netifaces.ifaddresses(interface)[netifaces.AF_INET])
self.logger.debug('[Registration] returning ip addresses from every interfaces')
return ip_address