From 897bd990f9e5ef1be358172c6804d49ffe7c453c Mon Sep 17 00:00:00 2001
From: Alejandro Soto <alejandro@34project.org>
Date: Wed, 4 Oct 2023 00:58:57 -0600
Subject: demo: implement run, halt cmds

---
 demo/main.c | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

(limited to 'demo/main.c')

diff --git a/demo/main.c b/demo/main.c
index f75f6f1..efc183b 100644
--- a/demo/main.c
+++ b/demo/main.c
@@ -4,6 +4,24 @@ static struct cpu cpu0, cpu1, cpu2, cpu3;
 
 struct cpu *__cpus[] = { &cpu0, &cpu1, &cpu2, &cpu3 };
 
+static void cmd_run(char **tokens)
+{
+	unsigned mask;
+	if (parse_cpu_mask(tokens, &mask) < 0)
+		return;
+
+	run_cpus(mask);
+}
+
+static void cmd_halt(char **tokens)
+{
+	unsigned mask;
+	if (parse_cpu_mask(tokens, &mask) < 0)
+		return;
+
+	halt_cpus(mask);
+}
+
 void bsp_main(void)
 {
 	run_cpu(1);
@@ -21,7 +39,12 @@ void bsp_main(void)
 		if (!cmd)
 			continue;
 
-		print("unknown command '%s'", cmd);
+		if (!strcmp(cmd, "run"))
+			cmd_run(&tokens);
+		else if (!strcmp(cmd, "halt"))
+			cmd_halt(&tokens);
+		else
+			print("unknown command '%s'", cmd);
 	}
 }
 
-- 
cgit v1.2.3