move more deinit stuff into preempt::disable()

esp-rs · Feb 7, 2025 · e4ac994 · e4ac994
1 parent eb7c31b
commit e4ac994
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/esp-wifi/src/lib.rs b/esp-wifi/src/lib.rs
@@ -449,10 +449,6 @@ pub unsafe fn deinit_unchecked() -> Result<(), InitializationError> {
     // This shuts down the task switcher and timer tick interrupt.
     preempt::disable();
 
-    crate::preempt::delete_all_tasks();
-
-    crate::preempt::timer::TIMER.with(|timer| timer.take());
-
     crate::flags::ESP_WIFI_INITIALIZED.store(false, Ordering::Release);
 
     Ok(())

diff --git a/esp-wifi/src/preempt/mod.rs b/esp-wifi/src/preempt/mod.rs
@@ -33,6 +33,9 @@ pub(crate) fn setup(timer: crate::TimeBase) {
 pub(crate) fn disable() {
     disable_timer();
     disable_multitasking();
+    delete_all_tasks();
+
+    timer::TIMER.with(|timer| timer.take());
 }
 
 fn allocate_main_task() -> *mut Context {