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

   1 #ifndef _TOOLS_UTIL_H
   2 #define _TOOLS_UTIL_H
   3
   4 #include <errno.h>
   5 #include <mntent.h>
   6 #include <stdbool.h>
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <sys/ioctl.h>
  10 #include <sys/stat.h>
  11 #include <sys/types.h>
  12 #include <unistd.h>
  13
  14 #include <linux/bug.h>
  15 #include <linux/byteorder.h>
  16 #include <linux/kernel.h>
  17 #include <linux/log2.h>
  18 #include <linux/string.h>
  19 #include <linux/types.h>
  20 #include <linux/uuid.h>
  21 #include "libbcachefs/bcachefs.h"
  22 #include "libbcachefs/bbpos.h"
  23 #include "libbcachefs/darray.h"
  24
  25 #define noreturn __attribute__((noreturn))
  26
  27 void die(const char *, ...)
  28         __attribute__ ((format (printf, 1, 2))) noreturn;
  29 char *mprintf(const char *, ...)
  30         __attribute__ ((format (printf, 1, 2)));
  31 void xpread(int, void *, size_t, off_t);
  32 void xpwrite(int, const void *, size_t, off_t, const char *);
  33 struct stat xfstatat(int, const char *, int);
  34 struct stat xfstat(int);
  35 struct stat xstat(const char *);
  36
  37 static inline void *xmalloc(size_t size)
  38 {
  39         void *p = malloc(size);
  40
  41         if (!p)
  42                 die("insufficient memory");
  43
  44         memset(p, 0, size);
  45         return p;
  46 }
  47
  48 static inline void *xcalloc(size_t count, size_t size)
  49 {
  50         void *p = calloc(count, size);
  51
  52         if (!p)
  53                 die("insufficient memory");
  54
  55         return p;
  56 }
  57
  58 static inline void *xrealloc(void *p, size_t size)
  59 {
  60         p = realloc(p, size);
  61         if (!p)
  62                 die("insufficient memory");
  63
  64         return p;
  65 }
  66
  67 #define xopenat(_dirfd, _path, ...)                                     \
  68 ({                                                                      \
  69         int _fd = openat((_dirfd), (_path), __VA_ARGS__);               \
  70         if (_fd < 0)                                                    \
  71                 die("Error opening %s: %m", (_path));                   \
  72         _fd;                                                            \
  73 })
  74
  75 #define xopen(...)      xopenat(AT_FDCWD, __VA_ARGS__)
  76
  77 #define xioctl(_fd, _nr, ...)                                           \
  78 ({                                                                      \
  79         int _ret = ioctl((_fd), (_nr), ##__VA_ARGS__);                  \
  80         if (_ret < 0)                                                   \
  81                 die(#_nr " ioctl error: %m");                           \
  82         _ret;                                                           \
  83 })
  84
  85 void write_file_str(int, const char *, const char *);
  86 char *read_file_str(int, const char *);
  87 u64 read_file_u64(int, const char *);
  88
  89 ssize_t read_string_list_or_die(const char *, const char * const[],
  90                                 const char *);
  91
  92 u64 get_size(int);
  93 unsigned get_blocksize(int);
  94 struct dev_opts;
  95 int open_for_format(struct dev_opts *, bool);
  96
  97 bool ask_yn(void);
  98
  99 struct range {
 100         u64             start;
 101         u64             end;
 102 };
 103
 104 typedef DARRAY(struct range) ranges;
 105
 106 static inline void range_add(ranges *data, u64 offset, u64 size)
 107 {
 108         darray_push(data, ((struct range) {
 109                 .start = offset,
 110                 .end = offset + size
 111         }));
 112 }
 113
 114 void ranges_sort_merge(ranges *);
 115 void ranges_roundup(ranges *, unsigned);
 116 void ranges_rounddown(ranges *, unsigned);
 117
 118 struct hole_iter {
 119         ranges          r;
 120         size_t          idx;
 121         u64             end;
 122 };
 123
 124 static inline struct range hole_iter_next(struct hole_iter *iter)
 125 {
 126         struct range r = {
 127                 .start  = iter->idx ? iter->r.data[iter->idx - 1].end : 0,
 128                 .end    = iter->idx < iter->r.nr
 129                         ? iter->r.data[iter->idx].start : iter->end,
 130         };
 131
 132         BUG_ON(r.start > r.end);
 133
 134         iter->idx++;
 135         return r;
 136 }
 137
 138 #define for_each_hole(_iter, _ranges, _end, _i)                         \
 139         for (_iter = (struct hole_iter) { .r = _ranges, .end = _end };  \
 140              (_iter.idx <= _iter.r.nr &&                                \
 141               (_i = hole_iter_next(&_iter), true));)
 142
 143 #include <linux/fiemap.h>
 144
 145 struct fiemap_iter {
 146         struct fiemap           *f;
 147         unsigned                idx;
 148         int                     fd;
 149 };
 150
 151 static inline void fiemap_iter_init(struct fiemap_iter *iter, int fd)
 152 {
 153         memset(iter, 0, sizeof(*iter));
 154
 155         iter->f = xmalloc(sizeof(struct fiemap) +
 156                           sizeof(struct fiemap_extent) * 1024);
 157
 158         iter->f->fm_extent_count        = 1024;
 159         iter->f->fm_length      = FIEMAP_MAX_OFFSET;
 160         iter->fd                = fd;
 161 }
 162
 163 static inline void fiemap_iter_exit(struct fiemap_iter *iter)
 164 {
 165         free(iter->f);
 166         memset(iter, 0, sizeof(*iter));
 167 }
 168
 169 struct fiemap_extent fiemap_iter_next(struct fiemap_iter *);
 170
 171 #define fiemap_for_each(fd, iter, extent)                               \
 172         for (fiemap_iter_init(&iter, fd);                               \
 173              (extent = fiemap_iter_next(&iter)).fe_length;)
 174
 175 char *strcmp_prefix(char *, const char *);
 176
 177 u32 crc32c(u32, const void *, size_t);
 178
 179 char *dev_to_name(dev_t);
 180 char *dev_to_path(dev_t);
 181 struct mntent *dev_to_mount(char *);
 182 int dev_mounted(char *);
 183
 184 #define args_shift(_nr)                                                 \
 185 do {                                                                    \
 186         unsigned _n = min((_nr), argc);                                 \
 187         argc -= _n;                                                     \
 188         argv += _n;                                                     \
 189 } while (0)
 190
 191 #define arg_pop()                                                       \
 192 ({                                                                      \
 193         char *_ret = argc ? argv[0] : NULL;                             \
 194         if (_ret)                                                       \
 195                 args_shift(1);                                          \
 196         _ret;                                                           \
 197 })
 198
 199 struct bpos bpos_parse(char *);
 200 struct bbpos bbpos_parse(char *);
 201
 202 darray_str get_or_split_cmdline_devs(int argc, char *argv[]);
 203
 204 #endif /* _TOOLS_UTIL_H */