Fix task cleanup

src/system/emscripten/system.c

@@ -58,7 +58,11 @@ z_result_t _z_task_cancel(_z_task_t *task) { _Z_CHECK_SYS_ERR(pthread_cancel(*ta
 
 void _z_task_exit(void) { pthread_exit(NULL); }
 
-void _z_task_free(_z_task_t **task) { *task = NULL; }
+void _z_task_free(_z_task_t **task) {
+    _z_task_t *ptr = *task;
+    z_free(ptr);
+    *task = NULL;
+}
 
 /*------------------ Mutex ------------------*/
 z_result_t _z_mutex_init(_z_mutex_t *m) { _Z_CHECK_SYS_ERR(pthread_mutex_init(m, 0)); }

src/system/espidf/system.c

@@ -129,8 +129,9 @@ z_result_t z_task_cancel(_z_task_t *task) {
 void _z_task_exit(void) { vTaskDelete(NULL); }
 
 void _z_task_free(_z_task_t **task) {
-    z_free((*task)->join_event);
-    z_free(*task);
+    _z_task_t *ptr = *task;
+    z_free(ptr->join_event);
+    z_free(ptr);
     *task = NULL;
 }
 

src/system/flipper/system.c

@@ -98,6 +98,7 @@ void _z_task_free(_z_task_t** task) {
         return;
     }
     furi_thread_free(**task);
+    z_free(*task);
     *task = NULL;
 }
 

src/system/freertos_plus_tcp/system.c

@@ -164,6 +164,7 @@ void _z_task_exit(void) { vTaskDelete(NULL); }
 void _z_task_free(_z_task_t **task) {
     _z_task_t *ptr = *task;
     vEventGroupDelete(ptr->join_event);
+    z_free(ptr);
     *task = NULL;
 }
 

src/system/rpi_pico/system.c

@@ -120,8 +120,9 @@ z_result_t _z_task_cancel(_z_task_t *task) {
 void _z_task_exit(void) { vTaskDelete(NULL); }
 
 void _z_task_free(_z_task_t **task) {
-    z_free((*task)->join_event);
-    z_free(*task);
+    _z_task_t *ptr = *task;
+    z_free(ptr->join_event);
+    z_free(ptr);
     *task = NULL;
 }
 

src/system/unix/system.c

@@ -117,7 +117,11 @@ z_result_t _z_task_cancel(_z_task_t *task) { _Z_CHECK_SYS_ERR(pthread_cancel(*ta
 
 void _z_task_exit(void) { pthread_exit(NULL); }
 
-void _z_task_free(_z_task_t **task) { *task = NULL; }
+void _z_task_free(_z_task_t **task) {
+    _z_task_t *ptr = *task;
+    z_free(ptr);
+    *task = NULL;
+}
 
 /*------------------ Mutex ------------------*/
 z_result_t _z_mutex_init(_z_mutex_t *m) { _Z_CHECK_SYS_ERR(pthread_mutex_init(m, 0)); }

src/system/windows/system.c

@@ -79,6 +79,7 @@ z_result_t _z_task_join(_z_task_t *task) {
 z_result_t _z_task_detach(_z_task_t *task) {
     z_result_t ret = _Z_RES_OK;
     CloseHandle(*task);
+    *task = 0;
     return ret;
 }
 
@@ -92,7 +93,9 @@ void _z_task_exit(void) { ExitThread(0); }
 
 void _z_task_free(_z_task_t **task) {
     _z_task_t *ptr = *task;
-    CloseHandle(*ptr);
+    if (*ptr != 0) {
+        CloseHandle(*ptr);
+    }
     z_free(ptr);
     *task = NULL;
 }

src/system/zephyr/system.c

@@ -92,7 +92,11 @@ z_result_t _z_task_cancel(_z_task_t *task) { _Z_CHECK_SYS_ERR(pthread_cancel(*ta
 
 void _z_task_exit(void) { pthread_exit(NULL); }
 
-void _z_task_free(_z_task_t **task) { *task = NULL; }
+void _z_task_free(_z_task_t **task) {
+    _z_task_t *ptr = *task;
+    z_free(ptr);
+    *task = NULL;
+}
 
 /*------------------ Mutex ------------------*/
 z_result_t _z_mutex_init(_z_mutex_t *m) { _Z_CHECK_SYS_ERR(pthread_mutex_init(m, 0)); }

src/transport/common/transport.c

@@ -30,7 +30,6 @@ void _z_common_transport_clear(_z_transport_common_t *ztc, bool detach_tasks) {
             _z_task_join(ztc->_read_task);
         }
         _z_task_free(&ztc->_read_task);
-        z_free(ztc->_read_task);
         ztc->_read_task = NULL;
     }
     if (ztc->_lease_task != NULL) {
@@ -41,7 +40,6 @@ void _z_common_transport_clear(_z_transport_common_t *ztc, bool detach_tasks) {
             _z_task_join(ztc->_lease_task);
         }
         _z_task_free(&ztc->_lease_task);
-        z_free(ztc->_lease_task);
         ztc->_lease_task = NULL;
     }
 

src/transport/unicast/lease.c

@@ -57,6 +57,8 @@ static void _zp_unicast_failed(_z_transport_unicast_t *ztu) {
         _Z_ERROR("Reopen failed: %i", ret);
     }
 #endif
+
+    _z_task_exit();
 }
 
 void *_zp_unicast_lease_task(void *ztu_arg) {