Do not read directory

ruby · Jul 18, 2024 · aec088c · aec088c
1 parent f514987
commit aec088c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/include/prism/util/pm_string.h b/include/prism/util/pm_string.h
@@ -9,6 +9,7 @@
 #include "prism/defines.h"
 
 #include <assert.h>
+#include <errno.h>
 #include <stdbool.h>
 #include <stddef.h>
 #include <stdlib.h>

diff --git a/src/util/pm_string.c b/src/util/pm_string.c
@@ -116,6 +116,13 @@ pm_string_mapped_init(pm_string_t *string, const char *filepath) {
         return false;
     }
 
+    // Ensure it is a file and not a directory
+    if (S_ISDIR(sb.st_mode)) {
+        errno = EISDIR;
+        close(fd);
+        return false;
+    }
+
     // mmap the file descriptor to virtually get the contents
     size_t size = (size_t) sb.st_size;
     uint8_t *source = NULL;