From 8e5ac5a87d8c10f4e94bb71ece3d83c78a53876e Mon Sep 17 00:00:00 2001 From: Kevin Newton Date: Thu, 18 Jul 2024 11:39:41 -0400 Subject: [PATCH] [PRISM] Ensure not opening directories --- prism/util/pm_string.c | 7 +++++++ prism/util/pm_string.h | 1 + 2 files changed, 8 insertions(+) diff --git a/prism/util/pm_string.c b/prism/util/pm_string.c index dfc121b6a2..e9e597113f 100644 --- a/prism/util/pm_string.c +++ b/prism/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; diff --git a/prism/util/pm_string.h b/prism/util/pm_string.h index d23792c0ba..e4a20558d3 100644 --- a/prism/util/pm_string.h +++ b/prism/util/pm_string.h @@ -9,6 +9,7 @@ #include "prism/defines.h" #include +#include #include #include #include