demo: implement run, halt cmds

1 files changed, 28 insertions, 0 deletions

diff --git a/demo/util.c b/demo/util.c
index 99566bd..6bc1920 100644
--- a/demo/util.c
+++ b/demo/util.c
@@ -2,6 +2,11 @@
 
 #include "demo.h"
 
+static void bad_input(void)
+{
+	print("bad input");
+}
+
 int strcmp(const char *s1, const char *s2)
 {
     while (*s1 && *s1 == *s2)
@@ -30,3 +35,26 @@ char *strtok_input(char **tokens)
 
 	return start;
 }
+
+int parse_cpu_mask(char **tokens, unsigned *mask)
+{
+	*mask = 0;
+
+	char *token;
+	while ((token = strtok_input(tokens))) {
+		if (token[0] != 'c' || token[1] != 'p' || token[2] != 'u'
+		|| !('0' <= token[3] && token[3] < '0' + NUM_CPUS) || token[4]) {
+			bad_input();
+			return -1;
+		}
+
+		*mask |= 1 << (token[3] - '0');
+	}
+
+	if (!*mask) {
+		print("must specify at least one cpu");
+		return -1;
+	}
+
+	return 0;
+}