Add TSI/TCP tests

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

Author: mtjhrc

tests/test_cases/src/lib.rs

@@ -4,6 +4,9 @@ use test_vm_config::TestVmConfig;
 mod test_vsock_guest_connect;
 use test_vsock_guest_connect::TestVsockGuestConnect;
 
+mod test_tsi_tcp_guest_connect;
+use test_tsi_tcp_guest_connect::TestTsiTcpGuestConnect;
+
 pub fn test_cases() -> Vec<TestCase> {
     // Register your test here:
     vec![
@@ -22,6 +25,7 @@ pub fn test_cases() -> Vec<TestCase> {
             }),
         ),
         TestCase::new("vsock-guest-connect", Box::new(TestVsockGuestConnect)),
+        TestCase::new("tsi-tcp-guest-connect", Box::new(TestTsiTcpGuestConnect)),
     ]
 }
 

tests/test_cases/src/test_tsi_tcp_guest_connect.rs

@@ -0,0 +1,105 @@
+use macros::{guest, host};
+use std::io::{ErrorKind, Read};
+use std::mem;
+use std::net::TcpStream;
+use std::time::Duration;
+
+pub struct TestTsiTcpGuestConnect;
+
+fn stream_expect_msg(stream: &mut TcpStream, expected: &[u8]) {
+    let mut buf = vec![0; expected.len()];
+    stream.read_exact(&mut buf[..]).unwrap();
+    assert_eq!(&buf[..], expected);
+}
+
+fn stream_expect_wouldblock(stream: &mut TcpStream) {
+    stream.set_nonblocking(true).unwrap();
+    let err = stream.read(&mut [0u8; 1]).unwrap_err();
+    stream.set_nonblocking(false).unwrap();
+    assert_eq!(err.kind(), ErrorKind::WouldBlock);
+}
+
+fn stream_set_timeouts(stream: &mut TcpStream) {
+    stream
+        .set_read_timeout(Some(Duration::from_millis(500)))
+        .unwrap();
+    stream
+        .set_write_timeout(Some(Duration::from_millis(500)))
+        .unwrap();
+}
+
+#[host]
+mod host {
+    use super::*;
+
+    use crate::common::setup_fs_and_enter;
+    use crate::{krun_call, krun_call_u32};
+    use crate::{Test, TestSetup};
+    use krun_sys::*;
+    use std::ffi::CStr;
+    use std::io::Write;
+    use std::net::TcpListener;
+    use std::thread;
+
+    fn server(listener: TcpListener) {
+        let (mut stream, _addr) = listener.accept().unwrap();
+        eprintln!("accepted a new connection");
+        stream_set_timeouts(&mut stream);
+        stream.write_all(b"ping!").unwrap();
+        stream_expect_msg(&mut stream, b"pong!");
+        stream_expect_wouldblock(&mut stream);
+        stream.write_all(b"bye!").unwrap();
+        mem::forget(listener);
+    }
+
+    impl Test for TestTsiTcpGuestConnect {
+        fn start_vm(self: Box<Self>, test_setup: TestSetup) -> anyhow::Result<()> {
+            unsafe {
+                krun_call!(krun_set_log_level(0));
+            }
+            let listener = TcpListener::bind("0.0.0.0:8121").unwrap();
+            thread::spawn(move || server(listener));
+            thread::sleep(Duration::from_secs(1));
+
+            eprintln!("spawned server");
+            unsafe {
+                let ctx = krun_call_u32!(krun_create_ctx())?;
+                krun_call!(krun_set_vm_config(ctx, 1, 512))?;
+                setup_fs_and_enter(ctx, test_setup)?;
+            }
+            Ok(())
+        }
+    }
+}
+
+#[guest]
+mod guest {
+    use super::*;
+    use crate::Test;
+    use std::io::{stderr, Write};
+    use std::net::SocketAddr;
+    use std::str::FromStr;
+
+    impl Test for TestTsiTcpGuestConnect {
+        fn in_guest(self: Box<Self>) {
+            eprintln!("In guest");
+            let addr = SocketAddr::from_str("127.0.0.1:8121").unwrap();
+            let mut stream = match TcpStream::connect(&addr) {
+                Ok(stream) => stream,
+                Err(err) => {
+                    let raw_err = err.raw_os_error().unwrap();
+                    eprintln!("connect error: {err}, ({raw_err})");
+                    return;
+                }
+            };
+            stream_set_timeouts(&mut stream);
+            eprintln!("guest connected!");
+            stderr().flush();
+            stream_expect_msg(&mut stream, b"ping!");
+            stream_expect_wouldblock(&mut stream);
+            stream.write_all(b"pong!").unwrap();
+            stream_expect_msg(&mut stream, b"bye!");
+            println!("OK");
+        }
+    }
+}