3 * - By John Hodge (thePowersGang)
5 * - Common Thread Control
9 #include <threads_int.h>
12 #include <semaphore.h>
13 #include <vfs_threads.h> // VFS Handle maintainence
17 #define DEBUG_TRACE_TICKETS 0 // Trace ticket counts
18 #define DEBUG_TRACE_STATE 0 // Trace state changes (sleep/wake)
22 #define SCHED_LOTTERY 1 // Lottery scheduler
23 #define SCHED_RR_SIM 2 // Single Queue Round Robin
24 #define SCHED_RR_PRI 3 // Multi Queue Round Robin
26 #define SCHEDULER_TYPE SCHED_RR_PRI
29 #define DEFAULT_QUANTUM 5
30 #define DEFAULT_PRIORITY 5
31 #define MIN_PRIORITY 10
43 void Threads_Init(void);
45 void Threads_Delete(tThread *Thread);
46 int Threads_SetName(const char *NewName);
48 char *Threads_GetName(tTID ID);
50 void Threads_SetPriority(tThread *Thread, int Pri);
51 tThread *Threads_CloneTCB(Uint *Err, Uint Flags);
52 int Threads_WaitTID(int TID, int *status);
53 tThread *Threads_GetThread(Uint TID);
55 tThread *Threads_int_DelFromQueue(tThreadList *List, tThread *Thread);
56 void Threads_int_AddToList(tThreadList *List, tThread *Thread);
58 void Threads_Exit(int TID, int Status);
59 void Threads_Kill(tThread *Thread, int Status);
60 void Threads_Yield(void);
61 void Threads_Sleep(void);
62 int Threads_Wake(tThread *Thread);
63 void Threads_AddActive(tThread *Thread);
64 tThread *Threads_RemActive(void);
66 void Threads_ToggleTrace(int TID);
67 void Threads_Fault(int Num);
68 void Threads_SegFault(tVAddr Addr);
70 int Threads_GetPID(void);
71 int Threads_GetTID(void);
72 tUID Threads_GetUID(void);
73 tGID Threads_GetGID(void);
74 int Threads_SetUID(Uint *Errno, tUID ID);
75 int Threads_SetGID(Uint *Errno, tUID ID);
77 void Threads_int_DumpThread(tThread *thread);
78 void Threads_Dump(void);
79 void Threads_DumpActive(void);
83 struct sProcess gProcessZero = {
85 // Only used for the core kernel
86 tThread gThreadZero = {
87 .Status = THREAD_STAT_ACTIVE, // Status
88 .ThreadName = (char*)"ThreadZero", // Name
89 .Quantum = DEFAULT_QUANTUM, // Default Quantum
90 .Remaining = DEFAULT_QUANTUM, // Current Quantum
91 .Priority = DEFAULT_PRIORITY // Number of tickets
95 tShortSpinlock glThreadListLock; ///\note NEVER use a heap function while locked
96 // --- Current State ---
97 volatile int giNumActiveThreads = 0; // Number of threads on the active queue
98 volatile Uint giNextTID = 1; // Next TID to allocate
99 // --- Thread Lists ---
100 tThread *gAllThreads = NULL; // All allocated threads
101 tThreadList gSleepingThreads; // Sleeping Threads
102 int giNumCPUs = 1; // Number of CPUs
103 BOOL gaThreads_NoTaskSwitch[MAX_CPUS]; // Disables task switches for each core (Pseudo-IF)
104 // --- Scheduler Types ---
105 #if SCHEDULER_TYPE == SCHED_LOTTERY
106 const int caiTICKET_COUNTS[MIN_PRIORITY+1] = {100,81,64,49,36,25,16,9,4,1,0};
107 volatile int giFreeTickets = 0; // Number of tickets held by non-scheduled threads
108 tThreadList gActiveThreads; // Currently Running Threads
109 #elif SCHEDULER_TYPE == SCHED_RR_SIM
110 tThreadList gActiveThreads; // Currently Running Threads
111 #elif SCHEDULER_TYPE == SCHED_RR_PRI
112 tThreadList gaActiveThreads[MIN_PRIORITY+1]; // Active threads for each priority level
114 # error "Unkown scheduler type"
119 * \fn void Threads_Init(void)
120 * \brief Initialse the thread list
122 void Threads_Init(void)
126 Log_Debug("Threads", "Offsets of tThread");
127 Log_Debug("Threads", ".Priority = %i", offsetof(tThread, Priority));
128 Log_Debug("Threads", ".KernelStack = %i", offsetof(tThread, KernelStack));
130 // Create Initial Task
131 gAllThreads = &gThreadZero;
132 giNumActiveThreads = 1;
133 gThreadZero.Process = &gProcessZero;
138 void Threads_Delete(tThread *Thread)
141 Thread->Status = THREAD_STAT_BURIED;
143 // Clear out process state
144 Proc_ClearThread(Thread);
146 Thread->Process->nThreads --;
148 if( Thread->Process->FirstThread == Thread )
150 Thread->Process->FirstThread = Thread->ProcessNext;
154 tThread *prev = Thread->Process->FirstThread;
155 while(prev && prev->ProcessNext != Thread)
156 prev = prev->ProcessNext;
158 Log_Error("Threads", "Thread %p(%i %s) is not on the process's list",
159 Thread, Thread->TID, Thread->ThreadName
162 prev->ProcessNext = Thread->ProcessNext;
165 // If the final thread is being terminated, clean up the process
166 if( Thread->Process->nThreads == 0 )
168 tProcess *proc = Thread->Process;
170 VFS_CloseAllUserHandles();
171 // Architecture cleanup
172 Proc_ClearProcess( proc );
173 // VFS Configuration strings
174 if( proc->CurrentWorkingDir)
175 free( proc->CurrentWorkingDir );
177 free( proc->RootDir );
178 // Process descriptor
183 if( IsHeap(Thread->ThreadName) )
184 free(Thread->ThreadName);
186 // Remove from global list
187 // TODO: Lock this too
188 if( Thread == gAllThreads )
189 gAllThreads = Thread->GlobalNext;
191 Thread->GlobalPrev->GlobalNext = Thread->GlobalNext;
197 * \fn void Threads_SetName(const char *NewName)
198 * \brief Sets the current thread's name
199 * \param NewName New name for the thread
200 * \return Boolean Failure
202 int Threads_SetName(const char *NewName)
204 tThread *cur = Proc_GetCurThread();
205 char *oldname = cur->ThreadName;
207 // NOTE: There is a possibility of non-thread safety here
208 // A thread could read the current name pointer before it is zeroed
210 cur->ThreadName = NULL;
212 if( IsHeap(oldname) ) free( oldname );
213 cur->ThreadName = strdup(NewName);
215 // Log_Debug("Threads", "Thread renamed to '%s'", NewName);
221 * \fn char *Threads_GetName(int ID)
222 * \brief Gets a thread's name
223 * \param ID Thread ID (-1 indicates current thread)
224 * \return Pointer to name
225 * \retval NULL Failure
227 char *Threads_GetName(tTID ID)
230 return Proc_GetCurThread()->ThreadName;
232 return Threads_GetThread(ID)->ThreadName;
236 * \fn void Threads_SetPriority(tThread *Thread, int Pri)
237 * \brief Sets the priority of a task
238 * \param Thread Thread to update ticket count (NULL means current thread)
239 * \param Pri New priority
241 void Threads_SetPriority(tThread *Thread, int Pri)
243 // Get current thread
244 if(Thread == NULL) Thread = Proc_GetCurThread();
246 // - If < 0, set to lowest priority
247 // - Minumum priority is actualy a high number, 0 is highest
248 if(Pri < 0) Pri = MIN_PRIORITY;
249 if(Pri > MIN_PRIORITY) Pri = MIN_PRIORITY;
251 // Do we actually have to do anything?
252 if( Pri == Thread->Priority ) return;
254 #if SCHEDULER_TYPE == SCHED_RR_PRI
255 if( Thread != Proc_GetCurThread() )
257 SHORTLOCK( &glThreadListLock );
258 // Remove from old priority
259 Threads_int_DelFromQueue( &gaActiveThreads[Thread->Priority], Thread );
261 Threads_int_AddToList( &gaActiveThreads[Pri], Thread );
262 Thread->Priority = Pri;
263 SHORTREL( &glThreadListLock );
266 Thread->Priority = Pri;
268 // If this isn't the current thread, we need to lock
269 if( Thread != Proc_GetCurThread() )
271 SHORTLOCK( &glThreadListLock );
273 #if SCHEDULER_TYPE == SCHED_LOTTERY
274 giFreeTickets -= caiTICKET_COUNTS[Thread->Priority] - caiTICKET_COUNTS[Pri];
275 # if DEBUG_TRACE_TICKETS
276 Log("Threads_SetTickets: new giFreeTickets = %i [-%i+%i]",
278 caiTICKET_COUNTS[Thread->Priority], caiTICKET_COUNTS[Pri]);
281 Thread->Priority = Pri;
282 SHORTREL( &glThreadListLock );
285 Thread->Priority = Pri;
288 #if DEBUG_TRACE_STATE
289 Log("Threads_SetPriority: %p(%i %s) pri set %i",
290 Thread, Thread->TID, Thread->ThreadName,
296 * \brief Clone the TCB of the current thread
297 * \param Flags Flags for something... (What is this for?)
299 tThread *Threads_CloneTCB(Uint Flags)
302 cur = Proc_GetCurThread();
304 // Allocate and duplicate
305 new = malloc(sizeof(tThread));
306 if(new == NULL) { errno = -ENOMEM; return NULL; }
307 memcpy(new, cur, sizeof(tThread));
311 memset( &new->IsLocked, 0, sizeof(new->IsLocked));
312 new->Status = THREAD_STAT_PREINIT;
316 new->TID = giNextTID++;
318 new->bInstrTrace = 0;
321 new->ThreadName = strdup(cur->ThreadName);
323 // Set Thread Group ID (PID)
324 if(Flags & CLONE_VM) {
325 tProcess *newproc, *oldproc;
326 oldproc = cur->Process;
327 new->Process = malloc( sizeof(struct sProcess) );
328 newproc = new->Process;
329 newproc->PID = new->TID;
330 if( Flags & CLONE_PGID )
331 newproc->PGID = oldproc->PGID;
333 newproc->PGID = newproc->PID;
334 newproc->UID = oldproc->UID;
335 newproc->GID = oldproc->GID;
336 newproc->MaxFD = oldproc->MaxFD;
337 if( oldproc->CurrentWorkingDir )
338 newproc->CurrentWorkingDir = strdup( oldproc->CurrentWorkingDir );
340 newproc->CurrentWorkingDir = NULL;
341 if( oldproc->RootDir )
342 newproc->RootDir = strdup( oldproc->RootDir );
344 newproc->RootDir = NULL;
345 newproc->nThreads = 1;
346 // Reference all handles in the VFS
347 VFS_ReferenceUserHandles();
349 newproc->FirstThread = new;
350 new->ProcessNext = NULL;
353 new->Process->nThreads ++;
354 new->Process = cur->Process;
356 new->ProcessNext = new->Process->FirstThread;
357 new->Process->FirstThread = new;
360 // Messages are not inherited
361 new->Messages = NULL;
362 new->LastMessage = NULL;
365 new->Remaining = new->Quantum = cur->Quantum;
366 new->Priority = cur->Priority;
369 // Set Signal Handlers
370 new->CurFaultNum = 0;
371 new->FaultHandler = cur->FaultHandler;
373 // Maintain a global list of threads
374 SHORTLOCK( &glThreadListLock );
375 new->GlobalPrev = NULL; // Protect against bugs
376 new->GlobalNext = gAllThreads;
377 gAllThreads->GlobalPrev = new;
379 SHORTREL( &glThreadListLock );
385 * \brief Clone the TCB of the kernel thread
387 tThread *Threads_CloneThreadZero(void)
391 // Allocate and duplicate
392 new = malloc(sizeof(tThread));
396 memcpy(new, &gThreadZero, sizeof(tThread));
398 new->Process->nThreads ++;
402 memset( &new->IsLocked, 0, sizeof(new->IsLocked));
403 new->Status = THREAD_STAT_PREINIT;
407 new->TID = giNextTID++;
411 new->ThreadName = NULL;
413 // Messages are not inherited
414 new->Messages = NULL;
415 new->LastMessage = NULL;
418 new->Remaining = new->Quantum = DEFAULT_QUANTUM;
419 new->Priority = DEFAULT_PRIORITY;
420 new->bInstrTrace = 0;
422 // Set Signal Handlers
423 new->CurFaultNum = 0;
424 new->FaultHandler = 0;
426 // Maintain a global list of threads
427 SHORTLOCK( &glThreadListLock );
428 new->GlobalPrev = NULL; // Protect against bugs
429 new->GlobalNext = gAllThreads;
430 gAllThreads->GlobalPrev = new;
432 SHORTREL( &glThreadListLock );
438 * \brief Wait for a task to change state
439 * \param TID Thread ID to wait on (-1: Any child thread, 0: Any Child/Sibling, <-1: -PID)
440 * \param Status Thread return status
441 * \return TID of child that changed state
443 tTID Threads_WaitTID(int TID, int *Status)
448 Uint32 ev = Threads_WaitEvents(THREAD_EVENT_DEADCHILD);
450 if( ev & THREAD_EVENT_DEADCHILD )
452 // A child died, get the TID
453 tThread *us = Proc_GetCurThread();
454 ASSERT(us->LastDeadChild);
455 ret = us->LastDeadChild->TID;
456 // - Mark as dead (as opposed to undead)
457 ASSERT(us->LastDeadChild->Status == THREAD_STAT_ZOMBIE);
458 us->LastDeadChild->Status = THREAD_STAT_DEAD;
459 // - Set return status
461 *Status = us->LastDeadChild->RetStatus;
462 us->LastDeadChild = NULL;
463 Mutex_Release(&us->DeadChildLock);
467 Log_Error("Threads", "TODO: Threads_WaitTID(TID=-1) - Any Child");
472 // Any peer/child thread
474 Log_Error("Threads", "TODO: Threads_WaitTID(TID=0) - Any Child/Sibling");
480 Log_Error("Threads", "TODO: Threads_WaitTID(TID<0) - TGID");
488 // NOTE: Race condition - Other child dies, desired child dies, first death is 'lost'
489 while( (ret = Threads_WaitTID(-1, Status)) != TID )
501 * \brief Gets a thread given its TID
502 * \param TID Thread ID
503 * \return Thread pointer
505 tThread *Threads_GetThread(Uint TID)
509 // Search global list
510 for( thread = gAllThreads; thread; thread = thread->GlobalNext )
512 if(thread->TID == TID)
516 Log_Notice("Threads", "Unable to find TID %i on main list\n", TID);
522 * \brief Deletes an entry from a list
523 * \param List Pointer to the list head
524 * \param Thread Thread to find
527 tThread *Threads_int_DelFromQueue(tThreadList *List, tThread *Thread)
529 tThread *ret, *prev = NULL;
531 for(ret = List->Head;
532 ret && ret != Thread;
533 prev = ret, ret = ret->Next
536 // Is the thread on the list
538 //LogF("%p(%s) is not on list %p\n", Thread, Thread->ThreadName, List);
543 List->Head = Thread->Next;
544 //LogF("%p(%s) removed from head of %p\n", Thread, Thread->ThreadName, List);
547 prev->Next = Thread->Next;
548 //LogF("%p(%s) removed from %p (prev=%p)\n", Thread, Thread->ThreadName, List, prev);
550 if( Thread->Next == NULL )
556 void Threads_int_AddToList(tThreadList *List, tThread *Thread)
559 List->Tail->Next = Thread;
567 * \brief Exit the current process (or another?)
568 * \param TID Thread ID to kill
569 * \param Status Exit status
571 void Threads_Exit(int TID, int Status)
574 Threads_Kill( Proc_GetCurThread(), (Uint)Status & 0xFF );
576 Threads_Kill( Threads_GetThread(TID), (Uint)Status & 0xFF );
578 // Halt forever, just in case
583 * \fn void Threads_Kill(tThread *Thread, int Status)
584 * \brief Kill a thread
585 * \param Thread Thread to kill
586 * \param Status Status code to return to the parent
588 void Threads_Kill(tThread *Thread, int Status)
591 int isCurThread = Thread == Proc_GetCurThread();
593 // TODO: Disown all children?
597 // TODO: I should keep a .Children list
598 for(child = gAllThreads;
600 child = child->GlobalNext)
602 if(child->Parent == Thread)
603 child->Parent = &gThreadZero;
608 ///\note Double lock is needed due to overlap of lock areas
610 // Lock thread (stop us recieving messages)
611 SHORTLOCK( &Thread->IsLocked );
613 // Clear Message Queue
614 while( Thread->Messages )
616 msg = Thread->Messages->Next;
617 free( Thread->Messages );
618 Thread->Messages = msg;
622 SHORTLOCK( &glThreadListLock );
624 switch(Thread->Status)
626 case THREAD_STAT_PREINIT: // Only on main list
629 // Currently active thread
630 case THREAD_STAT_ACTIVE:
631 if( Thread != Proc_GetCurThread() )
633 #if SCHEDULER_TYPE == SCHED_RR_PRI
634 tThreadList *list = &gaActiveThreads[Thread->Priority];
636 tThreadList *list = &gActiveThreads;
638 if( Threads_int_DelFromQueue( list, Thread ) )
643 Log_Warning("Threads",
644 "Threads_Kill - Thread %p(%i,%s) marked as active, but not on list",
645 Thread, Thread->TID, Thread->ThreadName
648 #if SCHEDULER_TYPE == SCHED_LOTTERY
649 giFreeTickets -= caiTICKET_COUNTS[ Thread->Priority ];
652 // Ensure that we are not rescheduled
653 Thread->Remaining = 0; // Clear Remaining Quantum
654 Thread->Quantum = 0; // Clear Quantum to indicate dead thread
656 // Update bookkeeping
657 giNumActiveThreads --;
659 // Kill it while it sleeps!
660 case THREAD_STAT_SLEEPING:
661 if( !Threads_int_DelFromQueue( &gSleepingThreads, Thread ) )
663 Log_Warning("Threads",
664 "Threads_Kill - Thread %p(%i,%s) marked as sleeping, but not on list",
665 Thread, Thread->TID, Thread->ThreadName
670 // Brains!... You cannot kill something that is already dead
671 case THREAD_STAT_ZOMBIE:
672 Log_Warning("Threads", "Threads_Kill - Thread %p(%i,%s) is undead, you cannot kill it",
673 Thread, Thread->TID, Thread->ThreadName);
674 SHORTREL( &glThreadListLock );
675 SHORTREL( &Thread->IsLocked );
679 Log_Warning("Threads", "Threads_Kill - BUG Un-checked status (%i)",
685 Thread->RetStatus = Status;
687 SHORTREL( &Thread->IsLocked );
689 Thread->Status = THREAD_STAT_ZOMBIE;
690 SHORTREL( &glThreadListLock );
691 // TODO: It's possible that we could be timer-preempted here, should disable that... somehow
692 Mutex_Acquire( &Thread->Parent->DeadChildLock ); // released by parent
693 Thread->Parent->LastDeadChild = Thread;
694 Threads_PostEvent( Thread->Parent, THREAD_EVENT_DEADCHILD );
696 Log("Thread %i went *hurk* (%i)", Thread->TID, Status);
707 * \brief Yield remainder of the current thread's timeslice
709 void Threads_Yield(void)
711 // Log("Threads_Yield: by %p", __builtin_return_address(0));
716 * \fn void Threads_Sleep(void)
717 * \brief Take the current process off the run queue
719 void Threads_Sleep(void)
721 tThread *cur = Proc_GetCurThread();
724 SHORTLOCK( &glThreadListLock );
726 // Don't sleep if there is a message waiting
727 if( cur->Messages ) {
728 SHORTREL( &glThreadListLock );
732 // Remove us from running queue
734 // Mark thread as sleeping
735 cur->Status = THREAD_STAT_SLEEPING;
737 // Add to Sleeping List (at the top)
738 Threads_int_AddToList( &gSleepingThreads, cur );
740 #if DEBUG_TRACE_STATE
741 Log("Threads_Sleep: %p (%i %s) sleeping", cur, cur->TID, cur->ThreadName);
745 SHORTREL( &glThreadListLock );
747 while(cur->Status != THREAD_STAT_ACTIVE) {
749 if( cur->Status != THREAD_STAT_ACTIVE )
750 Log("%i - Huh? why am I up? zzzz...", cur->TID);
756 * \brief Wakes a sleeping/waiting thread up
757 * \param Thread Thread to wake
758 * \return Boolean Failure (Returns ERRNO)
759 * \warning This should ONLY be called with task switches disabled
761 int Threads_Wake(tThread *Thread)
766 switch(Thread->Status)
768 case THREAD_STAT_ACTIVE:
769 Log("Threads_Wake - Waking awake thread (%i)", Thread->TID);
772 case THREAD_STAT_SLEEPING:
773 // Remove from sleeping queue
774 SHORTLOCK( &glThreadListLock );
775 Threads_int_DelFromQueue(&gSleepingThreads, Thread);
776 SHORTREL( &glThreadListLock );
778 Threads_AddActive( Thread );
780 #if DEBUG_TRACE_STATE
781 Log("Threads_Sleep: %p (%i %s) woken", Thread, Thread->TID, Thread->ThreadName);
785 case THREAD_STAT_SEMAPHORESLEEP: {
787 tThread *th, *prev=NULL;
789 sem = Thread->WaitPointer;
791 SHORTLOCK( &sem->Protector );
793 // Remove from sleeping queue
794 for( th = sem->Waiting; th; prev = th, th = th->Next )
795 if( th == Thread ) break;
799 prev->Next = Thread->Next;
801 sem->Waiting = Thread->Next;
802 if(sem->LastWaiting == Thread)
803 sem->LastWaiting = prev;
808 for( th = sem->Signaling; th; prev = th, th = th->Next )
809 if( th == Thread ) break;
811 Log_Warning("Threads", "Thread %p(%i %s) is not on semaphore %p(%s:%s)",
812 Thread, Thread->TID, Thread->ThreadName,
813 sem, sem->ModName, sem->Name);
818 prev->Next = Thread->Next;
820 sem->Signaling = Thread->Next;
821 if(sem->LastSignaling == Thread)
822 sem->LastSignaling = prev;
825 Thread->RetStatus = 0; // It didn't get anything
826 Threads_AddActive( Thread );
828 #if DEBUG_TRACE_STATE
829 Log("Threads_Sleep: %p(%i %s) woken from semaphore", Thread, Thread->TID, Thread->ThreadName);
831 SHORTREL( &sem->Protector );
834 case THREAD_STAT_WAITING:
835 Warning("Threads_Wake - Waiting threads are not currently supported");
838 case THREAD_STAT_DEAD:
839 Warning("Threads_Wake - Attempt to wake dead thread (%i)", Thread->TID);
843 Log_Warning("Threads", "Threads_Wake - Unknown process status (%i)", Thread->Status);
849 * \brief Wake a thread given the TID
850 * \param TID Thread ID to wake
851 * \return Boolean Faulure (errno)
853 int Threads_WakeTID(tTID TID)
855 tThread *thread = Threads_GetThread(TID);
859 ret = Threads_Wake( thread );
860 //Log_Debug("Threads", "TID %i woke %i (%p)", Threads_GetTID(), TID, thread);
864 void Threads_ToggleTrace(int TID)
866 tThread *thread = Threads_GetThread(TID);
868 thread->bInstrTrace = !thread->bInstrTrace;
872 * \brief Adds a thread to the active queue
874 void Threads_AddActive(tThread *Thread)
876 SHORTLOCK( &glThreadListLock );
878 if( Thread->Status == THREAD_STAT_ACTIVE ) {
879 tThread *cur = Proc_GetCurThread();
880 Log_Warning("Threads", "WTF, %p CPU%i %p (%i %s) is adding %p (%i %s) when it is active",
881 __builtin_return_address(0),
882 GetCPUNum(), cur, cur->TID, cur->ThreadName, Thread, Thread->TID, Thread->ThreadName);
883 SHORTREL( &glThreadListLock );
888 Thread->Status = THREAD_STAT_ACTIVE;
889 // Thread->CurCPU = -1;
890 // Add to active list
892 #if SCHEDULER_TYPE == SCHED_RR_PRI
893 tThreadList *list = &gaActiveThreads[Thread->Priority];
895 tThreadList *list = &gActiveThreads;
897 Threads_int_AddToList( list, Thread );
900 // Update bookkeeping
901 giNumActiveThreads ++;
903 #if SCHEDULER_TYPE == SCHED_LOTTERY
906 // Only change the ticket count if the thread is un-scheduled
907 if(Thread->CurCPU != -1)
910 delta = caiTICKET_COUNTS[ Thread->Priority ];
912 giFreeTickets += delta;
913 # if DEBUG_TRACE_TICKETS
914 Log("CPU%i %p (%i %s) added, new giFreeTickets = %i [+%i]",
915 GetCPUNum(), Thread, Thread->TID, Thread->ThreadName,
922 SHORTREL( &glThreadListLock );
926 * \brief Removes the current thread from the active queue
927 * \warning This should ONLY be called with the lock held
928 * \return Current thread pointer
930 tThread *Threads_RemActive(void)
932 giNumActiveThreads --;
933 return Proc_GetCurThread();
937 * \fn void Threads_SetFaultHandler(Uint Handler)
938 * \brief Sets the signal handler for a signal
940 void Threads_SetFaultHandler(Uint Handler)
942 //Log_Debug("Threads", "Threads_SetFaultHandler: Handler = %p", Handler);
943 Proc_GetCurThread()->FaultHandler = Handler;
947 * \fn void Threads_Fault(int Num)
948 * \brief Calls a fault handler
950 void Threads_Fault(int Num)
952 tThread *thread = Proc_GetCurThread();
956 Log_Log("Threads", "Threads_Fault: thread->FaultHandler = %p", thread->FaultHandler);
958 switch(thread->FaultHandler)
961 Threads_Kill(thread, -1);
964 case 1: // Dump Core?
965 Threads_Kill(thread, -1);
970 // Double Fault? Oh, F**k
971 if(thread->CurFaultNum != 0) {
972 Log_Warning("Threads", "Threads_Fault: Double fault on %i", thread->TID);
973 Threads_Kill(thread, -1); // For now, just kill
977 thread->CurFaultNum = Num;
979 Proc_CallFaultHandler(thread);
983 * \fn void Threads_SegFault(tVAddr Addr)
984 * \brief Called when a Segment Fault occurs
986 void Threads_SegFault(tVAddr Addr)
988 tThread *cur = Proc_GetCurThread();
989 cur->bInstrTrace = 0;
990 Log_Warning("Threads", "Thread #%i committed a segfault at address %p", cur->TID, Addr);
991 MM_DumpTables(0, USER_MAX);
993 //Threads_Exit( 0, -1 );
996 // --- Process Structure Access Functions ---
997 tPGID Threads_GetPGID(void)
999 return Proc_GetCurThread()->Process->PGID;
1001 tPID Threads_GetPID(void)
1003 return Proc_GetCurThread()->Process->PID;
1005 tTID Threads_GetTID(void)
1007 return Proc_GetCurThread()->TID;
1009 tUID Threads_GetUID(void)
1011 return Proc_GetCurThread()->Process->UID;
1013 tGID Threads_GetGID(void)
1015 return Proc_GetCurThread()->Process->GID;
1018 int Threads_SetUID(tUID ID)
1020 tThread *t = Proc_GetCurThread();
1021 if( t->Process->UID != 0 ) {
1025 Log_Debug("Threads", "PID %i's UID set to %i", t->Process->PID, ID);
1026 t->Process->UID = ID;
1030 int Threads_SetGID(tGID ID)
1032 tThread *t = Proc_GetCurThread();
1033 if( t->Process->UID != 0 ) {
1037 Log_Debug("Threads", "PID %i's GID set to %i", t->Process->PID, ID);
1038 t->Process->GID = ID;
1042 // --- Per-thread storage ---
1043 int *Threads_GetErrno(void)
1045 return &Proc_GetCurThread()->_errno;
1048 // --- Configuration ---
1049 int *Threads_GetMaxFD(void)
1051 return &Proc_GetCurThread()->Process->MaxFD;
1053 char **Threads_GetChroot(void)
1055 return &Proc_GetCurThread()->Process->RootDir;
1057 char **Threads_GetCWD(void)
1059 return &Proc_GetCurThread()->Process->CurrentWorkingDir;
1063 void Threads_int_DumpThread(tThread *thread)
1065 Log(" %p %i (%i) - %s (CPU %i) - %i (%s)",
1066 thread, thread->TID, thread->Process->PID, thread->ThreadName, thread->CurCPU,
1067 thread->Status, casTHREAD_STAT[thread->Status]
1069 switch(thread->Status)
1071 case THREAD_STAT_MUTEXSLEEP:
1072 Log(" Mutex Pointer: %p", thread->WaitPointer);
1074 case THREAD_STAT_SEMAPHORESLEEP:
1075 Log(" Semaphore Pointer: %p", thread->WaitPointer);
1076 Log(" Semaphore Name: %s:%s",
1077 ((tSemaphore*)thread->WaitPointer)->ModName,
1078 ((tSemaphore*)thread->WaitPointer)->Name
1081 case THREAD_STAT_ZOMBIE:
1082 Log(" Return Status: %i", thread->RetStatus);
1086 Log(" Priority %i, Quantum %i", thread->Priority, thread->Quantum);
1087 Log(" KStack %p", thread->KernelStack);
1088 if( thread->bInstrTrace )
1089 Log(" Tracing Enabled");
1090 Proc_DumpThreadCPUState(thread);
1094 * \fn void Threads_Dump(void)
1096 void Threads_DumpActive(void)
1100 #if SCHEDULER_TYPE == SCHED_RR_PRI
1104 Log("Active Threads: (%i reported)", giNumActiveThreads);
1106 #if SCHEDULER_TYPE == SCHED_RR_PRI
1107 for( i = 0; i < MIN_PRIORITY+1; i++ )
1109 list = &gaActiveThreads[i];
1111 list = &gActiveThreads;
1113 for(thread=list->Head;thread;thread=thread->Next)
1115 Threads_int_DumpThread(thread);
1116 if(thread->Status != THREAD_STAT_ACTIVE)
1117 Log(" ERROR State (%i) != THREAD_STAT_ACTIVE (%i)",
1118 thread->Status, THREAD_STAT_ACTIVE);
1121 #if SCHEDULER_TYPE == SCHED_RR_PRI
1127 * \fn void Threads_Dump(void)
1128 * \brief Dumps a list of currently running threads
1130 void Threads_Dump(void)
1134 Log("--- Thread Dump ---");
1135 Threads_DumpActive();
1137 Log("All Threads:");
1138 for(thread=gAllThreads;thread;thread=thread->GlobalNext)
1140 Threads_int_DumpThread(thread);
1145 * \brief Gets the next thread to run
1146 * \param CPU Current CPU
1147 * \param Last The thread the CPU was running
1149 tThread *Threads_GetNextToRun(int CPU, tThread *Last)
1153 // If this CPU has the lock, we must let it complete
1154 if( CPU_HAS_LOCK( &glThreadListLock ) )
1157 // Don't change threads if the current CPU has switches disabled
1158 if( gaThreads_NoTaskSwitch[CPU] )
1162 SHORTLOCK( &glThreadListLock );
1164 // Make sure the current (well, old) thread is marked as de-scheduled
1165 if(Last) Last->CurCPU = -1;
1167 // No active threads, just take a nap
1168 if(giNumActiveThreads == 0) {
1169 SHORTREL( &glThreadListLock );
1170 #if DEBUG_TRACE_TICKETS
1171 Log("No active threads");
1177 #if SCHEDULER_TYPE != SCHED_RR_PRI
1178 // Special case: 1 thread
1179 if(giNumActiveThreads == 1) {
1180 if( gActiveThreads.Head->CurCPU == -1 )
1181 gActiveThreads.Head->CurCPU = CPU;
1183 SHORTREL( &glThreadListLock );
1185 if( gActiveThreads.Head->CurCPU == CPU )
1186 return gActiveThreads.Head;
1188 return NULL; // CPU has nothing to do
1193 // Allow the old thread to be scheduled again
1195 if( Last->Status == THREAD_STAT_ACTIVE ) {
1197 #if SCHEDULER_TYPE == SCHED_LOTTERY
1198 giFreeTickets += caiTICKET_COUNTS[ Last->Priority ];
1199 # if DEBUG_TRACE_TICKETS
1200 LogF("Log: CPU%i released %p (%i %s) into the pool (%i [+%i] tickets in pool)\n",
1201 CPU, Last, Last->TID, Last->ThreadName, giFreeTickets,
1202 caiTICKET_COUNTS[ Last->Priority ]);
1206 #if SCHEDULER_TYPE == SCHED_RR_PRI
1207 list = &gaActiveThreads[ Last->Priority ];
1209 list = &gActiveThreads;
1211 // Add to end of list
1212 Threads_int_AddToList( list, Last );
1214 #if SCHEDULER_TYPE == SCHED_LOTTERY && DEBUG_TRACE_TICKETS
1216 LogF("Log: CPU%i released %p (%i %s)->Status = %i (Released,not in pool)\n",
1217 CPU, Last, Last->TID, Last->ThreadName, Last->Status);
1223 // Lottery Scheduler
1225 #if SCHEDULER_TYPE == SCHED_LOTTERY
1230 for(thread = gActiveThreads.Head; thread; thread = thread->Next)
1232 if(thread->Status != THREAD_STAT_ACTIVE)
1233 Panic("Bookkeeping fail - %p %i(%s) is on the active queue with a status of %i",
1234 thread, thread->TID, thread->ThreadName, thread->Status);
1235 if(thread->Next == thread) {
1236 Panic("Bookkeeping fail - %p %i(%s) loops back on itself",
1237 thread, thread->TID, thread->ThreadName, thread->Status);
1239 number += caiTICKET_COUNTS[ thread->Priority ];
1241 if(number != giFreeTickets) {
1242 Panic("Bookkeeping fail (giFreeTickets(%i) != number(%i)) - CPU%i",
1243 giFreeTickets, number, CPU);
1247 // No free tickets (all tasks delegated to cores)
1248 if( giFreeTickets == 0 ) {
1249 SHORTREL(&glThreadListLock);
1253 // Get the ticket number
1254 ticket = number = rand() % giFreeTickets;
1256 // Find the next thread
1257 for(thread = gActiveThreads.Head; thread; prev = thread, thread = thread->Next )
1259 if( caiTICKET_COUNTS[ thread->Priority ] > number) break;
1260 number -= caiTICKET_COUNTS[ thread->Priority ];
1263 // If we didn't find a thread, something went wrong
1267 for(thread=gActiveThreads;thread;thread=thread->Next) {
1268 if(thread->CurCPU >= 0) continue;
1269 number += caiTICKET_COUNTS[ thread->Priority ];
1271 Panic("Bookeeping Failed - giFreeTickets(%i) > true count (%i)",
1272 giFreeTickets, number);
1277 prev->Next = thread->Next;
1279 gActiveThreads.Head = thread->Next;
1281 gActiveThreads.Tail = prev;
1283 giFreeTickets -= caiTICKET_COUNTS[ thread->Priority ];
1284 # if DEBUG_TRACE_TICKETS
1285 LogF("Log: CPU%i allocated %p (%i %s), (%i [-%i] tickets in pool), \n",
1286 CPU, thread, thread->TID, thread->ThreadName,
1287 giFreeTickets, caiTICKET_COUNTS[ thread->Priority ]);
1292 // Priority based round robin scheduler
1294 #elif SCHEDULER_TYPE == SCHED_RR_PRI
1298 for( i = 0; i < MIN_PRIORITY + 1; i ++ )
1300 if( !gaActiveThreads[i].Head )
1303 thread = gaActiveThreads[i].Head;
1306 gaActiveThreads[i].Head = thread->Next;
1308 gaActiveThreads[i].Tail = NULL;
1309 thread->Next = NULL;
1315 SHORTREL(&glThreadListLock);
1318 if( thread->Status != THREAD_STAT_ACTIVE ) {
1319 LogF("Oops, Thread %i (%s) is not active\n", thread->TID, thread->ThreadName);
1322 #elif SCHEDULER_TYPE == SCHED_RR_SIM
1324 // Get the next thread off the list
1325 thread = gActiveThreads.Head;
1326 gActiveThreads.Head = thread->Next;
1328 gaActiveThreads.Tail = NULL;
1329 thread->Next = NULL;
1333 SHORTREL(&glThreadListLock);
1338 # error "Unimplemented scheduling algorithm"
1341 // Make the new thread non-schedulable
1342 thread->CurCPU = CPU;
1343 thread->Remaining = thread->Quantum;
1345 SHORTREL( &glThreadListLock );
1351 EXPORT(Threads_GetUID);
1352 EXPORT(Threads_GetGID);