MFC线程局部存储

MFC占用系统TLS的一个槽口，将它扩展成很多个。

 

从程序启动说起。

// appmodul.cpp #ifndef _AFXDLL char _afxInitAppState = (char)(AfxInitialize(FALSE, _MFC_VER), atexit(&_AfxTermAppState)); // goto 10 #else char _afxInitAppState = (char)(AfxInitialize(FALSE, _MFC_VER)); #endif 

上面的代码定义了一个全局变量_afxInitAppState，该全局变量初始化时调用函数AfxInitialize()，也就是它初始化了MFC的线程局部存储。

 

// initialize app state such that it points to this module's core state // appmodul.cpp BOOL AFXAPI AfxInitialize(BOOL bDLL, DWORD dwVersion) { AFX_MODULE_STATE* pModuleState = AfxGetModuleState(); // ... return TRUE; }

AfxInitialize函数调用AfxGetModuleState()函数。

 

// afxstate.cpp AFX_MODULE_STATE* AFXAPI AfxGetModuleState() { _AFX_THREAD_STATE* pState = _afxThreadState; // ... } // afxtls_.h CThreadLocal<_AFX_THREAD_STATE> _afxThreadState;

_afxThreadState是一个全局变量，类型为CThreadLocal<_AFX_THREAD_STATE>，赋值为_AFX_THREAD_STATE*类型，调用CThreadLocal的成员函数如下：

 

template<class TYPE> class CThreadLocal : public CThreadLocalObject { // Attributes public: AFX_INLINE TYPE* GetData() { TYPE* pData = (TYPE*)CThreadLocalObject::GetData(&CreateObject); return pData; } //... AFX_INLINE operator TYPE*() { return GetData(); } //... // Implementation public: static CNoTrackObject* AFXAPI CreateObject() { return new TYPE; } };

接下来调用父类中的CThreadLocalObject::GetData()函数，它的一个参数是一个函数指针，用于创建指定_AFX_THREAD_STATE对象。

// // afxtls.cpp 404 CNoTrackObject* CThreadLocalObject::GetData( CNoTrackObject* (AFXAPI* pfnCreateObject)()) { if (m_nSlot == 0) { // 到这里，说明还没有分配SLOT if (_afxThreadData == NULL) { // MFC通过CThreadSlotData来扩展系统TLS，构造该对象时分配系统槽口 // ,将槽口号存放在CThreadSlotData::m_tlsIndex成员中。同时， // _afxThreadData是全局唯一的对象，也就是它来管理MFC的TLS _afxThreadData = new(__afxThreadData) CThreadSlotData; } // 分配MFC槽口，分配完成后CThreadLocalObject::m_nSlot成员记录了SLOT号 m_nSlot = _afxThreadData->AllocSlot(); } // 获取SLOT数据，存放的是指针（类型为：CNoTrackObject*）。 CNoTrackObject* pValue = static_cast<CNoTrackObject*>(_afxThreadData->GetThreadValue(m_nSlot)); if (pValue == NULL) { // 当前线程的该槽口中没有数据，先创建数据。 pValue = (*pfnCreateObject)(); // 设置MFC槽口数据为新分配的值 _afxThreadData->SetValue(m_nSlot, pValue); } return pValue; }  
 

详细代码如下：

CThreadSlotData::CThreadSlotData() { // 以CThreadData作为节点构成的链表，每个节点记录 // 了一个线程的SLOT数组指针和SLOT数组项数。每个 // 线程的TLS槽口保存一个CThreadData实例指针。 m_list.Construct(offsetof(CThreadData, pNext)); // initialize state and allocate TLS index m_nAlloc = 0; // 记录了SLOT状态数组的大小。 m_nRover = 1; // first slot (0) is always reserved m_nMax = 0; // SLOT状态数组中被占用过的最大单元序号。 // 以CSlotData作为数据项构成的动态线性表。dwFlags标志 // 记录了每个SLOT单元的空闲和占用状态，hInst记录了客户 // 对象所属的模块。SLOT状态数组每次以32单位扩展。 m_pSlotData = NULL; // MFC线程独立对象管理子系统只占用一个TLS槽口 // , 申请得到的TLS槽口号保存在m_tlsIndex属性中。 m_tlsIndex = TlsAlloc(); if (m_tlsIndex == (DWORD)-1) AfxThrowMemoryException(); InitializeCriticalSection(&m_sect); } // // afxtls.cpp 157 int CThreadSlotData::AllocSlot() { EnterCriticalSection(&m_sect); int nAlloc = m_nAlloc; int nSlot = m_nRover; if (nSlot >= nAlloc || (m_pSlotData[nSlot].dwFlags & SLOT_USED)) { // search for first free slot, starting at beginning for (nSlot = 1; nSlot < nAlloc && (m_pSlotData[nSlot].dwFlags & SLOT_USED); nSlot++) ; // if none found, need to allocate more space if (nSlot >= nAlloc) { // realloc memory for the bit array and the slot memory int nNewAlloc = m_nAlloc+32; HGLOBAL hSlotData; if (m_pSlotData == NULL) { hSlotData = GlobalAlloc(GMEM_MOVEABLE, nNewAlloc * sizeof(CSlotData)); } else { hSlotData = GlobalHandle(m_pSlotData); GlobalUnlock(hSlotData); hSlotData = GlobalReAlloc(hSlotData, nNewAlloc * sizeof(CSlotData), GMEM_MOVEABLE|GMEM_SHARE); } CSlotData* pSlotData = (CSlotData*)GlobalLock(hSlotData); // always zero initialize after success memset(pSlotData+m_nAlloc, 0, (nNewAlloc-m_nAlloc)*sizeof(CSlotData)); m_nAlloc = nNewAlloc; m_pSlotData = pSlotData; } } // adjust m_nMax to largest slot ever allocated if (nSlot >= m_nMax) m_nMax = nSlot+1; m_pSlotData[nSlot].dwFlags |= SLOT_USED; // update m_nRover (likely place to find a free slot is next one) m_nRover = nSlot+1; LeaveCriticalSection(&m_sect); return nSlot; // slot can be used for FreeSlot, GetValue, SetValue } inline void* CThreadSlotData::GetThreadValue(int nSlot) { EnterCriticalSection(&m_sect); if( nSlot <= 0 || nSlot >= m_nMax ) // check for retail builds. { LeaveCriticalSection(&m_sect); return NULL; } CThreadData* pData = (CThreadData*)TlsGetValue(m_tlsIndex); if (pData == NULL || nSlot >= pData->nCount) { LeaveCriticalSection(&m_sect); return NULL; } void* pRetVal = pData->pData[nSlot]; LeaveCriticalSection(&m_sect); return pRetVal; }