92 lines
1.8 KiB
C
92 lines
1.8 KiB
C
|
#include "test/jemalloc_test.h"
|
||
|
|
|
||
|
|
#define NSENDERS 3
|
||
|
|
#define NMSGS 100000
|
||
|
|
|
||
|
|
typedef struct mq_msg_s mq_msg_t;
|
||
|
|
struct mq_msg_s {
|
||
|
|
mq_msg(mq_msg_t) link;
|
||
|
|
};
|
||
|
|
mq_gen(static, mq_, mq_t, mq_msg_t, link)
|
||
|
|
|
||
|
|
TEST_BEGIN(test_mq_basic)
|
||
|
|
{
|
||
|
|
mq_t mq;
|
||
|
|
mq_msg_t msg;
|
||
|
|
|
||
|
|
assert_false(mq_init(&mq), "Unexpected mq_init() failure");
|
||
|
|
assert_u_eq(mq_count(&mq), 0, "mq should be empty");
|
||
|
|
assert_ptr_null(mq_tryget(&mq),
|
||
|
|
"mq_tryget() should fail when the queue is empty");
|
||
|
|
|
||
|
|
mq_put(&mq, &msg);
|
||
|
|
assert_u_eq(mq_count(&mq), 1, "mq should contain one message");
|
||
|
|
assert_ptr_eq(mq_tryget(&mq), &msg, "mq_tryget() should return msg");
|
||
|
|
|
||
|
|
mq_put(&mq, &msg);
|
||
|
|
assert_ptr_eq(mq_get(&mq), &msg, "mq_get() should return msg");
|
||
|
|
|
||
|
|
mq_fini(&mq);
|
||
|
|
}
|
||
|
|
TEST_END
|
||
|
|
|
||
|
|
static void *
|
||
|
|
thd_receiver_start(void *arg)
|
||
|
|
{
|
||
|
|
mq_t *mq = (mq_t *)arg;
|
||
|
|
unsigned i;
|
||
|
|
|
||
|
|
for (i = 0; i < (NSENDERS * NMSGS); i++) {
|
||
|
|
mq_msg_t *msg = mq_get(mq);
|
||
|
|
assert_ptr_not_null(msg, "mq_get() should never return NULL");
|
||
|
|
assert_d_eq(jet_dallocm(msg, 0), ALLOCM_SUCCESS,
|
||
|
|
"Unexpected dallocm() failure");
|
||
|
|
}
|
||
|
|
return (NULL);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void *
|
||
|
|
thd_sender_start(void *arg)
|
||
|
|
{
|
||
|
|
mq_t *mq = (mq_t *)arg;
|
||
|
|
unsigned i;
|
||
|
|
|
||
|
|
for (i = 0; i < NMSGS; i++) {
|
||
|
|
mq_msg_t *msg;
|
||
|
|
assert_d_eq(jet_allocm((void **)&msg, NULL, sizeof(mq_msg_t),
|
||
|
|
0), ALLOCM_SUCCESS, "Unexpected allocm() failure");
|
||
|
|
mq_put(mq, msg);
|
||
|
|
}
|
||
|
|
return (NULL);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_BEGIN(test_mq_threaded)
|
||
|
|
{
|
||
|
|
mq_t mq;
|
||
|
|
thd_t receiver;
|
||
|
|
thd_t senders[NSENDERS];
|
||
|
|
unsigned i;
|
||
|
|
|
||
|
|
assert_false(mq_init(&mq), "Unexpected mq_init() failure");
|
||
|
|
|
||
|
|
thd_create(&receiver, thd_receiver_start, (void *)&mq);
|
||
|
|
for (i = 0; i < NSENDERS; i++)
|
||
|
|
thd_create(&senders[i], thd_sender_start, (void *)&mq);
|
||
|
|
|
||
|
|
thd_join(receiver, NULL);
|
||
|
|
for (i = 0; i < NSENDERS; i++)
|
||
|
|
thd_join(senders[i], NULL);
|
||
|
|
|
||
|
|
mq_fini(&mq);
|
||
|
|
}
|
||
|
|
TEST_END
|
||
|
|
|
||
|
|
int
|
||
|
|
main(void)
|
||
|
|
{
|
||
|
|
return (test(
|
||
|
|
test_mq_basic,
|
||
|
|
test_mq_threaded));
|
||
|
|
}
|
||
|
|
|