dump-vdso: Add error handling

1 files changed, 11 insertions, 3 deletions

diff --git a/dump-vdso.c b/dump-vdso.c
index 937c00c..dbd6439 100644
--- a/dump-vdso.c
+++ b/dump-vdso.c
@@ -1,19 +1,27 @@
 #include <stdio.h>
 #include <string.h>
 
-int
-main()
+int main()
 {
 	FILE *maps;
 	void *vdso_begin, *vdso_end;
+	int found_vdso = 0;
 
 	maps = fopen("/proc/self/maps", "r");
 	char buf[1024];
 	while (fgets(buf, 1024, maps)) {
-		if (strstr(buf, "[vdso]")) break;
+		if (strstr(buf, "[vdso]")) {
+			found_vdso = 1;
+			break;
+		}
 	}
 	fclose(maps);
 
+	if (!found_vdso) {
+		fprintf(stderr, "Could not find vdso mapping\n");
+		return 1;
+	}
+
 	sscanf(buf, "%p-%p", &vdso_begin, &vdso_end);
 	write(1, vdso_begin, vdso_end - vdso_begin);