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-23 22:27:29 +00:00
|
|
|
use quartz_relayer::types::InstantiateResponse;
|
2024-02-20 11:27:10 +00:00
|
|
|
use tonic::{Request, Response, Status};
|
|
|
|
|
2024-02-23 20:51:55 +00:00
|
|
|
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
|
|
|
}
|