feat(allocation-stats): support regularly printing allocation stats

Talc can optionally collect and expose some basic statistics about
allocations. Provide a feature that regularly prints these statistics
via another executor tassk.

Author: hcsch

Cargo.toml

@@ -69,6 +69,7 @@ trace = []
 udp = ["smoltcp", "smoltcp/socket-udp"]
 vga = []
 vsock = ["pci"]
+allocation-stats = ["talc/counters"]
 
 [lints.rust]
 rust_2018_idioms = "warn"

src/executor/alloc_stats.rs

@@ -0,0 +1,20 @@
+use core::task::Poll;
+
+use crate::executor::spawn;
+use crate::mm::ALLOCATOR;
+
+async fn print_alloc_stats() {
+	core::future::poll_fn::<(), _>(|_cx| {
+		let talc = ALLOCATOR.inner().lock();
+
+		debug!("<alloc-stats>\n{}", talc.get_counters());
+
+		Poll::Pending
+	})
+	.await;
+}
+
+pub(crate) fn init() {
+	info!("Spawning allocation stats printing task");
+	spawn(print_alloc_stats());
+}

src/executor/mod.rs

@@ -1,5 +1,7 @@
 #![allow(dead_code)]
 
+#[cfg(feature = "allocation-stats")]
+mod alloc_stats;
 #[cfg(any(feature = "tcp", feature = "udp"))]
 pub(crate) mod device;
 #[cfg(any(feature = "tcp", feature = "udp"))]
@@ -127,6 +129,8 @@ pub fn init() {
 	crate::executor::network::init();
 	#[cfg(feature = "vsock")]
 	crate::executor::vsock::init();
+	#[cfg(feature = "allocation-stats")]
+	crate::executor::alloc_stats::init();
 }
 
 /// Blocks the current thread on `f`, running the executor when idling.