From: Alan T. DeKok <aland@freeradius.org>
Date: Wed, 8 Jun 2022 23:58:24 +0000 (-0400)
Subject: add "cast" xlat, for cases where we don't have the full parser
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=32fd7b3bb17f087cb6ea12b47330d48882dde2b2;p=thirdparty%2Ffreeradius-server.git

add "cast" xlat, for cases where we don't have the full parser
---

diff --git a/src/lib/unlang/xlat_builtin.c b/src/lib/unlang/xlat_builtin.c
index 6fc69d21dbc..544add4dfc8 100644
--- a/src/lib/unlang/xlat_builtin.c
+++ b/src/lib/unlang/xlat_builtin.c
@@ -2040,6 +2040,81 @@ finish:
 	return XLAT_ACTION_DONE;
 }
 
+static xlat_arg_parser_t const xlat_func_cast_args[] = {
+	{ .required = true, .single = true, .type = FR_TYPE_VOID },
+	{ .required = true, .variadic = true, .type = FR_TYPE_VOID },
+	XLAT_ARG_PARSER_TERMINATOR
+};
+
+/** Cast one or more output value-boxes to the given type
+ *
+ * First argument of is type to cast to.
+ *
+ * Example:
+@verbatim
+%(cast:string %{request[*]}) results in all of the input boxes being cast to string/
+@endverbatim
+ *
+ * @ingroup xlat_functions
+ */
+static xlat_action_t xlat_func_cast(UNUSED TALLOC_CTX *ctx, fr_dcursor_t *out,
+				    UNUSED xlat_ctx_t const *xctx,
+				    request_t *request, fr_value_box_list_t *in)
+{
+	fr_value_box_t	*name = fr_dlist_head(in);
+	fr_value_box_t	*arg;
+	fr_type_t	type;
+
+	/*
+	 *	Get the type, which can be in one of a few formats.
+	 */
+	if (fr_type_is_numeric(name->type)) {
+		if (fr_value_box_cast_in_place(name, name, FR_TYPE_UINT8, NULL) < 0) {
+			RPEDEBUG("Failed parsing '%pV' as a numerical data type", name);
+			return XLAT_ACTION_FAIL;
+		}
+		type = name->vb_uint8;
+
+	} else {
+		if (name->type != FR_TYPE_STRING) {
+			if (fr_value_box_cast_in_place(name, name, FR_TYPE_STRING, NULL) < 0) {
+				RPEDEBUG("Failed parsing '%pV' as a string data type", name);
+				return XLAT_ACTION_FAIL;
+			}
+		}
+
+		type = fr_table_value_by_str(fr_type_table, name->vb_strvalue, FR_TYPE_NULL);
+		if (type == FR_TYPE_NULL) {
+			RDEBUG("Unknown data type '%s'", name->vb_strvalue);
+			return XLAT_ACTION_FAIL;
+		}
+	}
+
+	/*
+	 *	Copy inputs to outputs, casting them along the way.
+	 */
+	arg = name;
+	while ((arg = fr_dlist_next(in, arg)) != NULL) {
+		fr_value_box_t	*vb, *p;
+
+		fr_assert(arg->type == FR_TYPE_GROUP);
+
+		vb = fr_dlist_head(&arg->vb_group);
+		while (vb) {
+			p = fr_dlist_remove(&arg->vb_group, vb);
+
+			if (fr_value_box_cast_in_place(vb, vb, type, NULL) < 0) {
+				RPEDEBUG("Failed casting %pV to data type %s", vb, fr_type_to_str(type));
+				return XLAT_ACTION_FAIL;
+			}
+			fr_dcursor_append(out, vb);
+			vb = fr_dlist_next(&arg->vb_group, p);
+		}
+	}
+
+	return XLAT_ACTION_DONE;
+}
+
 static xlat_arg_parser_t const xlat_func_concat_args[] = {
 	{ .required = true, .type = FR_TYPE_VOID },
 	{ .concat = true, .type = FR_TYPE_STRING },
@@ -3686,6 +3761,7 @@ do { \
 	xlat_internal(xlat); \
 } while (0)
 
+	XLAT_REGISTER_ARGS("cast", xlat_func_cast, xlat_func_cast_args);
 	XLAT_REGISTER_ARGS("concat", xlat_func_concat, xlat_func_concat_args);
 	XLAT_REGISTER_ARGS("explode", xlat_func_explode, xlat_func_explode_args);
 	XLAT_REGISTER_ARGS("hmacmd5", xlat_func_hmac_md5, xlat_hmac_args);