src/afs/DARWIN/osi_sleep.c

   1 /*
   2  * Copyright 2000, International Business Machines Corporation and others.
   3  * All Rights Reserved.
   4  *
   5  * This software has been released under the terms of the IBM Public
   6  * License.  For details, see the LICENSE file in the top-level source
   7  * directory or online at http://www.openafs.org/dl/license10.html
   8  */
   9
  10 #include "../afs/param.h"       /* Should be always first */
  11 #include <afsconfig.h>
  12
  13 RCSID("$Header$");
  14
  15 #include "../afs/sysincludes.h" /* Standard vendor system headers */
  16 #include "../afs/afsincludes.h" /* Afs-based standard headers */
  17 #include "../afs/afs_stats.h"   /* afs statistics */
  18
  19
  20 static int osi_TimedSleep(char *event, afs_int32 ams, int aintok);
  21 void afs_osi_Wakeup(char *event);
  22 void afs_osi_Sleep(char *event);
  23
  24 static char waitV;
  25
  26
  27 void afs_osi_InitWaitHandle(struct afs_osi_WaitHandle *achandle)
  28 {
  29     AFS_STATCNT(osi_InitWaitHandle);
  30     achandle->proc = (caddr_t) 0;
  31 }
  32
  33 /* cancel osi_Wait */
  34 void afs_osi_CancelWait(struct afs_osi_WaitHandle *achandle)
  35 {
  36     caddr_t proc;
  37
  38     AFS_STATCNT(osi_CancelWait);
  39     proc = achandle->proc;
  40     if (proc == 0) return;
  41     achandle->proc = (caddr_t) 0;   /* so dude can figure out he was signalled */
  42     afs_osi_Wakeup(&waitV);
  43 }
  44
  45 /* afs_osi_Wait
  46  * Waits for data on ahandle, or ams ms later.  ahandle may be null.
  47  * Returns 0 if timeout and EINTR if signalled.
  48  */
  49 int afs_osi_Wait(afs_int32 ams, struct afs_osi_WaitHandle *ahandle, int aintok)
  50 {
  51     int code;
  52     afs_int32 endTime, tid;
  53     struct proc *p=current_proc();
  54
  55     AFS_STATCNT(osi_Wait);
  56     endTime = osi_Time() + (ams/1000);
  57     if (ahandle)
  58         ahandle->proc = (caddr_t)p;
  59     do {
  60         AFS_ASSERT_GLOCK();
  61         code = 0;
  62         code = osi_TimedSleep(&waitV, ams, aintok);
  63
  64         if (code) break;        /* if something happened, quit now */
  65         /* if we we're cancelled, quit now */
  66         if (ahandle && (ahandle->proc == (caddr_t) 0)) {
  67             /* we've been signalled */
  68             break;
  69         }
  70     } while (osi_Time() < endTime);
  71     return code;
  72 }
  73
  74
  75
  76 typedef struct afs_event {
  77     struct afs_event *next;     /* next in hash chain */
  78     char *event;                /* lwp event: an address */
  79     int refcount;               /* Is it in use? */
  80     int seq;                    /* Sequence number: this is incremented
  81                                    by wakeup calls; wait will not return until
  82                                    it changes */
  83 } afs_event_t;
  84
  85 #define HASHSIZE 128
  86 afs_event_t *afs_evhasht[HASHSIZE];/* Hash table for events */
  87 #define afs_evhash(event)       (afs_uint32) ((((long)event)>>2) & (HASHSIZE-1));
  88 int afs_evhashcnt = 0;
  89
  90 /* Get and initialize event structure corresponding to lwp event (i.e. address)
  91  * */
  92 static afs_event_t *afs_getevent(char *event)
  93 {
  94     afs_event_t *evp, *newp = 0;
  95     int hashcode;
  96
  97     AFS_ASSERT_GLOCK();
  98     hashcode = afs_evhash(event);
  99     evp = afs_evhasht[hashcode];
 100     while (evp) {
 101         if (evp->event == event) {
 102             evp->refcount++;
 103             return evp;
 104         }
 105         if (evp->refcount == 0)
 106             newp = evp;
 107         evp = evp->next;
 108     }
 109     if (!newp) {
 110         newp = (afs_event_t *) osi_AllocSmallSpace(sizeof (afs_event_t));
 111         afs_evhashcnt++;
 112         newp->next = afs_evhasht[hashcode];
 113         afs_evhasht[hashcode] = newp;
 114         newp->seq = 0;
 115     }
 116     newp->event = event;
 117     newp->refcount = 1;
 118     return newp;
 119 }
 120
 121 /* Release the specified event */
 122 #define relevent(evp) ((evp)->refcount--)
 123
 124
 125 void afs_osi_Sleep(char *event)
 126 {
 127     struct afs_event *evp;
 128     int seq;
 129
 130     evp = afs_getevent(event);
 131     seq = evp->seq;
 132     while (seq == evp->seq) {
 133         AFS_ASSERT_GLOCK();
 134         assert_wait((event_t)event, 0);
 135         AFS_GUNLOCK();
 136         thread_block(0);
 137         AFS_GLOCK();
 138     }
 139     relevent(evp);
 140 }
 141
 142 /* osi_TimedSleep
 143  *
 144  * Arguments:
 145  * event - event to sleep on
 146  * ams --- max sleep time in milliseconds
 147  * aintok - 1 if should sleep interruptibly
 148  *
 149  * Returns 0 if timeout and EINTR if signalled.
 150  */
 151 static int osi_TimedSleep(char *event, afs_int32 ams, int aintok)
 152 {
 153     int code = 0;
 154     struct afs_event *evp;
 155     int ticks,seq;
 156
 157     ticks = ( ams * afs_hz )/1000;
 158
 159
 160     evp = afs_getevent(event);
 161     seq=evp->seq;
 162     assert_wait((event_t)event, aintok ? THREAD_ABORTSAFE : 0);
 163     AFS_GUNLOCK();
 164     thread_set_timer(ticks, NSEC_PER_SEC / hz);
 165     thread_block(0);
 166     AFS_GLOCK();
 167 #if 0 /* thread_t structure only available if MACH_KERNEL_PRIVATE */
 168     if (current_thread()->wait_result != THREAD_AWAKENED)
 169         code = EINTR;
 170 #else
 171     if (seq == evp->seq)
 172         code = EINTR;
 173 #endif
 174
 175     relevent(evp);
 176     return code;
 177 }
 178
 179
 180 void afs_osi_Wakeup(char *event)
 181 {
 182     struct afs_event *evp;
 183
 184     evp = afs_getevent(event);
 185     if (evp->refcount > 1) {
 186         evp->seq++;
 187         thread_wakeup((event_t)event);
 188     }
 189     relevent(evp);
 190 }