rom/exec/settaskstorageslot.c

   1 /*
   2     Copyright © 2012, The AROS Development Team. All rights reserved.
   3     $Id$
   4 */
   5
   6 #include <aros/debug.h>
   7
   8 #include <exec/nodes.h>
   9 #include <exec/lists.h>
  10
  11 #include "exec_intern.h"
  12 #include "taskstorage.h"
  13
  14 /*****************************************************************************
  15
  16     NAME */
  17 #include <proto/exec.h>
  18
  19         AROS_LH2(BOOL, SetTaskStorageSlot,
  20
  21 /*  LOCATION */
  22         AROS_LHA(LONG, id, D0),
  23         AROS_LHA(IPTR , value, D1),
  24         struct ExecBase *, SysBase, 184, Exec)
  25
  26 /*  FUNCTION
  27         The will remember the current state of the task storage slots
  28
  29     INPUTS
  30         None.
  31
  32     RESULT
  33         An id will be returned whit which the current state can be restored
  34         using RestoreTaskStorageSlots().
  35         NULL is returned when not enough memory was available.
  36
  37     NOTES
  38
  39     EXAMPLE
  40
  41     BUGS
  42
  43     SEE ALSO
  44         AllocTaskStorageSlot()/FreeTaskStorageSlot()/GetTaskStorageSlot()
  45
  46     INTERNALS
  47
  48 ******************************************************************************/
  49 {
  50     AROS_LIBFUNC_INIT
  51
  52     struct ETask *et = GetETask(FindTask(NULL));
  53     IPTR *ts;
  54
  55     D(bug("SetTaskStorage: %p: Set TaskStorageSlot %d to %p\n", et, id, (APTR)value));
  56
  57     if (!et) {
  58         /* Only ETasks can do this */
  59         D(bug("SetTaskStorage: Not an ETask!\n"));
  60         return FALSE;
  61     }
  62
  63     /* Valid ID? */
  64     if (id < 1) {
  65         D(bug("SetTaskStorage: Invalid ID %d\n", id));
  66         return FALSE;
  67     }
  68
  69     ts = et->et_TaskStorage;
  70     if (ts == NULL) {
  71         /* No TaskStorage? Allocate some. */
  72         ULONG slots = ((id + TASKSTORAGEPUDDLE) / TASKSTORAGEPUDDLE) * TASKSTORAGEPUDDLE;
  73         ts = AllocMem(slots * sizeof(ts[0]), MEMF_ANY | MEMF_CLEAR);
  74         if (ts == NULL) {
  75             D(bug("SetTaskStorage: Can't allocate %d slots\n", slots));
  76             return FALSE;
  77         }
  78
  79         ts[__TS_FIRSTSLOT] = slots;
  80
  81         et->et_TaskStorage = ts;
  82     } else if (ts[__TS_FIRSTSLOT] <= id) {
  83         IPTR *newts;
  84         ULONG newslots = ((id + TASKSTORAGEPUDDLE) / TASKSTORAGEPUDDLE) * TASKSTORAGEPUDDLE;
  85         size_t oldslots = ts[__TS_FIRSTSLOT];
  86
  87         newts = AllocMem(newslots * sizeof(ts[0]), MEMF_ANY);
  88         if (newts == NULL) {
  89             D(bug("SetTaskStorage: Can't allocate %d new slots\n", newslots));
  90             return FALSE;
  91         }
  92
  93         newts[__TS_FIRSTSLOT] = newslots;
  94
  95         CopyMem(&ts[__TS_FIRSTSLOT+1], &newts[__TS_FIRSTSLOT+1],
  96                 (oldslots-1) * sizeof(ts[0]));
  97         memset(&newts[oldslots], 0, (newslots - oldslots) * sizeof(ts[0]));
  98
  99         et->et_TaskStorage = newts;
 100         FreeMem(ts, oldslots * sizeof(ts[0]));
 101     }
 102
 103     et->et_TaskStorage[id] = value;
 104
 105     return TRUE;
 106
 107     AROS_LIBFUNC_EXIT
 108 }