cycles-quartz/enclaves/quartz/src/server.rs

67 lines
1.8 KiB
Rust
Raw Normal View History

2024-02-23 20:51:55 +00:00
use std::{
fs::{read, File},
io::{Result as IoResult, Write},
};
2024-02-22 20:48:13 +00:00
2024-02-23 20:51:20 +00:00
use quartz_cw::{
msg::{execute::attested::HasUserData, instantiate::CoreInstantiate},
state::{Config, UserData},
};
2024-02-22 20:09:44 +00:00
use quartz_proto::quartz::{
2024-02-23 20:51:55 +00:00
core_server::Core, InstantiateRequest as RawInstantiateRequest,
InstantiateResponse as RawInstantiateResponse, SessionCreateRequest, SessionCreateResponse,
2024-02-22 20:09:44 +00:00
};
2024-02-20 11:27:10 +00:00
use tonic::{Request, Response, Status};
2024-02-23 20:51:55 +00:00
use crate::types::InstantiateResponse;
type TonicResult<T> = Result<T, Status>;
2024-02-22 20:48:13 +00:00
#[derive(Clone, Debug)]
2024-02-23 20:51:20 +00:00
pub struct CoreService {
config: Config,
}
impl CoreService {
pub fn new(config: Config) -> Self {
Self { config }
}
}
2024-02-20 11:27:10 +00:00
#[tonic::async_trait]
impl Core for CoreService {
2024-02-22 20:09:44 +00:00
async fn instantiate(
&self,
2024-02-23 20:51:55 +00:00
_request: Request<RawInstantiateRequest>,
) -> TonicResult<Response<RawInstantiateResponse>> {
let core_instantiate_msg = CoreInstantiate::new(self.config.clone());
2024-02-22 20:09:44 +00:00
2024-02-23 20:51:55 +00:00
let user_data = core_instantiate_msg.user_data();
let quote = attestion_quote(user_data).map_err(|e| Status::internal(e.to_string()))?;
2024-02-22 20:09:44 +00:00
2024-02-23 20:51:55 +00:00
let response = InstantiateResponse::new(self.config.clone(), quote);
Ok(Response::new(response.into()))
2024-02-22 20:09:44 +00:00
}
2024-02-20 11:27:10 +00:00
async fn session_create(
&self,
request: Request<SessionCreateRequest>,
2024-02-23 20:51:55 +00:00
) -> TonicResult<Response<SessionCreateResponse>> {
2024-02-20 11:27:10 +00:00
println!("Got a request: {:?}", request);
let reply = SessionCreateResponse {
message: "Hello!".to_string(),
};
Ok(Response::new(reply))
}
}
2024-02-22 20:48:13 +00:00
2024-02-23 20:51:20 +00:00
pub fn attestion_quote(user_data: UserData) -> IoResult<Vec<u8>> {
let mut user_report_data = File::create("/dev/attestation/user_report_data")?;
user_report_data.write_all(user_data.as_slice())?;
user_report_data.flush()?;
2024-02-22 20:48:13 +00:00
2024-02-23 20:51:20 +00:00
let quote = read("/dev/attestation/quote")?;
Ok(quote)
2024-02-22 20:48:13 +00:00
}