Changeset 6a57b93 in mainline

Timestamp:

2025-03-05T21:40:21Z (4 days ago)

Author:

Matěj Volf <git@…>

Children:

5208709

Parents:

df2b4ce7

git-author:

Matěj Volf <git@…> (2025-01-17 14:14:59)

git-committer:

Matěj Volf <git@…> (2025-03-05 21:40:21)

Message:

Add fibril exit hook

Location:

uspace/lib/c

Files:

• generic/private/fibril.h (modified) (1 diff)
• generic/thread/fibril.c (modified) (4 diffs)
• include/fibril.h (modified) (1 diff)

uspace/lib/c/generic/private/fibril.h

@@ -70 +70 @@
         fibril_owner_info_t *waits_for;
         fibril_event_t *sleep_event;
+
+        list_t exit_hooks;
 };
+
+typedef struct {
+        link_t link;
+        void (*func)(void);
+} fibril_hook_t;
 
 extern fibril_t *fibril_alloc(void);

uspace/lib/c/generic/thread/fibril.c

@@ -198 +198 @@
         if (fibril->is_freeable) {
                 tls_free(fibril->tcb);
+                list_foreach_safe(fibril->exit_hooks, cur, _next) {
+                        fibril_hook_t *hook = list_get_instance(cur, fibril_hook_t, link);
+                        free(hook);
+                }
                 free(fibril);
         }
@@ -562 +566 @@
         fibril->arg = arg;
 
+        list_initialize(&fibril->exit_hooks);
+
         context_create_t sctx = {
                 .fn = _fibril_main,
@@ -855 +861 @@
         (void) retval;
 
+        list_foreach(fibril_self()->exit_hooks, link, fibril_hook_t, hook) {
+                hook->func();
+        }
+
         fibril_t *f = _ready_list_pop_nonblocking(false);
         if (!f)
@@ -918 +928 @@
 }
 
+errno_t fibril_add_exit_hook(void (*hook)(void))
+{
+        fibril_hook_t *h = malloc(sizeof(fibril_hook_t));
+        if (!h)
+                return ENOMEM;
+
+        h->func = hook;
+        list_append(&h->link, &fibril_self()->exit_hooks);
+        return EOK;
+}
+
 errno_t fibril_ipc_wait(ipc_call_t *call, const struct timespec *expires)
 {

uspace/lib/c/include/fibril.h

@@ -74 +74 @@
 extern __noreturn void fibril_exit(long);
 
+/** Add a function to be called after fibril exits, just before it is destroyed */
+extern errno_t fibril_add_exit_hook(void (*)(void));
+
 __HELENOS_DECLS_END;