71cf22541bddfe93ebbb0881099440016d7ddc7a
[openafs.git] / src / afs / NBSD / 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
11 #include <afsconfig.h>
12 #include "afs/param.h"
13
14
15 #include "afs/sysincludes.h"    /* Standard vendor system headers */
16 #include "afsincludes.h"        /* Afs-based standard headers */
17 #include "afs/afs_stats.h"      /* afs statistics */
18
19
20
21 static int osi_TimedSleep(char *event, afs_int32 ams, int aintok);
22 static char waitV;
23
24
25 void
26 afs_osi_InitWaitHandle(struct afs_osi_WaitHandle *achandle)
27 {
28     AFS_STATCNT(osi_InitWaitHandle);
29     achandle->proc = (caddr_t) 0;
30 }
31
32 /* cancel osi_Wait */
33 void
34 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)
41         return;
42     achandle->proc = (caddr_t) 0;       /* so dude can figure out he was signalled */
43     afs_osi_Wakeup(&waitV);
44 }
45
46 /* afs_osi_Wait
47  * Waits for data on ahandle, or ams ms later.  ahandle may be null.
48  * Returns 0 if timeout and EINTR if signalled.
49  */
50 int
51 afs_osi_Wait(afs_int32 ams, struct afs_osi_WaitHandle *ahandle, int aintok)
52 {
53     int code;
54     afs_int32 endTime, tid;
55
56     AFS_STATCNT(osi_Wait);
57     endTime = osi_Time() + (ams / 1000);
58     if (ahandle)
59         ahandle->proc = (caddr_t) curproc;
60     do {
61         AFS_ASSERT_GLOCK();
62         code = 0;
63         code = osi_TimedSleep(&waitV, ams, aintok);
64
65         if (code)
66             break;              /* if something happened, quit now */
67         /* if we we're cancelled, quit now */
68         if (ahandle && (ahandle->proc == (caddr_t) 0)) {
69             /* we've been signalled */
70             break;
71         }
72     } while (osi_Time() < endTime);
73     return code;
74 }
75
76
77
78
79 typedef struct afs_event {
80     struct afs_event *next;     /* next in hash chain */
81     char *event;                /* lwp event: an address */
82     int refcount;               /* Is it in use? */
83     int seq;                    /* Sequence number: this is incremented
84                                  * by wakeup calls; wait will not return until
85                                  * it changes */
86     int cond;
87 } afs_event_t;
88
89 #define HASHSIZE 128
90 afs_event_t *afs_evhasht[HASHSIZE];     /* Hash table for events */
91 #define afs_evhash(event)       (afs_uint32) ((((long)event)>>2) & (HASHSIZE-1));
92 int afs_evhashcnt = 0;
93
94 /* Get and initialize event structure corresponding to lwp event (i.e. address)
95  * */
96 static afs_event_t *
97 afs_getevent(char *event)
98 {
99     afs_event_t *evp, *newp = 0;
100     int hashcode;
101
102     AFS_ASSERT_GLOCK();
103     hashcode = afs_evhash(event);
104     evp = afs_evhasht[hashcode];
105     while (evp) {
106         if (evp->event == event) {
107             evp->refcount++;
108             return evp;
109         }
110         if (evp->refcount == 0)
111             newp = evp;
112         evp = evp->next;
113     }
114     if (!newp) {
115         newp = (afs_event_t *) osi_AllocSmallSpace(sizeof(afs_event_t));
116         afs_evhashcnt++;
117         newp->next = afs_evhasht[hashcode];
118         afs_evhasht[hashcode] = newp;
119         newp->seq = 0;
120     }
121     newp->event = event;
122     newp->refcount = 1;
123     return newp;
124 }
125
126 /* Release the specified event */
127 #define relevent(evp) ((evp)->refcount--)
128
129
130 void
131 afs_osi_Sleep(void *event)
132 {
133     struct afs_event *evp;
134     int seq;
135
136     evp = afs_getevent(event);
137     seq = evp->seq;
138     while (seq == evp->seq) {
139         AFS_ASSERT_GLOCK();
140         assert_wait((vm_offset_t) (&evp->cond), 0);
141         AFS_GUNLOCK();
142         thread_block();
143         AFS_GLOCK();
144     }
145     relevent(evp);
146 }
147
148 int
149 afs_osi_SleepSig(void *event)
150 {
151     afs_osi_Sleep(event);
152     return 0;
153 }
154
155 /* osi_TimedSleep
156  * 
157  * Arguments:
158  * event - event to sleep on
159  * ams --- max sleep time in milliseconds
160  * aintok - 1 if should sleep interruptibly
161  *
162  * Returns 0 if timeout and EINTR if signalled.
163  */
164 static int
165 osi_TimedSleep(char *event, afs_int32 ams, int aintok)
166 {
167     int code = 0;
168     struct afs_event *evp;
169     int ticks;
170
171     ticks = (ams * afs_hz) / 1000;
172
173
174     evp = afs_getevent(event);
175
176     assert_wait((vm_offset_t) (&evp->cond), aintok);
177     AFS_GUNLOCK();
178     thread_set_timeout(ticks);
179     thread_block();
180     AFS_GLOCK();
181     /*    if (current_thread()->wait_result != THREAD_AWAKENED)
182      * code = EINTR; */
183
184     relevent(evp);
185     return code;
186 }
187
188
189 int
190 afs_osi_Wakeup(void *event)
191 {
192     int ret = 1;
193     struct afs_event *evp;
194
195     evp = afs_getevent(event);
196     if (evp->refcount > 1) {
197         evp->seq++;
198         thread_wakeup((vm_offset_t) (&evp->cond));
199         ret = 0;
200     }
201     relevent(evp);
202     return ret;
203 }