悟空Agent实战：LLaMA-Factory高危0day漏洞挖掘与修复、【论文速读】| FuncVul——基于LLM与代码块的高效函数级漏洞检测模型、

2026-10-28 19:30:54 +00:00 · 2025-07-01 12:41:04 +00:00
1 parent 7a17278333
commit 14b3fab6e3
@@ -15723,5 +15723,7 @@
    "https://mp.weixin.qq.com/s?__biz=MzI4MjkxNzY1NQ==&mid=2247486072&idx=2&sn=d8ec0d298fa5cdf2eec24a47363cad59": "网络摄像头漏洞扫描工具",
    "https://mp.weixin.qq.com/s?__biz=Mzg2MDg0ODg1NQ==&mid=2247546847&idx=2&sn=9b0673dd296bbe07d675025dfa319c78": "蓝牙零日漏洞：数百万耳机变身隐秘窃听站",
    "https://mp.weixin.qq.com/s?__biz=MzkwOTE5MDY5NA==&mid=2247506878&idx=1&sn=646896e03d930c8c4ed7eb5db4c3f087": "NASA开放公开的漏洞赏金计划",
-    "https://mp.weixin.qq.com/s?__biz=Mzk0Mzc1MTI2Nw==&mid=2247492309&idx=1&sn=971b6609659bdf8e5a1c6cd2c264177b": "SRC 视角下：渗透测试中的逻辑漏洞思路博弈"
+    "https://mp.weixin.qq.com/s?__biz=Mzk0Mzc1MTI2Nw==&mid=2247492309&idx=1&sn=971b6609659bdf8e5a1c6cd2c264177b": "SRC 视角下：渗透测试中的逻辑漏洞思路博弈",
+    "https://mp.weixin.qq.com/s?__biz=MjM5ODYwMjI2MA==&mid=2649794356&idx=1&sn=1807d8a8e309d53363046e89b71a51fe": "悟空Agent实战：LLaMA-Factory高危0day漏洞挖掘与修复",
+    "https://mp.weixin.qq.com/s?__biz=MzkzNDUxOTk2Mw==&mid=2247496737&idx=1&sn=370c593f1bce328f065b2ee3561a83bb": "【论文速读】| FuncVul——基于LLM与代码块的高效函数级漏洞检测模型"
 }
@@ -0,0 +1,120 @@
+#  【论文速读】| FuncVul——基于LLM与代码块的高效函数级漏洞检测模型  
+原创 知识分享者  安全极客   2025-07-01 10:22  
+  
+![](https://mmbiz.qpic.cn/mmbiz_jpg/vWuBpewLia8QmTLhv0jB8GS6Wtic69pG44V8Gib7ccD3FZolnOVkdOPafA3YULibw9S5AEkdO8sstRLGNFVDj7SgRg/640?wx_fmt=jpeg&from=appmsg "")  
+  
+**基本信息**  
+  
+  
+原文标题：FuncVul: An Effective Function Level Vulnerability Detection Model using LLM and Code Chunk  
+  
+原文作者：Sajal Halder, Muhammad Ejaz Ahmed, Seyit Camtepe  
+  
+作者单位：Data61, CSIRO, Australia  
+  
+关键词：函数级代码、漏洞检测、代码块、软件供应链、大语言模型  
+  
+原文链接：https://arxiv.org/abs/2506.19453  
+  
+开源代码：https://github.com/sajalhalder/FuncVul  
+  
+  
+**论文要点**  
+  
+  
+论文简介：本文关注于软件供应链环境下的源代码漏洞检测，提出了一种创新的函数级漏洞检测方法FuncVul。相较于现有的主要针对整体包或库级别进行检测的方法，FuncVul聚焦于函数级别并进一步定位到函数内部的代码块，以实现对具体漏洞行的精准发现。该方法利用了大语言模型（LLM）和补丁（patch）信息自动标注高质量数据集，并基于预训练且支持结构化代码输入的GraphCodeBERT模型进行微调，实现了对C/C++和Python代码中多个漏洞点的识别。论文构建了六套不同类型的数据集，采用基线方法和多维度指标系统地开展对比评测。  
+  
+实验表明，FuncVul在不同数据集上均获得了87%-92%的准确率和86%-92%的F1分数，并对基于整个函数的检测方法带来了显著提升。研究还探讨了泛化性、对未知项目漏洞的检测能力及最佳代码块长度等问题。该方法及其数据集均已开源，促进了后续研究与应用落地。  
+  
+研究目的：本研究旨在解决现有函数级漏洞检测难以精确定位内部多处漏洞、难以指示具体存在问题的代码行以及泛化能力有限等挑战。作者尝试通过“基于代码块的建模”而非整体函数粒度，利用LLM高效生成带标签的训练数据，结合GraphCodeBERT精准捕获代码语法与语义特征，实现以下目标：（1）准确检测函数级漏洞并定位漏洞代码块，（2）衡量代码块方法相较全函数分析的性能优势，（3）提升模型对不同项目和未知补丁场景的泛化能力，（4）探索最佳的块划分策略以及能否在单个函数内部发现多个漏洞点。  
+  
+研究贡献：  
+- 研究者提出了一种新颖的基于代码块的函数漏洞（FuncVul）检测模型，该模型能够识别函数内的多个漏洞，并且重要的是，能够定位导致这些漏洞的具体较小代码段。  
+  
+- 研究者从多种数据源收集并整理了四个数据集，例如来自 GitHub 的项目源代码和漏洞公告数据库（如 OSV）。此外，研究者开发了新方法，利用大语言模型（LLMs）对源代码进行分析、处理和整理。  
+  
+- 研究者采用了微调后的 GraphCodeBERT 模型进行函数级漏洞预测，因为它能有效捕捉代码中的语法和语义相似性。  
+  
+- 实验结果表明，所提出的 FuncVul 模型性能优于最先进的基线模型，在六个数据集上实现了平均 89.39% 的准确率和 88.94% 的 F1 分数。  
+  
+- 此外，研究者还证明了 FuncVul 模型具有高度通用性，能够处理多样化的代码块并有效识别新的漏洞模式。  
+  
+  
+  
+  
+**引言**  
+  
+  
+当前，随着技术快速发展和软件复杂性的持续提升，软件供应链安全在业界与学术界都成为广泛关注的研究重点。国家漏洞数据库（NVD）截至2024年10月已累计收录24万多个CVE漏洞，且该数字每年平均增长15-20%。对C/C++和Python等主流编程语言进行准确漏洞检测，对保障软件供应链安全具有重要意义。然而，传统方法通常集中于检测整个包或库是否存在漏洞，难以精确定位具体函数甚至是代码行级的风险。更进一步，即便能够识别到漏洞函数，往往也无法揭示函数内部具体存在问题的代码段。此类粒度粗、定位不精的特性，导致开发者在修复时需手动代码审查，增加了维护难度和时间成本。  
+  
+面对数量庞大、分布广泛的开源包和函数，人工分析无疑代价高昂且难以大规模应用。因此，急需能够自动、精确、高效定位关键函数及其漏洞代码块的检测方法。与此同时，现有学术界方法虽然采用了机器学习、深度学习与图神经网络等手段，但仍存在两个显著不足：一是只能给出函数级二分类结果（是否含漏洞），难以进一步指示漏洞具体行和数量；二是受限于模型粒度和数据采集方式，实际定位能力与泛化能力有限。  
+  
+鉴于此，本文提出了一种创新性的基于代码块的函数级漏洞检测框架FuncVul。作者设计了以“代码块（code chunk）”为基本检测单元的新颖方法，并通过LLM自动生成高质量带标签的数据集，结合对预训练模型的适配微调，实现了函数级漏洞多点、精定位检测与全面评估。论文系统讨论了该方法在提升检测准确性、缩短修复时间及提高泛化能力等方面的有效性，并通过实验证实了其显著优势。引言中还明确提出了六个关键研究问题（RQ1-RQ6），覆盖了核心建模策略、代码块方法优越性、泛化能力、复杂场景适应性与多点检测能力，彰显了本文系统、全面推进函数级漏洞检测技术前沿的学术追求。  
+  
+  
+**相关工作与背景**  
+  
+  
+软件漏洞检测是软件工程与网络安全领域的核心问题。相关研究主要沿两个方向展开：一是漏洞检测算法与模型，二是漏洞数据集的创建。现有方法涵盖了传统的特征工程与机器学习（如TF-IDF加轻量模型）、深度学习（如BERT、RoBERTa、LSTM等语言模型）、基于图的神经网络（如Graph Neural Networks, GNNs）、上下文嵌入、以及近年来兴起的大语言模型（LLMs）等。  
+  
+代表性工作如Hanif等提出的VulBERTa基于RoBERTa模型，针对C/C++代码训练，充分挖掘代码语法与语义；Warschinski等提出的VUDENC利用Word2Vec与LSTM完成Python代码的漏洞序列识别；Yuan等则结合GRU与图网络提取代码多维特征并用随机森林支持不平衡样本，获得优异检测性能。Wang等开发了大规模仓库级数据集ReposVul，并形成用于流程化语义抽取的工具链。Li等开发了VulPecker工具，采用基于补丁特征与代码相似度的模式检测策略。Transformer和BERT模型，如LineVul，能利用深度上下文建模进行更细粒度的漏洞预测。GNN相关工作（如Devign、AMPLE等）通过图结构深入挖掘源代码的数据流、控制流和依赖关系。  
+  
+然而，大部分方法主要关注于函数级二分类，即判断函数是否存在漏洞，对内部漏洞数量及具体位置尚无法充分挖掘，导致开发者补丁定位与修复代价高。此外，部分方法能达到更细粒度的行级检测（如LineVul），但受限于语义捕捉能力与模型泛化性，实际检测效果有限。  
+  
+随着LLM模型（如BERT、GPT、Gemini等）在各领域应用日益广泛，学界也开始将其用于漏洞检测，如GRACE 工作利用图结构与语境学习提升LLM检测精度；Akuthota等直接利用LLMs完成漏洞监测任务。与这些先前工作不同，本文首次提出利用LLM配合补丁自动高质量标注训练集，并以代码块为单元结合GraphCodeBERT微调，实现对函数内部多个漏洞、未知项目及新型漏洞的全面、精准检测，具有明显的创新优势。  
+  
+  
+**预备知识与问题定义**  
+  
+  
+为支持新型方法的提出，论文首先明确了若干基础概念。函数代码块（Function Code Chunk, FC）被定义为函数源码内一段连续代码行，通常围绕代码变更（如补丁更新）为中心，并包含变更前后各若干行，以便保留关键上下文。泛化代码块（Generic Code Chunk）则指将函数名、变量名等具体标识符统一用通用占位符（如F1、V1等）替换后的代码块，这样可消除由开发风格、命名差异导致的数据偏差，增强模型泛化性与鲁棒性。  
+  
+特别地，3-行扩展代码块是以代码编辑行为中心，上下各扩展三行的代码片段，用于最大程度保留漏洞行的语义上下文。若补丁修改远多于10行，则只取变更相关的全部行而不做进一步扩展。  
+  
+基于上述定义，本文将函数级代码块漏洞检测问题形式化为：给定C/C++或Python代码对应的带补丁修改信息的函数代码块，设计一个检测器V，使得V能自动判别该代码块是否包含漏洞（1为有，0为无）。论文以高质量的代码块构建与泛化、自动标签生成、以及精细建模为核心，系统推进了该难题的解决。  
+  
+  
+**数据集构建与模型方法**  
+  
+  
+论文在数据集构建与方法设计方面进行了系统创新。首先，在数据集方面，作者开发了一套半自动标注机制，综合利用补丁信息和LLM能力。通过从GitHub代码和OSV等漏洞库中爬取项目及补丁，提取单一修改的函数变更（确保只有一处相关修改提升标签精度），再利用Gemini 1.5 Pro等LLM模型对每一代码块生成“脆弱/安全”标签，并指示具体漏洞行。代码块分为原始块和泛化块两种，结合不同的LLM提示词，最终产出六套多样化数据集，覆盖不同语言、抽象层次和细分场景。  
+  
+![](https://mmbiz.qpic.cn/mmbiz_png/vWuBpewLia8Rh6q5DHRmSSxaSDic5Ig9IiafuQLvI3QSC4kLGZErdiaE1utnjALnjepzvWUjiaCiaaGdicrvR8C8EtpDQ/640?wx_fmt=png&from=appmsg "")  
+  
+对于标签生成，论文提出一套交叉验证机制：先用补丁信息过滤出高置信度的单点修改，再由LLM对照CVE描述自动判断违规代码行，并要求其与补丁删除行有重叠才判定为漏洞样本。未满足标准的样本则标记为“未知”并筛除。负样本则由修复后的代码块或随机关联无风险块构造而成，保证样本质量和分布平衡。  
+  
+![](https://mmbiz.qpic.cn/mmbiz_png/vWuBpewLia8Rh6q5DHRmSSxaSDic5Ig9Iia7ddial0xnthTb7IMQHAYvMQzRBoT30j4WKVJ1gbRFkxicByibnRl47vwQ/640?wx_fmt=png&from=appmsg "")  
+  
+在模型方法层面，核心为以GraphCodeBERT为主体的微调框架。GraphCodeBERT作为支持数据流建模的高性能代码预训练模型，能捕捉代码语义结构及上下文依赖。训练流程为：将生成的带标签代码块分为80%训练、20%测试，利用GraphCodeBERT的tokenizer和模型对块数据进行编码、训练与优化，最终得到名为FuncVul的专用检测模型。推理过程中，新样本亦经Token化后由微调后的FuncVul判别其漏洞风险，并可返回判定理由。整体流程包括数据集构建、自动标签生成、块处理、泛化、模型训练与推理全流程，支持批量、自动化漏洞检测与多代码块并行处理。  
+  
+  
+**实验设计与结果分析**  
+  
+  
+实验设计严格对标方法提出的六个关键研究问题（RQ1-RQ6）。实验平台为Python环境，MacBook Pro（Apple M3, 24GB RAM），模型关键参数为batch size=8、编码维度512、3轮训练、warmup steps=50、权重衰减0.05。所有数据均采用80/20划分并做5折交叉验证，评价指标包括准确率、精确率、召回率、F1分数和MCC，能全面反映模型性能。  
+  
+![](https://mmbiz.qpic.cn/mmbiz_png/vWuBpewLia8Rh6q5DHRmSSxaSDic5Ig9IiaxicPbUWtYNLSKKyR3GSRpuqfEEbAdqhAT74potM3P89xXibPibjq62TGg/640?wx_fmt=png&from=appmsg "")  
+  
+六套数据集覆盖原始代码块、泛化代码块、多种标注与负样本采样。其中前四套基于补丁与LLM双重标签，第五、第六套则直接用LLM提取漏洞行并自动分块，均衡覆盖不同主流编程语言与真实修复案例。  
+  
+![](https://mmbiz.qpic.cn/mmbiz_png/vWuBpewLia8Rh6q5DHRmSSxaSDic5Ig9IianM2d86FFEv5CG2DcTUuppBuEiaBCFvkwh5hib4jGiaD96iawB2eiaRkQVRQ/640?wx_fmt=png&from=appmsg "")  
+  
+与五个强劲基线（CodeBERT、CustomVulBERTa、BERT、VUDENC、LineVul）对比实验表明，FuncVul在绝大多数指标和实验场景下均稳居第一，准确率区间为87%-92%，F1分数区间为86%-92%，远优于函数整体建模方法。以Dataset 1为例，块级建模在准确率上提升高达53.9%，F1分数提升42%。泛化代码块的实验结果虽略逊于原始块，但仍达到高水平，且具更强实时适配性。对于完全“未知补丁”、“新项目ID”等泛化挑战集，FuncVul依然取得了81.95%-76.69%的准确率和90%以上的召回，显示了良好的领域迁移性能。扩展不同块长度的实验进一步显示，3-行扩展块为最佳折中，既提供足够上下文，又不引入冗余干扰。  
+  
+此外，通过分块方式可自动拆解同一函数内的多个漏洞点——这一能力为函数级检测带来了全新可能。实验中，FuncVul有效捕捉并定位了复杂函数内多处高风险区域，极大减轻了专家人工定位难度。  
+  
+  
+**论文结论**  
+  
+  
+本文提出了一种创新性的函数级漏洞检测模型FuncVul，结合了补丁感知的高质量数据生成机制、基于代码块的精细划分策略以及大规模预训练模型（GraphCodeBERT）的微调能力，实现在C/C++和Python代码环境下对函数内部多个风险点的精准检测。通过系统性实验，FuncVul在各类真实数据集上均显示出卓越性能，远超现有主流方法，尤其是在定位精度、泛化能力及多漏洞检测等方面展现了独特优势。论文亦表明三行代码扩展块的策略效果最佳，数据泛化与模型迁移兼具高效和准确。所有实验流程、数据与代码工具已公开，为软件供应链安全、自动化漏洞修复和学术研究开辟了新路径。  
+  
+研究亦坦陈当前尚未探究多类别漏洞风险分级问题，以及仅覆盖C/C++和Python两类主流语言。未来，作者计划进一步拓展到更多编程语言，尝试多分类检测、多标签风险评估，并结合漏洞严重度、优先级等需求，提升实际应用价值与自动化水平。FuncVul的提出和开源，将为软件安全、代码智能分析及训练数据共享提供持久支撑，推动自动化漏洞检测技术的实用落地与学科进步。  
+  
+[](https://mp.weixin.qq.com/s?__biz=MzkzNDUxOTk2Mw==&mid=2247495405&idx=1&sn=67249648d5c312b5c178b23b077d28f3&scene=21#wechat_redirect)  
+  
+![图片](https://mmbiz.qpic.cn/mmbiz_png/vWuBpewLia8R7Rm0KL55HCcIiasO8JJ7IibXzYxx3losWVb2eddxdClACzWxWtQLwl0wkAl1ZLibcESVWvx5dCeibtQ/640?wx_fmt=png&from=appmsg&wxfrom=5&wx_lazy=1&wx_co=1&retryload=1&tp=webp "")  
+  
+  
+![图片](https://mmbiz.qpic.cn/mmbiz_png/vWuBpewLia8QRqLMRicZIN6VJg0ue41W1HVSmDpDqkj86j5SNicNE3X5KkPgcdv1ZmxM7FXrFUdkBes8dpos7d27w/640?wx_fmt=png&from=appmsg&wxfrom=5&wx_lazy=1&retryload=1&tp=webp "")  
+  
@@ -0,0 +1,235 @@
+#  悟空Agent实战：LLaMA-Factory高危0day漏洞挖掘与修复  
+原创 腾讯悟空团队  腾讯技术工程   2025-07-01 09:38  
+  
+![图片](https://mmbiz.qpic.cn/sz_mmbiz_gif/j3gficicyOvasVeMDmWoZ2zyN8iaSc6XWYj79H3xfgvsqK9TDxOBlcUa6W0EE5KBdxacd2Ql6QBmuhBJKIUS4PSZQ/640?wx_fmt=gif&from=appmsg&tp=webp&wxfrom=5&wx_lazy=1 "")  
+  
+作者：腾讯悟空团队 — 新一代 AI 代码安全捉“妖”行者（原腾讯AI安全-啄木鸟团队)  
+  
+在  
+前文[《0day漏洞量产？AI Agent“生产线”曝光！》](https://mp.weixin.qq.com/s?__biz=MjM5ODYwMjI2MA==&mid=2649793732&idx=1&sn=59278d0b2074a9c856a0700dfac2b83d&scene=21#wechat_redirect)  
+中，我们揭示了悟空AI Agent的架构与自动化漏洞挖掘能力。本次，我们将以53K Star的开源明星项目LLaMA-Factory为战场，详细展示悟空AI Agent如何在实际场景中，精准挖掘高危远程代码执行  
+0  
+day  
+漏洞（CVE-2025-53002），并推动官方修复的技术实战。  
+  
+### 一、LLaMA-Factory简介  
+  
+[LLaMA-Factory](https://www.llamafactory.cn/)  
+  
+是一个功能强大且用户友好的开源框架，在 GitHub 上斩获了超过 53K Stars，专注于大语言模型（LLM）的高效微调（Fine-tuning）。凭借其易用性和灵活性，LLaMA-Factory 已成为许多开发者和研究团队进行模型训练与部署的首选工具。  
+  
+![](https://mmbiz.qpic.cn/sz_mmbiz_png/j3gficicyOvav75ZsABYYfS0icVlXoticev6A5fibBGiakicy7gp8hPqIjBqhibt9zLJhSOGe4xrGGJTDChuXxMswE1TRA/640?wx_fmt=png&from=appmsg "")  
+  
+  
+（图：LLaMA-Factory Github   
+Index  
+）  
+  
+然而，面对这样的大型开源社区项目，当我们使用悟空AI Agent对其进行深度安全审计时，也能发现潜在的安全风险，成功识别了一个严重的远程代码执行漏洞：CVE-2025-53002（CVSS v3.1: 8.3）。  
+  
+  
+![](https://mmbiz.qpic.cn/sz_mmbiz_png/j3gficicyOvav75ZsABYYfS0icVlXoticev6hLZuicgd9YDTOAHp82z5hTMb6icz8aRkNuIuC18FDAG7cSmZqs6FmmYg/640?wx_fmt=png&from=appmsg "")  
+  
+  
+（  
+图  
+：  
+LLaMA-Factory Github  
+  
+Star  
+  
+History  
+）  
+  
+### 二、悟空AI Agent的实战效能  
+  
+悟空AI Agent在此次漏洞挖掘中展示了其强大的自动化漏洞挖掘能力，特别是在多智能体协作架构（Multi-Agent）的支持下，突破了传统手段的限制。通过以下几个关键阶段，悟空AI Agent逐步发现了漏洞的核心  
+：  
+  
+●  
+   
+Audit Agent：   
+首先，  
+Audit Agent负责从  
+  
+Web UI  
+  
+输入追踪到后台的数据处理参数及接口  
+，  
+从入口出发追踪目标数据流，并自动决策对下层调用函数的展开分析，直至发现存在漏洞特征的危险函数调用（不安全的反序列化 API 调用）。  
+  
+●  
+   
+Review Agent： 随后，Review Agent  
+  
+对 Audit Agent  
+  
+所提取的漏洞相关代码信息进行进一步分析，通过多重校验与投票机制评估该漏洞触发的约束条件，确定其可利用性。  
+  
+●  
+   
+Fix Agent：  
+最终  
+，  
+Fix Agent基于CVEs数据库和内部知识库生成了漏洞修复建议，形成了具体的修复方案。  
+  
+![](https://mmbiz.qpic.cn/sz_mmbiz_png/j3gficicyOvav75ZsABYYfS0icVlXoticev6ZCZzlEXSZl2hFAAfZapMEwUrx0z3BicPFhgb0R3YEa4tYzkCVSBkUUQ/640?wx_fmt=png&from=appmsg "")  
+  
+  
+（图：悟空  
+  
+AI  
+ Agent   
+架构  
+图）  
+  
+悟空  
+  
+AI Agent  
+  
+在此过程中展现了卓越的  
+程序分析与漏洞发现  
+能力，能够穿透复杂逻辑链，精确定位到不安全的反序列化操作（  
+torch.load  
+）及其触发路径。  
+悟空  
+  
+AI Agent  
+  
+不仅为传统的漏洞挖掘方法提供了补充，也为自动化安全审计提供了可行的解决方案。  
+  
+### 三、官方响应与修复  
+  
+在确认该漏洞有效后，我们立即通过GitHub Security Advisories渠道向LLaMA-Factory官方团队提交了详尽的技术报告（含PoC复现步骤，已同步报送CNVD等监管单位）。  
+## 官方响应：  
+  
+ LLaMA-Factory团队对此高度重视，迅速确认了漏洞的有效性和严重性。并颁发CVE漏洞编号。  
+  
+![](https://mmbiz.qpic.cn/sz_mmbiz_png/j3gficicyOvav75ZsABYYfS0icVlXoticev6GgDiaP3bWNv2BWW5SP1WaTkJWGzgUN9OaaWPzI96pNRMdB9AE5hCAQQ/640?wx_fmt=png&from=appmsg "")  
+  
+  
+（图：  
+[Github官方项目Security公告](https://github.com/hiyouga/LLaMA-Factory/security/advisories/GHSA-xj56-p8mm-qmxj)  
+  
+）  
+## 修复方案：  
+  
+官方对  
+src/llamafactory/model/model_utils/valuehead.py  
+ 的关键代码进行了加固：在  
+torch.load()  
+调用中  
+显式添加了安全参数   
+weights_only=True  
+，从根本上阻断了利用Pickle反序列化执行任意代码的可能性。  
+```
+
+```  
+  
+```
+# src/llamafactory/model/model_utils/valuehead.py  - state_dict = torch.load(vhead_file, map_location="cpu")  + state_dict = torch.load(vhead_file, map_location="cpu", weights_only=True)  
+```  
+  
+  
+▲   
+[关键补丁：启用PyTorch安全模式阻断反序列化攻击](https://github.com/hiyouga/LLaMA-Factory/commit/bb7bf51554d4ba8432333c35a5e3b52705955ede)  
+  
+  
+### 四、 技术分析  
+## 漏洞摘要   
+  
+在 LLaMA Factory 的训练过程中  
+，  
+发现一处严重的远程命令执行（RCE）漏洞。该漏洞源于加载   
+vhead_file  
+ 时未采取充分的安全措施。恶意攻击者仅需通过 WebUI 界面传递一个恶意的   
+Checkpoint path  
+ 参数，即可在宿主系统上执行任意恶意代码。整个攻击过程具有隐蔽性，受害者难以察觉。漏洞的根本原因是加载   
+vhead_file  
+ 参数时，未设置安全参数   
+weights_only=True  
+。  
+> 注意：  
+ 在 torch 版本 < 2.6 中，  
+torch.load()  
+ 的默认设置为   
+weights_only=False  
+。而 LLaMA Factory 的   
+setup.py  
+ 仅要求   
+torch>=2.0.0  
+，使得受影响版本默认处于不安全状态。  
+>   
+  
+  
+影响范围  
+  
+该漏洞影响的范围包括所有使用LLaMA-Factory框架的版本（<=0.9.3）。受影响的产品涉及广泛的应用场景，包括大语言模型的训练与推理。  
+  
+风险等级  
+  
+CVSS 3.1评分为8.3（高危），表明该漏洞对安全性构成了极大的威胁。  
+  
+CVE-2025-53002漏洞的存在使得攻击者能够通过精心构造的恶意文件，在LLaMA-Factory框架中实现远程代码执行，严重威胁使用该框架进行大语言模型训练和推理的组织。  
+  
+成功利用此漏洞后，攻击者能够执行任意恶意代码，可能导致权限提升、信息泄露、数据篡改等一系列恶劣后果，甚至可能破坏整个AI基础设施的安全性。  
+## 漏洞触发流程  
+  
+1.   
+在 LLaMA Factory 的 WebUI 中，当用户设置   
+Checkpoint path  
+ 时，该值会修改传递给训练过程的   
+adapter_name_or_path  
+ 参数。  
+  
+![](https://mmbiz.qpic.cn/sz_mmbiz_png/j3gficicyOvav75ZsABYYfS0icVlXoticev6X3JIxuiaf49B2XaVra1mvJNiac3GJsEicSibibuzme4jjMrlYxbYG5CFYJg/640?wx_fmt=png&from=appmsg "")  
+  
+  
+（图：相关代码位置：  
+src/llamafactory/webui/runner.py  
+）  
+  
+2.   
+训练过程中使用的   
+adapter_name_or_path  
+ 参数，在   
+src/llamafactory/model/model_utils/valuehead.py  
+ 文件中用于从 Hugging Face 获取对应的   
+value_head.bin  
+ 文件。  
+  
+3.   
+随后，该文件通过   
+torch.load()  
+ 加载，  
+但未设置安全参数   
+weights_only=True  
+，最终导致远程代码执行。  
+  
+![](https://mmbiz.qpic.cn/sz_mmbiz_png/j3gficicyOvav75ZsABYYfS0icVlXoticev6gMkOpRjKicvXog7O4uIzibPAqtxYTmfBiaLxld1vlwvjP2IOv1GkWFicXQ/640?wx_fmt=png&from=appmsg "")  
+  
+  
+（图：  
+相关代码位置：  
+src/llamafactory/model/model_utils/valuehead.py  
+）  
+  
+### 结语  
+  
+本次对LLaMA-Factory的实战审计，展示了悟空AI Agent在真实、复杂开源项目中发现高危漏洞的效率与能力。它不仅缩短了漏洞从潜伏到被发现的时间窗口，也通过标准化的报告流程加速了厂商响应与修复。未来，AI Agent将在网络安全领域扮演越来越核心的角色：  
+  
+1.   
+自动化深度审计  
+：实现对大型项目、复杂代码的高效、深度、持续的全自动安全审查。  
+  
+2.   
+0day威胁预警  
+：主动挖掘未知漏洞，从漏洞利用链上精准识别0day漏洞，提升整体防御前置性。  
+  
+3.   
+安全开发左移  
+：集成至CI/CD流程，辅助开发者在早期规避安全缺陷，降低修复成本。  
+  
+  
+悟空AI Agent的探索将持续深化，致力于推动安全研究与实践迈入智能化新纪元。  
+