漏洞概述 该漏洞涉及在 项目中,对 和 参数的验证不足。具体来说, 验证被用于检查这些参数,但 NaN 和 Infinity 值可以通过 Python 的比较运算符( , )绕过,导致 IEEE 754 浮点语义问题,进而引发 GPU 采样内核中的未定义行为或 CUDA 崩溃。 影响范围 参数: 和 场景:GPU 采样内核 后果:未定义行为或 CUDA 崩溃 修复方案 1. 验证逻辑: - 使用 验证 和 参数。 - 确保 NaN 和 Infinity 值被正确拒绝。 2. 测试计划: - 添加测试用例 ,覆盖 NaN 和 -Inf、+Inf 的拒绝情况。 - 运行 确保所有测试通过。 - 运行 确保所有钩子通过。 3. 性能优化: - 使用 进行快速比较,避免不必要的计算。 POC 代码 总结 该漏洞通过绕过 验证,导致 GPU 采样内核中的未定义行为或 CUDA 崩溃。修复方案包括加强参数验证、添加测试用例以及优化性能。