mq.c revision 0f4f1efd94d33a4bbf766d3d4e7e349fa7c0d3b9 
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		assert_d_eq(jet_dallocm(msg, 0), ALLOCM_SUCCESS,
43		    "Unexpected dallocm() failure");
44	}
45	return (NULL);
46}
47
48static void *
49thd_sender_start(void *arg)
50{
51	mq_t *mq = (mq_t *)arg;
52	unsigned i;
53
54	for (i = 0; i < NMSGS; i++) {
55		mq_msg_t *msg;
56		assert_d_eq(jet_allocm((void **)&msg, NULL, sizeof(mq_msg_t),
57		    0), ALLOCM_SUCCESS, "Unexpected allocm() failure");
58		mq_put(mq, msg);
59	}
60	return (NULL);
61}
62
63TEST_BEGIN(test_mq_threaded)
64{
65	mq_t mq;
66	thd_t receiver;
67	thd_t senders[NSENDERS];
68	unsigned i;
69
70	assert_false(mq_init(&mq), "Unexpected mq_init() failure");
71
72	thd_create(&receiver, thd_receiver_start, (void *)&mq);
73	for (i = 0; i < NSENDERS; i++)
74		thd_create(&senders[i], thd_sender_start, (void *)&mq);
75
76	thd_join(receiver, NULL);
77	for (i = 0; i < NSENDERS; i++)
78		thd_join(senders[i], NULL);
79
80	mq_fini(&mq);
81}
82TEST_END
83
84int
85main(void)
86{
87	return (test(
88	    test_mq_basic,
89	    test_mq_threaded));
90}
91
92