Alexandre Julliard : kernel32: Implement the fiber local storage functions.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: b2ad268a8b7d7ac65029bbdd71d14f2c0f1b260d
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=b2ad268a8b7d7ac65029bbdd71...

Author: Alexandre Julliard 
Date:   Fri Feb 29 19:56:04 2008 +0100

kernel32: Implement the fiber local storage functions.

---

dlls/kernel32/fiber.c |   88 ++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 73 insertions(+), 15 deletions(-)

diff --git a/dlls/kernel32/fiber.c b/dlls/kernel32/fiber.c
index e95ef86..fda5f92 100644
--- a/dlls/kernel32/fiber.c
+++ b/dlls/kernel32/fiber.c
@@ -48,6 +48,7 @@ struct fiber_data
     sigjmp_buf            jmpbuf;            /* 14 setjmp buffer (on Windows: CONTEXT) */
     DWORD                 flags;             /*    fiber flags */
     LPFIBER_START_ROUTINE start;             /*    start routine */
+    void                **fls_slots;         /*    fiber storage slots */
 };

@@ -107,6 +108,7 @@ LPVOID WINAPI CreateFiberEx( SIZE_T stack_commit, SIZE_T stack_reserve, DWORD fl
     fiber->except      = (void *)-1;
     fiber->start       = start;
     fiber->flags       = flags;
+    fiber->fls_slots   = NULL;
     return fiber;
 }

@@ -125,6 +127,7 @@ void WINAPI DeleteFiber( LPVOID fiber_ptr )
         ExitThread(1);
     }
     VirtualFree( fiber->stack_allocation, 0, MEM_RELEASE );
+    HeapFree( GetProcessHeap(), 0, fiber->fls_slots );
     HeapFree( GetProcessHeap(), 0, fiber );
 }

@@ -157,6 +160,7 @@ LPVOID WINAPI ConvertThreadToFiberEx( LPVOID param, DWORD flags )
     fiber->stack_allocation = NtCurrentTeb()->DeallocationStack;
     fiber->start            = NULL;
     fiber->flags            = flags;
+    fiber->fls_slots        = NtCurrentTeb()->FlsSlots;
     NtCurrentTeb()->Tib.u.FiberData = fiber;
     return fiber;
 }
@@ -188,6 +192,7 @@ void WINAPI SwitchToFiber( LPVOID fiber )

current_fiber->except      = NtCurrentTeb()->Tib.ExceptionList;
     current_fiber->stack_limit = NtCurrentTeb()->Tib.StackLimit;
+    current_fiber->fls_slots   = NtCurrentTeb()->FlsSlots;
     /* stack_allocation and stack_base never change */

/* FIXME: should save floating point context if requested in fiber->flags */
@@ -198,6 +203,7 @@ void WINAPI SwitchToFiber( LPVOID fiber )
         NtCurrentTeb()->Tib.StackBase     = new_fiber->stack_base;
         NtCurrentTeb()->Tib.StackLimit    = new_fiber->stack_limit;
         NtCurrentTeb()->DeallocationStack = new_fiber->stack_allocation;
+        NtCurrentTeb()->FlsSlots          = new_fiber->fls_slots;
         if (new_fiber->start)  /* first time */
             wine_switch_to_stack( start_fiber, new_fiber, new_fiber->stack_base );
         else
@@ -210,10 +216,40 @@ void WINAPI SwitchToFiber( LPVOID fiber )
  */
 DWORD WINAPI FlsAlloc( PFLS_CALLBACK_FUNCTION callback )
 {
-    FIXME( "%p: stub!\n", callback );
+    DWORD index;
+    PEB * const peb = NtCurrentTeb()->Peb;

-    SetLastError( ERROR_CALL_NOT_IMPLEMENTED );
-    return FLS_OUT_OF_INDEXES;
+    RtlAcquirePebLock();
+    if (!peb->FlsCallback &&
+        !(peb->FlsCallback = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY,
+                                        8 * sizeof(peb->FlsBitmapBits) * sizeof(void*) )))
+    {
+        SetLastError( ERROR_NOT_ENOUGH_MEMORY );
+        index = FLS_OUT_OF_INDEXES;
+    }
+    else
+    {
+        index = RtlFindClearBitsAndSet( peb->FlsBitmap, 1, 0 );
+        if (index != ~0U)
+        {
+            if (!NtCurrentTeb()->FlsSlots &&
+                !(NtCurrentTeb()->FlsSlots = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY,
+                                                        8 * sizeof(peb->FlsBitmapBits) * sizeof(void*) )))
+            {
+                RtlClearBits( peb->FlsBitmap, index, 1 );
+                index = FLS_OUT_OF_INDEXES;
+                SetLastError( ERROR_NOT_ENOUGH_MEMORY );
+            }
+            else
+            {
+                NtCurrentTeb()->FlsSlots[index] = 0; /* clear the value */
+                peb->FlsCallback[index] = callback;
+            }
+        }
+        else SetLastError( ERROR_NO_MORE_ITEMS );
+    }
+    RtlReleasePebLock();
+    return index;
 }

/***********************************************************************
@@ -221,10 +257,20 @@ DWORD WINAPI FlsAlloc( PFLS_CALLBACK_FUNCTION callback )
  */
 BOOL WINAPI FlsFree( DWORD index )
 {
-    FIXME( "%x: stub!\n", index );
+    BOOL ret;

-    SetLastError( ERROR_CALL_NOT_IMPLEMENTED );
-    return FALSE;
+    RtlAcquirePebLock();
+    ret = RtlAreBitsSet( NtCurrentTeb()->Peb->FlsBitmap, index, 1 );
+    if (ret) RtlClearBits( NtCurrentTeb()->Peb->FlsBitmap, index, 1 );
+    if (ret)
+    {
+        /* FIXME: call Fls callback */
+        /* FIXME: add equivalent of ThreadZeroTlsCell here */
+        if (NtCurrentTeb()->FlsSlots) NtCurrentTeb()->FlsSlots[index] = 0;
+    }
+    else SetLastError( ERROR_INVALID_PARAMETER );
+    RtlReleasePebLock();
+    return TRUE;
 }

/***********************************************************************
@@ -232,10 +278,13 @@ BOOL WINAPI FlsFree( DWORD index )
  */
 PVOID WINAPI FlsGetValue( DWORD index )
 {
-    FIXME( "%x: stub!\n", index );
-
-    SetLastError( ERROR_CALL_NOT_IMPLEMENTED );
-    return NULL;
+    if (index >= 8 * sizeof(NtCurrentTeb()->Peb->FlsBitmapBits) || !NtCurrentTeb()->FlsSlots)
+    {
+        SetLastError( ERROR_INVALID_PARAMETER );
+        return NULL;
+    }
+    SetLastError( ERROR_SUCCESS );
+    return NtCurrentTeb()->FlsSlots[index];
 }

/***********************************************************************
@@ -243,9 +292,18 @@ PVOID WINAPI FlsGetValue( DWORD index )
  */
 BOOL WINAPI FlsSetValue( DWORD index, PVOID data )
 {
-
-    FIXME( "%x, %p: stub!\n", index, data );
-
-    SetLastError( ERROR_CALL_NOT_IMPLEMENTED );
-    return FALSE;
+    if (index >= 8 * sizeof(NtCurrentTeb()->Peb->FlsBitmapBits))
+    {
+        SetLastError( ERROR_INVALID_PARAMETER );
+        return FALSE;
+    }
+    if (!NtCurrentTeb()->FlsSlots &&
+        !(NtCurrentTeb()->FlsSlots = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY,
+                                        8 * sizeof(NtCurrentTeb()->Peb->FlsBitmapBits) * sizeof(void*) )))
+    {
+        SetLastError( ERROR_NOT_ENOUGH_MEMORY );
+        return FALSE;
+    }
+    NtCurrentTeb()->FlsSlots[index] = data;
+    return TRUE;
 }