[API:KERNEL32] Implement _Inline_Tls(Alloc/Free)

Author: RatinCN

Source/Include/KNSoft/NDK/Win32/API/Kernel32.inl

@@ -561,6 +561,93 @@ _Inline_DeleteCriticalSection(
 
 #pragma region TLS / FLS
 
+_Must_inspect_result_
+__inline
+DWORD
+WINAPI
+_Inline_TlsAlloc(VOID)
+{
+    PPEB Peb;
+    ULONG Index;
+    PVOID* Slots;
+
+    Peb = NtCurrentPeb();
+    _Inline_RtlAcquirePebLock();
+    while (TRUE)
+    {
+        Index = RtlFindClearBitsAndSet(Peb->TlsBitmap, 1, 0);
+        if (Index != -1)
+        {
+            _Inline_RtlReleasePebLock();
+            NtWriteTeb(TlsSlots[Index], NULL);
+            return Index;
+        }
+        Slots = NtReadTeb(TlsExpansionSlots);
+        if (Slots != NULL)
+        {
+            break;
+        }
+        _Inline_RtlReleasePebLock();
+        Slots = (PVOID*)RtlAllocateHeap(RtlProcessHeap(), HEAP_ZERO_MEMORY, TLS_EXPANSION_SLOTS * sizeof(PVOID));
+        if (Slots == NULL)
+        {
+            goto _Fail;
+        }
+        NtWriteTeb(TlsExpansionSlots, Slots);
+        _Inline_RtlAcquirePebLock();
+    }
+    Index = RtlFindClearBitsAndSet(Peb->TlsExpansionBitmap, 1, 0);
+    _Inline_RtlReleasePebLock();
+    if (Index != -1)
+    {
+        Slots[Index] = NULL;
+        return Index + TLS_MINIMUM_AVAILABLE;
+    }
+
+_Fail:
+    _Inline_BaseSetLastNTError(STATUS_NO_MEMORY);
+    return TLS_OUT_OF_INDEXES;
+}
+
+__inline
+BOOL
+WINAPI
+_Inline_TlsFree(
+    _In_ DWORD dwTlsIndex)
+{
+    PPEB Peb;
+    ULONG Index;
+    PRTL_BITMAP Bitmap;
+
+    Peb = NtCurrentPeb();
+    Index = dwTlsIndex;
+    if (dwTlsIndex >= TLS_MINIMUM_AVAILABLE)
+    {
+        Index = dwTlsIndex - TLS_MINIMUM_AVAILABLE;
+        if (Index >= TLS_EXPANSION_SLOTS)
+        {
+            goto _Fail;
+        }
+        Bitmap = Peb->TlsExpansionBitmap;
+    } else
+    {
+        Bitmap = Peb->TlsBitmap;
+    }
+    _Inline_RtlAcquirePebLock();
+    if (RtlAreBitsSet(Bitmap, Index, 1) &&
+        NT_SUCCESS(NtSetInformationThread(NtCurrentThread(), ThreadZeroTlsCell, &dwTlsIndex, sizeof(dwTlsIndex))))
+    {
+        RtlClearBits(Bitmap, Index, 1);
+        _Inline_RtlReleasePebLock();
+        return TRUE;
+    }
+    _Inline_RtlReleasePebLock();
+
+_Fail:
+    _Inline_BaseSetLastNTError(STATUS_INVALID_PARAMETER);
+    return FALSE;
+}
+
 __inline
 BOOL
 WINAPI
@@ -612,9 +699,9 @@ _Inline_TlsGetValue(
         _Inline_BaseSetLastNTError(STATUS_INVALID_PARAMETER);
         return NULL;
     }
-    if (NtReadTeb(LastErrorValue) != ERROR_SUCCESS)
+    if (_Inline_RtlGetLastWin32Error() != ERROR_SUCCESS)
     {
-        NtWriteTeb(LastErrorValue, ERROR_SUCCESS);
+        _Inline_RtlSetLastWin32Error(ERROR_SUCCESS);
     }
     return Value;
 }