-
Notifications
You must be signed in to change notification settings - Fork 0
/
onion-station.c
60 lines (47 loc) · 1.55 KB
/
onion-station.c
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
#ifdef ATOMIC_ALL_NIGHTERS
/* function annotations */
#define MAY_SLEEP __attribute__((atomic_all_nighters("might_sleep")))
#define WONT_SLEEP __attribute__((atomic_all_nighters("wont_sleep")))
/* context-changing annotations */
#define ENTER_ATOMIC __attribute__((atomic_all_nighters("wont_sleep","force_enable")))
#define EXIT_ATOMIC __attribute__((atomic_all_nighters("wont_sleep","force_disable")))
#define ENTER_ATOMIC_NESTED __attribute__((atomic_all_nighters("wont_sleep","enter_nested")))
#define EXIT_ATOMIC_NESTED __attribute__((atomic_all_nighters("wont_sleep","exit_nested")))
#else
#define MAY_SLEEP
#define WONT_SLEEP
#define ENTER_ATOMIC
#define EXIT_ATOMIC
#define ENTER_ATOMIC_NESTED
#define EXIT_ATOMIC_NESTED
#endif
struct mutex;
struct spinlock;
void mutex_lock(struct mutex *mp) MAY_SLEEP;
void mutex_unlock(struct mutex *mp) MAY_SLEEP;
void mutex_assert_is_locked(struct mutex *mp) WONT_SLEEP;
void spin_lock(struct spinlock *sp) ENTER_ATOMIC_NESTED;
void spin_unlock(struct spinlock *sp) EXIT_ATOMIC_NESTED;
struct spinlock *a;
struct mutex *m;
/* h2 <: h1 */
void (*h1)(struct mutex *) MAY_SLEEP;
void (*h2)(struct mutex *) WONT_SLEEP;
/* g1 <: g2 */
void WONT_SLEEP g1(void (*x)(struct mutex *) MAY_SLEEP)
{
h1 = x;
}
void MAY_SLEEP g2(void (*x)(struct mutex *) WONT_SLEEP)
{
spin_lock(a);
x(m);
spin_unlock(a);
}
int MAY_SLEEP main()
{
/* this should fail - have to use intersection on the argument type */
void (*f)(void (*)(struct mutex *) MAY_SLEEP) MAY_SLEEP = 0 ? g1 : g2;
f(mutex_lock);
// return 0;
}