util/nvmutil: massive code cleanup

Signed-off-by: Leah Rowe <leah@libreboot.org>

2 files changed, 40 insertions, 50 deletions

diff --git a/util/nvmutil/nvmutil.c b/util/nvmutil/nvmutil.c
index a94ac34..d047e2a 100644
--- a/util/nvmutil/nvmutil.c
+++ b/util/nvmutil/nvmutil.c
@@ -7,15 +7,9 @@ int
 main(int argc, char *argv[])
 {
 	xpledge("stdio rpath wpath unveil", NULL);
-	size_t nr = 128;
-	int fd, flags = O_RDWR;
+	int flags = O_RDWR;
 	void (*cmd)(void) = NULL;
 	const char *strMac = NULL, *strRMac = "??:??:??:??:??:??";
-	buf = (uint8_t *) &buf16;
-	gbe[1] = (gbe[0] = (size_t) buf) + SIZE_4KB;
-
-	test = 1;
-	big_endian = ((uint8_t *) &test)[0] ^ 1;
 
 	if (argc == 3) {
 		if (strcmp(COMMAND, "dump") == 0) {
@@ -26,7 +20,7 @@ main(int argc, char *argv[])
 			strMac = (char *) strRMac; /* random mac address */
 		} else if (strcmp(COMMAND, "swap") == 0) {
 			cmd = &cmd_swap;
-			nr = SIZE_4KB;
+			nf = SIZE_4KB;
 		}
 	} else if (argc == 4) {
 		if (strcmp(COMMAND, "setmac") == 0) {
@@ -40,7 +34,7 @@ main(int argc, char *argv[])
 			cmd = &cmd_brick;
 		} else if (strcmp(COMMAND, "copy") == 0) {
 			cmd = &cmd_copy;
-			nr = SIZE_4KB;
+			nf = SIZE_4KB;
 		}
 	}
 
@@ -48,7 +42,7 @@ main(int argc, char *argv[])
 
 	skipread[part ^ 1] = (cmd == &cmd_copy) |
 		(cmd == &cmd_setchecksum) | (cmd == &cmd_brick);
-	readGbeFile(&fd, FILENAME, flags, nr);
+	readGbeFile(FILENAME, flags);
 	(void)rhex();
 	xunveil("/dev/urandom", "r");
 	if (flags == O_RDONLY) {
@@ -62,33 +56,27 @@ main(int argc, char *argv[])
 		cmd_setmac(strMac); /* nvm gbe.bin setmac */
 	else if (cmd != NULL)
 		(*cmd)(); /* all other commands except setmac */
-	writeGbeFile(&fd, FILENAME, nr);
+	writeGbeFile(FILENAME);
 
 	err_if((errno != 0) && (cmd != &cmd_dump));
 	return errno;
 }
 
 void
-readGbeFile(int *fd, const char *path, int flags, size_t nr)
+readGbeFile(const char *path, int flags)
 {
-	struct stat st;
-	if (opendir(path) != NULL)
-		err(errno = EISDIR, "%s", path);
-	xopen(*fd, path, flags);
-	if (fstat((*fd), &st) == -1)
-		err(ERR(), "%s", path);
-	else if ((st.st_size != SIZE_8KB))
+	xopen(fd, path, flags);
+	if ((st.st_size != SIZE_8KB))
 		err(errno = ECANCELED, "File `%s` not 8KiB", path);
-	else if (errno == ENOTDIR)
-		errno = 0;
+	errno = errno != ENOTDIR ? errno : 0;
 
+	big_endian = ((uint8_t *) &test)[0] ^ 1;
+	gbe[1] = (gbe[0] = (size_t) buf) + SIZE_4KB;
 	for (int p = 0; p < 2; p++) {
 		if (skipread[p])
 			continue;
-		if (pread((*fd), (uint8_t *) gbe[p], nr, p << 12) == -1)
-			err(ERR(), "%s", path);
-		if (big_endian)
-			xorswap_buf(nr, p);
+		xpread(fd, (uint8_t *) gbe[p], nf, p << 12, path);
+		handle_endianness();
 	}
 }
 
@@ -153,10 +141,8 @@ rhex(void)
 	static int rfd = -1, n = 0;
 	static uint8_t rnum[16];
 	if (!n) {
-		if (rfd == -1)
-			xopen(rfd, "/dev/urandom", O_RDONLY);
-		if (read(rfd, (uint8_t *) &rnum, (n = 15) + 1) == -1)
-			err(ERR(), "/dev/urandom");
+		xopen(rfd, "/dev/urandom", O_RDONLY);
+		xpread(rfd, (uint8_t *) &rnum, (n = 15) + 1, 0, "/dev/urandom");
 	}
 	return rnum[n--] & 0xf;
 }
@@ -251,15 +237,15 @@ setWord(int pos16, int partnum, uint16_t val16)
 }
 
 void
-xorswap_buf(int n, int partnum)
+xorswap_buf(int partnum)
 {
 	uint8_t *nbuf = (uint8_t *) gbe[partnum];
-	for (int w = 0; w < (n >> 1); w++)
+	for (size_t w = 0; w < (nf >> 1); w++)
 		xorswap(nbuf[w << 1], nbuf[(w << 1) + 1]);
 }
 
 void
-writeGbeFile(int *fd, const char *filename, size_t nw)
+writeGbeFile(const char *filename)
 {
 	if (gbeFileModified)
 		errno = 0;
@@ -268,15 +254,12 @@ writeGbeFile(int *fd, const char *filename, size_t nw)
 			p ^= 1; /* speedhack: write sequentially on-disk */
 		if (!nvmPartModified[p])
 			goto next_part;
-		if (big_endian)
-			xorswap_buf(nw, p);
-		if (pwrite((*fd), (uint8_t *) gbe[p], nw, p << 12) == -1)
-			err(ERR(), "%s", filename);
+		handle_endianness();
+		xpwrite(fd, (uint8_t *) gbe[p], nf, p << 12, filename);
 next_part:
 		if (gbe[0] > gbe[1])
 			p ^= 1; /* speedhack: write sequentially on-disk */
 	}
-	if (close((*fd)))
-		err(ERR(), "%s", filename);
+	xclose(fd, filename);
 	xpledge("stdio", NULL);
 }
diff --git a/util/nvmutil/nvmutil.h b/util/nvmutil/nvmutil.h
index 780195c..595ad3b 100644
--- a/util/nvmutil/nvmutil.h
+++ b/util/nvmutil/nvmutil.h
@@ -13,8 +13,7 @@
 #include <string.h>
 #include <unistd.h>
 
-void readGbeFile(int *fd, const char *path, int flags,
-	size_t nr);
+void readGbeFile(const char *path, int flags);
 void cmd_setmac(const char *strMac);
 int invalidMacAddress(const char *strMac, uint16_t *mac);
 uint8_t hextonum(char chs);
@@ -28,8 +27,8 @@ void cmd_swap(void);
 void cmd_copy(void);
 int validChecksum(int partnum);
 void setWord(int pos16, int partnum, uint16_t val16);
-void xorswap_buf(int n, int partnum);
-void writeGbeFile(int *fd, const char *filename, size_t nw);
+void xorswap_buf(int partnum);
+void writeGbeFile(const char *filename);
 
 #define FILENAME argv[1]
 #define COMMAND argv[2]
@@ -39,22 +38,30 @@ void writeGbeFile(int *fd, const char *filename, size_t nw);
 #define SIZE_8KB 0x2000
 
 uint16_t buf16[SIZE_4KB];
-uint8_t *buf;
-size_t gbe[2];
+uint8_t *buf = (uint8_t *) &buf16;
+size_t nf = 128, gbe[2];
 uint8_t skipread[2] = {0, 0};
 
-int part, gbeFileModified = 0;
+int fd = -1, part, gbeFileModified = 0;
 uint8_t nvmPartModified[2] = {0, 0};
 
-uint16_t test;
-uint8_t big_endian;
+int test = 1;
+int big_endian;
 
-#define word(pos16, partnum) buf16[pos16 + (partnum << 11)]
 #define ERR() errno = errno ? errno : ECANCELED
-#define xorswap(x, y) x ^= y, y ^= x, x ^= y
-#define xopen(fd, loc, p) if ((fd = open(loc, p)) == -1) err(ERR(), "%s", loc)
 #define err_if(x) if (x) err(ERR(), NULL)
 
+#define xopen(f,l,p) if (opendir(l) != NULL) err(errno = EISDIR, "%s", l); \
+	if (f == -1) if ((f = open(l, p)) == -1) err(ERR(), "%s", l); \
+	struct stat st; if (fstat(f, &st) == -1) err(ERR(), "%s", l)
+#define xpread(f, b, n, o, l) if (pread(f, b, n, o) == -1) err(ERR(), "%s", l)
+#define handle_endianness() if (big_endian) xorswap_buf(p)
+#define xpwrite(f, b, n, o, l) if (pwrite(f, b, n, o) == -1) err(ERR(), "%s", l)
+#define xclose(f, l) if (close(f)) err(ERR(), "%s", l)
+
+#define xorswap(x, y) x ^= y, y ^= x, x ^= y
+#define word(pos16, partnum) buf16[pos16 + (partnum << 11)]
+
 void
 xpledge(const char *promises, const char *execpromises)
 {