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.c395
1 files changed, 243 insertions, 152 deletions
diff --git a/util/nvmutil/nvmutil.c b/util/nvmutil/nvmutil.c
index 6c1d929e..f5fc30cc 100644
--- a/util/nvmutil/nvmutil.c
+++ b/util/nvmutil/nvmutil.c
@@ -14,15 +14,15 @@
#include <string.h>
#include <unistd.h>
-void cmd_setchecksum(void), cmd_brick(void), swap(int partnum), writeGbe(void),
- cmd_dump(void), cmd_setmac(void), readGbe(void),
- checkdir(const char *path), macf(int partnum), hexdump(int partnum),
- parseMacString(const char *strMac, uint16_t *mac), cmd_swap(void),
- openFiles(void), cmd_copy(void), writeGbe_part(int),
- readGbe_part(int), usage(char*), set_io_flags(int, char **),
- set_cmd(int, char **), setWord(int, int, uint16_t), check_bounds(int, int),
- xopen (int *, const char *, int p, struct stat *);
-int goodChecksum(int partnum);
+void cmd_setchecksum(void), cmd_brick(void), swap(int partnum), cmd_copy(void),
+ set_cmd(int, char **), set_word(int, int, uint16_t), check_bound(int, int),
+ xopen(int *, const char *, int p, struct stat *), check_mac_separator(int),
+ cmd_dump(void), cmd_setmac(void), read_gbe(void), print_mac_address(int),
+ parse_mac_string(void), cmd_swap(void), err_if(int), write_gbe_part(int),
+ set_mac_byte(int, uint64_t *), usage(char*), set_io_flags(int, char **),
+ hexdump(int), set_mac_nib(int, int, uint8_t *), checkdir(const char *),
+ openFiles(void), read_gbe_part(int, int), write_gbe(void);
+int good_checksum(int partnum), write_mac_part(int), set_err(int);
uint8_t hextonum(char chs), rhex(void);
uint16_t word(int, int);
@@ -38,12 +38,22 @@ uint16_t word(int, int);
#define SIZE_16KB 0x4000
#define SIZE_128KB 0x20000
+#define items(x) (sizeof((x)) / sizeof((x)[0]))
+
uint8_t buf[SIZE_8KB];
-uint16_t mac[3] = {0, 0, 0};
+uint16_t macbuf[3] = {0, 0, 0};
size_t partsize;
-int flags, rfd, fd, part, e = 1;
-const char *strMac = NULL, *strRMac = "xx:xx:xx:xx:xx:xx", *fname = NULL;
+int flags;
+int rfd;
+int fd;
+int part;
+int invert = 0;
+int part_modified[2] = {0, 0};
+
+const char *mac = NULL;
+const char *rmac = "xx:xx:xx:xx:xx:xx";
+const char *fname = "";
typedef struct op {
char *str;
@@ -51,59 +61,65 @@ typedef struct op {
int args;
} op_t;
op_t op[] = {
-{ .str = "dump", .cmd = cmd_dump, .args = 3 },
-{ .str = "setmac", .cmd = cmd_setmac, .args = 3 },
-{ .str = "swap", .cmd = cmd_swap, .args = 3 },
-{ .str = "copy", .cmd = cmd_copy, .args = 4 },
-{ .str = "brick", .cmd = cmd_brick, .args = 4 },
-{ .str = "setchecksum", .cmd = cmd_setchecksum, .args = 4 },
+ { .str = "dump",
+ .cmd = cmd_dump, .args = 3 },
+ { .str = "setmac",
+ .cmd = cmd_setmac, .args = 3 },
+ { .str = "swap",
+ .cmd = cmd_swap, .args = 3 },
+ { .str = "copy",
+ .cmd = cmd_copy, .args = 4 },
+ { .str = "brick",
+ .cmd = cmd_brick, .args = 4 },
+ { .str = "setchecksum",
+ .cmd = cmd_setchecksum, .args = 4 },
};
-void (*cmd)(void) = NULL;
-#define set_err(x) errno = errno ? errno : x
-#define err_if(x) if (x) err(set_err(ECANCELED), "%s", fname)
+void (*cmd)(void) = NULL;
int
main(int argc, char *argv[])
{
+ if (argc >= 2)
+ fname = argv[1];
#ifdef __OpenBSD__
- err_if(pledge("stdio rpath wpath unveil", NULL) == -1);
- err_if(unveil("/dev/urandom", "r") == -1);
+ err_if (pledge("stdio rpath wpath unveil", NULL) == -1);
+ err_if (unveil("/dev/urandom", "r") == -1);
#endif
set_cmd(argc, argv);
-
- fname = argv[1];
set_io_flags(argc, argv);
#ifdef __OpenBSD__
if (flags == O_RDONLY) {
- err_if(unveil(fname, "r") == -1);
- err_if(unveil(NULL, NULL) == -1);
- err_if(pledge("stdio rpath", NULL) == -1);
+ err_if (unveil(fname, "r") == -1);
+ err_if (unveil(NULL, NULL) == -1);
+ err_if (pledge("stdio rpath", NULL) == -1);
} else {
- err_if(unveil(fname, "rw") == -1);
- err_if(unveil(NULL, NULL) == -1);
- err_if(pledge("stdio rpath wpath", NULL) == -1);
+ err_if (unveil(fname, "rw") == -1);
+ err_if (unveil(NULL, NULL) == -1);
+ err_if (pledge("stdio rpath wpath", NULL) == -1);
}
#endif
openFiles();
#ifdef __OpenBSD__
- err_if(pledge("stdio", NULL) == -1);
+ err_if (pledge("stdio", NULL) == -1);
#endif
- readGbe();
+ read_gbe();
(*cmd)();
- writeGbe();
+ write_gbe();
- err_if((errno != 0) && (cmd != cmd_dump));
- return errno;
+ err_if ((errno != 0) && (cmd != cmd_dump));
+ return errno ? EXIT_FAILURE : EXIT_SUCCESS;
}
void
set_cmd(int argc, char *argv[])
{
+ size_t i;
+
if (argc < 2) {
usage(argv[0]);
} else if (argc > 2) {
- for (int i = 0; (i < 6) && (cmd == NULL); i++) {
+ for (i = 0; (i < items(op)) && (cmd == NULL); i++) {
if (strcmp(COMMAND, op[i].str) != 0)
continue;
if (argc >= op[i].args) {
@@ -118,17 +134,17 @@ set_cmd(int argc, char *argv[])
}
if ((cmd == NULL) && (argc > 2)) { /* nvm gbe [MAC] */
- strMac = COMMAND;
+ mac = COMMAND;
cmd = cmd_setmac;
} else if (cmd == cmd_setmac) { /* nvm gbe setmac [MAC] */
- strMac = strRMac; /* random MAC */
+ mac = rmac; /* random MAC */
if (argc > 3)
- strMac = MAC_ADDRESS;
+ mac = MAC_ADDRESS;
} else if ((cmd != NULL) && (argc > 3)) { /* user-supplied partnum */
- err_if((errno = (!((part = PARTN[0] - '0') == 0 || part == 1))
+ err_if ((errno = (!((part = PARTN[0] - '0') == 0 || part == 1))
|| PARTN[1] ? EINVAL : errno)); /* only allow '0' or '1' */
}
- err_if((errno = (cmd == NULL) ? EINVAL : errno));
+ err_if ((errno = (cmd == NULL) ? EINVAL : errno));
}
void
@@ -141,24 +157,15 @@ set_io_flags(int argc, char *argv[])
}
void
-checkdir(const char *path)
-{
- if (opendir(path) != NULL)
- err(set_err(EISDIR), "%s", path);
- if (errno == ENOTDIR)
- errno = 0;
- err_if(errno);
-}
-
-void
openFiles(void)
{
struct stat st;
+ struct stat st_rfd;
checkdir("/dev/urandom");
checkdir(fname);
- xopen(&rfd, "/dev/urandom", O_RDONLY, &st);
+ xopen(&rfd, "/dev/urandom", O_RDONLY, &st_rfd);
xopen(&fd, fname, flags, &st);
switch(st.st_size) {
@@ -174,95 +181,131 @@ openFiles(void)
}
void
+checkdir(const char *path)
+{
+ struct stat st;
+ if (stat(path, &st) == -1)
+ err(set_err(ECANCELED), "%s", path);
+ if (S_ISDIR(st.st_mode))
+ err(set_err(EISDIR), "%s", path);
+}
+
+void
xopen(int *f, const char *l, int p, struct stat *st)
{
- if ((*f = open(l, p)) == -1) \
- err(set_err(ECANCELED), "%s", l); \
- if (fstat(*f, st) == -1) \
- err(set_err(ECANCELED), "%s", l);
+ if ((*f = open(l, p)) == -1)
+ err(set_err(ECANCELED), "%s", l);
+ if (fstat(*f, st) == -1)
+ err(set_err(ECANCELED), "%s", l);
}
void
-readGbe(void)
+read_gbe(void)
{
- for (int p = 0; p < 2; p++)
- readGbe_part(p);
+ int p;
+ int do_read[2] = {1, 1};
+
+ if ((cmd == cmd_copy) || (cmd == cmd_brick) ||
+ (cmd == cmd_setchecksum))
+ do_read[part ^ 1] = 0;
+
+ /*
+ * speedhack: if copy/swap, flip where data gets written to memory,
+ * so that cmd_copy and cmd_swap don't have to work on every word
+ */
+ if ((cmd == cmd_copy) || (cmd == cmd_swap))
+ invert = 1;
+
+ for (p = 0; p < 2; p++)
+ if (do_read[p])
+ read_gbe_part(p, invert);
}
void
-readGbe_part(int p)
+read_gbe_part(int p, int invert)
{
- if (pread(fd, buf + (SIZE_4KB * p), SIZE_4KB, p * partsize) != SIZE_4KB)
+ if (pread(fd, buf + (SIZE_4KB * (p ^ invert)), SIZE_4KB, p * partsize)
+ != SIZE_4KB)
err(set_err(ECANCELED),
"Can't read %d b from '%s' p%d", SIZE_4KB, fname, p);
- swap(p); /* handle big-endian host CPU */
+ swap(p ^ invert); /* handle big-endian host CPU */
}
void
cmd_setmac(void)
{
+ int partnum;
int mac_updated = 0;
- parseMacString(strMac, mac);
-
- printf("MAC address to be written: %s\n", strMac);
-
- for (int partnum = 0; partnum < 2; partnum++) {
- if (!goodChecksum(part = partnum))
- continue;
-
- for (int w = 0; w < 3; w++)
- setWord(w, partnum, mac[w]);
+ parse_mac_string();
- printf("Wrote MAC address to part %d: ", partnum);
- macf(partnum);
-
- cmd_setchecksum();
- mac_updated = 1;
- }
+ printf("MAC address to be written: %s\n", mac);
+ for (partnum = 0; partnum < 2; partnum++)
+ mac_updated |= write_mac_part(partnum);
if (mac_updated)
errno = 0;
}
void
-parseMacString(const char *strMac, uint16_t *mac)
+parse_mac_string(void)
{
+ int mac_pos;
uint64_t total = 0;
- if (strnlen(strMac, 20) != 17)
+
+ if (strnlen(mac, 20) != 17)
err(set_err(EINVAL), "Invalid MAC address string length");
- for (uint8_t h, i = 0; i < 16; i += 3) {
- if (i != 15)
- if (strMac[i + 2] != ':')
- err(set_err(EINVAL),
- "Invalid MAC address separator '%c'",
- strMac[i + 2]);
-
- int byte = i / 3;
-
- for (int nib = 0; nib < 2; nib++, total += h) {
- if ((h = hextonum(strMac[i + nib])) > 15)
- err(set_err(EINVAL), "Invalid character '%c'",
- strMac[i + nib]);
-
- /* If random, ensure that local/unicast bits are set */
- if ((byte == 0) && (nib == 1))
- if ((strMac[i + nib] == '?') ||
- (strMac[i + nib] == 'x') ||
- (strMac[i + nib] == 'X')) /* random */
- h = (h & 0xE) | 2; /* local, unicast */
-
- mac[byte >> 1] |= ((uint16_t ) h)
- << ((8 * (byte % 2)) + (4 * (nib ^ 1)));
- }
- }
+ for (mac_pos = 0; mac_pos < 16; mac_pos += 3)
+ set_mac_byte(mac_pos, &total);
if (total == 0)
err(set_err(EINVAL), "Invalid MAC (all-zero MAC address)");
- if (mac[0] & 1)
+ if (macbuf[0] & 1)
err(set_err(EINVAL), "Invalid MAC (multicast bit set)");
}
+void
+set_mac_byte(int mac_pos, uint64_t *total)
+{
+ int nib;
+ uint8_t h = 0;
+ check_mac_separator(mac_pos);
+
+ for (nib = 0; nib < 2; nib++, *total += h)
+ set_mac_nib(mac_pos, nib, &h);
+}
+
+void
+check_mac_separator(int mac_pos)
+{
+ if (mac_pos == 15)
+ return;
+ char separator = mac[mac_pos + 2];
+ if (separator == ':')
+ return;
+ err(set_err(EINVAL), "Invalid MAC address separator '%c'", separator);
+}
+
+void
+set_mac_nib(int mac_pos, int nib, uint8_t *h)
+{
+ int byte = mac_pos / 3;
+
+ if ((*h = hextonum(mac[mac_pos + nib])) > 15)
+ err(set_err(EINVAL), "Invalid character '%c'",
+ mac[mac_pos + nib]);
+
+ /* If random, ensure that local/unicast bits are set */
+ if ((byte == 0) && (nib == 1))
+ if ((mac[mac_pos + nib] == '?') ||
+ (mac[mac_pos + nib] == 'x') ||
+ (mac[mac_pos + nib] == 'X')) /* random */
+ *h = (*h & 0xE) | 2; /* local, unicast */
+
+ macbuf[byte >> 1] |= ((uint16_t ) *h) << ((8 * (byte % 2)) +
+ (4 * (nib ^ 1)));
+}
+
uint8_t
hextonum(char ch)
{
@@ -282,30 +325,53 @@ rhex(void)
{
static uint8_t n = 0, rnum[16];
if (!n)
- err_if(pread(rfd, (uint8_t *) &rnum, (n = 15) + 1, 0) == -1);
+ err_if (read(rfd, (uint8_t *) &rnum, (n = 15) + 1) == -1);
return rnum[n--] & 0xf;
}
+int
+write_mac_part(int partnum)
+{
+ int w;
+
+ part = partnum;
+ if (!good_checksum(partnum))
+ return 0;
+
+ for (w = 0; w < 3; w++)
+ set_word(w, partnum, macbuf[w]);
+
+ printf("Wrote MAC address to part %d: ", partnum);
+ print_mac_address(partnum);
+
+ cmd_setchecksum();
+ return 1;
+}
+
void
cmd_dump(void)
{
- for (int partnum = 0, numInvalid = 0; partnum < 2; partnum++) {
- if (!goodChecksum(partnum))
- ++numInvalid;
+ int partnum;
+ int num_invalid = 0;
+
+ for (partnum = 0; partnum < 2; partnum++) {
+ if (!good_checksum(partnum))
+ ++num_invalid;
printf("MAC (part %d): ", partnum);
- macf(partnum);
+ print_mac_address(partnum);
hexdump(partnum);
-
- if ((numInvalid < 2) && (partnum))
- errno = 0;
}
+
+ if ((num_invalid < 2))
+ errno = 0;
}
void
-macf(int partnum)
+print_mac_address(int partnum)
{
- for (int c = 0; c < 3; c++) {
+ int c;
+ for (c = 0; c < 3; c++) {
uint16_t val16 = word(c, partnum);
printf("%02x:%02x", val16 & 0xff, val16 >> 8);
if (c == 2)
@@ -318,9 +384,12 @@ macf(int partnum)
void
hexdump(int partnum)
{
- for (int row = 0; row < 8; row++) {
+ int c;
+ int row;
+
+ for (row = 0; row < 8; row++) {
printf("%08x ", row << 4);
- for (int c = 0; c < 8; c++) {
+ for (c = 0; c < 8; c++) {
uint16_t val16 = word((row << 3) + c, partnum);
if (c == 4)
printf(" ");
@@ -333,93 +402,97 @@ hexdump(int partnum)
void
cmd_setchecksum(void)
{
+ int c;
uint16_t val16 = 0;
- for (int c = 0; c < NVM_CHECKSUM_WORD; c++)
+
+ for (c = 0; c < NVM_CHECKSUM_WORD; c++)
val16 += word(c, part);
- setWord(NVM_CHECKSUM_WORD, part, NVM_CHECKSUM - val16);
+ set_word(NVM_CHECKSUM_WORD, part, NVM_CHECKSUM - val16);
}
void
cmd_brick(void)
{
- if (goodChecksum(part))
- setWord(NVM_CHECKSUM_WORD, part,
+ if (good_checksum(part))
+ set_word(NVM_CHECKSUM_WORD, part,
((word(NVM_CHECKSUM_WORD, part)) ^ 0xFF));
}
void
cmd_copy(void)
{
- err_if(!goodChecksum(part));
- for (int c = 0; c < (SIZE_4KB >> 1); c++)
- setWord(c, part ^ 1, word(c, part));
+ err_if (!good_checksum(part ^ 1));
+ part_modified[part ^ 1] = 1;
}
void
cmd_swap(void) {
- err_if(!(goodChecksum(0) || goodChecksum(1)));
+ err_if (!(good_checksum(0) || good_checksum(1)));
errno = 0;
- for (int c = 0; c < (SIZE_4KB >> 1); c++) {
- uint16_t chg = word(c, 0);
- setWord(c, 0, word(c, 1));
- setWord(c, 1, chg);
- }
+ part_modified[1] = part_modified[0] = 1;
}
int
-goodChecksum(int partnum)
+good_checksum(int partnum)
{
+ int w;
uint16_t total = 0;
- for(int w = 0; w <= NVM_CHECKSUM_WORD; w++)
+ for (w = 0; w <= NVM_CHECKSUM_WORD; w++)
total += word(w, partnum);
if (total == NVM_CHECKSUM)
return 1;
- fprintf(stderr, "WARNING: BAD checksum in part %d\n", partnum);
- set_err(ECANCELED);
+ fprintf(stderr, "WARNING: BAD checksum in part %d\n",
+ partnum ^ invert);
+ (void) set_err(ECANCELED);
return 0;
}
uint16_t
word(int pos16, int p)
{
- check_bounds(pos16, p);
+ check_bound(pos16, p);
return ((uint16_t *) (buf + (SIZE_4KB * p)))[pos16];
}
void
-setWord(int pos16, int p, uint16_t val16)
+set_word(int pos16, int p, uint16_t val16)
{
- check_bounds(pos16, p);
+ check_bound(pos16, p);
+ part_modified[p] = 1;
((uint16_t *) (buf + (SIZE_4KB * p)))[pos16] = val16;
}
void
-check_bounds(int c, int p)
+check_bound(int c, int p)
{
if ((p != 0) && (p != 1))
- err(set_err(EINVAL), "check_bounds: invalid partnum %d", p);
+ err(set_err(EINVAL), "check_bound: invalid partnum %d", p);
if ((c < 0) || (c >= (SIZE_4KB >> 1)))
- err(set_err(EINVAL), "check_bounds: out of bounds %d", c);
+ err(set_err(EINVAL), "check_bound: out of bounds %d", c);
}
void
-writeGbe(void)
+write_gbe(void)
{
+ int p;
+
if (flags != O_RDONLY)
- for (int p = 0; p < 2; p++)
- writeGbe_part(p);
- err_if(close(fd) == -1);
+ for (p = 0; p < 2; p++)
+ if (part_modified[p])
+ write_gbe_part(p);
+ err_if (close(fd) == -1);
}
void
-writeGbe_part(int p)
+write_gbe_part(int p)
{
swap(p); /* swap bytes on big-endian host CPUs */
- if(pwrite(fd, buf + (SIZE_4KB * p), SIZE_4KB, p * partsize) != SIZE_4KB)
+ if(pwrite(fd, buf + (SIZE_4KB * p), SIZE_4KB, p * partsize)
+ != SIZE_4KB)
err(set_err(ECANCELED),
"Can't write %d b to '%s' p%d", SIZE_4KB, fname, p);
}
@@ -427,13 +500,17 @@ writeGbe_part(int p)
void
swap(int partnum)
{
+ size_t w;
+ size_t x;
+
+ int e = 1;
uint8_t *n = buf + (SIZE_4KB * partnum);
- for (size_t w = NVM_SIZE * ((uint8_t *) &e)[0], x = 1;
+ for (w = NVM_SIZE * ((uint8_t *) &e)[0], x = 1;
w < NVM_SIZE; w += 2, x += 2) {
uint8_t chg = n[w];
- n[w] ^= n[x];
- n[x] ^= chg;
+ n[w] = n[x];
+ n[x] = chg;
}
}
@@ -441,7 +518,7 @@ void
usage(char *util)
{
#ifdef __OpenBSD__
- err_if(pledge("stdio", NULL) == -1);
+ err_if (pledge("stdio", NULL) == -1);
#endif
fprintf(stderr,
"Modify Intel GbE NVM images e.g. set MAC\n"
@@ -456,3 +533,17 @@ usage(char *util)
util, util, util, util, util, util, util);
err(set_err(ECANCELED), "Too few arguments");
}
+
+void
+err_if(int x)
+{
+ if (x)
+ err(set_err(ECANCELED), "%s", fname);
+}
+
+int
+set_err(int x)
+{
+ errno = errno ? errno : x;
+ return EXIT_FAILURE;
+}
generated by cgit v1.2.1 (git 2.18.0) at 2026-03-03 14:54:56 +0000