Liderahenk/Ahenk
1
0
Fork 0
mirror of https://github.com/Pardus-LiderAhenk/ahenk synced 2024-11-26 00:02:19 +03:00
Code Issues Projects Releases Packages Wiki Activity

fixes about merging message receiver-sender and new enities

Browse source
This commit is contained in:
Volkan Şahin 2016-03-18 18:42:36 +02:00
parent 8fabf0d47b
commit bfeb41c1e1
3 changed files with 21 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
opt/ahenk/base/execution/ExecutionManager.py
View file

@ -4,7 +4,6 @@
import subprocess import subprocess
from base.Scope import Scope from base.Scope import Scope
from base.messaging.MessageSender import MessageSender
from base.model.Task import Task from base.model.Task import Task
from base.model.Policy import Policy from base.model.Policy import Policy
import hashlib,json,os,stat,shutil import hashlib,json,os,stat,shutil
@ -19,13 +18,14 @@ class ExecutionManager(object):
self.config_manager = scope.getConfigurationManager() self.config_manager = scope.getConfigurationManager()
self.event_manager = scope.getEventManager() self.event_manager = scope.getEventManager()
self.task_manager = scope.getTaskManager() self.task_manager = scope.getTaskManager()
self.messager = scope.getMessager()
self.logger=scope.getLogger() self.logger=scope.getLogger()
self.db_service=scope.getDbService() self.db_service=scope.getDbService()
self.event_manager.register_event('EXECUTE_SCRIPT',self.execute_script) self.event_manager.register_event('EXECUTE_SCRIPT',self.execute_script)
self.event_manager.register_event('REQUEST_FILE',self.request_file) self.event_manager.register_event('REQUEST_FILE',self.request_file)
self.event_manager.register_event('MOVE_FILE',self.move_file) self.event_manager.register_event('MOVE_FILE',self.move_file)
self.event_manager.register_event('TASK',self.add_task) self.event_manager.register_event('EXECUTE_TASK',self.execute_task)
self.event_manager.register_event('POLICY',self.update_policies) self.event_manager.register_event('POLICY',self.update_policies)
def update_policies(self,arg): def update_policies(self,arg):
@ -47,8 +47,9 @@ class ExecutionManager(object):
self.db_service.update('policy',['version'],[str(policy.ahenk_policy_version)],'type=\'A\'') self.db_service.update('policy',['version'],[str(policy.ahenk_policy_version)],'type=\'A\'')
for profile in policy.ahenk_profiles: for profile in policy.ahenk_profiles:
profile_columns=['id','label','description','is_overridable','is_active','profile_data','modify_date'] profile_columns=['id','create_date','modify_date','label','description','overridable','active','deleted','profile_data','plugin']
args=[str(ahenk_policy_id[0][0]),str(profile.label),str(profile.description),str(profile.is_overridable),str(profile.is_active),str(profile.profile_data),str(profile.modify_date)] args=[str(ahenk_policy_id[0][0]),str(profile.create_date),str(profile.modify_date),str(profile.label),
str(profile.description),str(profile.overridable),str(profile.active),str(profile.deleted),str(profile.profile_data),str(profile.plugin)]
self.db_service.update('profile',profile_columns,args) self.db_service.update('profile',profile_columns,args)
if profile.plugin.name not in installed_plugins and profile.plugin.name not in missing_plugins: if profile.plugin.name not in installed_plugins and profile.plugin.name not in missing_plugins:
missing_plugins.append(profile.plugin.name) missing_plugins.append(profile.plugin.name)
@ -61,8 +62,9 @@ class ExecutionManager(object):
self.db_service.delete('profile','id='+str(user_policy_id[0][0])) self.db_service.delete('profile','id='+str(user_policy_id[0][0]))
self.db_service.update('policy',['version'],[str(policy.user_policy_version)],'type=\'U\' and name=\''+username+'\'') self.db_service.update('policy',['version'],[str(policy.user_policy_version)],'type=\'U\' and name=\''+username+'\'')
for profile in policy.user_profiles: for profile in policy.user_profiles:
profile_columns=['id','label','description','is_overridable','is_active','profile_data','modify_date'] profile_columns=['id','create_date','modify_date','label','description','overridable','active','deleted','profile_data','plugin']
args=[str(user_policy_id[0][0]),str(profile.label),str(profile.description),str(profile.is_overridable),str(profile.is_active),str(profile.profile_data),str(profile.modify_date)] args = [str(user_policy_id[0][0]),str(profile.create_date),str(profile.modify_date),str(profile.label),
str(profile.description),str(profile.overridable),str(profile.active),str(profile.deleted),str(profile.profile_data),str(profile.plugin)]
self.db_service.update('profile',profile_columns,args) self.db_service.update('profile',profile_columns,args)
if profile.plugin.name not in installed_plugins and profile.plugin.name not in missing_plugins: if profile.plugin.name not in installed_plugins and profile.plugin.name not in missing_plugins:
missing_plugins.append(profile.plugin.name) missing_plugins.append(profile.plugin.name)
@ -79,8 +81,7 @@ class ExecutionManager(object):
p_list.append(str(p[0])+'-'+str(p[1])) p_list.append(str(p[0])+'-'+str(p[1]))
return p_list return p_list
def execute_task(self,arg):
def add_task(self,arg):
self.logger.debug('[ExecutionManager] Adding new task...') self.logger.debug('[ExecutionManager] Adding new task...')
task = Task(arg) task = Task(arg)
self.task_manager.addTask(task) self.task_manager.addTask(task)
@ -111,8 +112,7 @@ class ExecutionManager(object):
file_path =str(j['filepath']).lower() file_path =str(j['filepath']).lower()
time_stamp=str(j['timestamp']).lower() time_stamp=str(j['timestamp']).lower()
self.logger.debug('[ExecutionManager] Requested file is '+file_path) self.logger.debug('[ExecutionManager] Requested file is '+file_path)
message_sender=MessageSender(None,file_path) self.messager.send_file(file_path)
message_sender.connect_to_server()
def get_md5_file(self,fname): def get_md5_file(self,fname):
self.logger.debug('[ExecutionManager] md5 hashing') self.logger.debug('[ExecutionManager] md5 hashing')

23
opt/ahenk/base/messaging/Messaging.py
View file

@ -19,32 +19,15 @@ class Messaging(object):
# TODO can use sh commands or api for getting username and timestamp # TODO can use sh commands or api for getting username and timestamp
def policy_request_msg(self): def policy_request_msg(self):
#TODO volkan #TODO volkan
self.logger.debug('[Messaging] Creating policy request message') self.logger.debug('[Messaging] Creating policy request message')
############# Create policy tables ######################### ahenk_version=self.db_service.select('policy',['version'],'type = \'A\'')
columns=['id INTEGER PRIMARY KEY AUTOINCREMENT','type TEXT','version TEXT','name TEXT']
self.db_service.check_and_create_table('policy',columns)
columns=['id INTEGER','label TEXT','description TEXT','is_overridable INTEGER','is_active INTEGER','profile_data BLOB','modify_date TEXT']
self.db_service.check_and_create_table('profile',columns)
columns=['version TEXT','name TEXT','description TEXT']
self.db_service.check_and_create_table('plugin',columns)
############################################################
#cols=['type','version','name']
#args=['U','1','2559305d-a415-38e7-8498-2dbc458662a7']
#self.db_service.update('policy',cols,args,None)
colz=['version']
ahenk_version=self.db_service.select('policy',colz,'type = \'A\'')
username='volkan' username='volkan'
user_version=self.db_service.select('policy',colz,'type = \'U\' and name = \''+username+'\'') user_version=self.db_service.select('policy',['version'],'type = \'U\' and name = \''+username+'\'')
if len(ahenk_version)==0: if len(ahenk_version)==0:
ahenk_version.append(-1) ahenk_version.append(-1)
if len(user_version)==0: if len(user_version)==0:

14
opt/ahenk/base/registration/Registration.py
View file

@ -3,7 +3,7 @@
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com> # Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
from base.Scope import Scope from base.Scope import Scope
from base.messaging.MessageSender import MessageSender from base.messaging.AnonymousMessager import AnonymousMessager
from uuid import getnode as get_mac from uuid import getnode as get_mac
import json, uuid, netifaces, socket, datetime import json, uuid, netifaces, socket, datetime
@ -17,6 +17,7 @@ class Registration():
self.logger=scope.getLogger() self.logger=scope.getLogger()
self.message_manager=scope.getMessageManager() self.message_manager=scope.getMessageManager()
self.event_manager = scope.getEventManager() self.event_manager = scope.getEventManager()
self.messager = scope.getMessager()
self.event_manager.register_event('confirm_registration',self.confirm_registration) self.event_manager.register_event('confirm_registration',self.confirm_registration)
@ -30,13 +31,12 @@ class Registration():
def registration_request(self): def registration_request(self):
self.logger.debug('[Registration] Requesting registration') self.logger.debug('[Registration] Requesting registration')
message_sender=MessageSender(self.message_manager.registration_msg(),None) anon_messager = AnonymousMessager(self.message_manager.registration_msg(),None)
message_sender.connect_to_server() anon_messager.connect_to_server()
def ldap_registration_request(self): def ldap_registration_request(self):
self.logger.debug('[Registration] Requesting LDAP registration') self.logger.debug('[Registration] Requesting LDAP registration')
message_sender=MessageSender(self.message_manager.ldap_registration_msg(),None) self.messager.send_Direct_message(self.message_manager.ldap_registration_msg())
message_sender.connect_to_server()
def confirm_registration(self,reg_reply): def confirm_registration(self,reg_reply):
self.logger.debug('[Registration] Reading registration reply') self.logger.debug('[Registration] Reading registration reply')
@ -65,6 +65,7 @@ class Registration():
self.conf_manager.set('CONNECTION', 'password',self.conf_manager.get('REGISTRATION', 'password')) self.conf_manager.set('CONNECTION', 'password',self.conf_manager.get('REGISTRATION', 'password'))
self.conf_manager.set('REGISTRATION', 'dn',dn) self.conf_manager.set('REGISTRATION', 'dn',dn)
self.conf_manager.set('REGISTRATION', 'registered','true') self.conf_manager.set('REGISTRATION', 'registered','true')
#TODO get file path?
with open('/etc/ahenk/ahenk.conf', 'w') as configfile: with open('/etc/ahenk/ahenk.conf', 'w') as configfile:
self.conf_manager.write(configfile) self.conf_manager.write(configfile)
self.logger.debug('[Registration] Registration configuration file is updated') self.logger.debug('[Registration] Registration configuration file is updated')
@ -98,7 +99,7 @@ class Registration():
self.conf_manager.set('REGISTRATION', 'dn','') self.conf_manager.set('REGISTRATION', 'dn','')
self.conf_manager.set('REGISTRATION', 'registered','false') self.conf_manager.set('REGISTRATION', 'registered','false')
#TODO self.conf_manager.configurationFilePath attribute error ? READ olacak #TODO get file path?
self.logger.debug('[Registration] Parameters were set up, section will write to configuration file') self.logger.debug('[Registration] Parameters were set up, section will write to configuration file')
with open('/etc/ahenk/ahenk.conf', 'w') as configfile: with open('/etc/ahenk/ahenk.conf', 'w') as configfile:
self.conf_manager.write(configfile) self.conf_manager.write(configfile)
@ -108,6 +109,7 @@ class Registration():
self.logger.debug('[Registration] Ahenk is unregistering...') self.logger.debug('[Registration] Ahenk is unregistering...')
if self.conf_manager.has_section('REGISTRATION'): if self.conf_manager.has_section('REGISTRATION'):
#TODO open this block if you want to be aware about unregistration #TODO open this block if you want to be aware about unregistration
#TODO messaging thread must be terminated
#message_sender=MessageSender(self.message_manager.unregister_msg(),None) #message_sender=MessageSender(self.message_manager.unregister_msg(),None)
#message_sender.connect_to_server() #message_sender.connect_to_server()