Example #1
0
    def plugin_installation_failure(self, plugin_name, plugin_version):

        self.logger.warning('{0} plugin installation failure '.format(plugin_name))

        if plugin_name in self.plugin_manager.delayed_profiles.keys():
            profile = self.plugin_manager.delayed_profiles[plugin_name]
            self.logger.warning('An error message sending with related profile properties...')
            related_policy = self.db_service.select('policy', ['version', 'execution_id'],
                                                    'id={0}'.format(profile.get_id()))
            data = dict()
            data['message'] = "Profil işletilirken eklenti bulunamadı "
            "ve eksik olan eklenti kurulmaya çalışırken hata ile karşılaşıldı. "
            "İlgili eklenti Ahenk'e yüklendiğinde, başarısız olan bu profil "
            "(Başka bir politika tarafından ezilmedikçe) "
            "çalıştırılacaktır"
            " Sorunu çözmek için Lider yapılandırma dosyasındaki eklenti dağıtım "
            "bilgilerinin doğruluğundan ve belirtilen dizinde geçerli eklenti paketinin "
            "bulunduğundan emin olun."
            response = Response(type=MessageType.POLICY_STATUS.value, id=profile.get_id(),
                                code=MessageCode.POLICY_ERROR.value,
                                message="Profil işletilirken eklenti bulunamadı "
                                        "ve eksik olan eklenti kurulurken hata oluştu",
                                execution_id=related_policy[0][1], policy_version=related_policy[0][0],
                                data=json.dumps(data), content_type=ContentType.APPLICATION_JSON.value)
            messenger = Scope.get_instance().get_messenger()
            messenger.send_direct_message(self.message_manager.policy_status_msg(response))
            self.logger.warning(
                'Error message was sent about {0} plugin installation failure while trying to run a profile')

        if plugin_name in self.plugin_manager.delayed_tasks.keys():
            task = self.plugin_manager.delayed_tasks[plugin_name]
            self.logger.warning('An error message sending with related task properties...')

            data = dict()
            data['message'] = "Görev işletilirken eklenti bulunamadı "
            "ve eksik olan eklenti kurulmaya çalışırken hata ile karşılaşıldı. "
            "İlgili eklenti Ahenk'e yüklendiğinde, başarısız olan bu görev "
            "çalıştırılacaktır"
            " Sorunu çözmek için Lider yapılandırma dosyasındaki eklenti dağıtım "
            "bilgilerinin doğruluğundan ve belirtilen dizinde geçerli eklenti paketinin "
            "bulunduğundan emin olun."
            response = Response(type=MessageType.TASK_STATUS.value, id=task.get_id(),
                                code=MessageCode.TASK_ERROR.value,
                                message="Görev işletilirken eklenti bulunamadı "
                                        "ve eksik olan eklenti kurulmaya çalışırken oluştu.",
                                data=json.dumps(data), content_type=ContentType.APPLICATION_JSON.value)
            messenger = Scope.get_instance().get_messenger()
            messenger.send_direct_message(self.message_manager.task_status_msg(response))
            self.logger.warning(
                'Error message was sent about {0} plugin installation failure while trying to run a task')
Example #2
0
    def run(self):

        while self.keep_run:
            try:
                try:
                    item_obj = self.in_queue.get(block=True)
                    obj_name = item_obj.obj_name
                except Exception as e:
                    self.logger.error(
                        '[Plugin] A problem occurred while executing process. Error Message: {0}'
                        .format(str(e)))

                if obj_name == "TASK":
                    self.logger.debug('[Plugin] Executing task')
                    command = Scope.get_instance().get_plugin_manager(
                    ).find_command(self.getName(),
                                   item_obj.get_plugin().get_version(),
                                   item_obj.get_command_cls_id().lower())
                    self.context.put('task_id', item_obj.get_id())

                    if item_obj.get_file_server(
                    ) is not None and item_obj.get_file_server() != 'null':
                        self.context.put(
                            'protocol',
                            json.loads(item_obj.get_file_server())['protocol'])
                        self.context.put(
                            'parameterMap',
                            json.loads(
                                item_obj.get_file_server())['parameterMap'])

                    task_data = item_obj.get_parameter_map()

                    self.logger.debug(
                        '[Plugin] Sending notify to user about task process')

                    if System.Sessions.user_name() is not None and len(
                            System.Sessions.user_name()) > 0:
                        for user in System.Sessions.user_name():
                            Util.send_notify(
                                "Lider Ahenk",
                                "{0} eklentisi şu anda bir görev çalıştırıyor."
                                .format(self.getName()),
                                System.Sessions.display(user), user)

                    self.logger.debug('[Plugin] Handling task')
                    command.handle_task(task_data, self.context)

                    if self.context.data is not None and self.context.get(
                            'responseCode') is not None:
                        self.logger.debug('[Plugin] Creating response')
                        response = Response(
                            type=MessageType.TASK_STATUS.value,
                            id=item_obj.get_id(),
                            code=self.context.get('responseCode'),
                            message=self.context.get('responseMessage'),
                            data=self.context.get('responseData'),
                            content_type=self.context.get('contentType'))

                        if response.get_data() and response.get_content_type(
                        ) != ContentType.APPLICATION_JSON.value:
                            success = False
                            try:
                                file_manager = FileTransferManager(
                                    json.loads(item_obj.get_file_server())
                                    ['protocol'],
                                    json.loads(item_obj.get_file_server())
                                    ['parameterMap'])
                                file_manager.transporter.connect()
                                md5 = str(
                                    json.loads(response.get_data())['md5'])
                                success = file_manager.transporter.send_file(
                                    System.Ahenk.received_dir_path() + md5,
                                    md5)
                                file_manager.transporter.disconnect()
                            except Exception as e:
                                self.logger.error(
                                    '[Plugin] A problem occurred while file transferring. Error Message :{0}'
                                    .format(str(e)))

                            self.logger.debug('[Plugin] Sending response')

                            message = self.messaging.task_status_msg(response)
                            if success is False:
                                response = Response(
                                    type=MessageType.TASK_STATUS.value,
                                    id=item_obj.get_id(),
                                    code=MessageCode.TASK_ERROR.value,
                                    message=
                                    'Task processed successfully but file transfer not completed. Check defined server conf'
                                )
                                message = self.messaging.task_status_msg(
                                    response)

                            Scope.get_instance().get_messenger(
                            ).send_direct_message(message)

                        else:
                            self.logger.debug('[Plugin] Sending task response')
                            Scope.get_instance().get_messenger(
                            ).send_direct_message(
                                self.messaging.task_status_msg(response))

                    else:
                        self.logger.error(
                            '[Plugin] There is no Response. Plugin must create response after run a task!'
                        )

                elif obj_name == "PROFILE":

                    self.logger.debug('[Plugin] Executing profile')
                    profile_data = item_obj.get_profile_data()
                    policy_module = Scope.get_instance().get_plugin_manager(
                    ).find_policy_module(item_obj.get_plugin().get_name(),
                                         item_obj.get_plugin().get_version())
                    self.context.put('username', item_obj.get_username())

                    execution_id = self.get_execution_id(item_obj.get_id())
                    policy_ver = self.get_policy_version(item_obj.get_id())

                    self.context.put('policy_version', policy_ver)
                    self.context.put('execution_id', execution_id)

                    # if item_obj.get_file_server() is not None  and item_obj.get_file_server() !='null':
                    #     self.context.put('protocol', json.loads(item_obj.get_file_server())['protocol'])
                    #     self.context.put('parameterMap', json.loads(item_obj.get_file_server())['parameterMap'])

                    self.logger.debug(
                        '[Plugin] Sending notify to user about profile process'
                    )

                    Util.send_notify(
                        "Lider Ahenk",
                        "{0} eklentisi şu anda bir profil çalıştırıyor.".
                        format(self.getName()),
                        System.Sessions.display(item_obj.get_username()),
                        item_obj.get_username())
                    self.logger.debug('[Plugin] Handling profile')
                    policy_module.handle_policy(profile_data, self.context)

                    if self.context.data is not None and self.context.get(
                            'responseCode') is not None:
                        self.logger.debug('[Plugin] Creating response')
                        response = Response(
                            type=MessageType.POLICY_STATUS.value,
                            id=item_obj.get_id(),
                            code=self.context.get('responseCode'),
                            message=self.context.get('responseMessage'),
                            data=self.context.get('responseData'),
                            content_type=self.context.get('contentType'),
                            execution_id=execution_id,
                            policy_version=policy_ver)

                        if response.get_data() and response.get_content_type(
                        ) != ContentType.APPLICATION_JSON.value:
                            success = False
                            try:
                                file_manager = FileTransferManager(
                                    json.loads(item_obj.get_file_server())
                                    ['protocol'],
                                    json.loads(item_obj.get_file_server())
                                    ['parameterMap'])
                                file_manager.transporter.connect()
                                md5 = str(
                                    json.loads(response.get_data())['md5'])
                                success = file_manager.transporter.send_file(
                                    System.Ahenk.received_dir_path() + md5,
                                    md5)
                                file_manager.transporter.disconnect()
                            except Exception as e:
                                self.logger.error(
                                    '[Plugin] A problem occurred while file transferring. Error Message :{0}'
                                    .format(str(e)))

                            self.logger.debug('[Plugin] Sending response')

                            message = self.messaging.task_status_msg(response)
                            if success is False:
                                response = Response(
                                    type=MessageType.POLICY_STATUS.value,
                                    id=item_obj.get_id(),
                                    code=MessageCode.POLICY_ERROR.value,
                                    message=
                                    'Policy processed successfully but file transfer not completed. Check defined server conf'
                                )
                                message = self.messaging.task_status_msg(
                                    response)
                            Scope.get_instance().get_messenger(
                            ).send_direct_message(message)
                        else:
                            self.logger.debug(
                                '[Plugin] Sending profile response')
                            Scope.get_instance().get_messenger(
                            ).send_direct_message(
                                self.messaging.policy_status_msg(response))
                    else:
                        self.logger.error(
                            '[Plugin] There is no Response. Plugin must create response after run a policy!'
                        )
                elif 'MODE' in obj_name:
                    module = Scope.get_instance().get_plugin_manager(
                    ).find_module(obj_name, self.name)
                    if module is not None:
                        if item_obj.obj_name in ('LOGIN_MODE', 'LOGOUT_MODE',
                                                 'SAFE_MODE'):
                            self.context.put('username', item_obj.username)
                        try:
                            self.logger.debug(
                                '[Plugin] {0} is running on {1} plugin'.format(
                                    str(item_obj.obj_name), str(self.name)))
                            module.handle_mode(self.context)
                        except Exception as e:
                            self.logger.error(
                                '[Plugin] A problem occurred while running {0} on {1} plugin. Error Message: {2}'
                                .format(str(obj_name), str(self.name), str(e)))

                    if item_obj.obj_name is 'SHUTDOWN_MODE':
                        self.logger.debug(
                            '[Plugin] {0} plugin is stopping...'.format(
                                str(self.name)))
                        self.keep_run = False
                else:
                    self.logger.warning(
                        "[Plugin] Not supported object type: {0}".format(
                            str(obj_name)))

                self.context.empty_data()
            except Exception as e:
                self.logger.error(
                    "[Plugin] Plugin running exception. Exception Message: {0} "
                    .format(str(e)))
Example #3
0
    def run(self):

        while self.keep_run:
            try:
                try:
                    item_obj = self.in_queue.get(block=True)
                    obj_name = item_obj.obj_name
                except Exception as e:
                    self.logger.error(
                        "[Plugin] A problem occurred while executing process. Error Message: {0}".format(str(e))
                    )

                if obj_name == "TASK":
                    self.logger.debug("[Plugin] Executing task")
                    command = (
                        Scope.get_instance()
                        .get_plugin_manager()
                        .find_command(
                            self.getName(), item_obj.get_plugin().get_version(), item_obj.get_command_cls_id().lower()
                        )
                    )
                    self.context.put("task_id", item_obj.get_id())

                    if item_obj.get_file_server() is not None and item_obj.get_file_server() != "null":
                        self.context.put("protocol", json.loads(item_obj.get_file_server())["protocol"])
                        self.context.put("parameterMap", json.loads(item_obj.get_file_server())["parameterMap"])

                    task_data = item_obj.get_parameter_map()

                    self.logger.debug("[Plugin] Sending notify to user about task process")

                    if System.Sessions.user_name() is not None and len(System.Sessions.user_name()) > 0:
                        for user in System.Sessions.user_name():
                            Util.send_notify(
                                "Lider Ahenk",
                                "{0} eklentisi şu anda bir görev çalıştırıyor.".format(self.getName()),
                                System.Sessions.display(user),
                                user,
                            )

                    self.logger.debug("[Plugin] Handling task")
                    command.handle_task(task_data, self.context)

                    if self.context.data is not None and self.context.get("responseCode") is not None:
                        self.logger.debug("[Plugin] Creating response")
                        response = Response(
                            type=MessageType.TASK_STATUS.value,
                            id=item_obj.get_id(),
                            code=self.context.get("responseCode"),
                            message=self.context.get("responseMessage"),
                            data=self.context.get("responseData"),
                            content_type=self.context.get("contentType"),
                        )

                        if response.get_data() and response.get_content_type() != ContentType.APPLICATION_JSON.value:
                            success = False
                            try:
                                file_manager = FileTransferManager(
                                    json.loads(item_obj.get_file_server())["protocol"],
                                    json.loads(item_obj.get_file_server())["parameterMap"],
                                )
                                file_manager.transporter.connect()
                                md5 = str(json.loads(response.get_data())["md5"])
                                success = file_manager.transporter.send_file(
                                    System.Ahenk.received_dir_path() + md5, md5
                                )
                                file_manager.transporter.disconnect()
                            except Exception as e:
                                self.logger.error(
                                    "[Plugin] A problem occurred while file transferring. Error Message :{0}".format(
                                        str(e)
                                    )
                                )

                            self.logger.debug("[Plugin] Sending response")

                            message = self.messaging.task_status_msg(response)
                            if success is False:
                                response = Response(
                                    type=MessageType.TASK_STATUS.value,
                                    id=item_obj.get_id(),
                                    code=MessageCode.TASK_ERROR.value,
                                    message="Task processed successfully but file transfer not completed. Check defined server conf",
                                )
                                message = self.messaging.task_status_msg(response)

                            Scope.get_instance().get_messenger().send_direct_message(message)

                        else:
                            self.logger.debug("[Plugin] Sending task response")
                            Scope.get_instance().get_messenger().send_direct_message(
                                self.messaging.task_status_msg(response)
                            )

                    else:
                        self.logger.error(
                            "[Plugin] There is no Response. Plugin must create response after run a task!"
                        )

                elif obj_name == "PROFILE":

                    self.logger.debug("[Plugin] Executing profile")
                    profile_data = item_obj.get_profile_data()
                    policy_module = (
                        Scope.get_instance()
                        .get_plugin_manager()
                        .find_policy_module(item_obj.get_plugin().get_name(), item_obj.get_plugin().get_version())
                    )
                    self.context.put("username", item_obj.get_username())

                    execution_id = self.get_execution_id(item_obj.get_id())
                    policy_ver = self.get_policy_version(item_obj.get_id())

                    self.context.put("policy_version", policy_ver)
                    self.context.put("execution_id", execution_id)

                    # if item_obj.get_file_server() is not None  and item_obj.get_file_server() !='null':
                    #     self.context.put('protocol', json.loads(item_obj.get_file_server())['protocol'])
                    #     self.context.put('parameterMap', json.loads(item_obj.get_file_server())['parameterMap'])

                    self.logger.debug("[Plugin] Sending notify to user about profile process")

                    Util.send_notify(
                        "Lider Ahenk",
                        "{0} eklentisi şu anda bir profil çalıştırıyor.".format(self.getName()),
                        System.Sessions.display(item_obj.get_username()),
                        item_obj.get_username(),
                    )
                    self.logger.debug("[Plugin] Handling profile")
                    policy_module.handle_policy(profile_data, self.context)

                    if self.context.data is not None and self.context.get("responseCode") is not None:
                        self.logger.debug("[Plugin] Creating response")
                        response = Response(
                            type=MessageType.POLICY_STATUS.value,
                            id=item_obj.get_id(),
                            code=self.context.get("responseCode"),
                            message=self.context.get("responseMessage"),
                            data=self.context.get("responseData"),
                            content_type=self.context.get("contentType"),
                            execution_id=execution_id,
                            policy_version=policy_ver,
                        )

                        if response.get_data() and response.get_content_type() != ContentType.APPLICATION_JSON.value:
                            success = False
                            try:
                                file_manager = FileTransferManager(
                                    json.loads(item_obj.get_file_server())["protocol"],
                                    json.loads(item_obj.get_file_server())["parameterMap"],
                                )
                                file_manager.transporter.connect()
                                md5 = str(json.loads(response.get_data())["md5"])
                                success = file_manager.transporter.send_file(
                                    System.Ahenk.received_dir_path() + md5, md5
                                )
                                file_manager.transporter.disconnect()
                            except Exception as e:
                                self.logger.error(
                                    "[Plugin] A problem occurred while file transferring. Error Message :{0}".format(
                                        str(e)
                                    )
                                )

                            self.logger.debug("[Plugin] Sending response")

                            message = self.messaging.task_status_msg(response)
                            if success is False:
                                response = Response(
                                    type=MessageType.POLICY_STATUS.value,
                                    id=item_obj.get_id(),
                                    code=MessageCode.POLICY_ERROR.value,
                                    message="Policy processed successfully but file transfer not completed. Check defined server conf",
                                )
                                message = self.messaging.task_status_msg(response)
                            Scope.get_instance().get_messenger().send_direct_message(message)
                        else:
                            self.logger.debug("[Plugin] Sending profile response")
                            Scope.get_instance().get_messenger().send_direct_message(
                                self.messaging.policy_status_msg(response)
                            )
                    else:
                        self.logger.error(
                            "[Plugin] There is no Response. Plugin must create response after run a policy!"
                        )
                elif "MODE" in obj_name:
                    module = Scope.get_instance().get_plugin_manager().find_module(obj_name, self.name)
                    if module is not None:
                        if item_obj.obj_name in ("LOGIN_MODE", "LOGOUT_MODE", "SAFE_MODE"):
                            self.context.put("username", item_obj.username)
                        try:
                            self.logger.debug(
                                "[Plugin] {0} is running on {1} plugin".format(str(item_obj.obj_name), str(self.name))
                            )
                            module.handle_mode(self.context)
                        except Exception as e:
                            self.logger.error(
                                "[Plugin] A problem occurred while running {0} on {1} plugin. Error Message: {2}".format(
                                    str(obj_name), str(self.name), str(e)
                                )
                            )

                    if item_obj.obj_name is "SHUTDOWN_MODE":
                        self.logger.debug("[Plugin] {0} plugin is stopping...".format(str(self.name)))
                        self.keep_run = False
                else:
                    self.logger.warning("[Plugin] Not supported object type: {0}".format(str(obj_name)))

                self.context.empty_data()
            except Exception as e:
                self.logger.error("[Plugin] Plugin running exception. Exception Message: {0} ".format(str(e)))