fix(simulator): satisfy formatting and clippy checks

Author: kvinwang

dstack/crates/mock-attestation/src/tdx.rs

@@ -3,17 +3,17 @@
 // SPDX-License-Identifier: Apache-2.0
 
 use anyhow::Result;
-use dcap_qvl::QuoteCollateralV3;
 use dcap_qvl::quote::{
     AuthData, AuthDataV4, CertificationData, Data, EnclaveReport, Header,
     QEReportCertificationData, Quote, Report, TDReport10,
 };
-use p256::ecdsa::{Signature, SigningKey, signature::Signer};
+use dcap_qvl::QuoteCollateralV3;
+use p256::ecdsa::{signature::Signer, Signature, SigningKey};
 use p256::pkcs8::{DecodePrivateKey, EncodePrivateKey};
 use rcgen::{
     BasicConstraints, Certificate, CertificateParams, CertificateRevocationListParams,
     CertifiedKey, CustomExtension, DnType, ExtendedKeyUsagePurpose, IsCa, KeyIdMethod, KeyPair,
-    KeyUsagePurpose, PKCS_ECDSA_P256_SHA256, RemoteKeyPair, SerialNumber, SignatureAlgorithm,
+    KeyUsagePurpose, RemoteKeyPair, SerialNumber, SignatureAlgorithm, PKCS_ECDSA_P256_SHA256,
 };
 use scale::Encode;
 use serde_json::json;
@@ -142,11 +142,11 @@ impl TdxGenerator {
     pub fn root_ca_pem(&self) -> String {
         self.root.pem()
     }
-    pub fn root_key_pem(&self) -> String {
-        self.root_signing_key
-            .to_pkcs8_pem(Default::default())
-            .expect("P-256 key serialization")
-            .to_string()
+    pub fn root_key_pem(&self) -> Result<String> {
+        Ok(self
+            .root_signing_key
+            .to_pkcs8_pem(Default::default())?
+            .to_string())
     }
 
     pub fn sample_collateral(&self) -> Result<QuoteCollateralV3> {
@@ -446,14 +446,13 @@ mod tests {
         );
         let mut tampered = evidence.quote.clone();
         tampered[100] ^= 1;
-        assert!(
-            verifier
-                .verify(&tampered, &evidence.collateral, now)
-                .is_err()
-        );
-        assert!(
-            crate::ensure_report_data(&verified.report.as_td10().unwrap().report_data, &[0x24; 64])
-                .is_err()
-        );
+        assert!(verifier
+            .verify(&tampered, &evidence.collateral, now)
+            .is_err());
+        assert!(crate::ensure_report_data(
+            &verified.report.as_td10().unwrap().report_data,
+            &[0x24; 64]
+        )
+        .is_err());
     }
 }

dstack/dstack-attest/src/attestation.rs

@@ -202,7 +202,6 @@ impl AttestationVerifier {
         self.external_trust_anchors
     }
 
-
     async fn verify_tdx_quote(&self, quote: &[u8]) -> Result<TdxVerifiedReport> {
         let collateral = self.tdx_collateral.fetch(quote).await?;
         let now = SystemTime::now()

dstack/guest-agent-simulator/src/main.rs

@@ -9,10 +9,9 @@ use std::sync::Arc;
 use anyhow::{Context, Result};
 use clap::Parser;
 use dstack_guest_agent::{
-    AppState,
     backend::PlatformBackend,
     config::{self, Config},
-    run_server,
+    run_server, AppState,
 };
 use dstack_guest_agent_rpc::{AttestResponse, GetQuoteResponse};
 use mock_attestation::tdx::TdxGenerator;
@@ -116,7 +115,7 @@ impl PlatformBackend for SimulatorPlatform {
 #[rocket::main]
 async fn main() -> Result<()> {
     {
-        use tracing_subscriber::{EnvFilter, fmt};
+        use tracing_subscriber::{fmt, EnvFilter};
         let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"));
         fmt().with_env_filter(filter).with_ansi(false).init();
     }

dstack/guest-agent-simulator/src/simulator.rs

@@ -4,7 +4,7 @@
 
 use std::path::Path;
 
-use anyhow::{Context, Result, anyhow};
+use anyhow::{anyhow, Context, Result};
 use dcap_qvl::quote::Quote;
 use dstack_guest_agent_rpc::{AttestResponse, GetQuoteResponse};
 use mock_attestation::tdx::TdxGenerator;