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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
use crate::{AccountCache, CertCache};
use async_trait::async_trait;
use ring::digest::{Context, SHA256};
use std::io::ErrorKind;
use std::path::Path;
pub struct DirCache<P: AsRef<Path> + Send + Sync> {
inner: P,
}
impl<P: AsRef<Path> + Send + Sync> DirCache<P> {
pub fn new(dir: P) -> Self {
Self { inner: dir }
}
async fn read_if_exist(
&self,
file: impl AsRef<Path>,
) -> Result<Option<Vec<u8>>, std::io::Error> {
let path = self.inner.as_ref().join(file);
match smol::fs::read(path).await {
Ok(content) => Ok(Some(content)),
Err(err) => match err.kind() {
ErrorKind::NotFound => Ok(None),
_ => Err(err.into()),
},
}
}
async fn write(
&self,
file: impl AsRef<Path>,
contents: impl AsRef<[u8]>,
) -> Result<(), std::io::Error> {
smol::fs::create_dir_all(&self.inner).await?;
let path = self.inner.as_ref().join(file);
Ok(smol::fs::write(path, contents).await?)
}
fn cached_account_file_name(contact: &[String], directory_url: impl AsRef<str>) -> String {
let mut ctx = Context::new(&SHA256);
for el in contact {
ctx.update(el.as_ref());
ctx.update(&[0])
}
ctx.update(directory_url.as_ref().as_bytes());
let hash = base64::encode_config(ctx.finish(), base64::URL_SAFE_NO_PAD);
format!("cached_account_{}", hash)
}
fn cached_cert_file_name(domains: &[String], directory_url: impl AsRef<str>) -> String {
let mut ctx = Context::new(&SHA256);
for domain in domains {
ctx.update(domain.as_ref());
ctx.update(&[0])
}
ctx.update(directory_url.as_ref().as_bytes());
let hash = base64::encode_config(ctx.finish(), base64::URL_SAFE_NO_PAD);
format!("cached_cert_{}", hash)
}
}
#[async_trait]
impl<P: AsRef<Path> + Send + Sync> CertCache for DirCache<P> {
type EC = std::io::Error;
async fn load_cert(
&self,
domains: &[String],
directory_url: &str,
) -> Result<Option<Vec<u8>>, Self::EC> {
let file_name = Self::cached_cert_file_name(&domains, directory_url);
self.read_if_exist(file_name).await
}
async fn store_cert(
&self,
domains: &[String],
directory_url: &str,
cert: &[u8],
) -> Result<(), Self::EC> {
let file_name = Self::cached_cert_file_name(&domains, directory_url);
self.write(file_name, cert).await
}
}
#[async_trait]
impl<P: AsRef<Path> + Send + Sync> AccountCache for DirCache<P> {
type EA = std::io::Error;
async fn load_account(
&self,
contact: &[String],
directory_url: &str,
) -> Result<Option<Vec<u8>>, Self::EA> {
let file_name = Self::cached_account_file_name(&contact, directory_url);
self.read_if_exist(file_name).await
}
async fn store_account(
&self,
contact: &[String],
directory_url: &str,
account: &[u8],
) -> Result<(), Self::EA> {
let file_name = Self::cached_account_file_name(&contact, directory_url);
self.write(file_name, account).await
}
}