11 #include <sys/types.h>
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 "linux/darray.h"
25 #define noreturn __attribute__((noreturn))
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 *);
37 static inline void *xmalloc(size_t size)
39 void *p = malloc(size);
42 die("insufficient memory");
48 static inline void *xcalloc(size_t count, size_t size)
50 void *p = calloc(count, size);
53 die("insufficient memory");
58 static inline void *xrealloc(void *p, size_t size)
62 die("insufficient memory");
67 #define xopenat(_dirfd, _path, ...) \
69 int _fd = openat((_dirfd), (_path), __VA_ARGS__); \
71 die("Error opening %s: %m", (_path)); \
75 #define xopen(...) xopenat(AT_FDCWD, __VA_ARGS__)
77 #define xioctl(_fd, _nr, ...) \
79 int _ret = ioctl((_fd), (_nr), ##__VA_ARGS__); \
81 die(#_nr " ioctl error: %m"); \
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 *);
89 ssize_t read_string_list_or_die(const char *, const char * const[],
93 unsigned get_blocksize(int);
95 int open_for_format(struct dev_opts *, bool);
104 typedef DARRAY(struct range) ranges;
106 static inline void range_add(ranges *data, u64 offset, u64 size)
108 darray_push(data, ((struct range) {
114 void ranges_sort_merge(ranges *);
115 void ranges_roundup(ranges *, unsigned);
116 void ranges_rounddown(ranges *, unsigned);
124 static inline struct range hole_iter_next(struct hole_iter *iter)
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,
132 BUG_ON(r.start > r.end);
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));)
143 #include <linux/fiemap.h>
151 static inline void fiemap_iter_init(struct fiemap_iter *iter, int fd)
153 memset(iter, 0, sizeof(*iter));
155 iter->f = xmalloc(sizeof(struct fiemap) +
156 sizeof(struct fiemap_extent) * 1024);
158 iter->f->fm_extent_count = 1024;
159 iter->f->fm_length = FIEMAP_MAX_OFFSET;
163 static inline void fiemap_iter_exit(struct fiemap_iter *iter)
166 memset(iter, 0, sizeof(*iter));
169 struct fiemap_extent fiemap_iter_next(struct fiemap_iter *);
171 #define fiemap_for_each(fd, iter, extent) \
172 for (fiemap_iter_init(&iter, fd); \
173 (extent = fiemap_iter_next(&iter)).fe_length;)
175 char *strcmp_prefix(char *, const char *);
177 u32 crc32c(u32, const void *, size_t);
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 *);
184 #define args_shift(_nr) \
186 unsigned _n = min((_nr), argc); \
193 char *_ret = argc ? argv[0] : NULL; \
199 struct bpos bpos_parse(char *);
200 struct bbpos bbpos_parse(char *);
207 struct bbpos_range bbpos_range_parse(char *);
209 darray_str get_or_split_cmdline_devs(int argc, char *argv[]);
211 #endif /* _TOOLS_UTIL_H */