Use reqwest to send activities

This commit is contained in:
Felix Ableitner 2020-09-29 15:10:55 +02:00
parent 927ab1f040
commit 0aa0ea19fb
4 changed files with 57 additions and 33 deletions

1
Cargo.lock generated
View File

@ -1865,6 +1865,7 @@ dependencies = [
"diesel", "diesel",
"futures", "futures",
"http", "http",
"http-signature-normalization",
"http-signature-normalization-actix", "http-signature-normalization-actix",
"itertools", "itertools",
"lazy_static", "lazy_static",

View File

@ -33,6 +33,7 @@ url = { version = "2.1", features = ["serde"] }
percent-encoding = "2.1" percent-encoding = "2.1"
openssl = "0.10" openssl = "0.10"
http = "0.2" http = "0.2"
http-signature-normalization = "0.5"
http-signature-normalization-actix = { version = "0.4", default-features = false, features = ["sha-2"] } http-signature-normalization-actix = { version = "0.4", default-features = false, features = ["sha-2"] }
base64 = "0.12" base64 = "0.12"
tokio = "0.2" tokio = "0.2"

View File

@ -4,7 +4,6 @@ use activitystreams::{
object::AsObject, object::AsObject,
}; };
use anyhow::{anyhow, Context, Error}; use anyhow::{anyhow, Context, Error};
use awc::Client;
use background_jobs::{ use background_jobs::{
create_server, create_server,
memory_storage::Storage, memory_storage::Storage,
@ -16,8 +15,9 @@ use background_jobs::{
}; };
use lemmy_utils::{location_info, settings::Settings, LemmyError}; use lemmy_utils::{location_info, settings::Settings, LemmyError};
use log::warn; use log::warn;
use reqwest::Client;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::{future::Future, pin::Pin}; use std::{collections::BTreeMap, future::Future, pin::Pin};
use url::Url; use url::Url;
pub fn send_activity<T, Kind>( pub fn send_activity<T, Kind>(
@ -50,6 +50,7 @@ where
actor_id: actor.actor_id()?, actor_id: actor.actor_id()?,
private_key: actor.private_key().context(location_info!())?, private_key: actor.private_key().context(location_info!())?,
}; };
activity_sender.queue::<SendActivityTask>(message)?; activity_sender.queue::<SendActivityTask>(message)?;
Ok(()) Ok(())
@ -74,18 +75,18 @@ impl ActixJob for SendActivityTask {
fn run(self, state: Self::State) -> Self::Future { fn run(self, state: Self::State) -> Self::Future {
Box::pin(async move { Box::pin(async move {
for to_url in &self.to { for to_url in &self.to {
let request = state let mut headers = BTreeMap::<String, String>::new();
.client headers.insert("Content-Type".into(), "application/json".into());
.post(to_url.as_str())
.header("Content-Type", "application/json");
let signed = sign( let signed = sign(
request, &state.client,
headers,
to_url,
self.activity.clone(), self.activity.clone(),
&self.actor_id, &self.actor_id,
self.private_key.to_owned(), self.private_key.to_owned(),
) )
.await; .await;
let signed = match signed { let signed = match signed {
Ok(s) => s, Ok(s) => s,
Err(e) => { Err(e) => {
@ -94,7 +95,7 @@ impl ActixJob for SendActivityTask {
return Ok(()); return Ok(());
} }
}; };
if let Err(e) = signed.send().await { if let Err(e) = state.client.execute(signed).await {
warn!("{}", e); warn!("{}", e);
return Err(anyhow!( return Err(anyhow!(
"Failed to send activity {} to {}", "Failed to send activity {} to {}",
@ -103,7 +104,6 @@ impl ActixJob for SendActivityTask {
)); ));
} }
} }
Ok(()) Ok(())
}) })
} }

View File

@ -1,12 +1,11 @@
use crate::ActorType; use crate::ActorType;
use activitystreams::unparsed::UnparsedMutExt; use activitystreams::unparsed::UnparsedMutExt;
use activitystreams_ext::UnparsedExtension; use activitystreams_ext::UnparsedExtension;
use actix_web::{client::ClientRequest, HttpRequest}; use actix_web::HttpRequest;
use anyhow::{anyhow, Context}; use anyhow::{anyhow, Context};
use http_signature_normalization_actix::{ use http::{header::HeaderName, HeaderMap, HeaderValue};
digest::{DigestClient, SignExt}, use http_signature_normalization::Config;
Config, use http_signature_normalization_actix::{digest::DigestCreate, Config as ConfigActix};
};
use lemmy_utils::{location_info, LemmyError}; use lemmy_utils::{location_info, LemmyError};
use log::debug; use log::debug;
use openssl::{ use openssl::{
@ -14,45 +13,68 @@ use openssl::{
pkey::PKey, pkey::PKey,
sign::{Signer, Verifier}, sign::{Signer, Verifier},
}; };
use reqwest::{Client, Request};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use sha2::{Digest, Sha256}; use sha2::{Digest, Sha256};
use std::{collections::BTreeMap, str::FromStr};
use url::Url; use url::Url;
lazy_static! { lazy_static! {
static ref CONFIG2: ConfigActix = ConfigActix::new();
static ref HTTP_SIG_CONFIG: Config = Config::new(); static ref HTTP_SIG_CONFIG: Config = Config::new();
} }
/// Signs request headers with the given keypair. /// Signs request headers with the given keypair.
pub async fn sign( pub async fn sign(
request: ClientRequest, client: &Client,
headers: BTreeMap<String, String>,
url: &Url,
activity: String, activity: String,
actor_id: &Url, actor_id: &Url,
private_key: String, private_key: String,
) -> Result<DigestClient<String>, LemmyError> { ) -> Result<Request, LemmyError> {
let signing_key_id = format!("{}#main-key", actor_id); let signing_key_id = format!("{}#main-key", actor_id);
let digest_client = request let mut path_and_query = url.path().to_string();
.signature_with_digest( if let Some(query) = url.query() {
HTTP_SIG_CONFIG.clone(), path_and_query = format!("{}?{}", path_and_query, query);
signing_key_id, }
Sha256::new(), let signature_header_value = HTTP_SIG_CONFIG
activity, .begin_sign("POST", &path_and_query, headers.clone())?
move |signing_string| { .sign(signing_key_id, |signing_string| {
let private_key = PKey::private_key_from_pem(private_key.as_bytes())?; let private_key = PKey::private_key_from_pem(private_key.as_bytes())?;
let mut signer = Signer::new(MessageDigest::sha256(), &private_key)?; let mut signer = Signer::new(MessageDigest::sha256(), &private_key)?;
signer.update(signing_string.as_bytes())?; signer.update(signing_string.as_bytes())?;
Ok(base64::encode(signer.sign_to_vec()?)) as Result<_, LemmyError> Ok(base64::encode(signer.sign_to_vec()?)) as Result<_, LemmyError>
}, })?
) .signature_header();
.await?; let digest = format!(
"{}={}",
Sha256::NAME,
Sha256::new().compute(activity.as_bytes())
);
Ok(digest_client) let mut header_map = HeaderMap::new();
for h in headers {
header_map.insert(
HeaderName::from_str(h.0.as_str())?,
HeaderValue::from_str(h.1.as_str())?,
);
}
let signed_request = client
.post(&url.to_string())
.headers(header_map)
.header("Signature", signature_header_value)
.header("Digest", digest)
.body(activity);
Ok(signed_request.build()?)
} }
pub fn verify(request: &HttpRequest, actor: &dyn ActorType) -> Result<(), LemmyError> { pub fn verify(request: &HttpRequest, actor: &dyn ActorType) -> Result<(), LemmyError> {
let public_key = actor.public_key().context(location_info!())?; let public_key = actor.public_key().context(location_info!())?;
let verified = HTTP_SIG_CONFIG let verified = CONFIG2
.begin_verify( .begin_verify(
request.method(), request.method(),
request.uri().path_and_query(), request.uri().path_and_query(),