Add Dockerfile, Makefile, start.sh

Author: DJAndries

Dockerfile

@@ -0,0 +1,19 @@
+FROM public.ecr.aws/docker/library/rust:1.72.0-slim-bookworm as rust-builder
+
+WORKDIR /src/
+
+COPY ./Cargo.toml ./Cargo.lock ./
+COPY ./src src/
+
+RUN cargo build --locked --release
+
+FROM public.ecr.aws/docker/library/debian:12.1-slim
+
+RUN apt update && apt install -y procps
+
+COPY start.sh /usr/local/bin
+RUN chmod +x /usr/local/bin/start.sh
+
+COPY --from=rust-builder /src/target/release/vsock-experiment /usr/local/bin/
+
+CMD ["/usr/local/bin/start.sh"]

Makefile

@@ -0,0 +1,21 @@
+all: run
+
+install-relay:
+	cargo install vsock-relay
+
+run-relay:
+	vsock-relay -s 0.0.0.0:8080 -l 4:8080
+
+build: src/*.rs Cargo.toml Dockerfile
+	docker build -t vsock-experiment .
+	nitro-cli build-enclave --docker-uri vsock-experiment:latest --output-file vsock-experiment.eif
+
+run: build
+	nitro-cli terminate-enclave --all
+	nitro-cli run-enclave \
+		--cpu-count 8 \
+		--memory 512 \
+		--enclave-cid 4 \
+		--eif-path "vsock-experiment.eif" \
+		--debug-mode \
+		--attach-console

src/main.rs

@@ -1,10 +1,20 @@
 use futures::StreamExt;
+use std::net::Shutdown;
+use std::process::exit;
+use std::time::Duration;
 use tokio::io::AsyncWriteExt;
+use tokio::time::sleep;
 use tokio_vsock::VsockListener;
 
 #[tokio::main]
 async fn main() {
-    let listener = VsockListener::bind(4, 1234).expect("Failed to bind to address");
+    let listener = VsockListener::bind(4, 8080).expect("Failed to bind to address");
+
+    tokio::spawn(async {
+        sleep(Duration::from_secs(30)).await;
+        println!("I'm out!");
+        exit(0);
+    });
 
     let mut incoming = listener.incoming();
     while let Some(result) = incoming.next().await {
@@ -13,6 +23,7 @@ async fn main() {
                 let response = "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\n\r\nHello, World!";
 
                 stream.write_all(response.as_bytes()).await.ok();
+                stream.shutdown(Shutdown::Both).ok();
             });
         }
     }

start.sh

@@ -0,0 +1,10 @@
+#!/bin/sh
+
+sh -c 'while true; do top -b -n 1 -o %MEM | head -n 13; sleep 2; done' &
+#sh -c 'while true; do cat /proc/meminfo; sleep 10; done' &
+#sh -c 'while true; do slabtop -o; sleep 10; done' &
+#sh -c 'while true; do cat /proc/meminfo; slabtop -o; echo "alloc calls:"; cat /sys/kernel/slab/kmalloc-4096/alloc_calls; echo "free calls:"; cat /sys/kernel/slab/kmalloc-4096/free_calls; echo 3 > /proc/sys/vm/drop_caches; sleep 10; done' &
+
+vsock-experiment &
+
+sleep infinity