[cuda 函数劫持] cuda12.2 版本新增了一个函数 cuGetProcAddress_v2，请问如何进行劫持？

› Caffe

This topic created in 914 days ago, the information mentioned may be changed or developed.

“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 这个结构体。请问这个函数该如何劫持呢？

函数

flags

劫持

const

1 replies • 2023-11-11 13:40:31 +08:00

hexdecode

Nov 11, 2023

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