mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-11-30 02:02:27 +03:00
93 lines
3.9 KiB
Python
93 lines
3.9 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
# Author: İsmail BAŞARAN <ismail.basaran@tubitak.gov.tr> <basaran.ismaill@gmail.com>
|
|
import threading
|
|
import subprocess
|
|
from base.Scope import Scope
|
|
from base.model.Response import Response
|
|
from base.model.MessageType import MessageType
|
|
from base.model.MessageCode import MessageCode
|
|
from base.messaging.Messaging import Messaging
|
|
|
|
|
|
class Context(object):
|
|
def __init__(self):
|
|
self.data = {}
|
|
|
|
def put(self, var_name, data):
|
|
self.data[var_name] = data
|
|
|
|
def get(self, var_name):
|
|
return self.data[var_name]
|
|
|
|
def empty_data(self):
|
|
self.data = {}
|
|
|
|
def execute(self, command):
|
|
return subprocess.Popen(command, shell=True)
|
|
|
|
|
|
class Plugin(threading.Thread):
|
|
"""
|
|
This is a thread inherit class and have a queue.
|
|
Plugin class responsible for processing TASK or USER PLUGIN PROFILE.
|
|
"""
|
|
|
|
def __init__(self, name, InQueue):
|
|
threading.Thread.__init__(self)
|
|
self.name = name
|
|
self.InQueue = InQueue
|
|
|
|
scope = Scope.getInstance()
|
|
self.logger = scope.getLogger()
|
|
self.response_queue = scope.getResponseQueue()
|
|
self.messaging = scope.getMessageManager()
|
|
self.messager =None
|
|
|
|
self.keep_run = True
|
|
self.context = Context()
|
|
|
|
def run(self):
|
|
|
|
while self.keep_run:
|
|
try:
|
|
item_obj = self.InQueue.get(block=True)
|
|
obj_name = item_obj.obj_name
|
|
print(obj_name)
|
|
if obj_name == "TASK":
|
|
command = Scope.getInstance().getPluginManager().findCommand(self.getName(), item_obj.command_cls_id)
|
|
command.handle_task(item_obj, self.context)
|
|
# TODO create response message from context and item_obj. item_obj is task
|
|
|
|
#TODO Message Code keep
|
|
response = Response(type=MessageType.TASK_STATUS, id=item_obj.id, code=MessageCode.TASK_PROCESSED, message='__message__', data=self.context.get('data'), content_type=self.context.get('content_type'))
|
|
#self.response_queue.put(self.messaging.response_msg(response)) #TODO DEBUG
|
|
Scope.getInstance().getMessager().send_direct_message(self.messaging.response_msg(response)) #TODO REMOVE
|
|
|
|
# Empty context for next use
|
|
self.context.empty_data()
|
|
|
|
elif obj_name == "PROFILE":
|
|
profile_data = item_obj.get_profile_data()
|
|
policy_module = Scope.getInstance().getPluginManager().findPolicyModule(item_obj.get_plugin().get_name())
|
|
self.context.put('username', item_obj.get_username())
|
|
policy_module.handle_policy(profile_data, self.context)
|
|
#TODO Message Code keep Set default message if not exist
|
|
response = Response(type=self.context.get('message_type'), id=item_obj.id, code=self.context.get('message_code'), message=self.context.get('message'), data=self.context.get('data'), content_type=self.context.get('content_type'), execution_id='get_execution_id')
|
|
#self.response_queue.put(self.messaging.response_msg(response)) #TODO DEBUG
|
|
Scope.getInstance().getMessager().send_direct_message(self.messaging.response_msg(response))#TODO REMOVE
|
|
|
|
# Empty context for next use
|
|
self.context.empty_data()
|
|
|
|
elif obj_name == "KILL_SIGNAL":
|
|
self.keep_run = False
|
|
self.logger.debug('[Plugin] Killing queue ! Plugin Name : ' + str(self.name))
|
|
else:
|
|
self.logger.warning("[Plugin] Not supported object type " + obj_name)
|
|
except Exception as e:
|
|
# TODO error log here
|
|
self.logger.error("[Plugin] Plugin running exception " + str(e))
|
|
|
|
def getName(self):
|
|
return self.name
|