tls.c

ReactOS是一些高手根据Windows XP的内核编写出的类XP。内核实现机理和API函数调用几乎相同。甚至可以兼容XP的程序。喜欢研究系统内核的人可以看一看。

#include <precomp.h>
#include <stdlib.h>
#include <internal/tls.h>
#include <internal/rterror.h>


static unsigned long TlsIndex = (unsigned long)-1;


static void InitThreadData(PTHREADDATA ThreadData)
{
   ThreadData->terrno = 0;
   ThreadData->tdoserrno = 0;

   ThreadData->fpecode = 0;

   ThreadData->tnext = 1;

   /* FIXME: init more thread local data */

}


int CreateThreadData(void)
{
   PTHREADDATA ThreadData;

   TlsIndex = TlsAlloc();
   if (TlsIndex == (unsigned long)-1)
     return FALSE;

   ThreadData = (PTHREADDATA)calloc(1, sizeof(THREADDATA));
   if (ThreadData == NULL)
     return FALSE;

   if(!TlsSetValue(TlsIndex, (LPVOID)ThreadData))
     return FALSE;

   InitThreadData(ThreadData);

   return TRUE;
}


void DestroyThreadData(void)
{
   if (TlsIndex != (unsigned long)-1)
     {
	TlsFree(TlsIndex);
	TlsIndex = (unsigned long)-1;
     }
}


void FreeThreadData(PTHREADDATA ThreadData)
{
   if (TlsIndex != (unsigned long)-1)
     {
	if (ThreadData == NULL)
	   ThreadData = TlsGetValue(TlsIndex);

	if (ThreadData != NULL)
	  {
	     /* FIXME: free more thread local data */

	     free(ThreadData);
	  }

	TlsSetValue(TlsIndex, NULL);
     }
}


PTHREADDATA GetThreadData(void)
{
   PTHREADDATA ThreadData;
   DWORD LastError;

   LastError = GetLastError();
   ThreadData = TlsGetValue(TlsIndex);
   if (ThreadData == NULL)
     {
	ThreadData = (PTHREADDATA)calloc(1, sizeof(THREADDATA));
	if (ThreadData != NULL)
	  {
	     TlsSetValue(TlsIndex, (LPVOID)ThreadData);

	     InitThreadData(ThreadData);
	  }
	else
	  {
	    _amsg_exit(_RT_THREAD); /* write message and die */
	  }
     }

   SetLastError(LastError);

   return ThreadData;
}

/* EOF */