Merge main

Author: ameba23

.github/workflows/test.yml

@@ -36,4 +36,4 @@ jobs:
       run: cargo clippy --workspace -- -D warnings
 
     - name: Run cargo test
-      run: cargo test --workspace --all-targets
+      run: cargo test --workspace --all-targets -- --test-threads=1

Cargo.lock

@@ -3,11 +3,12 @@ members = [".", "dummy-attestation-server"]
 
 [package]
 name = "attested-tls-proxy"
-description = "An HTTP attested TLS proxy server and client for secure communication with CVM services"
-version = "0.1.0"
+version = "0.0.1"
 edition = "2024"
 license = "MIT"
+description = "An HTTP attested TLS proxy server and client for secure communication with CVM services"
 repository = "https://github.com/flashbots/attested-tls-proxy"
+keywords = ["attested-TLS", "CVM", "TDX"]
 
 [dependencies]
 tokio = { version = "1.48.0", features = ["full"] }
@@ -49,16 +50,23 @@ time = "0.3.44"
 once_cell = "1.21.3"
 axum = "0.8.6"
 tower-http = { version = "0.6.7", features = ["fs"] }
+tokio-tungstenite = { version = "0.28.0", optional = true }
+futures-util = { version = "0.3.31", optional = true }
 
 [dev-dependencies]
 rcgen = "0.14.5"
 tempfile = "3.23.0"
-tdx-quote = { version = "0.0.4", features = ["mock"] }
+tdx-quote = { version = "0.0.5", features = ["mock"] }
 
 [features]
-default = ["azure"]
+default = ["azure", "ws"]
+
+# Adds support for Microsoft Azure attestation generation and verification
 azure = ["tss-esapi", "az-tdx-vtpm"]
 
+# Adds websocket support
+ws = ["tokio-tungstenite", "futures-util"]
+
 [package.metadata.deb]
 maintainer = "Flashbots Team <[email protected]>"
 depends = "$auto"

Cargo.toml

@@ -4,6 +4,7 @@ version = "0.1.0"
 edition = "2024"
 license = "MIT"
 publish = false
+repository = "https://github.com/flashbots/attested-tls-proxy"
 
 [dependencies]
 attested-tls-proxy = { path = ".." }

dummy-attestation-server/Cargo.toml

@@ -1,4 +1,4 @@
-//! Microsoft Azure Attestation (MAA) evidence generation and verification
+//! Microsoft Azure vTPM attestation evidence generation and verification
 mod ak_certificate;
 mod nv_index;
 use ak_certificate::{read_ak_certificate_from_tpm, verify_ak_cert_with_azure_roots};
@@ -245,6 +245,7 @@ impl RsaPubKey {
     }
 }
 
+/// An error when generating or verifying a Microsoft Azure vTPM attestation
 #[derive(Error, Debug)]
 pub enum MaaError {
     #[error("Report: {0}")]