#!/usr/bin/python3 # -*- coding: utf-8 -*- # Author: İsmail BAŞARAN # Author: Volkan Şahin import os import queue import signal import sys import threading import time import json from base.Scope import Scope from base.command.commander import Commander from base.config.ConfigManager import ConfigManager from base.database.AhenkDbService import AhenkDbService from base.deamon.BaseDeamon import BaseDaemon from base.event.EventManager import EventManager from base.execution.ExecutionManager import ExecutionManager from base.logger.AhenkLogger import Logger from base.messaging.MessageResponseQueue import MessageResponseQueue from base.messaging.Messenger import Messenger from base.messaging.Messaging import Messaging from base.plugin.plugin_manager_factory import PluginManagerFactory from base.registration.Registration import Registration from base.scheduler.scheduler_factory import SchedulerFactory from base.system.system import System from base.task.TaskManager import TaskManager ahenkdaemon = None class AhenkDeamon(BaseDaemon): """docstring for AhenkDeamon""" def reload(self): # reload service here pass def init_logger(self): logger = Logger() logger.info('[AhenkDeamon] Log was set') Scope.getInstance().setLogger(logger) return logger def init_config_manager(self, configFilePath, configfileFolderPath): configManager = ConfigManager(configFilePath, configfileFolderPath) config = configManager.read() Scope.getInstance().setConfigurationManager(config) return config def init_scheduler(self): scheduler_ins = SchedulerFactory.get_intstance() scheduler_ins.initialize() Scope.getInstance().set_scheduler(scheduler_ins) sc_thread = threading.Thread(target=scheduler_ins.run) sc_thread.setDaemon(True) sc_thread.start() return scheduler_ins def init_event_manager(self): eventManager = EventManager() Scope.getInstance().setEventManager(eventManager) return eventManager def init_ahenk_db(self): db_service = AhenkDbService() db_service.connect() db_service.initialize_table() Scope.getInstance().setDbService(db_service) return db_service def init_messaging(self): messageManager = Messaging() Scope.getInstance().setMessageManager(messageManager) return messageManager def init_plugin_manager(self): pluginManager = PluginManagerFactory.get_instance() pluginManager.loadPlugins() Scope.getInstance().setPluginManager(pluginManager) return pluginManager def init_task_manager(self): taskManager = TaskManager() Scope.getInstance().setTaskManager(taskManager) return taskManager def init_registration(self): registration = Registration() Scope.getInstance().setRegistration(registration) return registration def init_execution_manager(self): execution_manager = ExecutionManager() Scope.getInstance().setExecutionManager(execution_manager) return execution_manager def init_messenger(self): messenger = Messenger() messenger.connect_to_server() Scope.getInstance().setMessenger(messenger) return messenger def init_message_response_queue(self): responseQueue = queue.Queue() messageResponseQueue = MessageResponseQueue(responseQueue) messageResponseQueue.setDaemon(True) messageResponseQueue.start() Scope.getInstance().setResponseQueue(responseQueue) return responseQueue def check_registration(self): max_attemp_number = int(System.Hardware.Network.interface_size()) * 3 logger = Scope.getInstance().getLogger() try: while Scope.getInstance().getRegistration().is_registered() is False: max_attemp_number -= 1 logger.debug('[AhenkDeamon] Ahenk is not registered. Attempting for registration') # TODO 'Could not reach Registration response from Lider. Be sure Lider is awake and it is connected to XMPP server!' Scope.getInstance().getRegistration().registration_request() if max_attemp_number < 0: logger.warning('[AhenkDeamon] Number of Attempting for registration is over') self.registration_failed() break except Exception as e: logger.error('[AhenkDeamon] Registration failed. Error message: {}'.format(str(e))) def registration_failed(self): # TODO registration fail protocol implement pass def reload_plugins(self): Scope.getInstance().getPluginManager().reloadPlugins() def reload_configuration(self): # Not implemented yet pass def reload_messaging(self): # Not implemented yet pass def reload_logger(self): # Not implemented yet pass def update_plugin_manager(self): # TODO destroy plugin manager here self.init_plugin_manager() def run(self): print('Ahenk running...') self.signal_number = 0 globalscope = Scope() globalscope.setInstance(globalscope) configfileFolderPath = '/etc/ahenk/config.d/' # configuration manager must be first load self.init_config_manager(System.Ahenk.config_path(), configfileFolderPath) # Logger must be second self.logger = self.init_logger() self.init_event_manager() self.logger.info('[AhenkDeamon] Event Manager was set') self.init_ahenk_db() self.logger.info('[AhenkDeamon] DataBase Service was set') self.init_messaging() self.logger.info('[AhenkDeamon] Message Manager was set') self.init_plugin_manager() self.logger.info('[AhenkDeamon] Plugin Manager was set') self.init_task_manager() self.logger.info('[AhenkDeamon] Task Manager was set') self.init_registration() self.logger.info('[AhenkDeamon] Registration was set') self.init_execution_manager() self.logger.info('[AhenkDeamon] Execution Manager was set') self.check_registration() self.logger.info('[AhenkDeamon] Ahenk is registered') self.messenger = self.init_messenger() self.logger.info('[AhenkDeamon] Messager was set') self.init_message_response_queue() # if registration.is_ldap_registered() is False: # logger.debug('[AhenkDeamon] Attempting to registering ldap') # registration.ldap_registration_request() #TODO work on message self.logger.info('[AhenkDeamon] LDAP registration of Ahenk is completed') with open(System.Ahenk.pid_path(), 'w+') as config_file: config_file.write(str(os.getpid())) try: signal.signal(signal.SIGALRM, self.run_command_from_fifo) self.logger.info('[AhenkDeamon] Signal handler is set up') except Exception as e: self.logger.error('[AhenkDeamon] Signal handler could not set up. Error Message: {} '.format(str(e))) self.messenger.send_direct_message('test') while True: # if messager.is_connected() is False: # self.logger.debug('reconnecting') # Scope.getInstance().getLogger().warning('[AhenkDeamon] Connection is lost. Ahenk is trying for reconnection') # messager = self.init_messager() time.sleep(1) def run_command_from_fifo(self, num, stack): json_data = json.loads(Commander().get_event()) if json_data is not None: scope = Scope().getInstance() plugin_manager = scope.getPluginManager() message_manager = scope.getMessageManager() messenger = scope.getMessenger() self.logger.debug('[AhenkDeamon] Signal handled') self.logger.debug('[AhenkDeamon] Signal is :{}'.format(str(json_data['event']))) print('event:{}'.format(str(json_data['event']))) if 'login' == str(json_data['event']): self.logger.info('[AhenkDeamon] login event is handled for user: {}'.format(json_data['username'])) login_message = message_manager.login_msg(json_data['username']) messenger.send_direct_message(login_message) get_policy_message = message_manager.policy_request_msg(json_data['username']) messenger.send_direct_message(get_policy_message) elif 'logout' == str(json_data['event']): self.logger.info('[AhenkDeamon] logout event is handled for user: {}'.format(str(json_data['username']))) logout_message = message_manager.logout_msg(json_data['username']) messenger.send_direct_message(logout_message) plugin_manager.process_safe_mode(str(json_data['username'])) elif 'send' == str(json_data['event']): self.logger.info('[AhenkDeamon] Sending message over ahenkd command. Response Message: {}'.format(str(json_data['message']))) message = str(json.dumps(json_data['message'])) messenger.send_direct_message(message) else: self.logger.error('[AhenkDeamon] Unknown command error. Command:' + json_data['event']) self.logger.debug('[AhenkDeamon] Processing of handled event is completed') return True else: return False if __name__ == '__main__': ahenkdaemon = AhenkDeamon(System.Ahenk.pid_path()) try: if len(sys.argv) == 2 and (sys.argv[1] == 'start' or sys.argv[1] == 'stop' or sys.argv[1] == 'restart' or sys.argv[1] == 'status'): if sys.argv[1] == 'start': if System.Ahenk.is_running() is True: print('There is running Ahenk service. It will be killed.') print(str(System.Ahenk.get_pid_number())) System.Process.kill_by_pid(int(System.Ahenk.get_pid_number())) else: print('Ahenk starting...') ahenkdaemon.run() elif sys.argv[1] == 'stop': if System.Ahenk.is_running() is True: print('Ahenk stopping...') ahenkdaemon.stop() else: print('Ahenk not working!') elif sys.argv[1] == 'restart': if System.Ahenk.is_running() is True: print('Ahenk restarting...') ahenkdaemon.restart() else: print('Ahenk starting...') ahenkdaemon.run() elif sys.argv[1] == 'status': print(Commander().status()) else: print('Unknown command. Usage : %s start|stop|restart|status|clean' % sys.argv[0]) sys.exit(2) else: result = Commander().set_event(sys.argv) if result is None: print('Usage : {0} start|stop|restart|status|clean'.format(sys.argv[0])) sys.exit(2) elif result is True: if System.Ahenk.is_running() is True: os.kill(int(System.Ahenk.get_pid_number()), signal.SIGALRM) except(KeyboardInterrupt, SystemExit): if System.Ahenk.is_running() is True: print('Ahenk will be closed.') ahenkdaemon.stop()