kat, tak: handle open_map errors

2 files changed, 6 insertions, 0 deletions

diff --git a/src/kat.c b/src/kat.c
index 9c50aed..551a43d 100644
--- a/src/kat.c
+++ b/src/kat.c
@@ -19,6 +19,9 @@ int entry_kat(int argc, char **argv) {
   for (int i = 1; i < argc; i++) {
     const char *fname = argv[i];
     struct map *map = open_map(fname);
+    if (map == NULL) {
+      return 1;
+    }
 
     for (size_t i = 0; i < map->sb.st_size; i++) {
       putchar(map->addr[i]);
diff --git a/src/tak.c b/src/tak.c
index caf1837..06fc35e 100644
--- a/src/tak.c
+++ b/src/tak.c
@@ -19,6 +19,9 @@ int entry_tak(int argc, char **argv) {
   for (int i = argc-1; i >= 1; i--) {
     const char *fname = argv[i];
     struct map *map = open_map(fname);
+    if (map == NULL) {
+      return 1;
+    }
 
     char *lstart, *lend;
     lend = &map->addr[map->sb.st_size - 1];