[PATCH] lib/copperplate: Release main threadobj on exit

[PATCH] lib/copperplate: Release main threadobj on exit

[Jan Kiszka]

From: Jan Kiszka <jan.kiszka@siemens.com>

In case of CONFIG_XENO_PSHARED, the main thread's tcb is allocated on
the shared heap. Normally, threadobjs are released via the destructor
of threadobj_tskey. But this is not called when the main thread
terminates so that we leak the object. Fix that by installing an atexit
handler that only needs to perform the release now.

Reported-by: Mauro S. <mau.salvi@tin.it>
Signed-off-by: Jan Kiszka <jan.kiszka@siemens.com>
---
 lib/copperplate/threadobj.c | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/lib/copperplate/threadobj.c b/lib/copperplate/threadobj.c
index f4588a17a8..6808bcf164 100644
--- a/lib/copperplate/threadobj.c
+++ b/lib/copperplate/threadobj.c
@@ -1770,6 +1770,13 @@ int threadobj_set_schedprio(struct threadobj *thobj, int priority)
 	return threadobj_set_schedparam(thobj, policy, &param_ex);
 }
 
+#ifdef CONFIG_XENO_PSHARED
+static void main_exit(void)
+{
+	threadobj_free(threadobj_current());
+}
+#endif
+
 static inline int main_overlay(void)
 {
 	struct threadobj_init_data idata;
@@ -1806,6 +1813,10 @@ static inline int main_overlay(void)
 	threadobj_prologue(tcb, NULL);
 	pthread_setcanceltype(PTHREAD_CANCEL_DEFERRED, NULL);
 
+#ifdef CONFIG_XENO_PSHARED
+	atexit(main_exit);
+#endif
+
 	return 0;
 }
 
-- 
2.35.3