from tools.ToolBase import ToolBase

class EchoTool(ToolBase):
    def validate_instruction(self, instruction):
        #指令过滤
        timeout = 0
        return instruction,timeout

    def analyze_result(self, result,instruction,stderr,stdout):
        #指令结果分析
        if "GET / HTTP/1.1" in result and "X-Original-URL: /proc/self/environ" in result:
            #通过构造 ‌自定义HTTP请求头‌ 尝试利用服务器配置漏洞,访问敏感文件
            if "HTTP/1.1 200" in result and "PATH=" in result:
                #result = "存在安全问题"  #暂时保留结果
                pass
            else:
                result ="不存在安全问题"
        else:#未预处理的情况,暂时不返回LLM
            pass

        return result