rom/task/init.c

   1 /*
   2     Copyright © 2015, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #define DEBUG 0
   7
   8 #include <aros/debug.h>
   9 #include <aros/symbolsets.h>
  10
  11 #include <proto/exec.h>
  12 #include <proto/kernel.h>
  13 #include <resources/task.h>
  14
  15 #include <exec_intern.h>
  16 #include "etask.h"
  17
  18 #include "taskres_intern.h"
  19
  20 extern APTR AROS_SLIB_ENTRY(NewAddTask, Task, 176)();
  21 extern void AROS_SLIB_ENTRY(RemTask, Task, 48)();
  22
  23 static LONG taskres_Init(struct TaskResBase *TaskResBase)
  24 {
  25 #if defined(__AROSEXEC_SMP__)
  26     spinlock_t *listLock;
  27 #endif
  28     struct TaskListEntry *taskEntry = NULL;
  29     struct Task *curTask = NULL;
  30
  31     KernelBase = OpenResource("kernel.resource");
  32     if (!KernelBase)
  33         return FALSE;
  34
  35     TaskResBase->trb_UtilityBase = OpenLibrary("utility.library", 0);
  36     if (!TaskResBase->trb_UtilityBase)
  37         return FALSE;
  38
  39     NEWLIST(&TaskResBase->trb_TaskList);
  40     NEWLIST(&TaskResBase->trb_LockedLists);
  41
  42     SysBase->lb_TaskResBase = (struct Library *)TaskResBase;
  43
  44     TaskResBase->trb_NewAddTask = SetFunction((struct Library *)SysBase, -176*LIB_VECTSIZE, AROS_SLIB_ENTRY(NewAddTask, Task, 176));
  45     TaskResBase->trb_RemTask = SetFunction((struct Library *)SysBase, -48*LIB_VECTSIZE, AROS_SLIB_ENTRY(RemTask, Task, 48));
  46
  47     /*
  48        Add existing tasks to our internal list ..
  49     */
  50 #if defined(__AROSEXEC_SMP__)
  51     listLock = KrnSpinLock(&PrivExecBase(SysBase)->TaskRunningSpinLock, NULL, SPINLOCK_MODE_READ);
  52     Forbid();
  53     ForeachNode(&PrivExecBase(SysBase)->TaskRunning, curTask)
  54     {
  55         if ((taskEntry = AllocMem(sizeof(struct TaskListEntry), MEMF_CLEAR)) != NULL)
  56         {
  57             D(bug("[TaskRes] 0x%p [R  ] %02d %s\n", curTask, GetIntETask(curTask)->iet_CpuNumber, curTask->tc_Node.ln_Name));
  58             taskEntry->tle_Task = curTask;
  59             AddTail(&TaskResBase->trb_TaskList, &taskEntry->tle_Node);
  60         }
  61     }
  62     KrnSpinUnLock(listLock);
  63     Permit();
  64     listLock = KrnSpinLock(&PrivExecBase(SysBase)->TaskReadySpinLock, NULL, SPINLOCK_MODE_READ);
  65     Forbid();
  66     // TODO : list TaskSpinning tasks..
  67 #else
  68     Disable();
  69     if (SysBase->ThisTask)
  70     {
  71         if ((taskEntry = AllocMem(sizeof(struct TaskListEntry), MEMF_CLEAR)) != NULL)
  72         {
  73             D(bug("[TaskRes] 0x%p [R--] 00 %s\n", SysBase->ThisTask, SysBase->ThisTask->tc_Node.ln_Name));
  74             taskEntry->tle_Task = SysBase->ThisTask;
  75             AddTail(&TaskResBase->trb_TaskList, &taskEntry->tle_Node);
  76         }
  77     }
  78 #endif
  79     ForeachNode(&SysBase->TaskReady, curTask)
  80     {
  81         if ((taskEntry = AllocMem(sizeof(struct TaskListEntry), MEMF_CLEAR)) != NULL)
  82         {
  83             D(bug("[TaskRes] 0x%p [-R-] -- %s\n", curTask, curTask->tc_Node.ln_Name));
  84             taskEntry->tle_Task = curTask;
  85             AddTail(&TaskResBase->trb_TaskList, &taskEntry->tle_Node);
  86         }
  87     }
  88 #if defined(__AROSEXEC_SMP__)
  89     KrnSpinUnLock(listLock);
  90     Permit();
  91     listLock = KrnSpinLock(&PrivExecBase(SysBase)->TaskWaitSpinLock, NULL, SPINLOCK_MODE_READ);
  92     Forbid();
  93 #endif
  94     ForeachNode(&SysBase->TaskWait, curTask)
  95     {
  96         if ((taskEntry = AllocMem(sizeof(struct TaskListEntry), MEMF_CLEAR)) != NULL)
  97         {
  98             D(bug("[TaskRes] 0x%p [--W] -- %s\n", curTask, curTask->tc_Node.ln_Name));
  99             taskEntry->tle_Task = curTask;
 100             AddTail(&TaskResBase->trb_TaskList, &taskEntry->tle_Node);
 101         }
 102     }
 103
 104 #if defined(__AROSEXEC_SMP__)
 105     KrnSpinUnLock(listLock);
 106     Permit();
 107 #else
 108     Enable();
 109 #endif
 110
 111     return TRUE;
 112 }
 113
 114 static LONG taskres_Exit(struct TaskResBase *TaskResBase)
 115 {
 116     SetFunction((struct Library *)SysBase, -176*LIB_VECTSIZE, TaskResBase->trb_NewAddTask);
 117     SetFunction((struct Library *)SysBase, -48*LIB_VECTSIZE, TaskResBase->trb_RemTask);
 118
 119     CloseLibrary(TaskResBase->trb_UtilityBase);
 120
 121     return TRUE;
 122 }
 123
 124 ADD2INITLIB(taskres_Init, 0)
 125
 126 ADD2EXPUNGELIB(taskres_Exit, 0)