Added GetProcessByID and GetThreadByID

This commit is contained in:
Alex 2023-02-07 04:55:26 +02:00
parent a23d999183
commit b104511241
Signed by untrusted user who does not match committer: enderice2
GPG Key ID: EACC3AD603BAB4DD
2 changed files with 23 additions and 0 deletions

View File

@ -202,6 +202,23 @@ namespace Tasking
PCB *Task::GetCurrentProcess() { return GetCurrentCPU()->CurrentProcess; }
TCB *Task::GetCurrentThread() { return GetCurrentCPU()->CurrentThread; }
PCB *Task::GetProcessByID(UPID ID)
{
for (size_t i = 0; i < ListProcess.size(); i++)
if (ListProcess[i]->ID == ID)
return ListProcess[i];
return nullptr;
}
TCB *Task::GetThreadByID(UTID ID)
{
for (size_t i = 0; i < ListProcess.size(); i++)
for (size_t j = 0; j < ListProcess[i]->Threads.size(); j++)
if (ListProcess[i]->Threads[j]->ID == ID)
return ListProcess[i]->Threads[j];
return nullptr;
}
void Task::WaitForProcess(PCB *pcb)
{
if (!pcb)

View File

@ -267,6 +267,7 @@ namespace Tasking
void SignalShutdown();
void RevertProcessCreation(PCB *Process);
void RevertThreadCreation(TCB *Thread);
long GetUsage(int Core)
{
if (IdleProcess)
@ -274,6 +275,7 @@ namespace Tasking
else
return 0;
}
void KillThread(TCB *tcb, int Code)
{
tcb->Status = TaskStatus::Terminated;
@ -298,6 +300,10 @@ namespace Tasking
*/
TCB *GetCurrentThread();
PCB *GetProcessByID(UPID ID);
TCB *GetThreadByID(UTID ID);
/** @brief Wait for process to terminate */
void WaitForProcess(PCB *pcb);