def write_to_global_profile(self):
process = subprocess.Popen(self.command_get_users_currently_login,
stdin=None,
env=None,
cwd=None,
stderr=subprocess.PIPE,
stdout=subprocess.PIPE,
shell=True)
process.wait()
p_out = process.stdout.read().decode("unicode_escape")
users = []
if p_out != None:
users = str(p_out).split('\n')
users.pop()
if 'root' in users:
users.remove('root')
logging.debug('Logged-in users: {0}'.format(users))
if str(self.today) in self.days:
if not (self.start_minute < self.current_minute < self.end_minute
and self.current_date <= self.last_date):
for user in users:
logging.debug('User \'{0}\' will log out.'.format(user))
process = subprocess.Popen(
self.command_logout_user.format(user),
stdin=None,
env=None,
cwd=None,
stderr=subprocess.PIPE,
stdout=subprocess.PIPE,
shell=True)
process.wait()
elif int(self.end_minute) - int(self.current_minute) == int(
self.duration):
for user in users:
Util.send_notify(
'Lider Ahenk',
'Oturum {0} dakika sonra kapatılacak!'.format(
str(self.duration)), ':0', user)
pass
else:
for user in users:
Util.send_notify('Lider Ahenk', 'Oturum şimdi kapatılacak!',
':0', user)
logging.debug('User \'{0}\' will log out.'.format(user))
process = subprocess.Popen(
self.command_logout_user.format(user),
stdin=None,
env=None,
cwd=None,
stderr=subprocess.PIPE,
stdout=subprocess.PIPE,
shell=True)
process.wait()
def write_to_user_profile(self):
logging.debug("---->>> " + str(self.username))
if str(self.today) in self.days:
if not (self.start_minute < self.current_minute < self.end_minute
and self.current_date <= self.last_date):
logging.debug('User \'{0}\' will log out.'.format(
self.username))
process = subprocess.Popen(self.command_logout_user.format(
self.username),
stdin=None,
env=None,
cwd=None,
stderr=subprocess.PIPE,
stdout=subprocess.PIPE,
shell=True)
process.wait()
elif int(self.end_minute) - int(self.current_minute) == int(
self.duration):
Util.send_notify(
'Lider Ahenk',
'Oturum {0} dakika sonra kapatılacak!'.format(
str(self.duration)), ':0', self.username)
else:
Util.send_notify('Lider Ahenk', 'Oturum şimdi kapatılacak!', ':0',
self.username)
time.sleep(5)
logging.debug('User \'{0}\' will log out.'.format(self.username))
process = subprocess.Popen(self.command_logout_user.format(
self.username),
stdin=None,
env=None,
cwd=None,
stderr=subprocess.PIPE,
stdout=subprocess.PIPE,
shell=True)
process.wait()
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)))
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)))
