write rootfs

Author: tylerfanelli

init/nitro/main.c

@@ -203,7 +203,7 @@ rootfs_rcv(int sock_fd)
 {
     uint8_t len_b[8], *rootfs_archive;
     ssize_t read_len;
-    uint64_t len;
+    uint64_t len, idx = 0;
     int ret;
 
     read_len = read(sock_fd, len_b, 8);
@@ -215,6 +215,24 @@ rootfs_rcv(int sock_fd)
 
     memcpy(&len, len_b, sizeof(uint64_t));
 
+    rootfs_archive = (uint8_t *) malloc(len);
+    if (rootfs_archive == NULL) {
+        close(sock_fd);
+        perror("rootfs archive malloc");
+        return -1;
+    }
+
+    while (len) {
+        read_len = read(sock_fd, &rootfs_archive[idx], len);
+        if (read_len <= 0) {
+            close(sock_fd);
+            perror("vsock rootfs archive read");
+            return -1;
+        }
+        idx += read_len;
+        len -= read_len;
+    }
+
     return 0;
 }
 

src/nitro/src/enclaves.rs

@@ -137,6 +137,10 @@ impl NitroEnclave {
             .write_all(&len.to_ne_bytes())
             .map_err(NitroError::RootFsLenWrite)?;
 
+        stream
+            .write_all(&archive)
+            .map_err(NitroError::RootFsWrite)?;
+
         Ok(())
     }
 }

src/nitro/src/error.rs

@@ -24,6 +24,7 @@ pub enum NitroError {
     VsockConnect,
     IpcWrite(io::Error),
     RootFsLenWrite(io::Error),
+    RootFsWrite(io::Error),
     RootFsTooLarge,
 }
 
@@ -69,6 +70,9 @@ impl fmt::Display for NitroError {
             NitroError::RootFsLenWrite(e) => {
                 format!("unable to write rootfs archive length to enclave: {e}")
             }
+            NitroError::RootFsWrite(e) => {
+                format!("unable to write rootfs archive to enclave: {e}")
+            }
             NitroError::RootFsTooLarge => "rootfs size is larger than 64 bytes".to_string(),
         };