Dify 组合漏洞获取LLM KEY
Dify 组合漏洞获取LLM KEY
蓝云Sec 2025-05-07 12:28
项目地址:https://github.com/langgenius/dify
1、前言：
目前该漏洞大部分已经修复了，该文只用于学习，请勿用于非法。
以下测试均为本地环境试验。
最近正巧使用Nas+dify完成部分工作，dify搭建好发现了@e0mlja师傅刚发的文章
2、漏洞利用
dify的配置文件中其实有许多的默认密码不限于SQL、OpenSearch等，具体的可以自己去看一下，前面的攻击思路就是通过默认口令
difyai123456连接postgresql，修改用户名密码(当然也可以去创建一个用户，可以参考e0mlja师傅的文章)
进入到后台后，创建一个知识库，上传任意文件，到数据库中修改地址获取到private.pem(后面再说为啥要获取这个东西)
最后找到加密后的key，写个解密脚本即可获取到key内容
3、漏洞分析
前面的内容一直到读取的部分，这里都不细说了，接下来只介绍key分析的部分
3.1、分析加密方式
先放图
如果我们直接在后台查看API key会发现是被加密的，从页面提示可以看到是通过PKCS1_OAEP进行加密
PKCS1_OAEP
PKCS1_OAEP是RSA加密算法的一种填充方案，用于提高加密安全性。它通过哈希函数和随机数填充明文，使相同明文加密产生不同密文。其主要特点是安全性高，能有效抵御多种攻击，并能验证数据完整性。它常用于密钥传输、数字签名和安全存储等场景，是RSA加密中更安全的选择。
这里我们知道大概逻辑后，下一步就是找三个东西了，私钥、公钥和密文
找公钥只是为了验证我们私钥是否正确
3.2、公钥、密钥获取
公钥和密文这两个其实都比较好找
公钥
在数据库里面有个tenants的表，里面存放的我们工作空间的内容，当然也包含我们相关的公钥
除了公钥以外，这个id也很重要，可以先记住，后面再详细说明是干啥用的
密文
密文存放位置在数据库中的providers表中
其中的encryptd_config也就是我们的key
大概内容如下
到这里我们前期需要的两个东西就有了，接下来是找私钥
3.3、私钥获取
全过程中最麻烦的一步
3.3.1 获取私钥位置
首先我们要分析私钥存放在哪个位置，才能去获取
如果你仔细查看过官方文档会找到一个STORAGE_LOCAL_PATH的参数，而该参数的默认指向/app/api/storage在storage中则存放着我们的private.pem，具体可以参考官方文档内容
这里的内容需要好好看，不然接下来你会踩和我一样的坑
这里知道private.pem的地址后我们可以去尝试拼接一下
/app/api/storage/[工作空间id]/private.pem
坑点1
如果你也是用这样的方式去尝试进行读取，那么恭喜你成功踩了第一个坑
你大概会得到如下结果
因为
/app/api/storage/[工作空间id]/private.pem并不是一个有效的绝对路径，你并不知道它前面还有哪些内容，当然如果你能推出它前面路径当我没说
例如
/var/docker/dify/test/app/api/storage/[工作空间id]/private.pem
解决方法
其实在最初的官方介绍中，我们便可以找到这个位置
注意分析这句话，我们可以推断出upload_file和我们私钥应该处于同一目录下
在github的源码中我找到了以下内容
ok，到这里其实已经很明显了，我们要找的私钥文件应当处于的相对位置是
/storage/privkeys/[工作空间id]/private.pem
这里我们起一个本地环境可以验证一下
此处工作空间id和我们数据库中正好能够对应，由此推断为正确的private.pem
那么就可以去获取一下我们的private.pem内容
将获取到的密钥内容，按照.pem格式拼接即可
当然，你也可以利用我们获取到的公钥对任意数据进行加密，尝试用获取到的私钥进行解密验证
结果显而易见
到这里如果你觉得已经完了，那你会发现拿着这个私钥无法解密你获取到的密文
这里也就是第二个坑
为什么会无法解密呢？难道是因为私钥不对？如果是私钥不正确，那么我们刚才用公钥加密的内容便无法被成功解密。
这里我的判断就是可能不存在一种加密方式
3.4、解密脚本撰写及分析
在api/commands.py中我们可以分析reset_encrypt_key_pair()命令会为每个工作空间（tenant）生成一对新的非对称密钥（RSA 2048位），私钥以 PEM 格式存储在
privkeys/{tenant_id}/private.pem，公钥则保存在数据库字段 tenant.encrypt_public_key 中
在dify-main/api/libs/rsa.py通过encrypt函数实现AES加密
加密时，首先生成一个随机的 AES 密钥（16字节），用 AES（EAX模式）对明文进行加密，然后用租户的 RSA 公钥对 AES 密钥进行加密。
最终密文格式为：HYBRID: 前缀 + 加密后的 AES 密钥 + AES nonce + tag + 密文内容。
该加密数据会被 base64 编码后存储。
解密流程
解密时，先用工作空间的私钥解密出 AES 密钥，再用 AES 密钥解密数据内容，恢复明文。
解密相关逻辑在 dify-main/api/libs/rsa.py 和 dify-main/api/core/helper/encrypter.py 中实现
那么接下来，我们便可以通过上面的方法来实现
总共两种实现方式
1、可以直接调用 libs/rsa.py 里的 decrypt 方法。但是这种方式对我们来说并不是特别方便，需要本地安装环境
2、写一个解密脚本
大致内容如下
最后成功获取KEY
放到dify验证一下，确实可以使用
以上便是Dify key泄露全过程，如果上面还有看不明白的地方或者需要相应脚本和环境的，可以私信找我要完整版本，如有问题欢迎沟通。