Merge pull request #1 from mohamed-arm/mohamed_parsec_execute

Add parsec provisionsing example

Author: gbryant-arm

Cargo.toml

@@ -3,6 +3,7 @@ members = [
     "dpu/dpu-runtime-manager",
     "host/examples/attestation",
     "host/examples/provision_execute",
+    "host/examples/parsec_execute",
 ]
 
 resolver = "2"

host/examples/parsec_execute/Cargo.toml

@@ -0,0 +1,13 @@
+[package]
+authors = ["The Veracruz Development Team"]
+description = "The test program to demonstrate provisioning parsec-client and executing remotely with DPU runtime manager"
+name = "parsec_execute"
+version = "0.1.0"
+edition = "2018"
+
+[dependencies]
+anyhow = "1.0.14"
+env_logger = { version = "0.10.0" }
+log = "0.4.13"
+transport = { path = "../../../common/transport" }
+utils = { path = "../../../common/utils" }

host/examples/parsec_execute/src/main.rs

@@ -0,0 +1,107 @@
+//! Provision and execute parsec.
+
+use anyhow::anyhow;
+use log::{error, info};
+use utils::attestation;
+use transport::{messages::{Request, Response, Status}, session::Session};
+
+use std::fs;
+use std::fs::File;
+use std::io::Read;
+
+const DPU_SERVER_URL: &str = "127.0.0.1:6666";
+const ATTESTATION_SERVER_URL: &str = "127.0.0.1:3010";
+
+const PARSEC_APP_NAME: &str = "/tmp/dpu/parsec_app";
+const PARSEC_PERM_CMD: &str = "chmod +x /tmp/dpu/parsec_app";
+const PARSEC_EXECUTE_CMD: &str = "PARSEC_SERVICE_CONN_TYPE=tcp PARSEC_SERVICE_CONN_IP_ADDR=172.28.97.222  PARSEC_SERVICE_CONN_PORT_NO=8002 /tmp/dpu/parsec_app ping";
+
+// Read program file
+fn get_file_data(filename: &String) -> anyhow::Result<Vec<u8>> {
+
+    let mut file = File::open(&filename).unwrap();
+    let metadata = fs::metadata(&filename).unwrap();
+    let mut file_content = vec![0; metadata.len() as usize];
+    file.read(&mut file_content).unwrap();
+
+    Ok(file_content)
+}
+
+fn main() -> anyhow::Result<()> {
+    env_logger::init();
+
+    // TODO: Parse arguments with clap
+    let dpu_server_url = DPU_SERVER_URL;
+    let proxy_attestation_server_url = ATTESTATION_SERVER_URL;
+
+    let dpu_session_id = Session::from_url(dpu_server_url)?;
+
+    attestation::request_attestation(dpu_session_id, proxy_attestation_server_url)?;
+    info!("Successfully attested DPU.");
+
+    info!("Preparing parsec-tool executable data...");
+    let filename = PARSEC_APP_NAME.to_owned();
+    let data = get_file_data(&filename).unwrap();
+
+    info!("Provisioning parsec-tool to DPU...");
+    Session::send_message(dpu_session_id, &Request::UploadFile(filename, data))
+        .map_err(|e| {
+            error!("Failed to send provisioning message to DPU.  Error returned: {:?}.", e);
+            e
+        })?;
+    let response = Session::receive_message(dpu_session_id).map_err(|e| {
+        error!("Failed to receive response to provisioning message.  Error received: {:?}.", e);
+        e
+    })?;
+    match response {
+        Response::Status(Status::Success(_)) => {
+            info!("Successfully provisioned DPU.");
+        },
+        _ => {
+            error!("Error provisioning DPU.");
+            return Err(anyhow!("Provisioning failure."));
+        },
+    }
+
+    info!("Requesting permission change...");
+    Session::send_message(dpu_session_id, &Request::Execute(PARSEC_PERM_CMD.to_owned()))
+        .map_err(|e| {
+            error!("Failed to send execution message to DPU.  Error returned: {:?}.", e);
+            e
+        })?;
+    let response = Session::receive_message(dpu_session_id).map_err(|e| {
+        error!("Failed to receive response to execution message.  Error received: {:?}.", e);
+        e
+    })?;
+    match response {
+        Response::Status(Status::Success(output)) => {
+            info!("Successfully executed code remotely. Output: \n{}", output);
+        },
+        _ => {
+            error!("Error executing code remotely.");
+            return Err(anyhow!("Remote execution failure."));
+        },
+    }
+
+    info!("Requesting parsec-tool execution..");
+    Session::send_message(dpu_session_id, &Request::Execute(PARSEC_EXECUTE_CMD.to_owned()))
+        .map_err(|e| {
+            error!("Failed to send execution message to DPU.  Error returned: {:?}.", e);
+            e
+        })?;
+    let response = Session::receive_message(dpu_session_id).map_err(|e| {
+        error!("Failed to receive response to execution message.  Error received: {:?}.", e);
+        e
+    })?;
+    match response {
+        Response::Status(Status::Success(output)) => {
+            info!("Successfully executed code remotely. Output: \n{}", output);
+        },
+        _ => {
+            error!("Error executing code remotely.");
+            return Err(anyhow!("Remote execution failure."));
+        },
+    }
+
+    Ok(())
+}