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/darray.h"
  22
  23 #define noreturn __attribute__((noreturn))
  24
  25 void die(const char *, ...)
  26         __attribute__ ((format (printf, 1, 2))) noreturn;
  27 char *mprintf(const char *, ...)
  28         __attribute__ ((format (printf, 1, 2)));
  29 void *xcalloc(size_t, size_t);
  30 void *xmalloc(size_t);
  31 void *xrealloc(void *, size_t);
  32 void xpread(int, void *, size_t, off_t);
  33 void xpwrite(int, const void *, size_t, off_t, const char *);
  34 struct stat xfstatat(int, const char *, int);
  35 struct stat xfstat(int);
  36 struct stat xstat(const char *);
  37
  38 #define xopenat(_dirfd, _path, ...)                                     \
  39 ({                                                                      \
  40         int _fd = openat((_dirfd), (_path), __VA_ARGS__);               \
  41         if (_fd < 0)                                                    \
  42                 die("Error opening %s: %m", (_path));                   \
  43         _fd;                                                            \
  44 })
  45
  46 #define xopen(...)      xopenat(AT_FDCWD, __VA_ARGS__)
  47
  48 #define xioctl(_fd, _nr, ...)                                           \
  49 ({                                                                      \
  50         int _ret = ioctl((_fd), (_nr), ##__VA_ARGS__);                  \
  51         if (_ret < 0)                                                   \
  52                 die(#_nr " ioctl error: %m");                           \
  53         _ret;                                                           \
  54 })
  55
  56 void write_file_str(int, const char *, const char *);
  57 char *read_file_str(int, const char *);
  58 u64 read_file_u64(int, const char *);
  59
  60 ssize_t read_string_list_or_die(const char *, const char * const[],
  61                                 const char *);
  62
  63 u64 get_size(const char *, int);
  64 unsigned get_blocksize(const char *, int);
  65 int open_for_format(const char *, bool);
  66
  67 bool ask_yn(void);
  68
  69 struct range {
  70         u64             start;
  71         u64             end;
  72 };
  73
  74 typedef DARRAY(struct range) ranges;
  75
  76 static inline void range_add(ranges *data, u64 offset, u64 size)
  77 {
  78         darray_push(data, ((struct range) {
  79                 .start = offset,
  80                 .end = offset + size
  81         }));
  82 }
  83
  84 void ranges_sort_merge(ranges *);
  85 void ranges_roundup(ranges *, unsigned);
  86 void ranges_rounddown(ranges *, unsigned);
  87
  88 struct hole_iter {
  89         ranges          r;
  90         size_t          idx;
  91         u64             end;
  92 };
  93
  94 static inline struct range hole_iter_next(struct hole_iter *iter)
  95 {
  96         struct range r = {
  97                 .start  = iter->idx ? iter->r.data[iter->idx - 1].end : 0,
  98                 .end    = iter->idx < iter->r.nr
  99                         ? iter->r.data[iter->idx].start : iter->end,
 100         };
 101
 102         BUG_ON(r.start > r.end);
 103
 104         iter->idx++;
 105         return r;
 106 }
 107
 108 #define for_each_hole(_iter, _ranges, _end, _i)                         \
 109         for (_iter = (struct hole_iter) { .r = _ranges, .end = _end };  \
 110              (_iter.idx <= _iter.r.nr &&                                \
 111               (_i = hole_iter_next(&_iter), true));)
 112
 113 #include <linux/fiemap.h>
 114
 115 struct fiemap_iter {
 116         struct fiemap           f;
 117         struct fiemap_extent    fe[1024];
 118         unsigned                idx;
 119         int                     fd;
 120 };
 121
 122 static inline void fiemap_iter_init(struct fiemap_iter *iter, int fd)
 123 {
 124         memset(iter, 0, sizeof(*iter));
 125
 126         iter->f.fm_extent_count = ARRAY_SIZE(iter->fe);
 127         iter->f.fm_length       = FIEMAP_MAX_OFFSET;
 128         iter->fd                = fd;
 129 }
 130
 131 struct fiemap_extent fiemap_iter_next(struct fiemap_iter *);
 132
 133 #define fiemap_for_each(fd, iter, extent)                               \
 134         for (fiemap_iter_init(&iter, fd);                               \
 135              (extent = fiemap_iter_next(&iter)).fe_length;)
 136
 137 char *strcmp_prefix(char *, const char *);
 138
 139 u32 crc32c(u32, const void *, size_t);
 140
 141 char *dev_to_name(dev_t);
 142 char *dev_to_path(dev_t);
 143 struct mntent *dev_to_mount(char *);
 144 int dev_mounted(char *);
 145
 146 #define args_shift(_nr)                                                 \
 147 do {                                                                    \
 148         unsigned _n = min((_nr), argc);                                 \
 149         argc -= _n;                                                     \
 150         argv += _n;                                                     \
 151 } while (0)
 152
 153 #define arg_pop()                                                       \
 154 ({                                                                      \
 155         char *_ret = argc ? argv[0] : NULL;                             \
 156         if (_ret)                                                       \
 157                 args_shift(1);                                          \
 158         _ret;                                                           \
 159 })
 160
 161 struct bpos bpos_parse(char *);
 162
 163 #endif /* _TOOLS_UTIL_H */