Skip to content

Commit

Permalink
using a exponential backoff to check the network device
Browse files Browse the repository at this point in the history
if exponential backoff has completed, the task will be blocked
and waits for an network interrupt
  • Loading branch information
stlankes committed Jul 29, 2023
1 parent 24219de commit 0bd6269
Showing 1 changed file with 21 additions and 27 deletions.
48 changes: 21 additions & 27 deletions src/executor/network.rs
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ use core::ops::DerefMut;
use core::sync::atomic::{AtomicU16, Ordering};
use core::task::{Context, Poll};

use crossbeam_utils::Backoff;
use hermit_sync::InterruptTicketMutex;
use smoltcp::iface::{SocketHandle, SocketSet};
#[cfg(feature = "dhcpv4")]
Expand Down Expand Up @@ -240,8 +241,7 @@ where
// Enter polling mode => no NIC interrupts
set_polling_mode(true);

let mut counter: u16 = 0;
let mut blocking_time = 1000;
let backoff = Backoff::new();
let start = crate::executor::network::now();
let task_notify = Arc::new(TaskNotify::new());
let waker = task_notify.into();
Expand Down Expand Up @@ -277,56 +277,50 @@ where
}
}

counter += 1;
// besure that we are not interrupted by a timer, which is able
// be sure that we are not interrupted by a timer, which is able
// to call `reschedule`
interrupts::disable();
let now = crate::executor::network::now();
let delay = network_delay(now).map(|d| d.total_micros());
if counter > 200 && delay.unwrap_or(10_000_000) > 100_000 {
// add additional check before the task will block
if let Poll::Ready(t) = future.as_mut().poll(&mut cx) {
// allow network interrupts
set_polling_mode(false);
// enable interrupts
interrupts::enable();

return t;
}

if backoff.is_completed() && delay.unwrap_or(10_000_000) > 100_000 {
let ticks = crate::arch::processor::get_timer_ticks();
let wakeup_time = timeout
.map(|duration| {
core::cmp::min(
u64::try_from((start + duration).total_micros()).unwrap(),
ticks + delay.unwrap_or(blocking_time),
)
})
.or(Some(ticks + delay.unwrap_or(blocking_time)));
let wakeup_time =
timeout.map(|duration| u64::try_from((start + duration).total_micros()).unwrap());
let network_timer = delay.map(|d| ticks + d);
let core_scheduler = core_scheduler();
blocking_time *= 2;

core_scheduler.add_network_timer(network_timer);
core_scheduler.block_current_task(wakeup_time);

// allow network interrupts
set_polling_mode(false);

// add additional check if we miss a network interrupt
if let Poll::Ready(t) = future.as_mut().poll(&mut cx) {
// move task from the list of blocked task
let handle = core_scheduler.get_current_task_handle();
core_scheduler.custom_wakeup(handle);

// enable interrupts
interrupts::enable();

return t;
}

// enable interrupts
interrupts::enable();

// switch to another task
core_scheduler.reschedule();

// reset polling counter
counter = 0;
backoff.reset();

// Enter polling mode => no NIC interrupts
set_polling_mode(true);
} else {
// enable interrupts
interrupts::enable();

backoff.snooze();
}
}
}
Expand Down

0 comments on commit 0bd6269

Please sign in to comment.