Method register_cvm

dstack-util/src/system_setup.rs:445–494

(
        &self,
        gateway_url: &str,
        key_store: &GatewayKeyStore,
    )

    }

    async fn register_cvm(
        &self,
        gateway_url: &str,
        key_store: &GatewayKeyStore,
    ) -> Result<RegisterCvmResponse> {
        let port_policy = RpcPortPolicy {
            ports: self
                .shared
                .app_compose
                .port_policy
                .ports
                .iter()
                .map(|p| RpcPortAttrs {
                    port: p.port as u32,
                    pp: p.pp,
                })
                .collect(),
            restrict_mode: self.shared.app_compose.port_policy.restrict_mode,
        };
        let client =
            self.create_gateway_client(gateway_url, &key_store.client_key, &key_store.client_cert)?;
        let result = client
            .register_cvm(RegisterCvmRequest {
                client_public_key: key_store.wg_pk.clone(),
                port_policy: Some(port_policy.clone()),
            })
            .await
            .context("Failed to register CVM");
        let Err(err) = &result else {
            return result;
        };
        // If the error contains "no attestation provided", it's likely an older gateway version
        let is_legacy_gateway = format!("{err:#}").contains("no attestation provided");
        if !is_legacy_gateway {
            return result;
        }
        info!("Seems like the gateway is an older version, retrying with quote cert");
        let client = self.create_gateway_client(
            gateway_url,
            &key_store.client_key,
            &key_store.client_cert_with_quote,
        )?;
        client
            .register_cvm(RegisterCvmRequest {
                client_public_key: key_store.wg_pk.clone(),
                port_policy: Some(port_policy),
            })
            .await
            .context("Failed to register CVM")
    }

    async fn get_or_generate_key_store(&self) -> Result<GatewayKeyStore> {
        // Try to load existing cache