1 #include "test/jemalloc_test.h"
2 
3 #define	NSENDERS	3
4 #define	NMSGS		100000
5 
6 typedef struct mq_msg_s mq_msg_t;
7 struct mq_msg_s {
8 	mq_msg(mq_msg_t)	link;
9 };
mq_gen(static,mq_,mq_t,mq_msg_t,link)10 mq_gen(static, mq_, mq_t, mq_msg_t, link)
11 
12 TEST_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 }
31 TEST_END
32 
33 static void *
thd_receiver_start(void * arg)34 thd_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 
47 static void *
thd_sender_start(void * arg)48 thd_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 
TEST_BEGIN(test_mq_threaded)64 TEST_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 }
83 TEST_END
84 
85 int
main(void)86 main(void)
87 {
88 
89 	return (test(
90 	    test_mq_basic,
91 	    test_mq_threaded));
92 }
93 
94