+++ /dev/null
-/*
- * Acess2
- * threads.c
- * - Common Thread Control
- */
-#include <acess.h>
-#include <threads.h>
-#include <errno.h>
-
-#define DEBUG_TRACE_TICKETS 0 // Trace ticket counts
-#define DEBUG_TRACE_STATE 0 // Trace state changes (sleep/wake)
-
-// === CONSTANTS ===
-#define DEFAULT_QUANTUM 10
-#define DEFAULT_TICKETS 5
-#define MAX_TICKETS 10
-const enum eConfigTypes cCONFIG_TYPES[] = {
- CFGT_HEAPSTR, // e.g. CFG_VFS_CWD
- CFGT_INT, // e.g. CFG_VFS_MAXFILES
- CFGT_NULL
-};
-
-// === IMPORTS ===
-extern void ArchThreads_Init(void);
-extern void Proc_Start(void);
-extern tThread *Proc_GetCurThread(void);
-extern int Proc_Clone(Uint *Err, Uint Flags);
-extern void Proc_CallFaultHandler(tThread *Thread);
-
-// === PROTOTYPES ===
-void Threads_Init(void);
- int Threads_SetName(char *NewName);
-char *Threads_GetName(int ID);
-void Threads_SetTickets(tThread *Thread, int Num);
-tThread *Threads_CloneTCB(Uint *Err, Uint Flags);
- int Threads_WaitTID(int TID, int *status);
-tThread *Threads_GetThread(Uint TID);
-void Threads_AddToDelete(tThread *Thread);
-tThread *Threads_int_GetPrev(tThread **List, tThread *Thread);
-void Threads_Exit(int TID, int Status);
-void Threads_Kill(tThread *Thread, int Status);
-void Threads_Yield(void);
-void Threads_Sleep(void);
- int Threads_Wake(tThread *Thread);
-void Threads_AddActive(tThread *Thread);
-tThread *Threads_RemActive(void);
- int Threads_GetPID(void);
- int Threads_GetTID(void);
-tUID Threads_GetUID(void);
- int Threads_SetUID(Uint *Errno, tUID ID);
-tGID Threads_GetGID(void);
- int Threads_SetGID(Uint *Errno, tUID ID);
-void Threads_Dump(void);
-void Mutex_Acquire(tMutex *Mutex);
-void Mutex_Release(tMutex *Mutex);
- int Mutex_IsLocked(tMutex *Mutex);
-
-// === GLOBALS ===
-// -- Core Thread --
-// Only used for the core kernel
-tThread gThreadZero = {
- Status: THREAD_STAT_ACTIVE, // Status
- ThreadName: "ThreadZero", // Name
- Quantum: DEFAULT_QUANTUM, // Default Quantum
- Remaining: DEFAULT_QUANTUM, // Current Quantum
- NumTickets: DEFAULT_TICKETS // Number of tickets
- };
-// -- Processes --
-// --- Locks ---
-tShortSpinlock glThreadListLock; ///\note NEVER use a heap function while locked
-// --- Current State ---
-volatile int giNumActiveThreads = 0; // Number of threads on the active queue
-volatile int giFreeTickets = 0; // Number of tickets held by non-scheduled threads
-volatile Uint giNextTID = 1; // Next TID to allocate
-// --- Thread Lists ---
-tThread *gAllThreads = NULL; // All allocated threads
-tThread *gActiveThreads = NULL; // Currently Running Threads
-tThread *gSleepingThreads = NULL; // Sleeping Threads
-tThread *gDeleteThreads = NULL; // Threads to delete
- int giNumCPUs = 1; // Number of CPUs
-BOOL gaThreads_NoTaskSwitch[MAX_CPUS]; // Disables task switches for each core (Pseudo-IF)
-
-// === CODE ===
-/**
- * \fn void Threads_Init(void)
- * \brief Initialse the thread list
- */
-void Threads_Init(void)
-{
- ArchThreads_Init();
-
- // Create Initial Task
- gActiveThreads = &gThreadZero;
- gAllThreads = &gThreadZero;
- //giFreeTickets = gThreadZero.NumTickets; // Not needed, as ThreadZero is scheduled
- giNumActiveThreads = 1;
-
- Proc_Start();
-}
-
-/**
- * \fn void Threads_SetName(char *NewName)
- * \brief Sets the current thread's name
- * \param NewName New name for the thread
- * \return Boolean Failure
- */
-int Threads_SetName(char *NewName)
-{
- tThread *cur = Proc_GetCurThread();
- char *oldname = cur->ThreadName;
-
- // NOTE: There is a possibility of non-thread safety here
- // A thread could read the current name pointer before it is zeroed
-
- cur->ThreadName = NULL;
-
- if( IsHeap(oldname) ) free( oldname );
-
- cur->ThreadName = strdup(NewName);
- return 0;
-}
-
-/**
- * \fn char *Threads_GetName(int ID)
- * \brief Gets a thread's name
- * \param ID Thread ID (-1 indicates current thread)
- * \return Pointer to name
- * \retval NULL Failure
- */
-char *Threads_GetName(tTID ID)
-{
- if(ID == -1) {
- return Proc_GetCurThread()->ThreadName;
- }
- return Threads_GetThread(ID)->ThreadName;
-}
-
-/**
- * \fn void Threads_SetTickets(tThread *Thread, int Num)
- * \brief Sets the 'priority' of a task
- * \param Thread Thread to update ticket count (NULL means current thread)
- * \param Num New ticket count (must be >= 0, clipped to \a MAX_TICKETS)
- */
-void Threads_SetTickets(tThread *Thread, int Num)
-{
- // Get current thread
- if(Thread == NULL) Thread = Proc_GetCurThread();
- // Bounds checking
- if(Num < 0) return;
- if(Num > MAX_TICKETS) Num = MAX_TICKETS;
-
- // If this isn't the current thread, we need to lock
- if( Thread != Proc_GetCurThread() ) {
- SHORTLOCK( &glThreadListLock );
- giFreeTickets -= Thread->NumTickets - Num;
- Thread->NumTickets = Num;
- #if DEBUG_TRACE_TICKETS
- Log("Threads_SetTickets: new giFreeTickets = %i", giFreeTickets);
- #endif
- SHORTREL( &glThreadListLock );
- }
- else
- Thread->NumTickets = Num;
-}
-
-/**
- * \fn tThread *Threads_CloneTCB(Uint *Err, Uint Flags)
- * \brief Clone the TCB of the current thread
- * \param Err Error pointer
- * \param Flags Flags for something... (What is this for?)
- */
-tThread *Threads_CloneTCB(Uint *Err, Uint Flags)
-{
- tThread *cur, *new;
- int i;
- cur = Proc_GetCurThread();
-
- // Allocate and duplicate
- new = malloc(sizeof(tThread));
- if(new == NULL) {
- *Err = -ENOMEM;
- return NULL;
- }
- memcpy(new, cur, sizeof(tThread));
-
- new->CurCPU = -1;
- new->Next = NULL;
- memset( &new->IsLocked, 0, sizeof(new->IsLocked));
- new->Status = THREAD_STAT_PREINIT;
- new->RetStatus = 0;
-
- // Get Thread ID
- new->TID = giNextTID++;
- new->Parent = cur;
-
- // Clone Name
- new->ThreadName = strdup(cur->ThreadName);
-
- // Set Thread Group ID (PID)
- if(Flags & CLONE_VM)
- new->TGID = new->TID;
- else
- new->TGID = cur->TGID;
-
- // Messages are not inherited
- new->Messages = NULL;
- new->LastMessage = NULL;
-
- // Set State
- new->Remaining = new->Quantum = cur->Quantum;
- new->NumTickets = cur->NumTickets;
-
- // Set Signal Handlers
- new->CurFaultNum = 0;
- new->FaultHandler = cur->FaultHandler;
-
- for( i = 0; i < NUM_CFG_ENTRIES; i ++ )
- {
- switch(cCONFIG_TYPES[i])
- {
- default:
- new->Config[i] = cur->Config[i];
- break;
- case CFGT_HEAPSTR:
- if(cur->Config[i])
- new->Config[i] = (Uint) strdup( (void*)cur->Config[i] );
- else
- new->Config[i] = 0;
- break;
- }
- }
-
- // Maintain a global list of threads
- SHORTLOCK( &glThreadListLock );
- new->GlobalPrev = NULL; // Protect against bugs
- new->GlobalNext = gAllThreads;
- gAllThreads = new;
- SHORTREL( &glThreadListLock );
-
- return new;
-}
-
-/**
- * \brief Get a configuration pointer from the Per-Thread data area
- * \param ID Config slot ID
- * \return Pointer at ID
- */
-Uint *Threads_GetCfgPtr(int ID)
-{
- if(ID < 0 || ID >= NUM_CFG_ENTRIES) {
- Warning("Threads_GetCfgPtr: Index %i is out of bounds", ID);
- return NULL;
- }
-
- return &Proc_GetCurThread()->Config[ID];
-}
-
-/**
- * \brief Wait for a task to change state
- * \param TID Thread ID to wait on (-1: Any child thread, 0: Any Child/Sibling, <-1: -PID)
- * \param Status Thread return status
- * \return TID of child that changed state
- */
-tTID Threads_WaitTID(int TID, int *Status)
-{
- // Any Child
- if(TID == -1) {
- Log_Error("Threads", "TODO: Threads_WaitTID(TID=-1) - Any Child");
- return -1;
- }
-
- // Any peer/child thread
- if(TID == 0) {
- Log_Error("Threads", "TODO: Threads_WaitTID(TID=0) - Any Child/Sibling");
- return -1;
- }
-
- // TGID = abs(TID)
- if(TID < -1) {
- Log_Error("Threads", "TODO: Threads_WaitTID(TID<0) - TGID");
- return -1;
- }
-
- // Specific Thread
- if(TID > 0) {
- tThread *t = Threads_GetThread(TID);
- int initStatus = t->Status;
- tTID ret;
-
- // Wait for the thread to die!
- if(initStatus != THREAD_STAT_ZOMBIE) {
- // TODO: Handle child also being suspended if wanted
- while(t->Status != THREAD_STAT_ZOMBIE) {
- Threads_Sleep();
- Log_Debug("Threads", "%i waiting for %i, t->Status = %i",
- Threads_GetTID(), t->TID, t->Status);
- }
- }
-
- // Set return status
- Log_Debug("Threads", "%i waiting for %i, t->Status = %i",
- Threads_GetTID(), t->TID, t->Status);
- ret = t->TID;
- switch(t->Status)
- {
- case THREAD_STAT_ZOMBIE:
- // Kill the thread
- t->Status = THREAD_STAT_DEAD;
- // TODO: Child return value?
- if(Status) *Status = t->RetStatus;
- // add to delete queue
- Threads_AddToDelete( t );
- break;
- default:
- if(Status) *Status = -1;
- break;
- }
- return ret;
- }
-
- return -1;
-}
-
-/**
- * \brief Gets a thread given its TID
- * \param TID Thread ID
- * \return Thread pointer
- */
-tThread *Threads_GetThread(Uint TID)
-{
- tThread *thread;
-
- // Search global list
- for(thread = gAllThreads;
- thread;
- thread = thread->GlobalNext)
- {
- if(thread->TID == TID)
- return thread;
- }
-
- Log("Unable to find TID %i on main list\n", TID);
-
- return NULL;
-}
-
-/**
- * \brief Adds a thread to the delete queue
- * \param Thread Thread to delete
- */
-void Threads_AddToDelete(tThread *Thread)
-{
- // Add to delete queue
- // TODO: Is locking needed?
- if(gDeleteThreads) {
- Thread->Next = gDeleteThreads;
- gDeleteThreads = Thread;
- } else {
- Thread->Next = NULL;
- gDeleteThreads = Thread;
- }
-}
-
-/**
- * \brief Gets the previous entry in a thead linked list
- * \param List Pointer to the list head
- * \param Thread Thread to find
- * \return Thread before \a Thread on \a List
- * \note This uses a massive hack of assuming that the first field in the
- * structure is the .Next pointer. By doing this, we can return \a List
- * as a (tThread*) and simplify other code.
- */
-tThread *Threads_int_GetPrev(tThread **List, tThread *Thread)
-{
- tThread *ret;
- // First Entry
- if(*List == Thread) {
- return (tThread*)List;
- }
- // Or not
- else {
- for(ret = *List;
- ret->Next && ret->Next != Thread;
- ret = ret->Next
- );
- // Error if the thread is not on the list
- if(!ret->Next || ret->Next != Thread) {
- return NULL;
- }
- }
- return ret;
-}
-
-/**
- * \brief Exit the current process (or another?)
- * \param TID Thread ID to kill
- * \param Status Exit status
- */
-void Threads_Exit(int TID, int Status)
-{
- if( TID == 0 )
- Threads_Kill( Proc_GetCurThread(), (Uint)Status & 0xFF );
- else
- Threads_Kill( Threads_GetThread(TID), (Uint)Status & 0xFF );
-
- // Halt forever, just in case
- for(;;) HALT();
-}
-
-/**
- * \fn void Threads_Kill(tThread *Thread, int Status)
- * \brief Kill a thread
- * \param Thread Thread to kill
- * \param Status Status code to return to the parent
- */
-void Threads_Kill(tThread *Thread, int Status)
-{
- tThread *prev;
- tMsg *msg;
-
- // TODO: Kill all children
- #if 0
- {
- tThread *child;
- // TODO: I should keep a .Parent pointer, and a .Children list
- for(child = gActiveThreads;
- child;
- child = child->Next)
- {
- if(child->PTID == Thread->TID)
- Threads_Kill(child, -1);
- }
- }
- #endif
-
- ///\note Double lock is needed due to overlap of lock areas
-
- // Lock thread (stop us recieving messages)
- SHORTLOCK( &Thread->IsLocked );
-
- // Clear Message Queue
- while( Thread->Messages )
- {
- msg = Thread->Messages->Next;
- free( Thread->Messages );
- Thread->Messages = msg;
- }
-
- // Lock thread list
- SHORTLOCK( &glThreadListLock );
-
- // Get previous thread on list
- prev = Threads_int_GetPrev( &gActiveThreads, Thread );
- if(!prev) {
- Warning("Proc_Exit - Current thread is not on the active queue");
- SHORTREL( &glThreadListLock );
- SHORTREL( &Thread->IsLocked );
- return;
- }
-
- // Ensure that we are not rescheduled
- Thread->Remaining = 0; // Clear Remaining Quantum
- Thread->Quantum = 0; // Clear Quantum to indicate dead thread
- prev->Next = Thread->Next; // Remove from active
-
- // Update bookkeeping
- giNumActiveThreads --;
- if( Thread != Proc_GetCurThread() )
- giFreeTickets -= Thread->NumTickets;
-
- // Save exit status
- Thread->RetStatus = Status;
-
- // Don't Zombie if we are being killed because our parent is
- if(Status == -1)
- {
- Thread->Status = THREAD_STAT_DEAD;
- Threads_AddToDelete( Thread );
- } else {
- Thread->Status = THREAD_STAT_ZOMBIE;
- // Wake parent
- Threads_Wake( Thread->Parent );
- }
-
- Log("Thread %i went *hurk* (%i)", Thread->TID, Thread->Status);
-
- // Release spinlocks
- SHORTREL( &glThreadListLock );
- SHORTREL( &Thread->IsLocked ); // TODO: We may not actually be released...
-
- // And, reschedule
- if(Status != -1) {
- for( ;; )
- HALT();
- }
-}
-
-/**
- * \brief Yield remainder of the current thread's timeslice
- */
-void Threads_Yield(void)
-{
- tThread *thread = Proc_GetCurThread();
- thread->Remaining = 0;
- //while(thread->Remaining == 0)
- HALT();
-}
-
-/**
- * \fn void Threads_Sleep(void)
- * \brief Take the current process off the run queue
- */
-void Threads_Sleep(void)
-{
- tThread *cur = Proc_GetCurThread();
-
- // Acquire Spinlock
- SHORTLOCK( &glThreadListLock );
-
- // Don't sleep if there is a message waiting
- if( cur->Messages ) {
- SHORTREL( &glThreadListLock );
- return;
- }
-
- // Remove us from running queue
- Threads_RemActive();
- // Mark thread as sleeping
- cur->Status = THREAD_STAT_SLEEPING;
-
- // Add to Sleeping List (at the top)
- cur->Next = gSleepingThreads;
- gSleepingThreads = cur;
-
-
- #if DEBUG_TRACE_STATE
- Log("Threads_Sleep: %p (%i %s) sleeping", cur, cur->TID, cur->ThreadName);
- #endif
-
- // Release Spinlock
- SHORTREL( &glThreadListLock );
-
- while(cur->Status != THREAD_STAT_ACTIVE) HALT();
-}
-
-
-/**
- * \fn int Threads_Wake( tThread *Thread )
- * \brief Wakes a sleeping/waiting thread up
- * \param Thread Thread to wake
- * \return Boolean Failure (Returns ERRNO)
- * \warning This should ONLY be called with task switches disabled
- */
-int Threads_Wake(tThread *Thread)
-{
- tThread *prev;
-
- if(!Thread)
- return -EINVAL;
-
- switch(Thread->Status)
- {
- case THREAD_STAT_ACTIVE:
- Log("Thread_Wake: Waking awake thread (%i)", Thread->TID);
- return -EALREADY;
-
- case THREAD_STAT_SLEEPING:
- SHORTLOCK( &glThreadListLock );
- // Remove from sleeping queue
- prev = Threads_int_GetPrev(&gSleepingThreads, Thread);
- prev->Next = Thread->Next;
-
- Threads_AddActive( Thread );
-
- #if DEBUG_TRACE_STATE
- Log("Threads_Sleep: %p (%i %s) woken", Thread, Thread->TID, Thread->ThreadName);
- #endif
- SHORTREL( &glThreadListLock );
- return -EOK;
-
- case THREAD_STAT_WAITING:
- Warning("Thread_Wake - Waiting threads are not currently supported");
- return -ENOTIMPL;
-
- case THREAD_STAT_DEAD:
- Warning("Thread_Wake - Attempt to wake dead thread (%i)", Thread->TID);
- return -ENOTIMPL;
-
- default:
- Warning("Thread_Wake - Unknown process status (%i)\n", Thread->Status);
- return -EINTERNAL;
- }
-}
-
-/**
- * \brief Wake a thread given the TID
- * \param TID Thread ID to wake
- * \return Boolean Faulure (errno)
- */
-int Threads_WakeTID(tTID TID)
-{
- tThread *thread = Threads_GetThread(TID);
- int ret;
- if(!thread)
- return -ENOENT;
- ret = Threads_Wake( thread );
- //Log_Debug("Threads", "TID %i woke %i (%p)", Threads_GetTID(), TID, thread);
- return ret;
-}
-
-/**
- * \brief Adds a thread to the active queue
- */
-void Threads_AddActive(tThread *Thread)
-{
- SHORTLOCK( &glThreadListLock );
-
- #if 1
- {
- tThread *t;
- for( t = gActiveThreads; t; t = t->Next )
- {
- if( t == Thread ) {
- Panic("Threads_AddActive: Attempting a double add of TID %i (0x%x)",
- Thread->TID, __builtin_return_address(0));
- }
-
- if(t->Status != THREAD_STAT_ACTIVE) {
- Panic("Threads_AddActive: TID %i status != THREAD_STAT_ACTIVE",
- Thread->TID);
- }
- }
- }
- #endif
-
- // Add to active list
- Thread->Next = gActiveThreads;
- gActiveThreads = Thread;
- // Set state
- Thread->Status = THREAD_STAT_ACTIVE;
- Thread->CurCPU = -1;
-
- // Update bookkeeping
- giNumActiveThreads ++;
- giFreeTickets += Thread->NumTickets;
-
- #if DEBUG_TRACE_TICKETS
- Log("Threads_AddActive: %p %i (%s) added, new giFreeTickets = %i",
- Thread, Thread->TID, Thread->ThreadName, giFreeTickets);
- #endif
- SHORTREL( &glThreadListLock );
-}
-
-/**
- * \brief Removes the current thread from the active queue
- * \warning This should ONLY be called with task switches disabled
- * \return Current thread pointer
- */
-tThread *Threads_RemActive(void)
-{
- tThread *ret = Proc_GetCurThread();
- tThread *prev;
-
- SHORTLOCK( &glThreadListLock );
-
- prev = Threads_int_GetPrev(&gActiveThreads, ret);
- if(!prev) {
- SHORTREL( &glThreadListLock );
- return NULL;
- }
-
- ret->Remaining = 0;
- ret->CurCPU = -1;
-
- prev->Next = ret->Next;
- giNumActiveThreads --;
- // no need to decrement tickets, scheduler did it for us
-
- #if DEBUG_TRACE_TICKETS
- Log("Threads_RemActive: %p %i (%s) removed, giFreeTickets = %i",
- ret, ret->TID, ret->ThreadName, giFreeTickets);
- #endif
-
- SHORTREL( &glThreadListLock );
-
- return ret;
-}
-
-/**
- * \fn void Threads_SetFaultHandler(Uint Handler)
- * \brief Sets the signal handler for a signal
- */
-void Threads_SetFaultHandler(Uint Handler)
-{
- //Log_Debug("Threads", "Threads_SetFaultHandler: Handler = %p", Handler);
- Proc_GetCurThread()->FaultHandler = Handler;
-}
-
-/**
- * \fn void Threads_Fault(int Num)
- * \brief Calls a fault handler
- */
-void Threads_Fault(int Num)
-{
- tThread *thread = Proc_GetCurThread();
-
- Log_Log("Threads", "Threads_Fault: thread = %p", thread);
-
- if(!thread) return ;
-
- Log_Log("Threads", "Threads_Fault: thread->FaultHandler = %p", thread->FaultHandler);
-
- switch(thread->FaultHandler)
- {
- case 0: // Panic?
- Threads_Kill(thread, -1);
- HALT();
- return ;
- case 1: // Dump Core?
- Threads_Kill(thread, -1);
- HALT();
- return ;
- }
-
- // Double Fault? Oh, F**k
- if(thread->CurFaultNum != 0) {
- Threads_Kill(thread, -1); // For now, just kill
- HALT();
- }
-
- thread->CurFaultNum = Num;
-
- Proc_CallFaultHandler(thread);
-}
-
-// --- Process Structure Access Functions ---
-tPID Threads_GetPID(void)
-{
- return Proc_GetCurThread()->TGID;
-}
-tTID Threads_GetTID(void)
-{
- return Proc_GetCurThread()->TID;
-}
-tUID Threads_GetUID(void)
-{
- return Proc_GetCurThread()->UID;
-}
-tGID Threads_GetGID(void)
-{
- return Proc_GetCurThread()->GID;
-}
-
-int Threads_SetUID(Uint *Errno, tUID ID)
-{
- tThread *t = Proc_GetCurThread();
- if( t->UID != 0 ) {
- *Errno = -EACCES;
- return -1;
- }
- Log_Debug("Threads", "TID %i's UID set to %i", t->TID, ID);
- t->UID = ID;
- return 0;
-}
-
-int Threads_SetGID(Uint *Errno, tGID ID)
-{
- tThread *t = Proc_GetCurThread();
- if( t->UID != 0 ) {
- *Errno = -EACCES;
- return -1;
- }
- Log_Debug("Threads", "TID %i's GID set to %i", t->TID, ID);
- t->GID = ID;
- return 0;
-}
-
-/**
- * \fn void Threads_Dump(void)
- * \brief Dumps a list of currently running threads
- */
-void Threads_Dump(void)
-{
- tThread *thread;
-
- Log("--- Thread Dump ---");
- Log("Active Threads: (%i reported)", giNumActiveThreads);
- for(thread=gActiveThreads;thread;thread=thread->Next)
- {
- Log(" %i (%i) - %s (CPU %i)",
- thread->TID, thread->TGID, thread->ThreadName, thread->CurCPU);
- if(thread->Status != THREAD_STAT_ACTIVE)
- Log(" ERROR State (%i) != THREAD_STAT_ACTIVE (%i)", thread->Status, THREAD_STAT_ACTIVE);
- Log(" %i Tickets, Quantum %i", thread->NumTickets, thread->Quantum);
- Log(" KStack 0x%x", thread->KernelStack);
- }
-
- Log("All Threads:");
- for(thread=gAllThreads;thread;thread=thread->GlobalNext)
- {
- Log(" %i (%i) - %s (CPU %i)",
- thread->TID, thread->TGID, thread->ThreadName, thread->CurCPU);
- Log(" State %i", thread->Status);
- Log(" %i Tickets, Quantum %i", thread->NumTickets, thread->Quantum);
- Log(" KStack 0x%x", thread->KernelStack);
- }
-}
-/**
- * \fn void Threads_Dump(void)
- */
-void Threads_DumpActive(void)
-{
- tThread *thread;
-
- Log("Active Threads:");
- for(thread=gActiveThreads;thread;thread=thread->Next)
- {
- Log(" %i (%i) - %s (CPU %i)",
- thread->TID, thread->TGID, thread->ThreadName, thread->CurCPU);
- if(thread->Status != THREAD_STAT_ACTIVE)
- Log(" ERROR State (%i) != THREAD_STAT_ACTIVE (%i)", thread->Status, THREAD_STAT_ACTIVE);
- Log(" %i Tickets, Quantum %i", thread->NumTickets, thread->Quantum);
- Log(" KStack 0x%x", thread->KernelStack);
- }
-}
-
-/**
- * \brief Gets the next thread to run
- * \param CPU Current CPU
- * \param Last The thread the CPU was running
- */
-tThread *Threads_GetNextToRun(int CPU, tThread *Last)
-{
- tThread *thread;
- int ticket;
- int number;
-
- // If this CPU has the lock, we must let it complete
- if( CPU_HAS_LOCK( &glThreadListLock ) )
- return Last;
-
- // Same if the current CPU has any lock
- if( gaThreads_NoTaskSwitch[CPU] )
- return Last;
-
-
- // Lock thread list
- SHORTLOCK( &glThreadListLock );
-
- // Clear Delete Queue
- // - I should probably put this in a worker thread to avoid calling free() in the scheduler
- while(gDeleteThreads)
- {
- thread = gDeleteThreads->Next;
- if( IS_LOCKED(&gDeleteThreads->IsLocked) ) { // Only free if structure is unused
- // Set to dead
- gDeleteThreads->Status = THREAD_STAT_DEAD;
- // Free name
- if( IsHeap(gDeleteThreads->ThreadName) )
- free(gDeleteThreads->ThreadName);
- // Remove from global list
- if( gDeleteThreads == gAllThreads )
- gAllThreads = gDeleteThreads->GlobalNext;
- else
- gDeleteThreads->GlobalPrev->GlobalNext = gDeleteThreads->GlobalNext;
- free( gDeleteThreads );
- }
- gDeleteThreads = thread;
- }
-
- // No active threads, just take a nap
- if(giNumActiveThreads == 0) {
- SHORTREL( &glThreadListLock );
- #if DEBUG_TRACE_TICKETS
- Log("No active threads");
- #endif
- return NULL;
- }
-
- // Special case: 1 thread
- if(giNumActiveThreads == 1) {
- if( gActiveThreads->CurCPU == -1 )
- gActiveThreads->CurCPU = CPU;
-
- SHORTREL( &glThreadListLock );
-
- if( gActiveThreads->CurCPU == CPU )
- return gActiveThreads;
-
- return NULL; // CPU has nothing to do
- }
-
- // Allow the old thread to be scheduled again
- if( Last ) {
- if( Last->Status == THREAD_STAT_ACTIVE ) {
- giFreeTickets += Last->NumTickets;
- #if DEBUG_TRACE_TICKETS
- LogF(" CPU %i released %p (%i %s) into the pool (%i tickets in pool)\n",
- CPU, Last, Last->TID, Last->ThreadName, giFreeTickets);
- #endif
- }
- #if DEBUG_TRACE_TICKETS
- else
- LogF(" CPU %i released %p (%s)->Status = %i (Released)\n",
- CPU, Last, Last->ThreadName, Last->Status);
- #endif
- Last->CurCPU = -1;
- }
-
- #if DEBUG_TRACE_TICKETS
- //Threads_DumpActive();
- #endif
-
- #if 1
- number = 0;
- for(thread = gActiveThreads; thread; thread = thread->Next) {
- if(thread->CurCPU >= 0) continue;
- if(thread->Status != THREAD_STAT_ACTIVE)
- Panic("Bookkeeping fail - %p %i(%s) is on the active queue with a status of %i",
- thread, thread->TID, thread->ThreadName, thread->Status);
- if(thread->Next == thread) {
- Panic("Bookkeeping fail - %p %i(%s) loops back on itself",
- thread, thread->TID, thread->ThreadName, thread->Status);
- }
- number += thread->NumTickets;
- }
- if(number != giFreeTickets) {
- Panic("Bookkeeping fail (giFreeTickets(%i) != number(%i)) - CPU%i",
- giFreeTickets, number, CPU);
- }
- #endif
-
- // No free tickets (all tasks delegated to cores)
- if( giFreeTickets == 0 ) {
- SHORTREL(&glThreadListLock);
- return NULL;
- }
-
- // Get the ticket number
- ticket = number = rand() % giFreeTickets;
-
- // Find the next thread
- for(thread=gActiveThreads;thread;thread=thread->Next)
- {
- if(thread->CurCPU >= 0) continue;
- if(thread->NumTickets > number) break;
- number -= thread->NumTickets;
- }
- // Error Check
- if(thread == NULL)
- {
- number = 0;
- for(thread=gActiveThreads;thread;thread=thread->Next) {
- if(thread->CurCPU >= 0) continue;
- number += thread->NumTickets;
- }
- Panic("Bookeeping Failed - giFreeTickets(%i) > true count (%i)",
- giFreeTickets, number);
- }
- #if DEBUG_TRACE_TICKETS
- LogF(" CPU%i giFreeTickets = %i\n", CPU, giFreeTickets);
- #endif
-
- // Make the new thread non-schedulable
- giFreeTickets -= thread->NumTickets;
- thread->CurCPU = CPU;
-
- //Threads_Dump();
- #if DEBUG_TRACE_TICKETS
- LogF(" CPU%i giFreeTickets = %i, giving %p (%i %s CPU=%i)\n",
- CPU, giFreeTickets, thread, thread->TID, thread->ThreadName, thread->CurCPU);
- #endif
-
- SHORTREL( &glThreadListLock );
-
- return thread;
-}
-
-/**
- * \fn void Threads_SegFault(tVAddr Addr)
- * \brief Called when a Segment Fault occurs
- */
-void Threads_SegFault(tVAddr Addr)
-{
- Warning("Thread #%i committed a segfault at address %p", Proc_GetCurThread()->TID, Addr);
- Threads_Fault( 1 );
- //Threads_Exit( 0, -1 );
-}
-
-/**
- * \brief Acquire a heavy mutex
- * \param Mutex Mutex to acquire
- *
- * This type of mutex checks if the mutex is avaliable, and acquires it
- * if it is. Otherwise, the current thread is added to the mutex's wait
- * queue and the thread suspends. When the holder of the mutex completes,
- * the oldest thread (top thread) on the queue is given the lock and
- * restarted.
- */
-void Mutex_Acquire(tMutex *Mutex)
-{
- tThread *us = Proc_GetCurThread();
-
- // Get protector
- SHORTLOCK( &Mutex->Protector );
-
- //Log("Mutex_Acquire: (%p)", Mutex);
-
- // Check if the lock is already held
- if( Mutex->Owner ) {
- SHORTLOCK( &glThreadListLock );
- // - Remove from active list
- Threads_RemActive();
- // - Mark as sleeping
- us->Status = THREAD_STAT_OFFSLEEP;
-
- // - Add to waiting
- if(Mutex->LastWaiting) {
- Mutex->LastWaiting->Next = us;
- Mutex->LastWaiting = us;
- }
- else {
- Mutex->Waiting = us;
- Mutex->LastWaiting = us;
- }
- SHORTREL( &glThreadListLock );
- SHORTREL( &Mutex->Protector );
- while(us->Status == THREAD_STAT_OFFSLEEP) Threads_Yield();
- // We're only woken when we get the lock
- }
- // Ooh, let's take it!
- else {
- Mutex->Owner = us;
- SHORTREL( &Mutex->Protector );
- }
-}
-
-/**
- * \brief Release a held mutex
- * \param Mutex Mutex to release
- */
-void Mutex_Release(tMutex *Mutex)
-{
- SHORTLOCK( &Mutex->Protector );
- //Log("Mutex_Release: (%p)", Mutex);
- if( Mutex->Waiting ) {
- Mutex->Owner = Mutex->Waiting; // Set owner
- Mutex->Waiting = Mutex->Waiting->Next; // Next!
-
- // Wake new owner
- SHORTLOCK( &glThreadListLock );
- if( Mutex->Owner->Status != THREAD_STAT_ACTIVE )
- Threads_AddActive(Mutex->Owner);
- SHORTREL( &glThreadListLock );
- }
- else {
- Mutex->Owner = NULL;
- }
- SHORTREL( &Mutex->Protector );
-}
-
-/**
- * \brief Is this mutex locked?
- * \param Mutex Mutex pointer
- */
-int Mutex_IsLocked(tMutex *Mutex)
-{
- return Mutex->Owner != NULL;
-}
-
-// === EXPORTS ===
-EXPORT(Threads_GetUID);
-EXPORT(Mutex_Acquire);
-EXPORT(Mutex_Release);
-EXPORT(Mutex_IsLocked);
git.ucc.asn.au / tpg / acess2.git / blobdiff
UCC git Repository :: git.ucc.asn.au