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

   1 /*
   2     queue.h -- Thread-safe queue
   3     Copyright (C) 2014 Guus Sliepen <guus@meshlink.io>
   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 2 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 along
  16     with this program; if not, write to the Free Software Foundation, Inc.,
  17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 */
  19
  20 #ifndef MESHLINK_QUEUE_H
  21 #define MESHLINK_QUEUE_H
  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 bool meshlink_queue_push(meshlink_queue_t *queue, void *data) {
  40         meshlink_queue_item_t *item = malloc(sizeof *item);
  41         if(!item)
  42                 return false;
  43         item->data = data;
  44         item->next = NULL;
  45         pthread_mutex_lock(&queue->mutex);
  46         if(!queue->tail)
  47                 queue->head = queue->tail = item;
  48         else
  49                 queue->tail = queue->tail->next = item;
  50         pthread_mutex_unlock(&queue->mutex);
  51         return true;
  52 }
  53
  54 static inline void *meshlink_queue_pop(meshlink_queue_t *queue) {
  55         meshlink_queue_item_t *item;
  56         void *data;
  57         pthread_mutex_lock(&queue->mutex);
  58         if((item = queue->head)) {
  59                 queue->head = item->next;
  60                 if(!queue->head)
  61                         queue->tail = NULL;
  62         }
  63         pthread_mutex_unlock(&queue->mutex);
  64         data = item ? item->data : NULL;
  65         free(item);
  66         return data;
  67 }
  68
  69 #endif