.NET运行时如何获取可用栈空间？提前抛出自定义溢出前异常

可以在.NET运行时获取当前线程的剩余栈空间，以此在栈溢出发生前终止递归并抛出自定义异常，具体实现如下：

核心方法 #

.NET提供System.Runtime.CompilerServices.RuntimeHelpers.GetRemainingStackBytes()方法，直接返回当前线程剩余的栈空间字节数，可用于栈空间阈值检查。

实现步骤 #

1. 定义自定义异常

public class StackOverflowPreventedException : Exception
{
    public StackOverflowPreventedException(string message) : base(message)
    {
    }
}

2. 带栈检查的递归方法
   在递归逻辑执行前，先检查剩余栈空间是否低于预设阈值，若不足则抛出自定义异常终止递归：

using System.Runtime.CompilerServices;

public class StackSafeRecursion
{
    // 根据递归方法的栈帧大小调整阈值，示例设为1KB
    private const int StackThreshold = 1024;

    public void RecursiveMethod(int depth)
    {
        // 检查剩余栈空间，提前终止递归
        if (RuntimeHelpers.GetRemainingStackBytes() < StackThreshold)
        {
            throw new StackOverflowPreventedException("Stopped before stack overflow");
        }

        // 模拟递归业务逻辑
        Console.WriteLine($"当前递归深度: {depth}");
        RecursiveMethod(depth + 1);
    }
}

3. 调用并捕获自定义异常

try
{
    var recursionInstance = new StackSafeRecursion();
    recursionInstance.RecursiveMethod(1);
}
catch (StackOverflowPreventedException ex)
{
    Console.WriteLine(ex.Message);
}

注意事项 #

• 阈值需根据实际递归方法的栈帧占用调整：不同方法的局部变量、参数数量会影响栈帧大小，建议通过测试确定合适的阈值。
• 不同线程的栈大小可能不同：默认线程栈大小为1MB，若创建线程时指定了自定义栈大小，需对应调整阈值逻辑。
• StackOverflowException无法被常规try/catch捕获：栈溢出发生时栈空间已耗尽，无法执行异常处理代码，因此必须提前检查终止递归。

内容的提问来源于stack exchange，提问作者iLoveToCode