coja/lbmk
1
0
Fork 0
mirror of https://codeberg.org/libreboot/lbmk.git synced 2026-07-18 04:22:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

util/nvmutil: never allow cmd to be negative

Browse Source 
make cmd a size_t and make the equivalent to NULL
be the number of items in command[]

Signed-off-by: Leah Rowe <leah@libreboot.org>
This commit is contained in:
Leah Rowe
2026-03-08 04:16:05 +00:00
parent 0160b26aee
commit 8ce1cbe7f4
1 changed files with 14 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
util/nvmutil/nvmutil.c
+14 -17
View File
@@ -30,7 +30,7 @@
static void set_cmd(int argc, char *argv[]); static void set_cmd(int argc, char *argv[]);
static void set_cmd_args(int argc, char *argv[]); static void set_cmd_args(int argc, char *argv[]);
static size_t conv_argv_part_num(const char *part_str); static size_t conv_argv_part_num(const char *part_str);
static void run_cmd(ssize_t c); static void run_cmd(size_t c);
static void set_io_flags(int argc, char *argv[]); static void set_io_flags(int argc, char *argv[]);
static void open_gbe_file(void); static void open_gbe_file(void);
#ifndef HAVE_ARC4RANDOM_BUF #ifndef HAVE_ARC4RANDOM_BUF
@@ -167,7 +167,6 @@ enum {
CMD_BRICK, CMD_BRICK,
CMD_SETCHECKSUM CMD_SETCHECKSUM
}; };
#define CMD_NULL -1
struct commands { struct commands {
size_t chk; /* use by in later check on run_cmd, size_t chk; /* use by in later check on run_cmd,
@@ -194,7 +193,7 @@ static const struct commands command[] = {
/* /*
* Index in command[], will be set later * Index in command[], will be set later
*/ */
static ssize_t cmd = CMD_NULL; static size_t cmd = items(command);
int int
main(int argc, char *argv[]) main(int argc, char *argv[])
@@ -300,7 +299,7 @@ main(int argc, char *argv[])
static void static void
set_cmd(int argc, char *argv[]) set_cmd(int argc, char *argv[])
{ {
for (cmd = 0; cmd < (ssize_t)items(command); cmd++) { for (cmd = 0; cmd < items(command); cmd++) {
if (argc < 3) if (argc < 3)
break; break;
@@ -313,7 +312,7 @@ set_cmd(int argc, char *argv[])
err(EINVAL, "Too few args: command '%s'", command[cmd].str); err(EINVAL, "Too few args: command '%s'", command[cmd].str);
} }
cmd = CMD_NULL; cmd = items(command);
} }
static void static void
@@ -324,7 +323,7 @@ set_cmd_args(int argc, char *argv[])
* 4th arg e.g.: ./nvmutil gbe.bin setmac 00:xx:11:22:xx:xx * 4th arg e.g.: ./nvmutil gbe.bin setmac 00:xx:11:22:xx:xx
*/ */
mac_str = argv[3]; mac_str = argv[3];
} else if (cmd == CMD_NULL && argc >= 3) { } else if (cmd >= items(command) && argc >= 3) {
/* /*
* Example: ./nvmutil gbe.bin xx:1f:16:xx:xx:xx * Example: ./nvmutil gbe.bin xx:1f:16:xx:xx:xx
* Equivalent ./nvmutil gbe.bin setmac xx:1f:16:xx:xx:xx * Equivalent ./nvmutil gbe.bin setmac xx:1f:16:xx:xx:xx
@@ -338,7 +337,7 @@ set_cmd_args(int argc, char *argv[])
*/ */
mac_str = rmac; mac_str = rmac;
cmd = CMD_SETMAC; cmd = CMD_SETMAC;
} else if (cmd != CMD_NULL && argc > 3) { /* user-supplied partnum */ } else if (cmd < items(command) && argc > 3) { /* user-supplied partnum */
/* /*
* Example: ./nvmutil gbe.bin copy 0 * Example: ./nvmutil gbe.bin copy 0
*/ */
@@ -358,7 +357,7 @@ set_cmd_args(int argc, char *argv[])
* MAC address is used. * MAC address is used.
*/ */
if ((size_t)cmd >= items(command)) if (cmd >= items(command))
err(EINVAL, "Unhandled command error"); err(EINVAL, "Unhandled command error");
} }
@@ -384,18 +383,16 @@ conv_argv_part_num(const char *part_str)
} }
static void static void
run_cmd(ssize_t c) run_cmd(size_t c)
{ {
size_t d = (size_t)c; if (c >= items(command))
err(ECANCELED, "run_cmd: Invalid run_cmd arg: %zu", c);
if (d >= items(command)) if (c != command[c].chk)
err(ECANCELED, "run_cmd: Invalid run_cmd arg: %zd", c); err(ECANCELED, "run_cmd: Invalid chk value (%zu) vs arg: %zu",
command[c].chk, c);
if (d != command[d].chk) command[c].run();
err(ECANCELED, "run_cmd: Invalid chk value (%zu) vs arg: %zd",
command[d].chk, c);
command[d].run();
} }
static void static void