asonix
133e081740
Update actix library with new methods Build reqwest client with feature parity to actix client
111 lines
3.4 KiB
Rust
111 lines
3.4 KiB
Rust
use crate::{Config, Sign, SignError};
|
|
use reqwest::{Body, RequestBuilder};
|
|
use std::{fmt::Display, future::Future, pin::Pin};
|
|
|
|
mod sha2;
|
|
mod sha3;
|
|
|
|
/// A trait for creating digests of an array of bytes
|
|
pub trait DigestCreate {
|
|
/// The name of the digest algorithm
|
|
const NAME: &'static str;
|
|
|
|
/// Compute the digest of the input bytes
|
|
fn compute(&mut self, input: &[u8]) -> String;
|
|
}
|
|
|
|
/// Extend the Sign trait with support for adding Digest Headers to the request
|
|
///
|
|
/// It generates HTTP Signatures after the Digest header has been added, in order to have
|
|
/// verification that the body has not been tampered with, or that the request can't be replayed by
|
|
/// a malicious entity
|
|
pub trait SignExt: Sign {
|
|
fn authorization_signature_with_digest<F, E, K, D, V>(
|
|
self,
|
|
config: Config,
|
|
key_id: K,
|
|
digest: D,
|
|
v: V,
|
|
f: F,
|
|
) -> Pin<Box<dyn Future<Output = Result<reqwest::Response, E>>>>
|
|
where
|
|
F: FnOnce(&str) -> Result<String, E> + Send + 'static,
|
|
E: From<SignError> + From<reqwest::Error>,
|
|
K: Display + 'static,
|
|
D: DigestCreate + Send + 'static,
|
|
V: AsRef<[u8]> + Into<Body> + Send + 'static,
|
|
Self: Sized;
|
|
|
|
fn signature_with_digest<F, E, K, D, V>(
|
|
self,
|
|
config: Config,
|
|
key_id: K,
|
|
digest: D,
|
|
v: V,
|
|
f: F,
|
|
) -> Pin<Box<dyn Future<Output = Result<reqwest::Response, E>>>>
|
|
where
|
|
F: FnOnce(&str) -> Result<String, E> + Send + 'static,
|
|
E: From<SignError> + From<reqwest::Error>,
|
|
K: Display + 'static,
|
|
D: DigestCreate + Send + 'static,
|
|
V: AsRef<[u8]> + Into<Body> + Send + 'static,
|
|
Self: Sized;
|
|
}
|
|
|
|
impl SignExt for RequestBuilder {
|
|
fn authorization_signature_with_digest<F, E, K, D, V>(
|
|
self,
|
|
config: Config,
|
|
key_id: K,
|
|
mut digest: D,
|
|
v: V,
|
|
f: F,
|
|
) -> Pin<Box<dyn Future<Output = Result<reqwest::Response, E>>>>
|
|
where
|
|
F: FnOnce(&str) -> Result<String, E> + Send + 'static,
|
|
E: From<SignError> + From<reqwest::Error>,
|
|
K: Display + 'static,
|
|
D: DigestCreate + Send + 'static,
|
|
V: AsRef<[u8]> + Into<Body> + Send + 'static,
|
|
Self: Sized,
|
|
{
|
|
Box::pin(async move {
|
|
let digest = tokio::task::block_in_place(|| digest.compute(v.as_ref()));
|
|
|
|
let c = self
|
|
.header("Digest", format!("{}={}", D::NAME, digest))
|
|
.authorization_signature(&config, key_id, f)?;
|
|
|
|
c.body(v).send().await.map_err(E::from)
|
|
})
|
|
}
|
|
|
|
fn signature_with_digest<F, E, K, D, V>(
|
|
self,
|
|
config: Config,
|
|
key_id: K,
|
|
mut digest: D,
|
|
v: V,
|
|
f: F,
|
|
) -> Pin<Box<dyn Future<Output = Result<reqwest::Response, E>>>>
|
|
where
|
|
F: FnOnce(&str) -> Result<String, E> + Send + 'static,
|
|
E: From<SignError> + From<reqwest::Error>,
|
|
K: Display + 'static,
|
|
D: DigestCreate + Send + 'static,
|
|
V: AsRef<[u8]> + Into<Body> + Send + 'static,
|
|
Self: Sized,
|
|
{
|
|
Box::pin(async move {
|
|
let digest = tokio::task::block_in_place(|| digest.compute(v.as_ref()));
|
|
|
|
let c = self
|
|
.header("Digest", format!("{}={}", D::NAME, digest))
|
|
.signature(&config, key_id, f)?;
|
|
|
|
c.body(v).send().await.map_err(E::from)
|
|
})
|
|
}
|
|
}
|