quartz-enclave-core 0.5.1

A Rust framework for building Intel SGX enclaves managed by a CosmWasm contract.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use std::sync::Arc;

use tokio::sync::{RwLock, RwLockReadGuard};

use crate::{
    backup_restore::{Export, Import},
    key_manager::KeyManager,
};

/// A thread-safe wrapper for a key-manager.
#[derive(Clone, Debug)]
pub struct SharedKeyManager<K> {
    pub inner: Arc<RwLock<K>>,
}

impl<K> SharedKeyManager<K> {
    pub fn wrapping(key_manager: K) -> Self {
        Self {
            inner: Arc::new(RwLock::new(key_manager)),
        }
    }

    pub async fn read_lock(&self) -> RwLockReadGuard<'_, K> {
        self.inner.read().await
    }
}

#[async_trait::async_trait]
impl<K: KeyManager> KeyManager for SharedKeyManager<K> {
    type PubKey = K::PubKey;

    async fn pub_key(&self) -> Self::PubKey {
        self.inner.read().await.pub_key().await
    }
}

#[async_trait::async_trait]
impl<K: KeyManager + Import> Import for SharedKeyManager<K> {
    type Error = K::Error;

    async fn import(data: Vec<u8>) -> Result<Self, Self::Error> {
        let km = K::import(data).await?;
        Ok(Self::wrapping(km))
    }
}

#[async_trait::async_trait]
impl<K: KeyManager + Export> Export for SharedKeyManager<K> {
    type Error = K::Error;

    async fn export(&self) -> Result<Vec<u8>, Self::Error> {
        let guard = self.inner.read().await;
        guard.export().await
    }
}