deps/libretro-common/libco/scefiber.c

   1 /*
   2   libco.win (2016-09-06)
   3   authors: frangarcj
   4   license: public domain
   5 */
   6
   7 #define LIBCO_C
   8 #include <libco.h>
   9 #include <stdlib.h>
  10 #include <psp2/sysmodule.h>
  11
  12 #ifdef __cplusplus
  13 extern "C" {
  14 #endif
  15
  16 static thread_local cothread_t co_active_ = 0;
  17
  18 typedef struct SceFiber
  19 {
  20         char reserved[128];
  21 } SceFiber __attribute__( ( aligned ( 8 ) ) ) ;
  22
  23 /* Forward declarations */
  24 int32_t _sceFiberInitializeImpl(SceFiber *fiber, char *name, void *entry, uint32_t argOnInitialize,
  25       void* addrContext, int32_t sizeContext, void* params);
  26 int32_t sceFiberFinalize(SceFiber* fiber);
  27 int32_t sceFiberRun(SceFiber* fiber, uint32_t argOnRunTo, uint32_t* argOnRun);
  28 int32_t sceFiberSwitch(SceFiber* fiber, uint32_t argOnRunTo, uint32_t* argOnRun);
  29 int32_t sceFiberReturnToThread(uint32_t argOnReturn, uint32_t* argOnRun);
  30
  31 static void co_thunk(uint32_t argOnInitialize, uint32_t argOnRun)
  32 {
  33    ((void (*)(void))argOnInitialize)();
  34 }
  35
  36 cothread_t co_active(void)
  37 {
  38    if (!co_active_)
  39    {
  40       sceSysmoduleLoadModule(SCE_SYSMODULE_FIBER);
  41       co_active_ = (cothread_t)1;
  42    }
  43    return co_active_;
  44 }
  45
  46 cothread_t co_create(unsigned int heapsize, void (*coentry)(void))
  47 {
  48    int ret;
  49    SceFiber* tail_fiber   = malloc(sizeof(SceFiber));
  50    char * m_ctxbuf        = malloc(sizeof(char)*heapsize);
  51    if (!co_active_)
  52    {
  53       sceSysmoduleLoadModule(SCE_SYSMODULE_FIBER);
  54       co_active_          = (cothread_t)1;
  55    }
  56
  57    /* _sceFiberInitializeImpl */
  58    if ((ret = _sceFiberInitializeImpl(
  59                tail_fiber, "tailFiber", co_thunk,
  60                (uint32_t)coentry, (void*)m_ctxbuf, heapsize, NULL)) == 0)
  61       return (cothread_t)tail_fiber;
  62    return (cothread_t)ret;
  63 }
  64
  65 void co_delete(cothread_t cothread)
  66 {
  67    if (cothread != (cothread_t)1)
  68       sceFiberFinalize((SceFiber*)cothread);
  69 }
  70
  71 void co_switch(cothread_t cothread)
  72 {
  73    uint32_t argOnReturn  = 0;
  74    if (cothread == (cothread_t)1)
  75    {
  76       co_active_         = cothread;
  77       sceFiberReturnToThread(0, NULL);
  78    }
  79    else
  80    {
  81       SceFiber* theFiber = (SceFiber*)cothread;
  82       co_active_         = cothread;
  83       if (co_active_ == (cothread_t)1)
  84          sceFiberRun(theFiber, 0, &argOnReturn);
  85       else
  86          sceFiberSwitch(theFiber, 0, &argOnReturn);
  87    }
  88 }
  89
  90 #ifdef __cplusplus
  91 }
  92 #endif