12 #include <linux/bug.h>
13 #include <linux/byteorder.h>
14 #include <linux/kernel.h>
15 #include <linux/log2.h>
16 #include <linux/string.h>
17 #include <linux/types.h>
18 #include "ccan/darray/darray.h"
20 #define die(arg, ...) \
22 fprintf(stderr, arg "\n", ##__VA_ARGS__); \
26 #define mprintf(...) \
29 asprintf(&_str, __VA_ARGS__); \
33 static inline void *xcalloc(size_t count, size_t size)
35 void *p = calloc(count, size);
38 die("insufficient memory");
43 static inline void *xmalloc(size_t size)
45 void *p = malloc(size);
48 die("insufficient memory");
54 static inline void xpread(int fd, void *buf, size_t count, off_t offset)
56 ssize_t r = pread(fd, buf, count, offset);
59 die("read error (ret %zi)", r);
62 static inline void xpwrite(int fd, const void *buf, size_t count, off_t offset)
64 ssize_t r = pwrite(fd, buf, count, offset);
67 die("write error (ret %zi err %s)", r, strerror(errno));
70 #define xopenat(_dirfd, _path, ...) \
72 int _fd = openat((_dirfd), (_path), __VA_ARGS__); \
74 die("Error opening %s: %s", (_path), strerror(errno)); \
78 #define xopen(...) xopenat(AT_FDCWD, __VA_ARGS__)
80 static inline struct stat xfstatat(int dirfd, const char *path, int flags)
83 if (fstatat(dirfd, path, &stat, flags))
84 die("stat error: %s", strerror(errno));
88 static inline struct stat xfstat(int fd)
92 die("stat error: %s", strerror(errno));
96 #define xioctl(_fd, _nr, ...) \
98 if (ioctl((_fd), (_nr), ##__VA_ARGS__)) \
99 die(#_nr " ioctl error: %s", strerror(errno)); \
108 struct units_buf __pr_units(u64, enum units);
114 #define pr_units(_v, _u) __pr_units(_v, _u).b
116 char *read_file_str(int, const char *);
117 u64 read_file_u64(int, const char *);
119 ssize_t read_string_list_or_die(const char *, const char * const[],
122 u64 get_size(const char *, int);
123 unsigned get_blocksize(const char *, int);
124 int open_for_format(const char *, bool);
126 int bcachectl_open(void);
128 struct bcache_handle {
133 struct bcache_handle bcache_fs_open(const char *);
142 typedef darray(struct range) ranges;
144 static inline void range_add(ranges *data, u64 offset, u64 size)
146 darray_append(*data, (struct range) {
152 void ranges_sort_merge(ranges *);
153 void ranges_roundup(ranges *, unsigned);
154 void ranges_rounddown(ranges *, unsigned);
162 static inline struct range hole_iter_next(struct hole_iter *iter)
165 .start = iter->idx ? iter->r.item[iter->idx - 1].end : 0,
166 .end = iter->idx < iter->r.size
167 ? iter->r.item[iter->idx].start : iter->end,
170 BUG_ON(r.start > r.end);
176 #define for_each_hole(_iter, _ranges, _end, _i) \
177 for (_iter = (struct hole_iter) { .r = _ranges, .end = _end }; \
178 (_iter.idx <= _iter.r.size && \
179 (_i = hole_iter_next(&_iter), true));)
181 #include <linux/fiemap.h>
185 struct fiemap_extent fe[1024];
190 static inline void fiemap_iter_init(struct fiemap_iter *iter, int fd)
192 memset(iter, 0, sizeof(*iter));
194 iter->f.fm_extent_count = ARRAY_SIZE(iter->fe);
195 iter->f.fm_length = FIEMAP_MAX_OFFSET;
199 struct fiemap_extent fiemap_iter_next(struct fiemap_iter *);
201 #define fiemap_for_each(fd, iter, extent) \
202 for (fiemap_iter_init(&iter, fd); \
203 (extent = fiemap_iter_next(&iter)).fe_length;)
205 const char *strcmp_prefix(const char *, const char *);
207 unsigned hatoi_validate(const char *, const char *);
209 #endif /* _TOOLS_UTIL_H */