Java: System.exit() 与安全策略

Posted 2020-09-23 cynchanpin

说明

System.exit() 的本质是通知 JVM 关闭。

一般来说。有两种禁用 System.exit() 的办法：

• 安全管理器
• 安全策略

本质都是JRE 提供的本地实现，在执行之前进行权限推断。

由于System.exit() 是一种非常暴力的手段。假设在 Client 模式下自己写个小程序无所谓,可是在 Server 上多个程序、或者多线程时就会有非常大的麻烦。

底层源代码

1.先来看看静态方法 System.exit() 的源代码:

// System.exit()
public static void exit(int status) {
    Runtime.getRuntime().exit(status);
}

应该说非常easy, 仅仅是简单地调用执行时的 exit 方法.

2.然后我们看执行时的实例方法 exit:

// Runtime.exit()
public void exit(int status) {
    SecurityManager security = System.getSecurityManager();
    if (security != null) {
        security.checkExit(status);
    }
    Shutdown.exit(status);
}

假设有安全管理器，那么就让安全管理器执行 checkExit退出权限检查。

假设检查不通过，安全管理器就会抛出异常(这就是约定！

)。

然后当前线程就会往外一路抛异常，假设不捕获，那么该线程就会退出。

此时假设没有其它的前台线程正在执行，那么JVM也会跟着退出。

3.Shutdown 是 java.lang 包以下的一个类。

訪问权限是 default, 所以我们在API中是不能调用的。

// Shutdown.exit()
static void exit(int status) {
    boolean runMoreFinalizers = false;
    synchronized (lock) {
        if (status != 0) runFinalizersOnExit = false;
        switch (state) {
        case RUNNING:       /* Initiate shutdown */
            state = HOOKS;
            break;
        case HOOKS:         /* Stall and halt */
            break;
        case FINALIZERS:
            if (status != 0) {
                /* Halt immediately on nonzero status */
                halt(status);
            } else {
                /* Compatibility with old behavior:
                 * Run more finalizers and then halt
                 */
                runMoreFinalizers = runFinalizersOnExit;
            }
            break;
        }
    }
    if (runMoreFinalizers) {
        runAllFinalizers();
        halt(status);
    }
    synchronized (Shutdown.class) {
        /* Synchronize on the class object, causing any other thread
         * that attempts to initiate shutdown to stall indefinitely
         */
        sequence();
        halt(status);
    }
}

当中有一些同步方法进行锁定。

退出逻辑是调用了 halt 方法。

// Shutdown.halt()
static void halt(int status) {
    synchronized (haltLock) {
        halt0(status);
    }
}

static native void halt0(int status);

然后就是调用 native 的 halt0() 方法让 JVM “自杀“了。

演示样例

使用安全管理器的实现代码例如以下所看到的:

1.定义异常类, 继承自 SecurityException

ExitException.java

package com.cncounter.security;

public class ExitException extends SecurityException {
    private static final long serialVersionUID = 1L;
    public final int status;

    public ExitException(int status) {
        super("忽略 Exit方法调用!");
        this.status = status;
    }
}

2.定义安全管理器类, 继承自 SecurityManager

NoExitSecurityManager.java

package com.cncounter.security;

import java.security.Permission;

public class NoExitSecurityManager extends SecurityManager {
    @Override
    public void checkPermission(Permission perm) {
        // allow anything.
    }

    @Override
    public void checkPermission(Permission perm, Object context) {
        // allow anything.
    }

    @Override
    public void checkExit(int status) {
        super.checkExit(status);
        throw new ExitException(status);
    }
}

当中直接拒绝系统退出。

3.添加一个辅助和測试类，实际使用时你也能够自己进行控制。

NoExitHelper.java

package com.cncounter.security;

public class NoExitHelper {

    /**
     * 设置不同意调用 System.exit(status)
     * 
     * @throws Exception
     */
    public static void setNoExit() throws Exception {
        System.setSecurityManager(new NoExitSecurityManager());
    }

    public static void main(String[] args) throws Exception {
        setNoExit();
        testNoExit();
        testExit();
        testNoExit();
    }

    public static void testNoExit() throws Exception {
        System.out.println("Printing works");
    }

    public static void testExit() throws Exception {
        try {
            System.exit(42);
        } catch (ExitException e) {
            //
            System.out.println("退出的状态码为: " + e.status);
        }
    }
}

在当中。使用了一个 main 方法来做简单的測试。 控制台输出结果例如以下:

Printing works
退出的状态码为: 42
Printing works

原问题

原来的问题例如以下:

I’ve got a few methods that should call System.exit() on certain inputs. Unfortunately, testing these cases causes JUnit to terminate! Putting the method calls in a new Thread doesn’t seem to help, since System.exit() terminates the JVM, not just the current thread. Are there any common patterns for dealing with this? For example, can I subsitute a stub for System.exit()?

大意是:

有一些方法须要測试, 可是在某些特定的输入时就会调用 System.exit()。这就杯具了,这时候 JUnit 測试也跟着退出了! 用一个新线程来调用这样的方法也没什么用, 由于 System.exit() 会停止JVM , 而不是退出当前线程。

有什么通用的模式来处理这样的情况吗?

比如,我能替换掉 System.exit() 方法吗?

建议例如以下:

Instead of terminating with System.exit(whateverValue), why not throw an unchecked exception?

In normal use it will drift all the way out to the JVM’s last-ditch catcher and shut your script down (unless you decide to catch it somewhere along the way, which might be useful someday).

In the JUnit scenario it will be caught by the JUnit framework, which will report that such-and-such test failed and move smoothly along to the next.

翻译例如以下:

在程序中调用 System.exit(whateverValue) 是一种非常不好的编程习惯, 这样的情况为什么不抛出一个未检測的异常(unchecked exception)呢? 假设程序中不进行捕获(catch), 抛出的异常会一路漂移到 JVM , 然后就会退出程序(仅仅有主线程的话)。

在 JUnit 的測试场景中异常会被 JUnit 框架捕获, 然后就会报告说某某某測试执行失败,然后就继续下一个单元測试了。

当然,给出的解决方式就是前面的那段代码. 你还能够阅读以下的參考文章，查找其它的解决方式。

參考文章:

• Java: 怎样单元測试那种会调用 System.exit() 的方法?

• Java中怎样禁止 API 调用 System.exit()

• 怎样配置 Tomcat 的安全管理器

日期: 2015年08月25日

人员: 铁锚 http://blog.csdn.net/renfufei