]> git.meshlink.io Git - meshlink-tiny/blob - src/meshlink_queue.h
Add a metering test.
[meshlink-tiny] / 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