闭包实现递归防护的核心是封装深度计数器并前置拦截:外层闭包固定maxDepth,内层维护不可见的currentDepth,每次调用前检查超限,未超则+1后调用原函数,超限则按配置抛异常或降级;不修改原函数签名,通过重绑定闭包实例传递深度。

用闭包实现带调用深度限制的递归防护,核心是让每次递归调用都携带并更新一个“当前层数”计数器,并在超出预设阈值时主动中断,避免栈溢出或恶意深度攻击。它不修改原递归函数逻辑,而是通过包装层做守门人。
封装结构:闭包捕获最大深度与当前层数
外部函数接收目标递归函数和最大允许深度,内部返回一个新函数。该函数在每次调用时检查当前深度是否超限;未超限时递增层数、传入参数调用原函数;超限时直接抛异常或返回兜底值。
- 最大深度(maxDepth)作为常量,在外层闭包中固定,避免被意外篡改
- 当前深度(currentDepth)作为可变状态,由闭包内部维护,对外不可见
- 原函数(fn)被闭包引用,但仅通过受控路径调用
关键实现:非递归式深度传递
为避免在递归调用链中手动层层传参污染原函数签名,闭包采用“重绑定”策略:每次进入递归分支前,用新闭包替换当前执行上下文中的调用入口,使下一层自动继承更新后的 depth 值。
- 不依赖函数参数传递 depth,原函数签名完全保持不变
- 每层调用实际执行的是一个新闭包实例,其中 currentDepth 已 +1
- 终止条件判断放在最前端,确保无任何计算开销就拦截非法调用
安全边界:显式终止 + 可配置响应
超过深度限制时,不能仅靠 raise 异常——需支持多种处置方式以适配不同场景,例如返回 None、记录日志、触发告警或降级为迭代逻辑。
- 默认行为是抛出 RecursionDepthError(自定义异常),便于上层捕获处理
- 可选传入 on_exceed 回调,在超限时执行清理或审计逻辑
- 支持设置 enable_logging 开关,仅在调试或高风险服务中启用深度追踪
典型使用示例:保护树遍历函数
假设有一个易受深度攻击的 JSON 树遍历函数,直接递归可能因恶意嵌套结构崩溃:
- 原始函数:
def traverse(node): return [node] + sum([traverse(child) for child in node.get('children', [])], []) - 加防护后:
safe_traverse = depth_limited(traverse, max_depth=50) - 调用
safe_traverse(malicious_deep_tree)将在第 51 层前终止,不触发 C stack overflow

















