[cuda 函数劫持] cuda12.2 版本新增了一个函数 cuGetProcAddress_v2，请问如何进行劫持？
TimeNewRome · 2023-11-11 11:41:57 +08:00 · 1297 次点击这是一个创建于 386 天前的主题，其中的信息可能已经有所发展或是发生改变。
“static inline CUresult cuGetProcAddress_v2_ptsz(const char *symbol, void **funcPtr, int driverVersion, cuuint64_t flags, CUdriverProcAddressQueryResult *symbolStatus) {
const int procAddressMask = (CU_GET_PROC_ADDRESS_LEGACY_STREAM|
CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM);
if ((flags & procAddressMask) == 0) {
flags |= CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM;
}
return cuGetPr”
这个就是新增的函数结构。似乎跟之前的 cuGetProcAddress 函数差不多，只是新增了 CUdriverProcAddressQueryResult 这个结构体。请问这个函数该如何劫持呢？
const int procAddressMask = (CU_GET_PROC_ADDRESS_LEGACY_STREAM|
CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM);
if ((flags & procAddressMask) == 0) {
flags |= CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM;
}
return cuGetPr”
这个就是新增的函数结构。似乎跟之前的 cuGetProcAddress 函数差不多，只是新增了 CUdriverProcAddressQueryResult 这个结构体。请问这个函数该如何劫持呢？
 |
1
hexdecode 2023-11-11 13:40:31 +08:00
1. 函数签名复制：
首先，复制 cuGetProcAddress_v2_ptsz 的函数签名。这意味着你需要创建一个具有相同参数和返回类型的函数。 2. 创建自定义函数： static inline CUresult custom_cuGetProcAddress_v2_ptsz(const char *symbol, void **funcPtr, int driverVersion, cuuint64_t flags, CUdriverProcAddressQueryResult *symbolStatus) { // 在这里添加你的自定义逻辑，例如日志记录、参数修改等 // 调用原始函数 return cuGetProcAddress_v2_ptsz(symbol, funcPtr, driverVersion, flags, symbolStatus); } 3. 替换函数调用： 现在，你需要在运行时将对原 cuGetProcAddress_v2_ptsz 的调用替换为对 custom_cuGetProcAddress_v2_ptsz 的调用。这通常可以通过以下方法之一实现： • 修改动态链接库：如果是在 Windows 上，可能涉及到替换或修改 DLL 。 • 使用 LD_PRELOAD：在 Linux 上，可以使用 LD_PRELOAD 来劫持库函数。 • 直接修改函数指针：如果有机会修改指向 cuGetProcAddress_v2_ptsz 的函数指针，可以直接将其指向你的函数。 4. 测试和调试： 劫持后，需要在你的目标环境中进行充分测试，确保劫持成功且不会引入其他问题。 爱来自 GPTS |
Select Language