将给定的 vararg 参数传递给另一个函数的适当方法是什么?



我已经开始在C++中研究钩子/事件系统。该系统应该处理应用程序其他部分通知的各种事件。

我面临的问题是我希望它运行的方式。 通常,我希望它能够使用您希望传递的所有参数调用特定函数,然后该函数处理调用该特定事件的所有注册钩子,将它们传递给参数,检索它们的结果值并将其返回给原始调用者。

一般来说,它应该是这样的:

CHookReturn* bInitializationStatus = Hook::Run("Initialize", gGame);
CHookReturn* bThinkSuccessful = Hook::Run("Think");



但是,我遇到了一个问题。 我以这样的方式设置它,即 Hook 命名空间中的Run函数调用 CHookData_t 结构的 Run 函数,需要传递 varargs。我找不到任何其他方法。结果是这样的:

union CHookReturn
{
const char* m_pszValue;
int m_iValue;
float m_flValue;
double m_dlValue;
bool m_bValue;
};
struct CHookData_t
{
virtual void Run(CHookReturn* ret, ...) = 0;
};
namespace Hook
{
std::unordered_map<const char*, std::unordered_map<const char*, CHookData_t*>> umHookList;
bool Add(const char*, const char*, CHookData_t*);
bool Exists(const char*, const char*);
bool Remove(const char*, const char*);
int Count(const char*);
CHookReturn* Run(const char*, ...);
};

Hook::Run 函数的 CPP 文件段:

CHookReturn* Hook::Run(const char* eventName, ...)
{
// FIXME: Look into alternative execution.
// This code seems more like a workaround
// than what I originally wanted it to be.
int count = Hook::Count(eventName);
CHookReturn* returnValues = new CHookReturn[count];
int c = 0;
unordered_map<const char*, CHookData_t*>::iterator itr;
unordered_map<const char*, CHookData_t*> res;
res = umHookList.at(eventName);
va_list valist;
void* args;
va_copy(args, valist);
for (itr = res.begin(); itr != res.end(); itr++)
{
CHookReturn returnData;
itr->second->Run(&returnData, args);
returnValues[c] = returnData;
++c;
}
return returnValues;
}

上面的代码提出了两个警告,让我质疑以这种方式执行它是否是一个好主意,以及我是否应该研究任何替代方案。

我收到的警告是:
Warning C6001 Using uninitialized memory 'valist'.
Warning C6386 Buffer overrun while writing to 'returnValues': the writable size is 'count*8' bytes, but '16' bytes might be written.

有没有更好的方法可以做到这一点?

使用va_list修复代码:

struct CHookData_t
{
virtual ~CHookData_t() {}
virtual void Run(CHookReturn* ret, va_list arg) = 0;
};
namespace Hook
{
using HooksList = std::unordered_map<
std::string, 
std::unordered_map<std::string, std::unique_ptr<CHookData_t>>;
HooksList umHookList;
...
}
std::vector<CHookReturn> Hook::Run(const std::string& eventName, ....)
{
va_list valist;
va_start(valist, eventName);
auto result = Hook::RunVarg(eventName, valist);
va_end(valist);
return result;
}
std::vector<CHookReturn> Hook::RunVarg(const std::string& eventName, va_list arg)
{
int count = Hook::Count(eventName);
std::vector<CHookReturn> returnValues(count);
size_t c = 0;
for (auto& item : umHookList.at(eventName))
{
va_list arg_copy;
va_copy(arg_copy, arg);
item.second->Run(&returnValues[c], arg_copy);
va_end(arg_copy);
++c;
}
return returnValues;
}

我不知道va_list指向Hook::Run的论据是什么,所以我不能提供很好的C++解决方案。

请注意,循环内部需要va_copy,因为某些编译器(不记得哪个,可能是 msvc)va_list的行为类似于指针,从中读取参数将对每次迭代产生影响。在其他编译器上,va_list的行为类似于值,并且va_copy不会更改任何内容。

题外话:你的代码是C,你不应该使用const char*,而应该使用std::stringstd::string_view如果你使用的是C++17,而不是va_args最好使用可变参数模板或std::initializer_list,避免使用原始指针以支持std::unique_ptrstd::shared_ptr。我已经稍微调整了您的代码以涵盖这一点。
此外,Hook不应该是一个namespace,从它包含的函数和变量的外观来看,它应该是一个类,所以你也应该修复它。

相关内容

最新更新