cmd: Exploit autodata for registration and enumeration

Autodata is much more elegant solution, and we're already exploiting it for other things. Spot tested various commands. Signed-off-by: Andrew Jeffery <[email protected]>
amboar · Dec 30, 2024 · e2d541a · e2d541a
1 parent b2238b3
commit e2d541a
Show file tree

Hide file tree

Showing 17 changed files with 145 additions and 57 deletions.
diff --git a/src/cmd.h b/src/cmd.h
@@ -0,0 +1,17 @@
+/* SPDX-License-Identifier: Apache-2.0 */
+/* Copyright (C) 2024 Andrew Jeffery */
+
+#ifndef CULVERT_CMD_H
+#define CULVERT_CMD_H
+
+#include "ccan/autodata/autodata.h"
+
+struct cmd {
+    const char *name;
+    int (*fn)(const char *, int, char *[]);
+};
+
+AUTODATA_TYPE(cmds, struct cmd);
+#define REGISTER_CMD(cmd) AUTODATA_SYM(cmds, cmd);
+
+#endif
diff --git a/src/cmd/console.c b/src/cmd/console.c
@@ -3,6 +3,7 @@
 
 #include "ahb.h"
 #include "ast.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "host.h"
 #include "log.h"
@@ -16,7 +17,7 @@
 #include <string.h>
 #include <unistd.h>
 
-int cmd_console(const char *name __unused, int argc, char *argv[])
+static int do_console(const char *name __unused, int argc, char *argv[])
 {
     struct suart _suart, *suart = &_suart;
     struct host _host, *host = &_host;
@@ -179,3 +180,8 @@ int cmd_console(const char *name __unused, int argc, char *argv[])
     return rc;
 }
 
+static const struct cmd console_cmd = {
+    "console",
+    do_console
+};
+REGISTER_CMD(console_cmd);
diff --git a/src/cmd/coprocessor.c b/src/cmd/coprocessor.c
@@ -2,6 +2,7 @@
 // Copyright (C) 2024 Code Construct
 
 #include "bits.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "host.h"
 #include "log.h"
@@ -274,7 +275,7 @@ static int cmd_coprocessor_stop(const char *name __unused, int argc, char *argv[
     return rc;
 }
 
-int cmd_coprocessor(const char *name __unused, int argc, char *argv[])
+static int do_coprocessor(const char *name __unused, int argc, char *argv[])
 {
     const char *arg_subcmd;
 
@@ -295,3 +296,9 @@ int cmd_coprocessor(const char *name __unused, int argc, char *argv[])
 
     return EXIT_FAILURE;
 }
+
+static const struct cmd coprocessor_cmd = {
+    "coprocessor",
+    do_coprocessor
+};
+REGISTER_CMD(coprocessor_cmd);
diff --git a/src/cmd/debug.c b/src/cmd/debug.c
@@ -6,9 +6,10 @@
 #include "ahb.h"
 #include "ast.h"
 #include "bridge/debug.h"
+#include "cmd.h"
 #include "log.h"
 
-int cmd_debug(const char *name, int argc, char *argv[])
+static int do_debug(const char *name, int argc, char *argv[])
 {
     struct debug _debug, *debug = &_debug;
     struct ahb *ahb;
@@ -73,3 +74,9 @@ int cmd_debug(const char *name, int argc, char *argv[])
 
     return rc;
 }
+
+static const struct cmd debug_cmd = {
+    "debug",
+    do_debug
+};
+REGISTER_CMD(debug_cmd);
diff --git a/src/cmd/devmem.c b/src/cmd/devmem.c
@@ -7,9 +7,10 @@
 #include "ahb.h"
 #include "ast.h"
 #include "bridge/devmem.h"
+#include "cmd.h"
 #include "priv.h"
 
-int cmd_devmem(const char *name, int argc, char *argv[])
+static int do_devmem(const char *name, int argc, char *argv[])
 {
     struct devmem _devmem, *devmem = &_devmem;
     struct ahb *ahb;
@@ -46,3 +47,8 @@ int cmd_devmem(const char *name, int argc, char *argv[])
     return 0;
 }
 
+static const struct cmd devmem_cmd = {
+    "devmem",
+    do_devmem,
+};
+REGISTER_CMD(devmem_cmd);
diff --git a/src/cmd/ilpc.c b/src/cmd/ilpc.c
@@ -6,9 +6,10 @@
 #include "ahb.h"
 #include "ast.h"
 #include "bridge/ilpc.h"
+#include "cmd.h"
 #include "priv.h"
 
-int cmd_ilpc(const char *name, int argc, char *argv[])
+static int do_ilpc(const char *name, int argc, char *argv[])
 {
     struct ilpcb _ilpcb, *ilpcb = &_ilpcb;
     struct ahb *ahb;
@@ -44,3 +45,9 @@ int cmd_ilpc(const char *name, int argc, char *argv[])
 
     return 0;
 }
+
+static const struct cmd ilpc_cmd = {
+    "ilpc",
+    do_ilpc,
+};
+REGISTER_CMD(ilpc_cmd);
diff --git a/src/cmd/jtag.c b/src/cmd/jtag.c
@@ -3,6 +3,7 @@
 
 #include "ahb.h"
 #include "ast.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "host.h"
 #include "log.h"
@@ -147,7 +148,7 @@ static int run_openocd_bitbang_server(struct jtag *jtag, uint16_t port)
         }
 }
 
-int cmd_jtag(const char *name, int argc, char *argv[])
+static int do_jtag(const char *name, int argc, char *argv[])
 {
         struct host _host, *host = &_host;
         struct soc _soc, *soc = &_soc;
@@ -260,3 +261,9 @@ int cmd_jtag(const char *name, int argc, char *argv[])
 done:
         exit(rc);
 }
+
+static const struct cmd jtag_cmd = {
+        "jtag",
+        do_jtag,
+};
+REGISTER_CMD(jtag_cmd);
diff --git a/src/cmd/otp.c b/src/cmd/otp.c
@@ -2,6 +2,7 @@
 // Copyright (C) 2018,2021 IBM Corp.
 #include "ahb.h"
 #include "ast.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "host.h"
 #include "log.h"
@@ -13,7 +14,7 @@
 #include <stdlib.h>
 #include <string.h>
 
-int cmd_otp(const char *name __unused, int argc, char *argv[])
+static int do_otp(const char *name __unused, int argc, char *argv[])
 {
     enum otp_region reg = otp_region_conf;
     struct host _host, *host = &_host;
@@ -104,3 +105,9 @@ int cmd_otp(const char *name __unused, int argc, char *argv[])
 
     return rc;
 }
+
+static const struct cmd otp_cmd = {
+    "otp",
+    do_otp,
+};
+REGISTER_CMD(otp_cmd);
diff --git a/src/cmd/p2a.c b/src/cmd/p2a.c
@@ -8,10 +8,11 @@
 #include "ahb.h"
 #include "ast.h"
 #include "bridge/p2a.h"
+#include "cmd.h"
 #include "log.h"
 #include "priv.h"
 
-int cmd_p2a(const char *name, int argc, char *argv[])
+static int do_p2a(const char *name, int argc, char *argv[])
 {
     struct p2ab _p2ab, *p2ab = &_p2ab;
     struct ahb *ahb;
@@ -55,3 +56,9 @@ int cmd_p2a(const char *name, int argc, char *argv[])
 
     return 0;
 }
+
+static const struct cmd p2a_cmd = {
+    "p2a",
+    do_p2a,
+};
+REGISTER_CMD(p2a_cmd);
diff --git a/src/cmd/probe.c b/src/cmd/probe.c
@@ -1,8 +1,8 @@
 // SPDX-License-Identifier: Apache-2.0
 // Copyright (C) 2018,2021 IBM Corp.
 
+#include "cmd.h"
 #include "compiler.h"
-
 #include "host.h"
 #include "log.h"
 #include "soc.h"
@@ -24,7 +24,7 @@ cmd_probe_help(const char *name, int argc __unused, char *argv[] __unused)
     printf(probe_help, name, name, name, name);
 }
 
-int cmd_probe(const char *name, int argc, char *argv[])
+static int do_probe(const char *name, int argc, char *argv[])
 {
     enum bridge_mode required = bm_permissive;
     struct host _host, *host = &_host;
@@ -123,3 +123,9 @@ int cmd_probe(const char *name, int argc, char *argv[])
 done:
     exit(rc);
 }
+
+static const struct cmd probe_cmd = {
+    "probe",
+    do_probe,
+};
+REGISTER_CMD(probe_cmd);
diff --git a/src/cmd/read.c b/src/cmd/read.c
@@ -3,6 +3,7 @@
 
 #include "ahb.h"
 #include "ast.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "flash.h"
 #include "host.h"
@@ -194,7 +195,7 @@ static int cmd_read_ram(int argc, char *argv[])
     return rc;
 }
 
-int cmd_read(const char *name __unused, int argc, char *argv[])
+static int do_read(const char *name __unused, int argc, char *argv[])
 {
     int rc;
 
@@ -214,3 +215,9 @@ int cmd_read(const char *name __unused, int argc, char *argv[])
 
     return rc < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
 }
+
+static const struct cmd read_cmd = {
+    "read",
+    do_read,
+};
+REGISTER_CMD(read_cmd);
diff --git a/src/cmd/replace.c b/src/cmd/replace.c
@@ -4,6 +4,7 @@
 
 #include "ahb.h"
 #include "ast.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "host.h"
 #include "log.h"
@@ -18,7 +19,7 @@
 
 #define DUMP_RAM_WIN  (8 << 20)
 
-int cmd_replace(const char *name __unused, int argc, char *argv[])
+static int do_replace(const char *name __unused, int argc, char *argv[])
 {
     struct host _host, *host = &_host;
     struct soc _soc, *soc = &_soc;
@@ -128,3 +129,9 @@ int cmd_replace(const char *name __unused, int argc, char *argv[])
 
     return rc;
 }
+
+static const struct cmd replace_cmd = {
+    "replace",
+    do_replace,
+};
+REGISTER_CMD(replace_cmd);
diff --git a/src/cmd/reset.c b/src/cmd/reset.c
@@ -2,6 +2,7 @@
 // Copyright (C) 2018,2021 IBM Corp.
 #include "ahb.h"
 #include "ast.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "host.h"
 #include "log.h"
@@ -16,7 +17,7 @@
 #include <string.h>
 #include <unistd.h>
 
-int cmd_reset(const char *name __unused, int argc, char *argv[])
+static int do_reset(const char *name __unused, int argc, char *argv[])
 {
     struct host _host, *host = &_host;
     struct soc _soc, *soc = &_soc;
@@ -99,3 +100,9 @@ int cmd_reset(const char *name __unused, int argc, char *argv[])
 
     return rc;
 }
+
+static const struct cmd reset_cmd = {
+    "reset",
+    do_reset,
+};
+REGISTER_CMD(reset_cmd);
diff --git a/src/cmd/sfc.c b/src/cmd/sfc.c
@@ -2,6 +2,7 @@
 // Copyright (C) 2018,2021 IBM Corp.
 #include "ahb.h"
 #include "ast.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "flash.h"
 #include "host.h"
@@ -19,7 +20,7 @@
 
 enum flash_op { flash_op_read, flash_op_write, flash_op_erase };
 
-int cmd_sfc(const char *name __unused, int argc, char *argv[])
+static int do_sfc(const char *name __unused, int argc, char *argv[])
 {
     struct host _host, *host = &_host;
     struct soc _soc, *soc = &_soc;
@@ -132,3 +133,9 @@ int cmd_sfc(const char *name __unused, int argc, char *argv[])
 
     return rc;
 }
+
+static const struct cmd sfc_cmd = {
+    "sfc",
+    do_sfc,
+};
+REGISTER_CMD(sfc_cmd);
diff --git a/src/cmd/trace.c b/src/cmd/trace.c
@@ -4,6 +4,7 @@
 
 #include "ahb.h"
 #include "ast.h"
+#include "cmd.h"
 #include "compiler.h"
 #include "host.h"
 #include "log.h"
@@ -23,7 +24,7 @@
 //culvert trace 0x1e788000 1:0 read
 //culvert trace 0x1e788000 2:2 read
 //culvert trace 0x1e788000 4:0 write
-int cmd_trace(const char *name __unused, int argc, char *argv[])
+static int do_trace(const char *name __unused, int argc, char *argv[])
 {
     struct host _host, *host = &_host;
     struct soc _soc, *soc = &_soc;
@@ -139,3 +140,9 @@ int cmd_trace(const char *name __unused, int argc, char *argv[])
 
     return rc;
 }
+
+static const struct cmd trace_cmd = {
+    "trace",
+    do_trace,
+};
+REGISTER_CMD(trace_cmd);