From: Arran Cudbard-Bell <a.cudbardb@freeradius.org>
Date: Mon, 25 Oct 2021 18:14:44 +0000 (-0400)
Subject: Add macro for automatic thread-local sbuffs
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=d4b26304375059fa6110533b78bf1f63d7043764;p=thirdparty%2Ffreeradius-server.git

Add macro for automatic thread-local sbuffs
---

diff --git a/src/lib/util/sbuff.h b/src/lib/util/sbuff.h
index b2c03e375ce..2e3c26736ac 100644
--- a/src/lib/util/sbuff.h
+++ b/src/lib/util/sbuff.h
@@ -499,6 +499,38 @@ do { \
 	.is_const	= IS_CONST(char *, _start) \
 })
 
+/** Structure to encapsulate a thread local sbuff information
+ *
+ */
+typedef struct {
+	fr_sbuff_t			sbuff;		//!< Thread local sbuff.
+	fr_sbuff_uctx_talloc_t		tctx;		//!< Thread local tctx.
+} fr_sbuff_thread_local_t;
+
+static inline void _sbuff_thread_local_free(void *sbtl)
+{
+	talloc_free(sbtl);
+}
+
+/** Create a function local and thread local extensible sbuff
+ *
+ * @param[out] _sbuff_out	Where to write a pointer to the thread local sbuff
+ * @param[in] _init		Initial size for the sbuff buffer.
+ * @param[in] _max		Maximum size of the sbuff buffer.
+ */
+#define FR_SBUFF_TALLOC_THREAD_LOCAL(_out, _init, _max) \
+{ \
+	static _Thread_local fr_sbuff_thread_local_t *_sbuff_t_local; \
+	if (!_sbuff_t_local) { \
+		fr_sbuff_thread_local_t *sbtl = talloc_zero(NULL, fr_sbuff_thread_local_t); \
+		fr_sbuff_init_talloc(sbtl, &sbtl->sbuff, &sbtl->tctx, _init, _max); \
+		fr_atexit_thread_local(_sbuff_t_local, _sbuff_thread_local_free, sbtl); \
+		*(_out) = &_sbuff_t_local->sbuff; \
+	} else { \
+		fr_sbuff_reset_talloc(&_sbuff_t_local->sbuff); \
+		*(_out) = &_sbuff_t_local->sbuff; \
+	} \
+}
 
 void	fr_sbuff_update(fr_sbuff_t *sbuff, char *new_buff, size_t new_len);