From e24756294b2e8f39c5de718ae2f58b9db1449e5a Mon Sep 17 00:00:00 2001
From: Masatake YAMATO <yamato@redhat.com>
Date: Fri, 23 Sep 2022 10:32:56 +0900
Subject: [PATCH] tests: (mkfds) add boolean, a new parameter type

Signed-off-by: Masatake YAMATO <yamato@redhat.com>
---
 tests/helpers/test_mkfds.c | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/tests/helpers/test_mkfds.c b/tests/helpers/test_mkfds.c
index b32584a8a5..b688633668 100644
--- a/tests/helpers/test_mkfds.c
+++ b/tests/helpers/test_mkfds.c
@@ -74,11 +74,13 @@ static void __attribute__((__noreturn__)) usage(FILE *out, int status)
 union value {
 	const char *string;
 	long integer;
+	bool boolean;
 };
 
 enum ptype {
 	PTYPE_STRING,
 	PTYPE_INTEGER,
+	PTYPE_BOOLEAN,
 };
 
 struct ptype_class {
@@ -99,6 +101,7 @@ struct ptype_class {
 
 #define ARG_STRING(A) (A.v.string)
 #define ARG_INTEGER(A) (A.v.integer)
+#define ARG_BOOLEAN(A) (A.v.boolean)
 struct arg {
 	union value v;
 	void (*free)(union value value);
@@ -155,6 +158,32 @@ static void integer_free(union value value _U_)
 	/* Do nothing */
 }
 
+static char *boolean_sprint(const union value *value)
+{
+	return xstrdup(value->boolean? "true": "false");
+}
+
+static union value boolean_read(const char *arg, const union value *defv)
+{
+	union value r;
+
+	if (!arg)
+		return *defv;
+
+	if (strcasecmp(arg, "true") == 0
+	    || strcmp(arg, "1") == 0
+	    || strcasecmp(arg, "yes") == 0
+	    || strcasecmp(arg, "y") == 0)
+		r.boolean = true;
+	else
+		r.boolean = false;
+	return r;
+}
+
+static void boolean_free(union value value _U_)
+{
+	/* Do nothing */
+}
 
 struct ptype_class ptype_classes [] = {
 	[PTYPE_STRING] = {
@@ -169,6 +198,12 @@ struct ptype_class ptype_classes [] = {
 		.read   = integer_read,
 		.free   = integer_free,
 	},
+	[PTYPE_BOOLEAN] = {
+		.name = "boolean",
+		.sprint = boolean_sprint,
+		.read   = boolean_read,
+		.free   = boolean_free,
+	},
 };
 
 static struct arg decode_arg(const char *pname,
-- 
2.47.3