dllmain

Bertha 。 2022-11-11 10:51 128阅读 0赞

`win32`提供了一个特殊函数名叫`DllMain`，这个函数写在dll中会根据四种情况进行回调。

1.  某个线程加载了dll
2.  某个线程卸载了dll
3.  主进程加载了dll
4.  主进程卸载了dll

[微软dllmain官方文档][dllmain]

我们在`dll`工程随意新建一个类

//My.cpp
    #include<iostream>
    #include<process.h>
    
    #include <windows.h>
    using namespace std;
    BOOL WINAPI DllMain(
    	_In_ HINSTANCE hinstDLL,
    	_In_ DWORD     fdwReason,//调用的原因，
    	_In_ LPVOID    lpvReserved //保留使用
    ) {
        
    	switch (fdwReason)
    	{
        
    	//主进程加载了dll
    	case DLL_PROCESS_ATTACH:
    		cout <<"DLL_PROCESS_ATTACH" <<endl;
    		break;
    	//主进程卸载了dll
    	case DLL_PROCESS_DETACH:
    		cout << "DLL_PROCESS_DETACH" << endl;
    
    		break;
    	}
    	//如果动态加载的dll的话，返回false 那么LoadLibrary返回null
    	//如果静态加载的dll的话  返回false 进程直接奔溃
    	return TRUE;
    }

我们看下使用库的地方

// UseLib.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
    //
    
    #include <iostream>
    
    #include <windows.h>
    
    using namespace std;
    
    typedef int(*myAdd)(int a, int b);
    int main()
    {
        
    	//加载我们的动态库 请放入exe运行路径或者放入系统`C:\Windows\SysWOW64\`
    	HMODULE loadLibrary = LoadLibrary("MyDll.dll");
    	//通过name
    	myAdd procAddress = (myAdd)GetProcAddress(loadLibrary, (LPCSTR)1);
    	//通过ordinal
    	//myAdd procAddress2 = (myAdd)GetProcAddress(loadLibrary, (LPCSTR)12);
    
    	int result = procAddress(1, 2);
    	//int result2 = procAddress2(1, 2);
    	cout << result  << endl;
    	//从程序中卸载dll
    	FreeLibrary(loadLibrary);
    	
    }

输出：

DLL_PROCESS_ATTACH
    3
    DLL_PROCESS_DETACH

[dllmain]: https://docs.microsoft.com/en-us/windows/win32/dlls/dllmain