tests: Add test to connect to the guest using TCP (vsock TSI backend)

Signed-off-by: Matej Hrica <[email protected]>

Author: mtjhrc

tests/test_cases/src/lib.rs

@@ -7,6 +7,9 @@ use test_vsock_guest_connect::TestVsockGuestConnect;
 mod test_tsi_tcp_guest_connect;
 use test_tsi_tcp_guest_connect::TestTsiTcpGuestConnect;
 
+mod test_tsi_tcp_guest_listen;
+use test_tsi_tcp_guest_listen::TestTsiTcpGuestListen;
+
 pub fn test_cases() -> Vec<TestCase> {
     // Register your test here:
     vec![
@@ -26,6 +29,10 @@ pub fn test_cases() -> Vec<TestCase> {
         ),
         TestCase::new("vsock-guest-connect", Box::new(TestVsockGuestConnect)),
         TestCase::new("tsi-tcp-guest-connect", Box::new(TestTsiTcpGuestConnect::new())),
+        TestCase::new(
+            "tsi-tcp-guest-listen",
+            Box::new(TestTsiTcpGuestListen::new()),
+        ),
     ]
 }
 

tests/test_cases/src/test_tsi_tcp_guest_listen.rs

@@ -0,0 +1,64 @@
+use crate::tcp_tester::TcpTester;
+use macros::{guest, host};
+
+const PORT: u16 = 8001;
+
+pub struct TestTsiTcpGuestListen {
+    tcp_tester: TcpTester,
+}
+
+impl TestTsiTcpGuestListen {
+    pub fn new() -> Self {
+        Self {
+            tcp_tester: TcpTester::new(PORT),
+        }
+    }
+}
+
+#[host]
+mod host {
+    use super::*;
+    use crate::common::setup_fs_and_enter;
+    use crate::{krun_call, krun_call_u32, Test, TestSetup};
+    use krun_sys::*;
+    use std::ffi::CString;
+    use std::ptr::null;
+    use std::thread;
+    use std::time::Duration;
+
+    impl Test for TestTsiTcpGuestListen {
+        fn start_vm(self: Box<Self>, test_setup: TestSetup) -> anyhow::Result<()> {
+            unsafe {
+                thread::spawn(move || {
+                    thread::sleep(Duration::from_secs(1));
+                    self.tcp_tester.run_client();
+                });
+
+                let ctx = krun_call_u32!(krun_create_ctx())?;
+                let port_mapping = format!("{PORT}:{PORT}");
+                let port_mapping = CString::new(port_mapping).unwrap();
+                let port_map = [port_mapping.as_ptr(), null()];
+
+                krun_call!(krun_set_port_map(ctx, port_map.as_ptr()))?;
+                krun_call!(krun_set_vm_config(ctx, 1, 512))?;
+                setup_fs_and_enter(ctx, test_setup)?;
+                println!("OK");
+            }
+            Ok(())
+        }
+    }
+}
+
+#[guest]
+mod guest {
+    use super::*;
+    use crate::Test;
+
+    impl Test for TestTsiTcpGuestListen {
+        fn in_guest(self: Box<Self>) {
+            let listener = self.tcp_tester.create_server_socket();
+            self.tcp_tester.run_server(listener);
+            println!("OK");
+        }
+    }
+}