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()
|