-/* $Id: fiber.c,v 1.3 2002/09/08 10:22:46 chorns Exp $
- *
- * FILE: lib/kernel32/thread/fiber.c
- *
- * ReactOS Kernel32.dll
- *
+/*
+ * COPYRIGHT: See COPYING in the top level directory
+ * PROJECT: ReactOS System Libraries
+ * FILE: lib/kernel32/thread/fiber.c
+ * PURPOSE: Fiber Implementation
+ * PROGRAMMERS:
+ * Alex Ionescu (alex@relsoft.net)
+ * KJK::Hyperion <noog@libero.it>
*/
-#include <windows.h>
+#include <k32.h>
+#define NDEBUG
+#include "../include/debug.h"
-/**********************************************************************
- * ConvertThreadToFiber
+typedef struct _FIBER /* Field offsets: */
+{ /* 32 bit 64 bit */
+ /* this must be the first field */
+ LPVOID Parameter; /* 0x00 0x00 */
+ struct _EXCEPTION_REGISTRATION_RECORD * ExceptionList; /* 0x04 0x08 */
+ LPVOID StackBase; /* 0x08 0x10 */
+ LPVOID StackLimit; /* 0x0C 0x18 */
+ LPVOID DeallocationStack; /* 0x10 0x20 */
+ CONTEXT Context; /* 0x14 0x28 */
+ ULONG GuaranteedStackBytes; /* 0x2E0 */
+ PVOID FlsData; /* 0x2E4 */
+ PVOID ActivationContextStack; /* 0x2E8 */
+} FIBER, *PFIBER;
+
+/*
+ * @implemented
*/
-LPVOID
-STDCALL
-ConvertThreadToFiber(
- LPVOID lpArgument
- )
+BOOL
+WINAPI
+ConvertFiberToThread(VOID)
{
- SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
- return NULL;
-}
+ PTEB pTeb = NtCurrentTeb();
+ DPRINT1("Converting Fiber to Thread\n");
+
+ /* the current thread isn't running a fiber: failure */
+ if(!pTeb->HasFiberData)
+ {
+ SetLastError(ERROR_INVALID_PARAMETER);
+ return FALSE;
+ }
+
+ /* this thread won't run a fiber anymore */
+ pTeb->HasFiberData = FALSE;
+ /* free the fiber */
+ if(pTeb->Tib.FiberData != NULL)
+ {
+ RtlFreeHeap(GetProcessHeap(), 0, pTeb->Tib.FiberData);
+ }
-/**********************************************************************
- * CreateFiber
+ /* success */
+ return TRUE;
+}
+
+/*
+ * @implemented
*/
LPVOID
-STDCALL
-CreateFiber(
- DWORD dwStackSize,
- LPFIBER_START_ROUTINE lpStartAddress,
- LPVOID lpArgument
- )
+WINAPI
+ConvertThreadToFiberEx(LPVOID lpParameter,
+ DWORD dwFlags)
{
- SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
- return NULL;
-}
+ PTEB pTeb = NtCurrentTeb();
+ PFIBER pfCurFiber;
+ DPRINT1("Converting Thread to Fiber\n");
+
+ /* the current thread is already a fiber */
+ if(pTeb->HasFiberData && pTeb->Tib.FiberData) return pTeb->Tib.FiberData;
+ /* allocate the fiber */
+ pfCurFiber = (PFIBER)RtlAllocateHeap(GetProcessHeap(),
+ 0,
+ sizeof(FIBER));
-/**********************************************************************
- * DeleteFiber
+ /* failure */
+ if(pfCurFiber == NULL)
+ {
+ SetLastError(ERROR_NOT_ENOUGH_MEMORY);
+ return NULL;
+ }
+
+ /* copy some contextual data from the thread to the fiber */
+ pfCurFiber->ExceptionList = pTeb->Tib.ExceptionList;
+ pfCurFiber->StackBase = pTeb->Tib.StackBase;
+ pfCurFiber->StackLimit = pTeb->Tib.StackLimit;
+ pfCurFiber->DeallocationStack = pTeb->DeallocationStack;
+ pfCurFiber->FlsData = pTeb->FlsData;
+ pfCurFiber->GuaranteedStackBytes = pTeb->GuaranteedStackBytes;
+ pfCurFiber->ActivationContextStack = pTeb->ActivationContextStackPointer;
+ pfCurFiber->Context.ContextFlags = CONTEXT_FULL;
+
+ /* Save FPU State if requsted */
+ if (dwFlags & FIBER_FLAG_FLOAT_SWITCH)
+ {
+ pfCurFiber->Context.ContextFlags |= CONTEXT_FLOATING_POINT;
+ }
+
+ /* associate the fiber to the current thread */
+ pTeb->Tib.FiberData = pfCurFiber;
+ pTeb->HasFiberData = TRUE;
+
+ /* success */
+ return (LPVOID)pfCurFiber;
+}
+
+/*
+ * @implemented
*/
-VOID
-STDCALL
-DeleteFiber(
- LPVOID lpFiber
- )
+LPVOID
+WINAPI
+ConvertThreadToFiber(LPVOID lpParameter)
{
- SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
- return;
+ /* Call the newer function */
+ return ConvertThreadToFiberEx(lpParameter, 0);
}
+/*
+ * @implemented
+ */
+LPVOID
+WINAPI
+CreateFiber(SIZE_T dwStackSize,
+ LPFIBER_START_ROUTINE lpStartAddress,
+ LPVOID lpParameter)
+{
+ /* Call the Newer Function */
+ return CreateFiberEx(dwStackSize, 0, 0, lpStartAddress, lpParameter);
+}
-/**********************************************************************
- * GetCurrentFiber
+/*
+ * @implemented
*/
-PVOID
-STDCALL
-GetCurrentFiber(VOID)
+LPVOID
+WINAPI
+CreateFiberEx(SIZE_T dwStackCommitSize,
+ SIZE_T dwStackReserveSize,
+ DWORD dwFlags,
+ LPFIBER_START_ROUTINE lpStartAddress,
+ LPVOID lpParameter)
{
- SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
- return NULL;
+ PFIBER pfCurFiber;
+ NTSTATUS nErrCode;
+ INITIAL_TEB usFiberInitialTeb;
+ CONTEXT ctxFiberContext;
+ PVOID ActivationContextStack = NULL;
+ DPRINT1("Creating Fiber\n");
+
+ #ifdef SXS_SUPPORT_ENABLED
+ /* Allocate the Activation Context Stack */
+ nErrCode = RtlAllocateActivationContextStack(&ActivationContextStack);
+ #endif
+
+ /* Allocate the fiber */
+ pfCurFiber = (PFIBER)RtlAllocateHeap(GetProcessHeap(),
+ 0,
+ sizeof(FIBER));
+ /* Failure */
+ if(pfCurFiber == NULL)
+ {
+ SetLastError(ERROR_NOT_ENOUGH_MEMORY);
+ return NULL;
+ }
+
+ /* Create the stack for the fiber */
+ nErrCode = BasepCreateStack(NtCurrentProcess(),
+ dwStackCommitSize,
+ dwStackReserveSize,
+ &usFiberInitialTeb);
+ /* Failure */
+ if(!NT_SUCCESS(nErrCode))
+ {
+ /* Free the fiber */
+ RtlFreeHeap(GetProcessHeap(), 0, pfCurFiber);
+
+ /* Failure */
+ SetLastErrorByStatus(nErrCode);
+ return NULL;
+ }
+
+ /* Clear the context */
+ RtlZeroMemory(&pfCurFiber->Context, sizeof(CONTEXT));
+
+ /* copy the data into the fiber */
+ pfCurFiber->StackBase = usFiberInitialTeb.StackBase;
+ pfCurFiber->StackLimit = usFiberInitialTeb.StackLimit;
+ pfCurFiber->DeallocationStack = usFiberInitialTeb.AllocatedStackBase;
+ pfCurFiber->Parameter = lpParameter;
+ pfCurFiber->ExceptionList = (struct _EXCEPTION_REGISTRATION_RECORD *)-1;
+ pfCurFiber->GuaranteedStackBytes = 0;
+ pfCurFiber->FlsData = NULL;
+ pfCurFiber->ActivationContextStack = ActivationContextStack;
+ pfCurFiber->Context.ContextFlags = CONTEXT_FULL;
+
+ /* Save FPU State if requsted */
+ if (dwFlags & FIBER_FLAG_FLOAT_SWITCH)
+ {
+ pfCurFiber->Context.ContextFlags |= CONTEXT_FLOATING_POINT;
+ }
+
+ /* initialize the context for the fiber */
+ BasepInitializeContext(&ctxFiberContext,
+ lpParameter,
+ lpStartAddress,
+ usFiberInitialTeb.StackBase,
+ 2);
+
+ /* Return the Fiber */
+ return pfCurFiber;
}
+/*
+ * @implemented
+ */
+VOID
+WINAPI
+DeleteFiber(LPVOID lpFiber)
+{
+ SIZE_T nSize = 0;
+ PVOID pStackAllocBase = ((PFIBER)lpFiber)->DeallocationStack;
+
+ /* free the fiber */
+ RtlFreeHeap(GetProcessHeap(), 0, lpFiber);
+
+ /* the fiber is deleting itself: let the system deallocate the stack */
+ if(NtCurrentTeb()->Tib.FiberData == lpFiber) ExitThread(1);
-/**********************************************************************
- * GetFiberData
+ /* deallocate the stack */
+ NtFreeVirtualMemory(NtCurrentProcess(),
+ &pStackAllocBase,
+ &nSize,
+ MEM_RELEASE);
+}
+
+/*
+ * @implemented
*/
-PVOID
-STDCALL
-GetFiberData(VOID)
+BOOL
+WINAPI
+IsThreadAFiber(VOID)
{
- SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
- return NULL;
+ return NtCurrentTeb()->HasFiberData;
}
-/**********************************************************************
- * SwitchToFiber
- */
+__declspec(noreturn)
VOID
-STDCALL
-SwitchToFiber(
- LPVOID lpFiber
- )
+WINAPI
+BaseFiberStartup(VOID)
{
- SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
- return;
+ PFIBER Fiber = GetFiberData();
+
+ /* Call the Thread Startup Routine */
+ DPRINT1("Starting Fiber\n");
+ BaseThreadStartup((LPTHREAD_START_ROUTINE)Fiber->Context.Eax,
+ (LPVOID)Fiber->Context.Ebx);
}
-
/* EOF */