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
|
/*
* mbox.c
*
* Simple thread mailbox interface
*/
#include "thread.h"
#include "mbox.h"
#include <errno.h>
void mbox_init(struct mailbox *mbox, size_t size)
{
if (!!mbox) {
sem_init(&mbox->prod_sem, size); /* All slots empty */
sem_init(&mbox->cons_sem, 0); /* No slots full */
sem_init(&mbox->head_sem, 1); /* Head mutex */
sem_init(&mbox->tail_sem, 1); /* Tail mutex */
mbox->wrap = &mbox->data[size];
mbox->head = &mbox->data[0];
mbox->tail = &mbox->data[0];
}
};
int mbox_post(struct mailbox *mbox, void *msg, mstime_t timeout)
{
if (!mbox_is_valid(mbox))
return ENOMEM;
if (sem_down(&mbox->prod_sem, timeout) == (mstime_t)-1)
return ENOMEM;
sem_down(&mbox->head_sem, 0);
*mbox->head = msg;
mbox->head++;
if (mbox->head == mbox->wrap)
mbox->head = &mbox->data[0];
sem_up(&mbox->head_sem);
sem_up(&mbox->cons_sem);
return 0;
}
mstime_t mbox_fetch(struct mailbox *mbox, void **msg, mstime_t timeout)
{
mstime_t t;
if (!mbox)
return -1;
t = sem_down(&mbox->cons_sem, timeout);
if (t == (mstime_t)-1)
return -1;
t += sem_down(&mbox->tail_sem, 0);
if (msg)
*msg = *mbox->tail;
mbox->tail++;
if (mbox->tail == mbox->wrap)
mbox->tail = &mbox->data[0];
sem_up(&mbox->tail_sem);
sem_up(&mbox->prod_sem);
return t;
}
|