rom/exec/alloctaskstorageslot.c

   1 /*
   2     Copyright © 2011-2015, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <aros/debug.h>
   7
   8 #include "exec_intern.h"
   9 #include "exec_util.h"
  10 #include "taskstorage.h"
  11
  12 /*****************************************************************************
  13
  14     NAME */
  15 #include <proto/exec.h>
  16
  17         AROS_LH0(LONG, AllocTaskStorageSlot,
  18
  19 /*  LOCATION */
  20         struct ExecBase *, SysBase, 180, Exec)
  21
  22 /*  FUNCTION
  23         This function will allocate a slot in the taskstorage.
  24
  25     INPUTS
  26         None.
  27
  28     RESULT
  29         slot - The allocated slot, or 0 if no slot could be allocated.
  30
  31     NOTES
  32         After this function SetTaskStorageSlot(slot) may be used to store
  33         values with each slot.
  34
  35     EXAMPLE
  36
  37     BUGS
  38
  39     SEE ALSO
  40         FreeTaskStorageSlot(), GetTaskStorageSlot(), SetTaskStorageSlot()
  41
  42     INTERNALS
  43
  44 ******************************************************************************/
  45 {
  46     AROS_LIBFUNC_INIT
  47
  48     struct Task *ThisTask = GET_THIS_TASK;
  49     struct TaskStorageFreeSlot *tsfs;
  50     LONG slot;
  51     struct IntETask *iet = ThisTask ? GetIntETask(ThisTask) : NULL;
  52
  53     if (!iet)
  54         return 0;
  55
  56     Forbid();
  57     tsfs = (struct TaskStorageFreeSlot *)
  58         GetHead(&PrivExecBase(SysBase)->TaskStorageSlots);
  59     if (!tsfs)
  60     {
  61         Alert(AT_DeadEnd|AN_MemoryInsane);
  62         __builtin_unreachable();
  63     }
  64
  65     slot = tsfs->FreeSlot;
  66
  67     D(bug("[TSS] Task 0x%p (%s): Allocated slot %d\n", ThisTask, ThisTask->tc_Node.ln_Name, slot);)
  68
  69     if (GetSucc(tsfs) == NULL)
  70     {
  71         /* Last element always points to highest element and is a new slot */
  72         tsfs->FreeSlot++;
  73     }
  74     else
  75     {
  76         Remove((struct Node *) tsfs);
  77         FreeMem(tsfs, sizeof(struct TaskStorageFreeSlot));
  78     }
  79     Permit();
  80
  81     return slot;
  82
  83     AROS_LIBFUNC_EXIT
  84 }