1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef _BCACHEFS_DARRAY_H
3 #define _BCACHEFS_DARRAY_H
8 * Inspired by CCAN's darray
12 #include <linux/slab.h>
14 #define DARRAY(type) \
20 typedef DARRAY(void) darray_void;
22 static inline int __darray_make_room(darray_void *d, size_t t_size, size_t more, gfp_t gfp)
24 if (d->nr + more > d->size) {
25 size_t new_size = roundup_pow_of_two(d->nr + more);
26 void *data = krealloc_array(d->data, new_size, t_size, gfp);
38 #define darray_make_room_gfp(_d, _more, _gfp) \
39 __darray_make_room((darray_void *) (_d), sizeof((_d)->data[0]), (_more), _gfp)
41 #define darray_make_room(_d, _more) \
42 darray_make_room_gfp(_d, _more, GFP_KERNEL)
44 #define darray_top(_d) ((_d).data[(_d).nr])
46 #define darray_push_gfp(_d, _item, _gfp) \
48 int _ret = darray_make_room_gfp((_d), 1, _gfp); \
51 (_d)->data[(_d)->nr++] = (_item); \
55 #define darray_push(_d, _item) darray_push_gfp(_d, _item, GFP_KERNEL)
57 #define darray_pop(_d) ((_d)->data[--(_d)->nr])
59 #define darray_first(_d) ((_d).data[0])
60 #define darray_last(_d) ((_d).data[(_d).nr - 1])
62 #define darray_insert_item(_d, _pos, _item) \
64 size_t pos = (_pos); \
65 int _ret = darray_make_room((_d), 1); \
68 array_insert_item((_d)->data, (_d)->nr, pos, (_item)); \
72 #define darray_for_each(_d, _i) \
73 for (_i = (_d).data; _i < (_d).data + (_d).nr; _i++)
75 #define darray_init(_d) \
78 (_d)->nr = (_d)->size = 0; \
81 #define darray_exit(_d) \
87 #endif /* _BCACHEFS_DARRAY_H */