1 | /* SPDX-License-Identifier: GPL-2.0 */ |
2 | #ifndef _BCACHEFS_FIFO_H |
3 | #define _BCACHEFS_FIFO_H |
4 | |
5 | #include "util.h" |
6 | |
7 | #define FIFO(type) \ |
8 | struct { \ |
9 | size_t front, back, size, mask; \ |
10 | type *data; \ |
11 | } |
12 | |
13 | #define DECLARE_FIFO(type, name) FIFO(type) name |
14 | |
15 | #define fifo_buf_size(fifo) \ |
16 | ((fifo)->size \ |
17 | ? roundup_pow_of_two((fifo)->size) * sizeof((fifo)->data[0]) \ |
18 | : 0) |
19 | |
20 | #define init_fifo(fifo, _size, _gfp) \ |
21 | ({ \ |
22 | (fifo)->front = (fifo)->back = 0; \ |
23 | (fifo)->size = (_size); \ |
24 | (fifo)->mask = (fifo)->size \ |
25 | ? roundup_pow_of_two((fifo)->size) - 1 \ |
26 | : 0; \ |
27 | (fifo)->data = kvmalloc(fifo_buf_size(fifo), (_gfp)); \ |
28 | }) |
29 | |
30 | #define free_fifo(fifo) \ |
31 | do { \ |
32 | kvfree((fifo)->data); \ |
33 | (fifo)->data = NULL; \ |
34 | } while (0) |
35 | |
36 | #define fifo_swap(l, r) \ |
37 | do { \ |
38 | swap((l)->front, (r)->front); \ |
39 | swap((l)->back, (r)->back); \ |
40 | swap((l)->size, (r)->size); \ |
41 | swap((l)->mask, (r)->mask); \ |
42 | swap((l)->data, (r)->data); \ |
43 | } while (0) |
44 | |
45 | #define fifo_move(dest, src) \ |
46 | do { \ |
47 | typeof(*((dest)->data)) _t; \ |
48 | while (!fifo_full(dest) && \ |
49 | fifo_pop(src, _t)) \ |
50 | fifo_push(dest, _t); \ |
51 | } while (0) |
52 | |
53 | #define fifo_used(fifo) (((fifo)->back - (fifo)->front)) |
54 | #define fifo_free(fifo) ((fifo)->size - fifo_used(fifo)) |
55 | |
56 | #define fifo_empty(fifo) ((fifo)->front == (fifo)->back) |
57 | #define fifo_full(fifo) (fifo_used(fifo) == (fifo)->size) |
58 | |
59 | #define fifo_peek_front(fifo) ((fifo)->data[(fifo)->front & (fifo)->mask]) |
60 | #define fifo_peek_back(fifo) ((fifo)->data[((fifo)->back - 1) & (fifo)->mask]) |
61 | |
62 | #define fifo_entry_idx_abs(fifo, p) \ |
63 | ((((p) >= &fifo_peek_front(fifo) \ |
64 | ? (fifo)->front : (fifo)->back) & ~(fifo)->mask) + \ |
65 | (((p) - (fifo)->data))) |
66 | |
67 | #define fifo_entry_idx(fifo, p) (((p) - &fifo_peek_front(fifo)) & (fifo)->mask) |
68 | #define fifo_idx_entry(fifo, i) ((fifo)->data[((fifo)->front + (i)) & (fifo)->mask]) |
69 | |
70 | #define fifo_push_back_ref(f) \ |
71 | (fifo_full((f)) ? NULL : &(f)->data[(f)->back++ & (f)->mask]) |
72 | |
73 | #define fifo_push_front_ref(f) \ |
74 | (fifo_full((f)) ? NULL : &(f)->data[--(f)->front & (f)->mask]) |
75 | |
76 | #define fifo_push_back(fifo, new) \ |
77 | ({ \ |
78 | typeof((fifo)->data) _r = fifo_push_back_ref(fifo); \ |
79 | if (_r) \ |
80 | *_r = (new); \ |
81 | _r != NULL; \ |
82 | }) |
83 | |
84 | #define fifo_push_front(fifo, new) \ |
85 | ({ \ |
86 | typeof((fifo)->data) _r = fifo_push_front_ref(fifo); \ |
87 | if (_r) \ |
88 | *_r = (new); \ |
89 | _r != NULL; \ |
90 | }) |
91 | |
92 | #define fifo_pop_front(fifo, i) \ |
93 | ({ \ |
94 | bool _r = !fifo_empty((fifo)); \ |
95 | if (_r) \ |
96 | (i) = (fifo)->data[(fifo)->front++ & (fifo)->mask]; \ |
97 | _r; \ |
98 | }) |
99 | |
100 | #define fifo_pop_back(fifo, i) \ |
101 | ({ \ |
102 | bool _r = !fifo_empty((fifo)); \ |
103 | if (_r) \ |
104 | (i) = (fifo)->data[--(fifo)->back & (fifo)->mask]; \ |
105 | _r; \ |
106 | }) |
107 | |
108 | #define fifo_push_ref(fifo) fifo_push_back_ref(fifo) |
109 | #define fifo_push(fifo, i) fifo_push_back(fifo, (i)) |
110 | #define fifo_pop(fifo, i) fifo_pop_front(fifo, (i)) |
111 | #define fifo_peek(fifo) fifo_peek_front(fifo) |
112 | |
113 | #define fifo_for_each_entry(_entry, _fifo, _iter) \ |
114 | for (typecheck(typeof((_fifo)->front), _iter), \ |
115 | (_iter) = (_fifo)->front; \ |
116 | ((_iter != (_fifo)->back) && \ |
117 | (_entry = (_fifo)->data[(_iter) & (_fifo)->mask], true)); \ |
118 | (_iter)++) |
119 | |
120 | #define fifo_for_each_entry_ptr(_ptr, _fifo, _iter) \ |
121 | for (typecheck(typeof((_fifo)->front), _iter), \ |
122 | (_iter) = (_fifo)->front; \ |
123 | ((_iter != (_fifo)->back) && \ |
124 | (_ptr = &(_fifo)->data[(_iter) & (_fifo)->mask], true)); \ |
125 | (_iter)++) |
126 | |
127 | #endif /* _BCACHEFS_FIFO_H */ |
128 | |