src/jemalloc/test/unit/mq.c

   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 };
  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 *
  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 *
  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
  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
  86 main(void)
  87 {
  88
  89         return (test(
  90             test_mq_basic,
  91             test_mq_threaded));
  92 }
  93