summaryrefslogtreecommitdiff
path: root/util/nvmutil/nvmutil.c
diff options
context:
space:
mode:
authorLeah Rowe <leah@libreboot.org>2023-06-01 13:35:34 +0100
committerLeah Rowe <leah@libreboot.org>2023-06-01 13:35:34 +0100
commit69fa333e25c070689b707e261b31b08be82d9fc4 (patch)
tree21a981e6dea9bcd4d0f896eb348e2f76dbb4a989 /util/nvmutil/nvmutil.c
parentadf3aece6faa134379bde6f72ae0521b7012f707 (diff)
util/nvmutil: harden pledge/unveil calls (OpenBSD)
*Open* files at the start, then unveil. The same overall behaviour is observed. In the case that invalid arguments are given, simply opening a file does not cause much performance impact (if any). Restrict operations as early as possible in code. Bonus: writeGbeFile also hardened; if flags is O_RDONLY, it aborts. Signed-off-by: Leah Rowe <leah@libreboot.org>
Diffstat (limited to 'util/nvmutil/nvmutil.c')
-rw-r--r--util/nvmutil/nvmutil.c36
1 files changed, 22 insertions, 14 deletions
diff --git a/util/nvmutil/nvmutil.c b/util/nvmutil/nvmutil.c
index 38d34436..e8abe516 100644
--- a/util/nvmutil/nvmutil.c
+++ b/util/nvmutil/nvmutil.c
@@ -7,14 +7,18 @@ int
main(int argc, char *argv[])
{
xpledge("stdio rpath wpath unveil", NULL);
- int flags = O_RDWR;
+ if (argc < 3)
+ err(errno = EINVAL, NULL);
+ if (strcmp(COMMAND, "dump") == 0)
+ flags = O_RDONLY;
+ openFiles(FILENAME);
+
void (*cmd)(void) = NULL;
const char *strMac = NULL, *strRMac = "??:??:??:??:??:??";
if (argc == 3) {
if (strcmp(COMMAND, "dump") == 0) {
- xpledge("stdio rpath unveil", NULL);
- flags = O_RDONLY;
+ xpledge("stdio", NULL);
cmd = &cmd_dump;
} else if (strcmp(COMMAND, "setmac") == 0) {
strMac = (char *) strRMac; /* random mac address */
@@ -42,16 +46,7 @@ main(int argc, char *argv[])
skipread[part ^ 1] = (cmd == &cmd_copy) | (cmd == &cmd_setchecksum)
| (cmd == &cmd_brick);
- readGbeFile(FILENAME, flags);
-
- (void)rhex();
- xunveil("/dev/urandom", "r");
- if (flags == O_RDONLY) {
- xpledge("stdio", NULL);
- } else {
- xpledge("stdio wpath unveil", NULL);
- xunveil(FILENAME, "w");
- }
+ readGbeFile(FILENAME);
if (strMac != NULL)
cmd_setmac(strMac); /* nvm gbe.bin setmac */
@@ -64,13 +59,24 @@ main(int argc, char *argv[])
}
void
-readGbeFile(const char *path, int flags)
+openFiles(const char *path)
{
+ (void)rhex();
xopen(fd, path, flags);
if ((st.st_size != SIZE_8KB))
err(errno = ECANCELED, "File `%s` not 8KiB", path);
errno = errno != ENOTDIR ? errno : 0;
+ xunveil("/dev/urandom", "r");
+ if (flags != O_RDONLY) {
+ xunveil(path, "w");
+ xpledge("stdio wpath", NULL);
+ } else
+ xpledge("stdio", NULL);
+}
+void
+readGbeFile(const char *path)
+{
big_endian = ((uint8_t *) &test)[0] ^ 1;
gbe[1] = (gbe[0] = (size_t) buf) + SIZE_4KB;
for (int p = 0; p < 2; p++) {
@@ -252,6 +258,8 @@ xorswap_buf(int partnum)
void
writeGbeFile(const char *filename)
{
+ if (flags == O_RDONLY)
+ err(ERR(), "Write aborted due to read-only mode: %s", filename);
if (gbeFileModified)
errno = 0;
for (int p = 0; p < 2; p++) {