import os
import shlex
from tools.ToolBase import ToolBase

class SmugglerTool(ToolBase):
    def validate_instruction(self, instruction_old):
        timeout = 0
        #指令过滤
        # 获取当前程序所在目录
        current_path = os.path.dirname(os.path.realpath(__file__))
        # 分割指令为参数列表
        newcmd = f"python {current_path}/../payload/smuggler-master/smuggler.py"
        instruction = instruction_old.replace("smuggler",newcmd)
        return instruction,timeout

    def analyze_result(self, result,instruction,stderr,stdout):
        #指令结果分析
        return result