call wasm function to trace the network traffic

Author: stlankes

src/drivers/net/virtio/mod.rs

@@ -265,6 +265,12 @@ impl smoltcp::phy::TxToken for TxToken<'_> {
 			result
 		};
 
+		#[cfg(feature = "wasm")]
+		if let Some(ref mut wasm_manager) = crate::wasm::WASM_MANAGER.lock().as_mut() {
+			crate::wasm::INPUT.lock().push_back(packet.to_vec());
+			let _ = wasm_manager.call_func::<(), ()>("pcap_writer", ());
+		}
+
 		let mut header = Box::new_in(<Hdr as Default>::default(), DeviceAlloc);
 
 		// If a checksum calculation by the host is necessary, we have to inform the host within the header
@@ -361,6 +367,14 @@ impl smoltcp::phy::RxToken for RxToken<'_> {
 				.unwrap();
 		}
 
+		#[cfg(feature = "wasm")]
+		if let Some(ref mut wasm_manager) = crate::wasm::WASM_MANAGER.lock().as_mut() {
+			crate::wasm::INPUT
+				.lock()
+				.push_back(combined_packets.to_vec());
+			let _ = wasm_manager.call_func::<(), ()>("pcap_writer", ());
+		}
+
 		f(&combined_packets)
 	}
 }

src/wasm/mod.rs

@@ -579,39 +579,37 @@ impl WasmManager {
 }
 
 async fn wasm_run() {
-	loop {
-		future::poll_fn(|cx| {
-			if let Some(mut guard) = OUTPUT.try_lock() {
-				if let Some((fd, data)) = guard.data.pop_front() {
-					let obj = match fd {
-						Descriptor::Stdout => crate::core_scheduler()
-							.get_object(fd::STDOUT_FILENO)
-							.unwrap(),
-						Descriptor::Stderr => crate::core_scheduler()
-							.get_object(fd::STDERR_FILENO)
-							.unwrap(),
-						Descriptor::RawFd(raw_fd) => {
-							crate::core_scheduler().get_object(raw_fd).unwrap()
-						}
-						_ => panic!("Unsuppted {fd:?}"),
-					};
+	future::poll_fn(|cx| {
+		if let Some(mut guard) = OUTPUT.try_lock() {
+			if let Some((fd, data)) = guard.data.pop_front() {
+				let obj = match fd {
+					Descriptor::Stdout => crate::core_scheduler()
+						.get_object(fd::STDOUT_FILENO)
+						.unwrap(),
+					Descriptor::Stderr => crate::core_scheduler()
+						.get_object(fd::STDERR_FILENO)
+						.unwrap(),
+					Descriptor::RawFd(raw_fd) => {
+						crate::core_scheduler().get_object(raw_fd).unwrap()
+					}
+					_ => panic!("Unsuppted {fd:?}"),
+				};
 
-					drop(guard);
-					while let Poll::Pending = pin!(obj.write(&data)).poll(cx) {}
+				drop(guard);
+				while let Poll::Pending = pin!(obj.write(&data)).poll(cx) {}
 
-					cx.waker().wake_by_ref();
-					Poll::<()>::Pending
-				} else {
-					guard.waker.register(cx.waker());
-					Poll::<()>::Pending
-				}
-			} else {
 				cx.waker().wake_by_ref();
 				Poll::<()>::Pending
+			} else {
+				guard.waker.register(cx.waker());
+				Poll::<()>::Pending
 			}
-		})
-		.await;
-	}
+		} else {
+			cx.waker().wake_by_ref();
+			Poll::<()>::Pending
+		}
+	})
+	.await;
 }
 
 #[hermit_macro::system]