49 lines
1.5 KiB
Rust
49 lines
1.5 KiB
Rust
|
use base64;
|
||
|
|
use diesel::PgConnection;
|
||
|
|
use hex;
|
||
|
|
use openssl::hash::{Hasher, MessageDigest};
|
||
|
|
use openssl::sha::sha256;
|
||
|
|
use reqwest::header::{Date, Headers, UserAgent};
|
||
|
|
use std::time::SystemTime;
|
||
|
|
|
||
|
|
use activity_pub::sign::Signer;
|
||
|
|
|
||
|
|
const USER_AGENT: &'static str = "Plume/0.1.0";
|
||
|
|
|
||
|
|
header! {
|
||
|
|
(Signature, "Signature") => [String]
|
||
|
|
}
|
||
|
|
|
||
|
|
header! {
|
||
|
|
(Digest, "Digest") => [String]
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn headers() -> Headers {
|
||
|
|
let mut headers = Headers::new();
|
||
|
|
headers.set(UserAgent::new(USER_AGENT));
|
||
|
|
headers.set(Date(SystemTime::now().into()));
|
||
|
|
headers
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn signature<S: Signer>(signer: S, headers: Headers, conn: &PgConnection) -> Signature {
|
||
|
|
let signed_string = headers.iter().map(|h| format!("{}: {}", h.name().to_lowercase(), h.value_string())).collect::<Vec<String>>().join("\n");
|
||
|
|
let signed_headers = headers.iter().map(|h| h.name().to_string()).collect::<Vec<String>>().join(" ").to_lowercase();
|
||
|
|
|
||
|
|
let data = signer.sign(signed_string);
|
||
|
|
let sign = hex::encode(sha256(&data[..]));
|
||
|
|
|
||
|
|
Signature(format!(
|
||
|
|
"keyId=\"{key_id}\",algorithm=\"rsa-sha256\",headers=\"#{signed_headers}\",signature=\"#{signature}\"",
|
||
|
|
key_id = signer.get_key_id(conn),
|
||
|
|
signed_headers = signed_headers,
|
||
|
|
signature = sign
|
||
|
|
))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn digest(body: String) -> Digest {
|
||
|
|
let mut hasher = Hasher::new(MessageDigest::sha256()).unwrap();
|
||
|
|
hasher.update(&body.into_bytes()[..]);
|
||
|
|
let res = base64::encode(&hasher.finish().unwrap());
|
||
|
|
Digest(format!("SHA-256={}", res))
|
||
|
|
}
|