1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
#include <aio.h>
#include <errno.h>
#include <time.h>
#include "atomic.h"
#include "libc.h"
#include "pthread_impl.h"
extern volatile int __aio_fut;
int aio_suspend(const struct aiocb *const cbs[], int cnt, const struct timespec *ts)
{
int i, tid = 0, ret, expect = 0;
struct timespec at;
volatile int dummy_fut, *pfut;
int nzcnt = 0;
const struct aiocb *cb = 0;
if (cnt<0) {
errno = EINVAL;
return -1;
}
for (i=0; i<cnt; i++) if (cbs[i]) {
if (aio_error(cbs[i]) != EINPROGRESS) return 0;
nzcnt++;
cb = cbs[i];
}
if (ts) {
clock_gettime(CLOCK_MONOTONIC, &at);
at.tv_sec += ts->tv_sec;
if ((at.tv_nsec += ts->tv_nsec) >= 1000000000) {
at.tv_nsec -= 1000000000;
at.tv_sec++;
}
}
for (;;) {
for (i=0; i<cnt; i++)
if (cbs[i] && aio_error(cbs[i]) != EINPROGRESS)
return 0;
switch (nzcnt) {
case 0:
pfut = &dummy_fut;
break;
case 1:
pfut = (void *)&cb->__err;
expect = EINPROGRESS | 0x80000000;
a_cas(pfut, EINPROGRESS, expect);
break;
default:
pfut = &__aio_fut;
if (!tid) tid = __pthread_self()->tid;
expect = a_cas(pfut, 0, tid);
if (!expect) expect = tid;
/* Need to recheck the predicate before waiting. */
for (i=0; i<cnt; i++)
if (cbs[i] && aio_error(cbs[i]) != EINPROGRESS)
return 0;
break;
}
ret = __timedwait(pfut, expect, CLOCK_MONOTONIC, ts?&at:0, 0, 0, 1);
if (ret) {
errno = ret==ETIMEDOUT ? EAGAIN : ret;
return -1;
}
}
}
LFS64(aio_suspend);
|