阻止 API 中的 System.exit()

Question

我正在使用一个第三方库，如果遇到异常，它会执行 System.exit() 。我正在使用 jar 中的 API。无论如何，我是否可以阻止 System.exit() 调用，因为它会导致我的应用程序关闭？由于许多其他许可问题，在删除 System.exit() 后我无法反编译和重新编译 jar。我曾经在 stackoverflow 中[对其他一些我不记得的问题]找到了一个答案，我们可以在 Java 中使用 SecurityManager 来做这样的事情。

Answer 1

您可以安装禁用 System.exit() 的安全管理器：

  private static class ExitTrappedException extends SecurityException { }

  private static void forbidSystemExitCall() {
    final SecurityManager securityManager = new SecurityManager() {
      public void checkPermission( Permission permission ) {
        if( "exitVM".equals( permission.getName() ) ) {
          throw new ExitTrappedException() ;
        }
      }
    } ;
    System.setSecurityManager( securityManager ) ;
  }

  private static void enableSystemExitCall() {
    System.setSecurityManager( null ) ;
  }

---

编辑： Max 在下面的评论中指出

从 Java 6 开始，权限名称实际上是“exitVM.”+状态，例如“exitVM.0”。

但是，权限exitVM.*指的是所有退出状态，并且exitVM被保留作为exitVM.*的简写，所以上面的代码仍然有效（请参阅 RuntimePermission< 的文档/代码>）。

Answer 2

请参阅我对 如何避免 JFrame EXIT_ON_CLOSE 操作退出整个应用程序？。

编辑1：链接的来源。演示如何使用 SecurityManager 来阻止 System.exit(n)。

import java.awt.*;
import java.awt.event.*;
import java.security.Permission;

/** NoExit demonstrates how to prevent 'child'
applications from ending the VM with a call
to System.exit(0).
@author Andrew Thompson */
public class NoExit extends Frame implements ActionListener {

  Button frameLaunch = new Button("Frame"),
     exitLaunch = new Button("Exit");

  /** Stores a reference to the original security manager. */
  ExitManager sm;

  public NoExit() {
     super("Launcher Application");

     sm = new ExitManager( System.getSecurityManager() );
     System.setSecurityManager(sm);

     setLayout(new GridLayout(0,1));

     frameLaunch.addActionListener(this);
     exitLaunch.addActionListener(this);

     add( frameLaunch );
     add( exitLaunch );

     pack();
     setSize( getPreferredSize() );
  }

  public void actionPerformed(ActionEvent ae) {
     if ( ae.getSource()==frameLaunch ) {
        TargetFrame tf = new TargetFrame();
     } else {
        // change back to the standard SM that allows exit.
        System.setSecurityManager(
           sm.getOriginalSecurityManager() );
        // exit the VM when *we* want
        System.exit(0);
     }
  }

  public static void main(String[] args) {
     NoExit ne = new NoExit();
     ne.setVisible(true);
  }
}

/** This example frame attempts to System.exit(0)
on closing, we must prevent it from doing so. */
class TargetFrame extends Frame {
  static int x=0, y=0;

  TargetFrame() {
     super("Close Me!");
     add(new Label("Hi!"));

     addWindowListener( new WindowAdapter() {
        public void windowClosing(WindowEvent we) {
           System.out.println("Bye!");
           System.exit(0);
        }
     });

     pack();
     setSize( getPreferredSize() );
     setLocation(++x*10,++y*10);
     setVisible(true);
  }
}

/** Our custom ExitManager does not allow the VM
to exit, but does allow itself to be replaced by
the original security manager.
@author Andrew Thompson */
class ExitManager extends SecurityManager {

  SecurityManager original;

  ExitManager(SecurityManager original) {
     this.original = original;
  }

  /** Deny permission to exit the VM. */
   public void checkExit(int status) {
     throw( new SecurityException() );
  }

  /** Allow this security manager to be replaced,
  if fact, allow pretty much everything. */
  public void checkPermission(Permission perm) {
  }

  public SecurityManager getOriginalSecurityManager() {
     return original;
  }
}

Answer 3

前面的代码示例部分正确，但我发现它最终阻止了我的代码对文件的访问。为了解决这个问题，我编写的 SecurityManager 有点不同：

public class MySecurityManager extends SecurityManager {

private SecurityManager baseSecurityManager;

public MySecurityManager(SecurityManager baseSecurityManager) {
    this.baseSecurityManager = baseSecurityManager;
}

@Override
public void checkPermission(Permission permission) {
    if (permission.getName().startsWith("exitVM")) {
        throw new SecurityException("System exit not allowed");
    }
    if (baseSecurityManager != null) {
        baseSecurityManager.checkPermission(permission);
    } else {
        return;
    }
}

}

在我的例子中，我需要防止第三方库终止虚拟机。但也有一些 grails 测试调用 System.exit。因此，我编写了代码，以便它仅在调用第 3 方库之前立即激活自定义安全管理器（不是常见事件），然后立即恢复原始安全管理器（如果有）。

这一切都有点丑陋。理想情况下，我更愿意简单地删除 System.exit 代码，但我无权访问第 3 方库的源代码。

Answer 4

使用 SecurityManager 禁止 System.exit() 调用并不完美，至少有两个原因：

1. 启用和不启用 SecurityManager 运行的 Java 应用程序有很大不同。这就是为什么它最终需要关闭，但不能用 System.setSecurityManager(null) 来完成。此调用将导致另一次安全权限检查，这将不可避免地失败，因为应用程序代码（SecurityManager 子类）位于调用堆栈的顶部。

2. 所有Java应用程序都是多线程的，其他线程可以在forbidSystemExitCall()和enableSystemExitCall()之间执行各种操作。其中一些内容可以通过安全权限检查来保护，但由于与上述相同的原因，安全权限检查也会失败。如果重写 checkExit() 而不是 [更通用] checkPermission()，那么它将涵盖大多数情况。

（据我所知）解决此问题的唯一方法是将所有特权授予 SecurityManager 子类。它很可能需要由单独的类加载器加载，例如引导（空）类加载器。