让 DLL 卸载自身
今天的问题是:有没有可能让一个 DLL 自己卸载自己?
这个问题可以分成两个部分:
卸载一个 DLL。
卸载 DLL 的代码应该是放在 DLL之中的。
当然,如果不考虑后果的话,这个代码并不难写,如下:
#include <Windows.h>;
HMODULE g_hDll = NULL;
DWORD WINAPI UnloadProc(PVOID param)
{
MessageBox(NULL, TEXT("Press ok to unload me."),
TEXT("MsgBox in dll"), MB_OK);
FreeLibrary(g_hDll);
// oops!
return 0;
}
BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, PVOID lpvReserved)
{
if (DLL_PROCESS_ATTACH == fdwReason)
{
g_hDll = (HMODULE)hinstDLL;
HANDLE hThread = CreateThread(NULL, 0, UnloadProc, NULL, 0, NULL);
CloseHandle(hThread);
}
return TRUE;
}
简单说明一下:在 DllMain 初始化的时候保存 DLL 的实例句柄(即模块句柄)供 FreeLibrary 调用,然后开启一个线程,在适当的时机调用 FreeLibrary 销毁 DLL。
但是,如果实际运行起来的话,我们会遇到一个很实际的问题:在 FreeLibrary 之后,该 DLL 的地址空间就不再可用了,但这时 EIP 指针仍然会指向 FreeLibrary 的下面一句,于是程序崩溃。
所幸,Win32 提供了另外的一个 API——FreeLibraryAndExitThread,这个函数能够在销毁 DLL 之后直接调用 ExitThread,这样一来 EIP 指针就不会指向非法的地址了。因此,我们只需要把 FreeLibrary 的一句替换为:
FreeLibraryAndExitThread(g_hDll, 0);