summaryrefslogtreecommitdiff
path: root/util/nvmutil/nvmutil.c
diff options
context:
space:
mode:
Diffstat (limited to 'util/nvmutil/nvmutil.c')
-rw-r--r--util/nvmutil/nvmutil.c53
1 files changed, 30 insertions, 23 deletions
diff --git a/util/nvmutil/nvmutil.c b/util/nvmutil/nvmutil.c
index 26c31d2c..e75ba650 100644
--- a/util/nvmutil/nvmutil.c
+++ b/util/nvmutil/nvmutil.c
@@ -58,6 +58,7 @@ void byteswap(int n, int partnum);
void writeGbeFile(int *fd, const char *filename, size_t nw);
void xpledge(const char *promises, const char *execpromises);
void xunveil(const char *path, const char *permissions);
+void err_if(int condition);
#define FILENAME argv[1]
#define COMMAND argv[2]
@@ -118,31 +119,27 @@ main(int argc, char *argv[])
}
}
- if ((strMac == NULL) && (cmd == NULL))
- errno = EINVAL;
-
- if (errno == 0) {
- skipread[part ^ 1] = (cmd == &cmd_copy) |
- (cmd == &cmd_setchecksum) | (cmd == &cmd_brick);
- readGbeFile(&fd, FILENAME, flags, nr);
- (void)rhex();
- xunveil("/dev/urandom", "r");
- if (flags == O_RDONLY) {
- xpledge("stdio", NULL);
- } else {
- xpledge("stdio wpath unveil", NULL);
- xunveil(FILENAME, "w");
- }
- if (strMac != NULL)
- cmd_setmac(strMac); /* nvm gbe.bin setmac */
- else if (cmd != NULL)
- (*cmd)(); /* all other commands except setmac */
- writeGbeFile(&fd, FILENAME, nr);
- } else
+ err_if(errno = ((strMac == NULL) && (cmd == NULL)) ? EINVAL : errno);
+
+ skipread[part ^ 1] = (cmd == &cmd_copy) |
+ (cmd == &cmd_setchecksum) | (cmd == &cmd_brick);
+ readGbeFile(&fd, FILENAME, flags, nr);
+ (void)rhex();
+ xunveil("/dev/urandom", "r");
+ if (flags == O_RDONLY) {
xpledge("stdio", NULL);
+ } else {
+ xpledge("stdio wpath unveil", NULL);
+ xunveil(FILENAME, "w");
+ }
- if ((errno != 0) && (cmd != &cmd_dump))
- err(errno, NULL);
+ if (strMac != NULL)
+ cmd_setmac(strMac); /* nvm gbe.bin setmac */
+ else if (cmd != NULL)
+ (*cmd)(); /* all other commands except setmac */
+ writeGbeFile(&fd, FILENAME, nr);
+
+ err_if((errno != 0) && (cmd != &cmd_dump));
return errno;
}
@@ -396,3 +393,13 @@ xunveil(const char *path, const char *permissions)
err(errno, NULL);
#endif
}
+
+void
+err_if(int condition)
+{
+ if (!condition)
+ return;
+ if (!errno)
+ errno = ECANCELED;
+ err(errno, NULL);
+}