From 39eff20bd938bae630270a3f06b17f787663826c Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@beefcake.parabola.nu>
Date: Fri, 18 May 2018 12:27:35 -0400
Subject: use return codes from fiemap

---
 lib/extent-map.c | 10 ++++++----
 lib/extent-map.h |  2 +-
 2 files changed, 7 insertions(+), 5 deletions(-)

(limited to 'lib')

diff --git a/lib/extent-map.c b/lib/extent-map.c
index b698fa1..297448c 100644
--- a/lib/extent-map.c
+++ b/lib/extent-map.c
@@ -2,13 +2,15 @@
 #include <error.h>        /* for error(3gnu) */
 #include <linux/fiemap.h> /* for all of the other fiemap stuff */
 #include <linux/fs.h>     /* for FS_IOC_FIEMAP */
-#include <stdlib.h>       /* for calloc(3p), exit(3p), EXIT_SUCCESS, EXIT_FAILURE */
+#include <stdlib.h>       /* for calloc(3p), EXIT_SUCCESS, EXIT_FAILURE */
 #include <sys/ioctl.h>    /* for ioctl(2) */
 #include <unistd.h>       /* for sysconf(3p), _SC_PAGESIZE */
 
 #include "extent-map.h"
 
-void fiemap(int fd, uint32_t flags, int (*handle_extent)(struct fiemap_extent)) {
+#define error(_exit_code, ...) do { error(0, __VA_ARGS__); if (_exit_code) return _exit_code; } while(0)
+
+int fiemap(int fd, uint32_t flags, int (*handle_extent)(struct fiemap_extent)) {
 		const size_t fm_size = sysconf(_SC_PAGESIZE);;
 	struct fiemap *fm = calloc(1, fm_size);
 	if (!fm)
@@ -29,12 +31,12 @@ void fiemap(int fd, uint32_t flags, int (*handle_extent)(struct fiemap_extent))
 			int r = handle_extent(fm->fm_extents[i]);
 			if (r) {
 				free(fm);
-				exit(r);
+				return r;
 			}
 
 			if (fm->fm_extents[i].fe_flags & FIEMAP_EXTENT_LAST) {
 				free(fm);
-				exit(EXIT_SUCCESS);
+				return EXIT_SUCCESS;
 			}
 		}
 	}		
diff --git a/lib/extent-map.h b/lib/extent-map.h
index 4f873a0..8b5e585 100644
--- a/lib/extent-map.h
+++ b/lib/extent-map.h
@@ -1,4 +1,4 @@
 #include <stdint.h>       /* for uint32_t */
 #include <linux/fiemap.h> /* for struct fiemap_extent */
 
-void fiemap(int fd, uint32_t flags, int (*handle_extent)(struct fiemap_extent));
+int fiemap(int fd, uint32_t flags, int (*handle_extent)(struct fiemap_extent));
-- 
cgit v1.2.3-2-g168b