fls.c

winNT技术操作系统,国外开放的原代码和LIUX一样

/* $Id: fls.c 21253 2006-03-08 21:33:04Z audit $
 *
 * COPYRIGHT:  See COPYING in the top level directory
 * PROJECT:    ReactOS system libraries
 * FILE:       lib/kernel32/thread/fls.c
 * PURPOSE:    Fiber local storage functions
 * PROGRAMMER: KJK::Hyperion <[e-mail]>
 *
 * UPDATE HISTORY:
 *             28/05/2003 - created. Stubs only
 *
 */

#include <k32.h>

#include "../include/debug.h"

/*
 * @unimplemented
 */
DWORD WINAPI FlsAlloc(PFLS_CALLBACK_FUNCTION lpCallback)
{
 (void)lpCallback;

 UNIMPLEMENTED;
 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
 return FLS_OUT_OF_INDEXES;
}


/*
 * @unimplemented
 */
BOOL WINAPI FlsFree(DWORD dwFlsIndex)
{
 (void)dwFlsIndex;

 UNIMPLEMENTED;
 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
 return FALSE;
}


/*
 * @implemented
 */
PVOID WINAPI FlsGetValue(DWORD dwFlsIndex)
{
 PVOID * ppFlsSlots;
 PVOID pRetVal;

 if(dwFlsIndex >= 128) goto l_InvalidParam;

 ppFlsSlots = NtCurrentTeb()->FlsData;

 if(ppFlsSlots == NULL) goto l_InvalidParam;

 SetLastError(0);
 pRetVal = ppFlsSlots[dwFlsIndex + 2];

 return pRetVal;

l_InvalidParam:
 SetLastError(ERROR_INVALID_PARAMETER);
 return NULL;
}


/*
 * @implemented
 */
BOOL WINAPI FlsSetValue(DWORD dwFlsIndex, PVOID lpFlsData)
{
 PVOID * ppFlsSlots;
 TEB * pTeb = NtCurrentTeb();

 if(dwFlsIndex >= 128) goto l_InvalidParam;

 ppFlsSlots = pTeb->FlsData;

 if(ppFlsSlots == NULL)
 {
  PEB * pPeb = pTeb->ProcessEnvironmentBlock;

  ppFlsSlots = RtlAllocateHeap
  (
   pPeb->ProcessHeap,
   HEAP_ZERO_MEMORY,
   (128 + 2) * sizeof(PVOID)
  );

  if(ppFlsSlots == NULL) goto l_OutOfMemory;

  pTeb->FlsData = ppFlsSlots;

  RtlAcquirePebLock();

  /* TODO: initialization */

  RtlReleasePebLock();
 }

 ppFlsSlots[dwFlsIndex + 2] = lpFlsData;

 return TRUE;

l_OutOfMemory:
 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
 goto l_Fail;

l_InvalidParam:
 SetLastError(ERROR_INVALID_PARAMETER);

l_Fail:
 return FALSE;
}

/* EOF */