捕获库的退出指令

Question

令我惊讶和恐惧的是，我刚刚在我使用的库中遇到了 System.exit(1); 行。我计划联系该库的作者并询问提供了什么，但同时有什么方法可以防止该库杀死我的代码（更糟糕的是，使用我的代码杀死应用程序）？

也许以某种方式强制库抛出一个 SecurityException，我看到 exit(int) 可能会抛出这个异常？

Answer 1

我只知道一种方法，那就是编写一个 SecurityManager 来禁止使用 System.exit。

public class MySecurityMgr extends SecurityManager {
   ... 

   @Override
    public void checkExit(int status) {
            throw new RuntimeException(Integer.toString(status));
    }
}

Answer 2

我认为你唯一的选择（除了用湿毛巾殴打作者直到他重写该死的东西）是更改库的字节代码。

漂亮的方法是在加载库时使用 AspectJ 和加载时编织，丑陋的选择是使用像 asm 这样的工具来删除或更改方法调用。

这是一个 AspectJ 方面，它会偏转对 System.exit() 的调用：

public aspect SystemExitEvader{

    pointcut systemExitCall() : call(* java.lang.System.exit(*));

    void around() : systemExitCall(){
        System.out.println("Call to System.exit() attempted");
        // note: there is no call to proceed()
    }

}

---

好的，我已经看到了有关 SecurityManagers 的其他答案，我同意这可能是要走的路，但我将在这里留下我的答案作为替代方案。

Answer 3

是的，您可以安装覆盖 checkExit。

我发现这对于单元测试代码特别有用，可以确保测试的代码不会以 0 退出，这表明在测试有机会运行完成之前出现了虚假的成功。

如果 SecurityException 最终杀死了一些通常寿命较长的基本线程，这将无济于事。

Answer 4

是的，安全经理可以解决这个问题

// install security manager to avoid System.exit() call from lib
SecurityManager       previousSecurityManager = System.getSecurityManager();
final SecurityManager securityManager         = new SecurityManager() {
    @Override public void checkPermission(final Permission permission) {
      if (permission.getName() != null && permission.getName().startsWith("exitVM")) {
        throw new SecurityException();
      }
    }
  };
System.setSecurityManager(securityManager);

try {
   // Call here your lib code
} catch (SecurityException e) {
  // Say hi to your favorite creator of closed source software that includes System.exit() in his code.
} finally {
  System.setSecurityManager(previousSecurityManager);
}