From: Nick Porter <nick@portercomputing.co.uk>
Date: Fri, 2 Feb 2024 14:07:35 +0000 (+0000)
Subject: Add sql escape uctx alloc / free functions for SQL IP Pool queries
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=b4658e657faf3069a2af746df6153b5168f9fe27;p=thirdparty%2Ffreeradius-server.git

Add sql escape uctx alloc / free functions for SQL IP Pool queries
---

diff --git a/src/modules/rlm_sqlippool/rlm_sqlippool.c b/src/modules/rlm_sqlippool/rlm_sqlippool.c
index 6497c77ce92..a1b54a9f623 100644
--- a/src/modules/rlm_sqlippool/rlm_sqlippool.c
+++ b/src/modules/rlm_sqlippool/rlm_sqlippool.c
@@ -110,6 +110,27 @@ static conf_parser_t module_config[] = {
 	CONF_PARSER_TERMINATOR
 };
 
+static int _sql_escape_uxtx_free(void *uctx)
+{
+	return talloc_free(uctx);
+}
+
+static void *sql_escape_uctx_alloc(request_t *request, void const *uctx)
+{
+	static _Thread_local rlm_sql_escape_uctx_t	*t_ctx;
+
+	if (unlikely(t_ctx == NULL)) {
+		rlm_sql_escape_uctx_t *ctx;
+
+		MEM(ctx = talloc_zero(NULL, rlm_sql_escape_uctx_t));
+		fr_atexit_thread_local(t_ctx, _sql_escape_uxtx_free, ctx);
+	}
+	t_ctx->sql = uctx;
+	t_ctx->handle = request_data_reference(request, (void *)sql_escape_uctx_alloc, 0);
+
+	return t_ctx;
+}
+
 /** Perform a single sqlippool query
  *
  * Mostly wrapper around sql_query which does some special sqlippool sequence substitutions and expands