#include <threads.h>
#include <errno.h>
+#define DEBUG_TRACE_TICKETS 0
+
// === CONSTANTS ===
#define DEFAULT_QUANTUM 10
#define DEFAULT_TICKETS 5
void Threads_Init(void);
int Threads_SetName(char *NewName);
char *Threads_GetName(int ID);
-void Threads_SetTickets(int Num);
+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_Kill(tThread *Thread, int Status);
void Threads_Yield(void);
void Threads_Sleep(void);
-void Threads_Wake(tThread *Thread);
+ int Threads_Wake(tThread *Thread);
void Threads_AddActive(tThread *Thread);
int Threads_GetPID(void);
int Threads_GetTID(void);
// -- Core Thread --
// Only used for the core kernel
tThread gThreadZero = {
- NULL, 0, // Next, Lock
- THREAD_STAT_ACTIVE, // Status
- 0, // Exit Status
- 0, 0, // TID, TGID
- 0, 0, // UID, GID
- 0, // Parent Thread ID
- "ThreadZero", // Name
-
- 0, // Kernel Stack
- {0}, // Saved State
- {0}, // VM State
-
- 0, 0, // Current Fault, Fault Handler
-
- NULL, NULL, // Messages, Last Message
- DEFAULT_QUANTUM, DEFAULT_QUANTUM, // Quantum, Remaining
- DEFAULT_TICKETS,
- {0} // Default config to zero
+ 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 ---
gActiveThreads = &gThreadZero;
//giFreeTickets = gThreadZero.NumTickets;
giNumActiveThreads = 1;
-
- #if 1
- // Create Idle Task
- if(Proc_Clone(0, 0) == 0)
- {
- tThread *cur = Proc_GetCurThread();
- cur->ThreadName = "Idle Thread";
- Threads_SetTickets(0); // Never called randomly
- cur->Quantum = 1; // 1 slice quantum
- for(;;) HALT(); // Just yeilds
- }
- #endif
-
+
Proc_Start();
}
int Threads_SetName(char *NewName)
{
tThread *cur = Proc_GetCurThread();
- if( IsHeap(cur->ThreadName) )
- free( cur->ThreadName );
- // TODO: Possible concurrency issue
- cur->ThreadName = malloc(strlen(NewName)+1);
- strcpy(cur->ThreadName, NewName);
+ char *oldname = cur->ThreadName;
+
+ cur->ThreadName = NULL;
+
+ if( IsHeap(oldname) ) free( oldname );
+
+ cur->ThreadName = strdup(NewName);
return 0;
}
}
/**
- * \fn void Threads_SetTickets(int Num)
+ * \fn void Threads_SetTickets(tThread *Thread, int Num)
* \brief Sets the 'priority' of a task
*/
-void Threads_SetTickets(int Num)
+void Threads_SetTickets(tThread *Thread, int Num)
{
- tThread *cur = Proc_GetCurThread();
+ if(Thread == NULL)
+ Thread = Proc_GetCurThread();
if(Num < 0) return;
if(Num > MAX_TICKETS) Num = MAX_TICKETS;
- cur->NumTickets = Num;
+ if( Thread != Proc_GetCurThread() ) {
+ LOCK( &glThreadListLock );
+ giFreeTickets -= Thread->NumTickets - Num;
+ Thread->NumTickets = Num;
+ RELEASE( &glThreadListLock );
+ }
+ else
+ Thread->NumTickets = Num;
}
/**
giNumActiveThreads --;
if( Thread != Proc_GetCurThread() )
giFreeTickets -= Thread->NumTickets;
+ //Log("Threads_Kill: giFreeTickets = %i", giFreeTickets);
// Mark thread as a zombie
Thread->RetStatus = Status;
// Reduce the active count & ticket count
giNumActiveThreads --;
- //giTotalTickets -= cur->NumTickets;
// Mark thread as sleeping
cur->Status = THREAD_STAT_SLEEPING;
/**
- * \fn void Threads_Wake( tThread *Thread )
+ * \fn int Threads_Wake( tThread *Thread )
* \brief Wakes a sleeping/waiting thread up
+ * \param Thread Thread to wake
+ * \return Boolean Failure (Returns ERRNO)
*/
-void Threads_Wake(tThread *Thread)
+int Threads_Wake(tThread *Thread)
{
tThread *prev;
+
+ if(!Thread)
+ return -EINVAL;
+
switch(Thread->Status)
{
- case THREAD_STAT_ACTIVE: break;
+ case THREAD_STAT_ACTIVE:
+ Log("Thread_Wake: Waking awake thread (%i)", Thread->TID);
+ return -EALREADY;
case THREAD_STAT_SLEEPING:
- //Log_Log("Threads", "Waking %i (%p) from sleeping", Thread->TID, Thread);
+ //Log_Log("Threads", "Waking %i (%p) from sleeping (CPU=%i)",
+ // Thread->TID, Thread, Thread->CurCPU);
LOCK( &glThreadListLock );
prev = Threads_int_GetPrev(&gSleepingThreads, Thread);
prev->Next = Thread->Next; // Remove from sleeping queue
gActiveThreads = Thread;
giNumActiveThreads ++;
// Thread can't be the current, so no need to check
+ Thread->CurCPU = -1;
giFreeTickets += Thread->NumTickets;
+ #if DEBUG_TRACE_TICKETS
+ Log("Threads_Wake: giFreeTickets = %i", giFreeTickets);
+ #endif
Thread->Status = THREAD_STAT_ACTIVE;
RELEASE( &glThreadListLock );
- break;
+ return -EOK;
case THREAD_STAT_WAITING:
Warning("Thread_Wake - Waiting threads are not currently supported");
- break;
+ return -ENOTIMPL;
case THREAD_STAT_DEAD:
Warning("Thread_Wake - Attempt to wake dead thread (%i)", Thread->TID);
- break;
+ return -ENOTIMPL;
default:
Warning("Thread_Wake - Unknown process status (%i)\n", Thread->Status);
- break;
+ return -EINTERNAL;
}
}
/**
* \brief Wake a thread given the TID
+ * \param TID Thread ID to wake
+ * \return Boolean Faulure (errno)
*/
-void Threads_WakeTID(tTID Thread)
+int Threads_WakeTID(tTID TID)
{
- Threads_Wake( Threads_GetThread(Thread) );
+ tThread *thread = Threads_GetThread(TID);
+ if(!thread)
+ return -ENOENT;
+ return Threads_Wake( thread );
}
/**
giNumActiveThreads ++;
// Thread can't be the current, so no need to check
giFreeTickets += Thread->NumTickets;
- //Log("Threads_AddActive: giNumActiveThreads = %i, giTotalTickets = %i",
- // giNumActiveThreads, giTotalTickets);
+ #if DEBUG_TRACE_TICKETS
+ Log("Threads_AddActive: giFreeTickets = %i", giFreeTickets);
+ #endif
RELEASE( &glThreadListLock );
}
void Threads_Dump(void)
{
tThread *thread;
- tThread *cur = Proc_GetCurThread();
Log("Active Threads:");
for(thread=gActiveThreads;thread;thread=thread->Next)
Log("Sleeping Threads:");
for(thread=gSleepingThreads;thread;thread=thread->Next)
{
- Log("%c%i (%i) - %s",
- (thread==cur?'*':' '),
+ Log(" %i (%i) - %s",
thread->TID, thread->TGID, thread->ThreadName);
Log(" %i Tickets, Quantum %i", thread->NumTickets, thread->Quantum);
Log(" KStack 0x%x", thread->KernelStack);
int ticket;
int number;
- // Note: Enable the code to tell if the switch code has the lock, or
- // if it's the other code.
+ // TODO: Enable the code to tell if the current CPU has the lock or
+ // another does.
// Check if the thread list is locked by other code
// - If so, don't switch (give it a chance to complete)
if( IS_LOCKED(&glThreadListLock) )
return Last;
+ // Clear Delete Queue
+ while(gDeleteThreads)
+ {
+ thread = gDeleteThreads->Next;
+ if(gDeleteThreads->IsLocked) { // Only free if structure is unused
+ gDeleteThreads->Status = THREAD_STAT_NULL;
+ free( gDeleteThreads );
+ }
+ gDeleteThreads = thread;
+ }
+
// No active threads, just take a nap
if(giNumActiveThreads == 0) {
+ #if DEBUG_TRACE_TICKETS
+ Log("No active threads");
+ #endif
return NULL;
}
// Allow the old thread to be scheduled again
if( Last ) {
- if( Last->Status == THREAD_STAT_ACTIVE )
+ if( Last->Status == THREAD_STAT_ACTIVE ) {
giFreeTickets += Last->NumTickets;
+ #if DEBUG_TRACE_TICKETS
+ LogF(" CPU %i released %p (%s) into the pool (%i tickets in pool)\n",
+ CPU, Last, Last->ThreadName, Last->NumTickets);
+ #endif
+ }
+ #if DEBUG_TRACE_TICKETS
+ else
+ LogF(" %p (%s)->Status = %i\n", Last, Last->ThreadName, Last->Status);
+ #endif
Last->CurCPU = -1;
}
-
+
+ #if 1
+ number = 0;
+ for(thread=gActiveThreads;thread;thread=thread->Next) {
+ if(thread->CurCPU >= 0) continue;
+ 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 ) {
+ RELEASE(&glThreadListLock);
+ return NULL;
+ }
+
// Get the ticket number
ticket = number = rand() % giFreeTickets;
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 (%s CPU=%i)\n",
+ CPU, giFreeTickets, thread, thread->ThreadName, thread->CurCPU);
+ #endif
+
RELEASE( &glThreadListLock );
return thread;