From 8ba54af71d0777f11d4358108eba2a9f56111a8f Mon Sep 17 00:00:00 2001
From: fazlamehrab <fazla.mehrab.akm@intel.com>
Date: Mon, 22 Jul 2019 12:32:24 -0700
Subject: [PATCH] vm-virtio: Add integration test for virtio console device

Two integration tests are added for testing the implemented virtio
console device for single port operation. One checks the presence
and the simple stdout operation. The other test checks the stdout
on file (option: file) using virtio console.

Signed-off-by: A K M Fazla Mehrab <fazla.mehrab.akm@intel.com>
---
 src/main.rs | 129 ++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 125 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 6aebcd7de..be5888fdf 100755
--- a/src/main.rs
+++ b/src/main.rs
@@ -172,7 +172,7 @@ mod tests {
     use std::fs::{self, read, OpenOptions};
     use std::io::{Read, Write};
     use std::net::TcpStream;
-    use std::process::Command;
+    use std::process::{Command, Stdio};
     use std::string::String;
     use std::sync::Mutex;
     use std::thread;
@@ -411,6 +411,45 @@ mod tests {
                 .trim()
                 .to_string()
         }
+        fn get_pci_device_ids(&self) -> String {
+            self.ssh_command("cat /sys/bus/pci/devices/*/device")
+                .trim()
+                .to_string()
+        }
+
+        fn get_pci_vendor_ids(&self) -> String {
+            self.ssh_command("cat /sys/bus/pci/devices/*/vendor")
+                .trim()
+                .to_string()
+        }
+
+        fn is_console_detected(&self) -> bool {
+            !(self
+                .ssh_command("dmesg | grep \"hvc0] enabled\"")
+                .trim()
+                .to_string()
+                .is_empty())
+        }
+
+        fn does_device_vendor_pair_match(&self, device_id: &str, vendor_id: &str) -> bool {
+            // We are checking if console device's device id and vendor id pair matches
+            let devices = self.get_pci_device_ids();
+            let devices: Vec<&str> = devices.split('\n').collect();
+            let vendors = self.get_pci_vendor_ids();
+            let vendors: Vec<&str> = vendors.split('\n').collect();
+
+            for (index, d_id) in devices.iter().enumerate() {
+                if *d_id == device_id {
+                    if let Some(v_id) = vendors.get(index) {
+                        if *v_id == vendor_id {
+                            return true;
+                        }
+                    }
+                }
+            }
+
+            false
+        }
     }
 
     #[test]
@@ -515,7 +554,7 @@ mod tests {
                     .trim()
                     .parse::<u32>()
                     .unwrap(),
-                8
+                10
             );
 
             guest.ssh_command("sudo reboot");
@@ -558,7 +597,7 @@ mod tests {
                     .trim()
                     .parse::<u32>()
                     .unwrap(),
-                8
+                10
             );
 
             guest.ssh_command("sudo reboot");
@@ -601,7 +640,7 @@ mod tests {
                     .trim()
                     .parse::<u32>()
                     .unwrap(),
-                8
+                10
             );
 
             guest.ssh_command("sudo reboot");
@@ -965,4 +1004,86 @@ mod tests {
             Ok(())
         });
     }
+
+    #[test]
+    fn test_virtio_console() {
+        test_block!(tb, "", {
+            let guest = Guest::new();
+            let mut child = Command::new("target/debug/cloud-hypervisor")
+                .args(&["--cpus", "1"])
+                .args(&["--memory", "size=512M"])
+                .args(&["--kernel", guest.fw_path.as_str()])
+                .args(&["--disk", guest.disks[0].as_str(), guest.disks[1].as_str()])
+                .args(&["--net", guest.default_net_string().as_str()])
+                .args(&["--console", "tty"])
+                .stdout(Stdio::piped())
+                .stderr(Stdio::piped())
+                .spawn()
+                .unwrap();
+
+            thread::sleep(std::time::Duration::new(20, 0));
+
+            aver!(tb, guest.does_device_vendor_pair_match("0x1043", "0x1af4"));
+            aver!(tb, guest.is_console_detected());
+
+            let text = String::from("On a branch floating down river a cricket, singing.");
+            let cmd = format!("sudo -E bash -c 'echo {} > /dev/hvc0'", text);
+            guest.ssh_command(&cmd);
+
+            guest.ssh_command("sudo reboot");
+            thread::sleep(std::time::Duration::new(10, 0));
+            let _ = child.kill();
+
+            match child.wait_with_output() {
+                Ok(out) => {
+                    aver!(tb, String::from_utf8_lossy(&out.stdout).contains(&text));
+                }
+                Err(_) => aver!(tb, false),
+            }
+
+            Ok(())
+        });
+    }
+
+    #[test]
+    fn test_console_file() {
+        test_block!(tb, "", {
+            let guest = Guest::new();
+
+            let console_path = guest.tmp_dir.path().join("/tmp/console-output");
+            let mut child = Command::new("target/debug/cloud-hypervisor")
+                .args(&["--cpus", "1"])
+                .args(&["--memory", "size=512M"])
+                .args(&["--kernel", guest.fw_path.as_str()])
+                .args(&["--disk", guest.disks[0].as_str(), guest.disks[1].as_str()])
+                .args(&["--net", guest.default_net_string().as_str()])
+                .args(&[
+                    "--console",
+                    format!("file={}", console_path.to_str().unwrap()).as_str(),
+                ])
+                .spawn()
+                .unwrap();
+
+            thread::sleep(std::time::Duration::new(20, 0));
+
+            // Test that there is a ttyS0
+            aver!(tb, guest.is_console_detected());
+
+            guest.ssh_command("sudo reboot");
+            thread::sleep(std::time::Duration::new(10, 0));
+
+            // Do this check after shutdown of the VM as an easy way to ensure
+            // all writes are flushed to disk
+            let mut f = std::fs::File::open(console_path).unwrap();
+            let mut buf = String::new();
+            f.read_to_string(&mut buf).unwrap();
+            aver!(tb, buf.contains("cloud login:"));
+
+            let _ = child.kill();
+            let _ = child.wait();
+
+            Ok(())
+        });
+    }
+
 }