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