git.sesse.net Git - bcachefs-tools-debian/blob - libbcachefs/darray.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #ifndef _BCACHEFS_DARRAY_H
   3 #define _BCACHEFS_DARRAY_H
   4
   5 /*
   6  * Dynamic arrays:
   7  *
   8  * Inspired by CCAN's darray
   9  */
  10
  11 #include <linux/slab.h>
  12
  13 #define DARRAY(type)                                                    \
  14 struct {                                                                \
  15         size_t nr, size;                                                \
  16         type *data;                                                     \
  17 }
  18
  19 typedef DARRAY(void) darray_void;
  20 typedef DARRAY(char *) darray_str;
  21
  22 int __bch2_darray_resize(darray_void *, size_t, size_t, gfp_t);
  23
  24 static inline int __darray_resize(darray_void *d, size_t element_size,
  25                                   size_t new_size, gfp_t gfp)
  26 {
  27         return unlikely(new_size > d->size)
  28                 ? __bch2_darray_resize(d, element_size, new_size, gfp)
  29                 : 0;
  30 }
  31
  32 #define darray_resize_gfp(_d, _new_size, _gfp)                          \
  33         __darray_resize((darray_void *) (_d), sizeof((_d)->data[0]), (_new_size), _gfp)
  34
  35 #define darray_resize(_d, _new_size)                                    \
  36         darray_resize_gfp(_d, _new_size, GFP_KERNEL)
  37
  38 static inline int __darray_make_room(darray_void *d, size_t t_size, size_t more, gfp_t gfp)
  39 {
  40         return __darray_resize(d, t_size, d->nr + more, gfp);
  41 }
  42
  43 #define darray_make_room_gfp(_d, _more, _gfp)                           \
  44         __darray_make_room((darray_void *) (_d), sizeof((_d)->data[0]), (_more), _gfp)
  45
  46 #define darray_make_room(_d, _more)                                     \
  47         darray_make_room_gfp(_d, _more, GFP_KERNEL)
  48
  49 #define darray_room(_d)         ((_d).size - (_d).nr)
  50
  51 #define darray_top(_d)          ((_d).data[(_d).nr])
  52
  53 #define darray_push_gfp(_d, _item, _gfp)                                \
  54 ({                                                                      \
  55         int _ret = darray_make_room_gfp((_d), 1, _gfp);                 \
  56                                                                         \
  57         if (!_ret)                                                      \
  58                 (_d)->data[(_d)->nr++] = (_item);                       \
  59         _ret;                                                           \
  60 })
  61
  62 #define darray_push(_d, _item)  darray_push_gfp(_d, _item, GFP_KERNEL)
  63
  64 #define darray_pop(_d)          ((_d)->data[--(_d)->nr])
  65
  66 #define darray_first(_d)        ((_d).data[0])
  67 #define darray_last(_d)         ((_d).data[(_d).nr - 1])
  68
  69 #define darray_insert_item(_d, pos, _item)                              \
  70 ({                                                                      \
  71         size_t _pos = (pos);                                            \
  72         int _ret = darray_make_room((_d), 1);                           \
  73                                                                         \
  74         if (!_ret)                                                      \
  75                 array_insert_item((_d)->data, (_d)->nr, _pos, (_item)); \
  76         _ret;                                                           \
  77 })
  78
  79 #define darray_remove_item(_d, _pos)                                    \
  80         array_remove_item((_d)->data, (_d)->nr, (_pos) - (_d)->data)
  81
  82 #define __darray_for_each(_d, _i)                                               \
  83         for ((_i) = (_d).data; _i < (_d).data + (_d).nr; _i++)
  84
  85 #define darray_for_each(_d, _i)                                         \
  86         for (typeof(&(_d).data[0]) _i = (_d).data; _i < (_d).data + (_d).nr; _i++)
  87
  88 #define darray_for_each_reverse(_d, _i)                                 \
  89         for (typeof(&(_d).data[0]) _i = (_d).data + (_d).nr - 1; _i >= (_d).data; --_i)
  90
  91 #define darray_init(_d)                                                 \
  92 do {                                                                    \
  93         (_d)->data = NULL;                                              \
  94         (_d)->nr = (_d)->size = 0;                                      \
  95 } while (0)
  96
  97 #define darray_exit(_d)                                                 \
  98 do {                                                                    \
  99         kvfree((_d)->data);                                             \
 100         darray_init(_d);                                                \
 101 } while (0)
 102
 103 #endif /* _BCACHEFS_DARRAY_H */