git.meshlink.io Git - meshlink/blob - src/meshlink_queue.h

   1 #ifndef MESHLINK_QUEUE_H
   2 #define MESHLINK_QUEUE_H
   3
   4 /*
   5     queue.h -- Thread-safe queue
   6     Copyright (C) 2014, 2017 Guus Sliepen <guus@meshlink.io>
   7
   8     This program is free software; you can redistribute it and/or modify
   9     it under the terms of the GNU General Public License as published by
  10     the Free Software Foundation; either version 2 of the License, or
  11     (at your option) any later version.
  12
  13     This program is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License along
  19     with this program; if not, write to the Free Software Foundation, Inc.,
  20     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21 */
  22
  23 #include <pthread.h>
  24 #include <stdbool.h>
  25 #include <stddef.h>
  26 #include <unistd.h>
  27
  28 typedef struct meshlink_queue {
  29         struct meshlink_queue_item *head;
  30         struct meshlink_queue_item *tail;
  31         pthread_mutex_t mutex;
  32 } meshlink_queue_t;
  33
  34 typedef struct meshlink_queue_item {
  35         void *data;
  36         struct meshlink_queue_item *next;
  37 } meshlink_queue_item_t;
  38
  39 static inline void meshlink_queue_init(meshlink_queue_t *queue) {
  40         queue->head = NULL;
  41         queue->tail = NULL;
  42         pthread_mutex_init(&queue->mutex, NULL);
  43 }
  44
  45 static inline void meshlink_queue_exit(meshlink_queue_t *queue) {
  46         pthread_mutex_destroy(&queue->mutex);
  47 }
  48
  49 static inline __attribute__((__warn_unused_result__)) bool meshlink_queue_push(meshlink_queue_t *queue, void *data) {
  50         meshlink_queue_item_t *item = malloc(sizeof(*item));
  51
  52         if(!item) {
  53                 return false;
  54         }
  55
  56         item->data = data;
  57         item->next = NULL;
  58
  59         if(pthread_mutex_lock(&queue->mutex) != 0) {
  60                 abort();
  61         }
  62
  63         if(!queue->tail) {
  64                 queue->head = queue->tail = item;
  65         } else {
  66                 queue->tail = queue->tail->next = item;
  67         }
  68
  69         pthread_mutex_unlock(&queue->mutex);
  70         return true;
  71 }
  72
  73 static inline __attribute__((__warn_unused_result__)) void *meshlink_queue_pop(meshlink_queue_t *queue) {
  74         meshlink_queue_item_t *item;
  75
  76         if(pthread_mutex_lock(&queue->mutex) != 0) {
  77                 abort();
  78         }
  79
  80         if((item = queue->head)) {
  81                 queue->head = item->next;
  82
  83                 if(!queue->head) {
  84                         queue->tail = NULL;
  85                 }
  86         }
  87
  88         pthread_mutex_unlock(&queue->mutex);
  89
  90         void *data = item ? item->data : NULL;
  91         free(item);
  92         return data;
  93 }
  94
  95 static inline __attribute__((__warn_unused_result__)) void *meshlink_queue_pop_cond(meshlink_queue_t *queue, pthread_cond_t *cond) {
  96         meshlink_queue_item_t *item;
  97
  98         if(pthread_mutex_lock(&queue->mutex) != 0) {
  99                 abort();
 100         }
 101
 102         while(!queue->head) {
 103                 pthread_cond_wait(cond, &queue->mutex);
 104         }
 105
 106         item = queue->head;
 107         queue->head = item->next;
 108
 109         if(!queue->head) {
 110                 queue->tail = NULL;
 111         }
 112
 113         pthread_mutex_unlock(&queue->mutex);
 114
 115         void *data = item->data;
 116         free(item);
 117         return data;
 118 }
 119
 120 #endif