Java运行自定义的groovy代码，如何保证groovy安全？

Question

如何阻止用户输入:System.exit(0) 和无限循环

例如:

public class TestGroovy {
    public static ScriptEngineManager factory = new ScriptEngineManager();

    public static void main(String[] args) throws ScriptException, NoSuchMethodException {
        TestGroovy testGroovy = new TestGroovy();
        testGroovy.runGroovy("    def executeMethod(){\n" +
                "        System.exit(0);\n" +
                "    }");
    }


    public void runGroovy(String groovyCode) throws ScriptException, NoSuchMethodException {
        ScriptEngine engine = factory.getEngineByName("groovy");
        engine.eval(groovyCode);
        Invocable invokeEngine = (Invocable) engine;
        invokeEngine.invokeFunction("executeMethod");
    }

}

如果人们输入System.exit(0)或无限循环，那就太糟糕了。

我想阻止用户输入这些内容。

谢谢~

<小时/>

我使用这个方法的一个答案:

public class MySecurityManager extends SecurityManager{

    @Override
    public void checkExit(int status) {
        throw new SecurityException();
    }

}

    public static void main(String[] args) {
        ScriptEngineManager factory = new ScriptEngineManager();
        ScriptEngine engine = factory.getEngineByName("groovy");

        try {
            Map<String,Object> output=new HashMap<>();
            Bindings binding = engine.createBindings();
            binding.put("input",1);
            binding.put("output",output);

            String fact = "def executeMethod(){\n" +
                    "  \tSystem.exit(0);\n" +
                    "    output.put('your key','your value');\n" +
                    "    return output;\n" +
                    "}";
            engine.eval(fact,binding);
            Invocable inv = (Invocable) engine;
            MySecurityManager secManager = new MySecurityManager();
            System.setSecurityManager(secManager);
            output  = (Map<String, Object>) inv.invokeFunction("executeMethod",null);
            System.out.println(output);
        } catch (SecurityException | ScriptException | NoSuchMethodException e) {
            System.out.println("can't input System.exit(0)");
        }
    }

谢谢~

Answer 1

如果您使用单独的线程来运行用户定义的代码，我可以建议以下内容:

对于 System.exit() 调用，您可以定义和设置您自己的 SecurityManager 。它有 checkExit() 方法，用于判断 System.exit 是否可行。覆盖它并提供你的逻辑。此解决方案的缺点是您必须确保您的应用程序始终在启用 SecurityManager 的情况下运行。

对于无限循环或任何其他长时间运行的代码，我只能建议一个外部监视线程，该线程在某些条件下(超时或其他)只会停止线程。但这个解决方案有很多问题:如果您停止的线程拥有未关闭的资源(打开的文件、数据库连接等)或锁，您可能会遇到资源泄漏或死锁。在这种情况下，中断调用将不起作用，因为代码可能会忽略它们。

作为替代解决方案，您可以使用单独的 JVM 来运行用户的代码。

这由你决定。