執行緒區域性儲存（Thread Local Storage，TLS）用來將資料與一個正在執行的指定執行緒關聯起來。我們在應用程式和DLL中可以使用兩種型別的TLS：動態TLS和靜態TLS。但一般來說，這兩項技術在建立DLL的時候更加有用，這是因為DLL通常並不知道它們被連結到的應用程式的結構是什麼樣的。但是在編寫應用程式時，我們一般都知道自己要建立多少執行緒，自己會如何使用這些執行緒。然後我們就可以設計一些替代方案來為每個執行緒關聯資料，或者設計得好一點的話，可以使用基於棧的方法（區域性變數）來為每個執行緒關聯資料。

動態TLS：

一般通過呼叫一組4個API函式來使用動態TLS，這些函式實際上最經常為DLL

使用。系統中每個程序都有一組正在使用標誌（in-use flags），每個標誌可以被設為FREE或INUSE，表示該TLS元素是否正在被使用。Microsoft保證至少TLS_MINIMUM_AVAILABLE個位標誌可供使用。其中TLS_MINIMUM_AVAILABLE在WinNT.h中被定義為64，系統會在需要時分配更多的TLS元素，最多可達1000多個！

1）要使用動態TLS，必須先呼叫TlsAlloc函式：

DWORD WINAPI TlsAlloc(void);

這個函式讓系統對程序中的位標誌進行檢索並找到一個FREE標誌，然後系統會將該標誌從FREE改為INUSE並讓TlsAlloc

返回該標誌在位陣列中的索引。一個DLL（或應用程式）通常將這個索引儲存在一個全域性變數中。由於這個值會在整個程序地址範圍內使用，而不是線上程範圍內使用，因此這種情況下全域性變數是一個更好的選擇。

如果TlsAlloc無法在列表中找到一個FREE標誌，那麼它會返回TLS_OUT_OF_INDEXES（在WinBase.h中被定義為0xFFFFFFFF）。

當系統建立一個執行緒的時候，會分配TLS_MINIMUM_AVAILABLE個PVOID值，將它們都初始化為0，並與執行緒關聯起來。每個執行緒都有自己的PVOID陣列，陣列中的每個PVOID可以儲存任意值。在能夠將資訊儲存到執行緒的PVOID陣列中之前，我們必須知道陣列中的哪個索引可供使用---

這就是呼叫TlsAlloc的目的。TlsAlloc為我們預定了一個索引，如果為2，即TlsAlloc返回值為2，那麼無論是程序中當前正在執行的執行緒，還是今後可能會建立的執行緒，都不能再使用該索引2了。

2）為了把一個值放到執行緒的PVOID陣列中，應該呼叫TlsSetValue函式：

BOOL WINAPI TlsSetValue(

__inDWORD dwTlsIndex, //索引值，表示在陣列中的具體位置

__in_optLPVOID lpTlsValue //要設定的值

);

當一個執行緒呼叫TlsSetValue函式成功時，它會修改自己的PVOID陣列，但它無法修改另一個執行緒的TLS值。在呼叫TlsSetValue時，我們應該總是傳入前面在呼叫TlsAlloc時返回的索引。因為Windows為了效率犧牲了對輸入值的錯誤檢測。

3）為了從執行緒的陣列中取回一個值，應該呼叫函式TlsGetValue：

LPVOID WINAPI TlsGetValue(

__inDWORD dwTlsIndex //索引值

);

這個函式會返回在索引為dwTlsIndex的TLS元素中儲存的值。TlsGetValue只會檢視屬於呼叫執行緒的陣列。

4）當不再需要一個已經預定的TLS元素時，應該呼叫TlsFree函式：

BOOL WINAPI TlsFree(

__inDWORD dwTlsIndex //索引值

);

這個函式告訴系統已經預定的這個TLS元素現在不需要了，函式會將程序內的位標誌陣列中對應的INUSE標誌重新設回FREE。此外，函式還會將所有執行緒中該元素的內容設為0.

使用動態TLS：

通常，如果DLL要使用TLS，那它會在DllMain函式處理DLL_PROCESS_ATTACH的時候呼叫TlsAlloc，在DllMain處理DLL_PROCESS_DETACH的時候呼叫TlsFree。而TlsSetValue和TlsGetValue的呼叫則最有可能發生在DLL所提供的其他函式中。

而嚮應用程式中新增TLS的一種方法是直到需要時才新增。

下面是在應用程式中使用動態TLS的例項程式碼：

#include<windows.h>

#include<stdio.h>

#define THREADCOUNT 4

DWORD dwTlsIndex;

VOID ErrorExit(LPSTR);

VOID CommonFunc(VOID)

{

LPVOID lpvData;

// Retrieve a data pointer for the current thread.

lpvData = TlsGetValue(dwTlsIndex);

if ((lpvData == 0) && (GetLastError() != ERROR_SUCCESS))

ErrorExit("TlsGetValue error");

// Use the data stored for the current thread.

printf("common: thread %d: lpvData=%lx/n",

GetCurrentThreadId(), lpvData);

Sleep(5000);

}

DWORD WINAPI ThreadFunc(VOID)

{

LPVOID lpvData;

// Initialize the TLS index for this thread.

lpvData = (LPVOID) LocalAlloc(LPTR, 256);

if (! TlsSetValue(dwTlsIndex, lpvData))

ErrorExit("TlsSetValue error");

printf("thread %d: lpvData=%lx/n", GetCurrentThreadId(), lpvData);

CommonFunc();

// Release the dynamic memory before the thread returns.

lpvData = TlsGetValue(dwTlsIndex);

if (lpvData != 0)

LocalFree((HLOCAL) lpvData);

return 0;

}

int main(VOID)

{

DWORD IDThread;

HANDLE hThread[THREADCOUNT];

int i;

// Allocate a TLS index.

if ((dwTlsIndex = TlsAlloc()) == TLS_OUT_OF_INDEXES)

ErrorExit("TlsAlloc failed");

// Create multiple threads.

for (i = 0; i < THREADCOUNT; i++)

{

hThread[i] = CreateThread(NULL, // default security attributes

0,// use default stack size

(LPTHREAD_START_ROUTINE) ThreadFunc, // thread function

NULL,// no thread function argument

0,// use default creation flags

&IDThread);// returns thread identifier

// Check the return value for success.

if (hThread[i] == NULL)

ErrorExit("CreateThread error/n");

}

for (i = 0; i < THREADCOUNT; i++)

WaitForSingleObject(hThread[i], INFINITE);

TlsFree(dwTlsIndex);

return 0;

}

VOID ErrorExit (LPSTR lpszMessage)

{

fprintf(stderr, "%s/n", lpszMessage);

ExitProcess(0);

}

下面的例項程式碼是在DLL中使用TLS的：

========================DLL程式碼==========================

// The DLL code

#include<windows.h>

static DWORD dwTlsIndex; // address of shared memory

// DllMain() is the entry-point function for this DLL.

BOOL WINAPI DllMain(HINSTANCE hinstDLL, // DLL module handle

DWORD fdwReason,// reason called

LPVOID lpvReserved)// reserved

{

LPVOID lpvData;

BOOL fIgnore;

switch (fdwReason)

{

// The DLL is loading due to process

// initialization or a call to LoadLibrary.

case DLL_PROCESS_ATTACH:

// Allocate a TLS index.

if ((dwTlsIndex = TlsAlloc()) == TLS_OUT_OF_INDEXES)

return FALSE;

// No break: Initialize the index for first thread.

// The attached process creates a new thread.

相關推薦