在任何全局对象之前初始化

Initialize before any global objects

本文关键字:初始化 对象 任何全      更新时间:2023-10-16

我正在写一个c++内存分析器。

众所周知,一个可执行程序可能包含许多全局对象,和它的DLL模块可以包含全局对象好。这些全局对象将用CRT初始化初始化——在入口点之后,在WinMain之前;dll,入口点是_DllMainCRTStartup,在进入DllMain之前,DLL的所有全局对象都被初始化。

全局对象可以分配内存。一个内存分析器必须在任何全局对象初始化之前初始化。后在做了大量的研究之后,我发现这不是一件容易的工作。

一个想法是使用CREATE_SUSPENDED标志的CreateProcess - try为了获得第一次机会。之后,使用CreateRemoteThread来调用LoadLibrary在目标进程中加载注入DLL,以及初始化DLL。但它不起作用,因为它会加载首先是可执行程序的所有隐式链接dll。也许CreateRemoteThread触发了这个行为?

那么,我们如何获得第一个机会呢?

可能有一种方法可以使用非常特定于平台的方法来做到这一点,但是解决这个问题的一种方法是将延迟初始化与dylib加载结合起来。

例如,假设您的内存分配器函数像这样导出:

API void* exported_alloc();
API void exported_free(void* mem);

…在一个名为mem.dll的dylib中。

在这种情况下,为了确保所有其他dylib在加载时都可以访问它,我们可以创建一个中央静态链接的库(例如:sdk.lib),所有dylib都可以链接到这个库,并使用如下标题:

#ifndef MEMORY_H
#define MEMORY_H
// Memory.h
void* my_alloc();
void my_free(void* mem);
#endif

…我们可以这样实现:

static void* (exported_alloc)() = 0;
static void (exported_free)(void* mem) = 0;
static void initialize()
{
    if (!exported_alloc)
    {
         // Load 'mem.dll' (ex: 'LoadLibrary') and look up
         // symbols for `exported_alloc` and `exported_free`
         // (ex: GetProcAddress).
    }
}
void* my_alloc()
{      
    initialize();
    return exported_alloc();
}
void my_free(void* mem)
{
    initialize();
    exported_free(mem);
}

. .然后在完成DLL后的适当时间调用FreeLibrary。这会产生一些运行时开销(类似于访问单例的开销),但这是一个跨平台的解决方案(如果您有在运行时加载/卸载dylibs/shared libs的跨平台方法)。

使用这个解决方案,所有在全局范围内分配内存的dll将在以惰性初始化的方式执行任何内存分配之前加载mem.dll,确保它们都可以在适当的时间访问您的内存函数。