DHCPv6 Server / Router Adv Daemon / Vsock Support / GRPC Client (#49)

Author: guvenc

Cargo.lock

@@ -44,11 +44,21 @@ chrono = "0.4"
 futures-util = "0.3"
 futures-core = "0.3"
 async-stream = "0.3"
+tokio-vsock = { git = "https://github.com/rust-vsock/tokio-vsock", rev = "3a41323", features = ["tonic-conn"]}
+tower = "0.4.13"
+hyper-util = "0.1"
+socket2 = "0.5.7"
 libc = "0.2"
 
+pelite = "0.9"
+regex = "1.10.6"
+
 [build-dependencies]
 tonic-build = "0.12.1"
 
+[dev-dependencies]
+serial_test = "3.1.1"
+
 [workspace]
 members = [
     "client",

Cargo.toml

@@ -1,6 +1,9 @@
 fn main() {
     tonic_build::configure()
         .build_server(false)
-        .compile(&["../proto/feos.proto"], &["../proto"])
+        .compile(
+            &["../proto/feos.proto", "../proto/container.proto"],
+            &["../proto"],
+        )
         .unwrap();
 }

client/build.rs

@@ -6,6 +6,7 @@ use tokio_stream::wrappers::ReceiverStream;
 use tonic::transport::Endpoint;
 use tonic::Request;
 
+use crate::client_container::ContainerCommand;
 use feos_grpc::feos_grpc_client::FeosGrpcClient;
 use feos_grpc::*;
 
@@ -45,6 +46,12 @@ pub enum Command {
     BootVM {
         uuid: String,
     },
+    ShutdownVM {
+        uuid: String,
+    },
+    PingVM {
+        uuid: String,
+    },
     AttachNicVM {
         uuid: String,
         mac_address: String,
@@ -55,6 +62,7 @@ pub enum Command {
     ConsoleVM {
         uuid: String,
     },
+    Container(ContainerCommand),
 }
 
 fn format_address(ip: &str, port: u16) -> String {
@@ -77,6 +85,10 @@ pub async fn run_client(opt: Opt) -> Result<(), Box<dyn std::error::Error>> {
     let mut client = FeosGrpcClient::new(channel);
 
     match opt.cmd {
+        Command::Container(container_cmd) => {
+            crate::client_container::run_container_client(opt.server_ip, opt.port, container_cmd)
+                .await?;
+        }
         Command::Reboot => {
             let request = Request::new(RebootRequest {});
             let response = client.reboot(request).await?;
@@ -127,6 +139,16 @@ pub async fn run_client(opt: Opt) -> Result<(), Box<dyn std::error::Error>> {
             let response = client.boot_vm(request).await?;
             println!("BOOT VM RESPONSE={:?}", response);
         }
+        Command::PingVM { uuid } => {
+            let request = Request::new(PingVmRequest { uuid });
+            let response = client.ping_vm(request).await?;
+            println!("BOOT VM RESPONSE={:?}", response);
+        }
+        Command::ShutdownVM { uuid } => {
+            let request = Request::new(ShutdownVmRequest { uuid });
+            let response = client.shutdown_vm(request).await?;
+            println!("SHUTDOWN VM RESPONSE={:?}", response);
+        }
         Command::AttachNicVM {
             uuid,
             mac_address,