From 464299c202f3ba963aed821b777075397e843856 Mon Sep 17 00:00:00 2001
From: S-P Chan <shihping.chan@gmail.com>
Date: Mon, 4 Mar 2024 22:00:14 +0800
Subject: [PATCH] tls: new option tls_threads_mode = 2

- use pthread_atfork to force all thread-locals
  to 0x0 after fork()
---
 src/modules/tls/tls_mod.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/src/modules/tls/tls_mod.c b/src/modules/tls/tls_mod.c
index 1dc9357f56a..6dfde00c93a 100644
--- a/src/modules/tls/tls_mod.c
+++ b/src/modules/tls/tls_mod.c
@@ -337,6 +337,20 @@ static tls_domains_cfg_t* tls_use_modparams(void)
 }
 #endif
 
+/* global config tls_threads_mode = 2
+ *  - force all thread-locals to be 0x0 after fork()
+ *  - with OpenSSL loaded the largest value observed
+ *     is < 10
+ *
+ */
+static void fork_child(void)
+{
+	for(int k = 0; k < 16; k++) {
+		if(pthread_getspecific(k) != 0)
+			pthread_setspecific(k, 0x0);
+	}
+}
+
 static int mod_init(void)
 {
 	int method;
@@ -446,6 +460,9 @@ static int mod_init(void)
 		ksr_module_set_flag(KSRMOD_FLAG_POSTCHILDINIT);
 	}
 #endif
+	if(ksr_tls_threads_mode == 2) {
+		pthread_atfork(NULL, NULL, &fork_child);
+	}
 	return 0;
 error:
 	tls_h_mod_destroy_f();