test/spsc_queue_m.c

   1 /*
   2         libnbds
   3         Copyright (C) 2013 Paweł Dziepak
   4
   5         This program is free software: you can redistribute it and/or modify
   6         it under the terms of the GNU General Public License as published by
   7         the Free Software Foundation, either version 3 of the License, or
   8         (at your option) any later version.
   9
  10         This program is distributed in the hope that it will be useful,
  11         but WITHOUT ANY WARRANTY; without even the implied warranty of
  12         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13         GNU General Public License for more details.
  14
  15         You should have received a copy of the GNU General Public License
  16         along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #include <stdint.h>
  20 #include <stdlib.h>
  21
  22 #include <pthread.h>
  23
  24 #include "spsc_queue.h"
  25
  26 struct spsc_queue queue;
  27
  28 pthread_barrier_t start;
  29
  30 #define ITERATIONS      (1024 * 1024 * 8u)
  31
  32 void* thread_func(void* ptr)
  33 {
  34         uintptr_t i;
  35
  36         pthread_barrier_wait(&start);
  37
  38         for (i = 1; i < ITERATIONS; i++) {
  39                 do {
  40                         ptr = spsc_queue_dequeue(&queue);
  41                 } while (!ptr);
  42
  43                 if ((uintptr_t)ptr != i)
  44                         exit(1);
  45         }
  46
  47         return NULL;
  48 }
  49
  50 int main(int argc, char** argv)
  51 {
  52         uintptr_t i;
  53         int error;
  54         pthread_t thread;
  55
  56         (void)argc;
  57         (void)argv;
  58
  59         error = spsc_queue_init(&queue);
  60         if (error)
  61                 return 1;
  62
  63         if (spsc_queue_dequeue(&queue))
  64                 return 1;
  65
  66         pthread_barrier_init(&start, NULL, 2);
  67         pthread_create(&thread, NULL, thread_func, NULL);
  68
  69         pthread_barrier_wait(&start);
  70         for (i = 1; i < ITERATIONS; i++) {
  71                 error = spsc_queue_enqueue(&queue, (void*)i);
  72                 if (error)
  73                         return 1;
  74         }
  75
  76         pthread_join(thread, NULL);
  77
  78         if (spsc_queue_dequeue(&queue))
  79                 return 1;
  80
  81         spsc_queue_destroy(&queue);
  82 }