学步园

DLL 里面使用TLS (Local Thread Storage
)
的常见做法是：在DLLMain的DLL_PROCESS_ATTACH/DLL_THREAD_ATTACH
被调用的时候为每个线程（Thread）分配内存，而在DLL_THREAD_DETACH/DLL_PROCESS_DETACH
被调用的时候释放内存。 MSDN文章《Using Thread Local Storage in a Dynamic-Link Library》
上有这样的示例代码。

BOOL WINAPI DllMain(HINSTANCE hinstDLL, // DLL module handle<br /> DWORD fdwReason, // reason called<br /> LPVOID lpvReserved) // reserved<br /> {<br /> LPVOID lpvData;<br /> BOOL fIgnore;<br /> switch (fdwReason)<br /> {<br /> case DLL_PROCESS_ATTACH:<br /> // Allocate a TLS index.<br /> if ((dwTlsIndex = TlsAlloc()) == TLS_OUT_OF_INDEXES) return FALSE;<br /> case DLL_THREAD_ATTACH:<br /> lpvData = (LPVOID) LocalAlloc(LPTR, 256); //为每个Thread分配内存<br /> if (lpvData != NULL)<br /> fIgnore = TlsSetValue(dwTlsIndex, lpvData);<br /> break;<br /> case DLL_THREAD_DETACH:<br /> lpvData = TlsGetValue(dwTlsIndex);<br /> if (lpvData != NULL)<br /> LocalFree((HLOCAL) lpvData); //释放内存<br /> break;<br /> case DLL_PROCESS_DETACH:<br /> lpvData = TlsGetValue(dwTlsIndex);<br /> if (lpvData != NULL)<br /> LocalFree((HLOCAL) lpvData); //释放内存<br /> TlsFree(dwTlsIndex);<br /> break;<br /> default:<br /> break;<br /> }<br /> return TRUE;<br /> }

这段代码认为DLL_THREAD_DETACH 总是会被调用， 但实际情况并非如此。在某些情况下DLL_THREAD_DETACH并不会被调用， 结果造成内存泄漏。 接下来做2个简单实验说明这个问题。

实验代码：

typedef void (__stdcall *FNSLEEP)();<br /> void CallTestDLL()<br /> {<br /> FNSLEEP pfnSleep = (FNSLEEP)::GetProcAddress(g_hDLLModule, "DoSleep");<br /> ATLASSERT(pfnSleep);<br /> (*pfnSleep)();<br /> }<br /> DWORD WINAPI ThreadProc( LPVOID lpParam)<br /> {<br /> CallTestDLL();<br /> return 0;<br /> }<br /> g_hDLLModule = ::LoadLibrary(_T("TestDLL.dll"));<br /> ATLTRACE("[Thread %d] LoadLibrary=0x%.8x/n", ::GetCurrentThreadId());<br /> CallTestDLL();<br /> const int MAX_THREAD = 2;<br /> HANDLE hThread[MAX_THREAD];<br /> for (int i=0; i < MAX_THREAD; i++)<br /> {<br /> hThread[i] = ::CreateThread(NULL, 0, ThreadProc, 0, 0, NULL);<br /> }<br /> Sleep(MAX_THREAD * 1000);<br /> ::FreeLibrary(g_hDLLModule);<br />

输出结果1：

[Thread 4976] DLL_PROCESS_ATTACH //主线程<br /> [Thread 4976] LoadLibrary=0x0ecbf9d4<br /> [Thread 4976] DoSleep() in DLL<br /> [Thread 7860] DLL_THREAD_ATTACH //CreateThread 产生的线程<br /> [Thread 736] DLL_THREAD_ATTACH //CreateThread 产生的线程<br /> [Thread 736] DoSleep() in DLL<br /> [Thread 7860] DoSleep() in DLL<br /> [Thread 736] DLL_THREAD_DETACH<br /> [Thread 7860] DLL_THREAD_DETACH<br /> [Thread 4976] DLL_PROCESS_DETACH //主线程<br />

以上输入结果我们看到每个Thread 调用DLL函数DoSleep 立即结束，这时候DLL_THREAD_DETACH 被正常调用。 这时只要候稍微改一下代码，会看到完全不同的结果。

DWORD WINAPI ThreadProc( LPVOID lpParam)<br /> {<br /> CallTestDLL();<br /> DoSomethingElse(); // 延迟线程结束<br /> return 0;<br /> }<br />

输出结果2：

[Thread 7448] DLL_PROCESS_ATTACH //主线程<br /> [Thread 7448] LoadLibrary=0x0b1cf9d4<br /> [Thread 7448] DoSleep() in DLL<br /> [Thread 6872] DLL_THREAD_ATTACH<br /> [Thread 6556] DLL_THREAD_ATTACH<br /> [Thread 6556] DoSleep() in DLL<br /> [Thread 6872] DoSleep() in DLL<br /> [Thread 7448] DLL_PROCESS_DETACH //主线

我们发现，CreateThread 产生的线程并没有调用DLL_THREAD_DETACH 。

结论
：如果是线程在DLL被卸载（调用FreeLibrary） 之前结束，则DLL_THREAD_DETACH 会被调用。 如果线程在DLL卸载之后结束，则DLL_THREAD_DETACH 不会被调用。

本文转自Hellocpp: http://www.hellocpp.net/Articles/Article/590.aspx