103 lines
4.0 KiB
Python
103 lines
4.0 KiB
Python
# !/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
# Author: Hasan Kara
|
||
|
||
import json
|
||
|
||
from base.plugin.abstract_plugin import AbstractPlugin
|
||
|
||
|
||
class Script(AbstractPlugin):
|
||
def __init__(self, data, context):
|
||
super(Script, self).__init__()
|
||
self.data = data
|
||
self.context = context
|
||
self.logger = self.get_logger()
|
||
self.machine_profile = True
|
||
self.data = data
|
||
self.context = context
|
||
self.logger = self.get_logger()
|
||
self.message_code = self.get_message_code()
|
||
|
||
self.temp_file_name = str(self.generate_uuid())
|
||
self.base_file_path = '{0}{1}'.format(str(self.Ahenk.received_dir_path()), self.temp_file_name)
|
||
|
||
def handle_policy(self):
|
||
try:
|
||
json_data = json.loads(self.data)
|
||
|
||
script_type = str(json_data['SCRIPT_TYPE'])
|
||
script_contents = self.format_contents(str(json_data['SCRIPT_CONTENTS']))
|
||
script_params = str(json_data['SCRIPT_PARAMS'])
|
||
file_path = self.base_file_path
|
||
|
||
# Determine script extension and command
|
||
command = ''
|
||
if script_type == 'BASH':
|
||
file_path += '.sh'
|
||
command += 'bash'
|
||
elif script_type == 'PYTHON':
|
||
file_path += '.py'
|
||
command += 'python'
|
||
elif script_type == 'PERL':
|
||
file_path += '.pl'
|
||
command += 'perl'
|
||
elif script_type == 'RUBY':
|
||
file_path += '.rb'
|
||
command += 'ruby'
|
||
|
||
self.create_script_file(file_path, script_contents)
|
||
|
||
result_code, p_out, p_err = self.execute_script_file(command, file_path, script_params)
|
||
if result_code != 0:
|
||
self.logger.error("Error occurred while executing script: " + str(p_err))
|
||
self.context.create_response(code=self.message_code.TASK_ERROR.value,
|
||
message='Betik çalıştırılırken hata oluştu',
|
||
data=json.dumps({'Result': p_err}),
|
||
content_type=self.get_content_type().APPLICATION_JSON.value)
|
||
else:
|
||
self.logger.debug("Executed script file.")
|
||
self.context.create_response(code=self.message_code.TASK_PROCESSED.value,
|
||
message='Betik başarıyla çalıştırıldı.',
|
||
data=json.dumps({'Result': p_out}),
|
||
content_type=self.get_content_type().APPLICATION_JSON.value)
|
||
|
||
except Exception as e:
|
||
self.logger.error(str(e))
|
||
self.context.create_response(code=self.message_code.TASK_ERROR.value,
|
||
message='Betik çalıştırılırken hata oluştu:' + str(e),
|
||
content_type=self.get_content_type().APPLICATION_JSON.value)
|
||
|
||
def create_script_file(self, file_path, script_contents):
|
||
self.logger.debug("Creating script file.")
|
||
# Create temporary script file with the provided content
|
||
self.write_file(file_path, script_contents)
|
||
|
||
# Make the file executable
|
||
self.make_executable(file_path)
|
||
self.logger.debug("Created script file: {0}".format(file_path))
|
||
|
||
def execute_script_file(self, command, file_path, script_params):
|
||
self.logger.debug("Executing script file.")
|
||
# Execute the file
|
||
if not script_params:
|
||
return self.execute('{0} {1}'.format(command, file_path))
|
||
else:
|
||
return self.execute('{0} {1} {2}'.format(command, file_path, script_params))
|
||
|
||
@staticmethod
|
||
def format_contents(contents):
|
||
tmp = contents
|
||
replacements = list()
|
||
replacements.append(('&dbq;', '\"'))
|
||
replacements.append(('&sgq;', '\''))
|
||
for r, n in replacements:
|
||
tmp = tmp.replace(r, n)
|
||
return tmp
|
||
|
||
|
||
|
||
def handle_policy(profile_data, context):
|
||
plugin = Script(profile_data, context)
|
||
plugin.handle_policy()
|