]> git.proxmox.com Git - rustc.git/blame - src/jemalloc/test/unit/mq.c
Imported Upstream version 1.9.0+dfsg1
[rustc.git] / src / jemalloc / test / unit / mq.c
CommitLineData
1a4d82fc
JJ
1#include "test/jemalloc_test.h"
2
3#define NSENDERS 3
4#define NMSGS 100000
5
6typedef struct mq_msg_s mq_msg_t;
7struct mq_msg_s {
8 mq_msg(mq_msg_t) link;
9};
10mq_gen(static, mq_, mq_t, mq_msg_t, link)
11
12TEST_BEGIN(test_mq_basic)
13{
14 mq_t mq;
15 mq_msg_t msg;
16
17 assert_false(mq_init(&mq), "Unexpected mq_init() failure");
18 assert_u_eq(mq_count(&mq), 0, "mq should be empty");
19 assert_ptr_null(mq_tryget(&mq),
20 "mq_tryget() should fail when the queue is empty");
21
22 mq_put(&mq, &msg);
23 assert_u_eq(mq_count(&mq), 1, "mq should contain one message");
24 assert_ptr_eq(mq_tryget(&mq), &msg, "mq_tryget() should return msg");
25
26 mq_put(&mq, &msg);
27 assert_ptr_eq(mq_get(&mq), &msg, "mq_get() should return msg");
28
29 mq_fini(&mq);
30}
31TEST_END
32
33static void *
34thd_receiver_start(void *arg)
35{
36 mq_t *mq = (mq_t *)arg;
37 unsigned i;
38
39 for (i = 0; i < (NSENDERS * NMSGS); i++) {
40 mq_msg_t *msg = mq_get(mq);
41 assert_ptr_not_null(msg, "mq_get() should never return NULL");
42 dallocx(msg, 0);
43 }
44 return (NULL);
45}
46
47static void *
48thd_sender_start(void *arg)
49{
50 mq_t *mq = (mq_t *)arg;
51 unsigned i;
52
53 for (i = 0; i < NMSGS; i++) {
54 mq_msg_t *msg;
55 void *p;
56 p = mallocx(sizeof(mq_msg_t), 0);
57 assert_ptr_not_null(p, "Unexpected mallocx() failure");
58 msg = (mq_msg_t *)p;
59 mq_put(mq, msg);
60 }
61 return (NULL);
62}
63
64TEST_BEGIN(test_mq_threaded)
65{
66 mq_t mq;
67 thd_t receiver;
68 thd_t senders[NSENDERS];
69 unsigned i;
70
71 assert_false(mq_init(&mq), "Unexpected mq_init() failure");
72
73 thd_create(&receiver, thd_receiver_start, (void *)&mq);
74 for (i = 0; i < NSENDERS; i++)
75 thd_create(&senders[i], thd_sender_start, (void *)&mq);
76
77 thd_join(receiver, NULL);
78 for (i = 0; i < NSENDERS; i++)
79 thd_join(senders[i], NULL);
80
81 mq_fini(&mq);
82}
83TEST_END
84
85int
86main(void)
87{
54a0048b 88
1a4d82fc
JJ
89 return (test(
90 test_mq_basic,
91 test_mq_threaded));
92}
93