减少 Java 运行时的最大堆栈大小

Question

我正在尝试编写一个junit测试来防止一段代码陷入无休止的迭代中，这最终会导致StackOverflow。

所以我正在寻找一种方法来减少运行时的堆栈大小，以便 Junittest 更快地失败。

将最大堆栈设置为 jvm 参数是不可能的，因为该测试是更大的测试套件的一部分。

Answer 1

您可以运行一个递归方法，该方法将运行给定的次数，然后执行给定的操作。 不过听起来很不稳定:(

类似这样的东西：

public void eatStackThenExecute(int depth, Runnable action)
{
    // Maybe put some locals here (and use them) to eat more stack per iteration?
    if (depth == 0)
    {
        action();
    }
    else
    {
        eatStackThenExecute(depth - 1, action);
    }
}

编辑：智能 JVM 可能会在这里优化尾部调用，所以我们可能需要在递归调用之后做“一些事情”来阻止这种情况的发生...

Ick 'n东西:(

Answer 2

不可能在运行时设置堆栈大小，但也许您可以：

• 在不同线程内调用该代码段 - 保留对其的引用；
• 定期轮询thread.getStackTrace()，如果其大小大于x，则失败；
• 取消执行是否正确终止的检查。

---

未编译的概念验证代码（未正确检查所有边缘条件）：

AtomicBoolean success = new AtomicBoolean(false);

Thread t= new Thread(new Runnable() {
   public void run() {
       codeToTest();
       success.set(true);
   }

});

t.start();

while ( t.isAlive() ) {
     if ( t.getStackTrace().length > 50 )
          fail("Stack trace too large");

     Thread.sleep(50);
}

assertTrue(sucess.get());

Answer 3

您只能在启动时设置这样的参数，但是您可以从 Java 启动另一个进程。 因此，您可以让单元测试启动具有较小堆栈大小的第二个进程来执行测试。

Answer 4

这有点……好吧，有趣……但这可能是值得的。

1. 获取您的 .class 文件并使用 jasper 反编译它。
2. 编辑生成的 JVM 汇编式代码，并为要以此方式测试的例程添加或更改“.limit stack x”参数。
3. 使用 Jasmin 重新编译。
4. 运行并测试。