disable network interrupts in polling mode

hermit-os · Aug 3, 2023 · 8f36730 · 8f36730
1 parent 850f267
commit 8f36730
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/executor/network.rs b/src/executor/network.rs
@@ -334,6 +334,9 @@ where
 	// be sure that we are not interrupted by a timer, which is able
 	// to call `reschedule`
 	without_interrupts(|| {
+		// avoid network interrupts
+		get_network_driver().unwrap().lock().set_polling_mode(true);
+
 		let start = now();
 		let waker = core::task::Waker::noop();
 		let mut cx = Context::from_waker(&waker);
@@ -349,6 +352,9 @@ where
 					.map(|d| crate::arch::processor::get_timer_ticks() + d.total_micros());
 				core_scheduler().add_network_timer(wakeup_time);
 
+				// allow network interrupts
+				get_network_driver().unwrap().lock().set_polling_mode(false);
+
 				return t;
 			}
 
@@ -358,6 +364,9 @@ where
 						.map(|d| crate::arch::processor::get_timer_ticks() + d.total_micros());
 					core_scheduler().add_network_timer(wakeup_time);
 
+					// allow network interrupts
+					get_network_driver().unwrap().lock().set_polling_mode(false);
+
 					return Err(-crate::errno::ETIME);
 				}
 			}