chore: update dependencies

Cargo.toml

@@ -14,7 +14,7 @@ edition = "2018"
 [dependencies]
 # Used to handle requests
 # TODO: This can become optional as soon as proper configs are supported
-rocket = { git = "https://github.com/SergioBenitez/Rocket.git", rev = "93e62c86eddf7cc9a7fc40b044182f83f0d7d92a", features = ["tls"] } # Used to handle requests
+rocket = { git = "https://github.com/SergioBenitez/Rocket.git", rev = "e1307ddf48dac14e6a37e526098732327bcb86f0", features = ["tls"] } # Used to handle requests
 #rocket = { git = "https://github.com/timokoesters/Rocket.git", branch = "empty_parameters", default-features = false, features = ["tls"] }
 
 # Used for matrix spec type definitions and helpers

src/main.rs

@@ -21,7 +21,6 @@ pub use ruma_wrapper::{ConduitResult, Ruma, RumaResponse};
 
 use rocket::{
     catch, catchers,
-    fairing::AdHoc,
     figment::{
         providers::{Env, Format, Toml},
         Figment,
@@ -31,9 +30,9 @@ use rocket::{
 use tracing::span;
 use tracing_subscriber::{prelude::*, Registry};
 
-fn setup_rocket() -> (rocket::Rocket, Config) {
+async fn setup_rocket() -> (rocket::Rocket<rocket::Build>, Config) {
     // Force log level off, so we can use our own logger
-    std::env::set_var("CONDUIT_LOG_LEVEL", "off");
+    //std::env::set_var("CONDUIT_LOG_LEVEL", "off");
 
     let config =
         Figment::from(rocket::Config::release_default())
@@ -48,9 +47,15 @@ fn setup_rocket() -> (rocket::Rocket, Config) {
     let parsed_config = config
         .extract::<Config>()
         .expect("It looks like your config is invalid. Please take a look at the error");
-    let parsed_config2 = parsed_config.clone();
+
+    let data = Database::load_or_create(parsed_config.clone())
+        .await
+        .expect("config is valid");
+
+    data.sending.start_handler(&data);
 
     let rocket = rocket::custom(config)
+        .manage(data)
         .mount(
             "/",
             routes![
@@ -176,29 +181,23 @@ fn setup_rocket() -> (rocket::Rocket, Config) {
                 server_server::get_profile_information_route,
             ],
         )
-        .register(catchers![
+        .register(
+            "/",
+            catchers![
                 not_found_catcher,
                 forbidden_catcher,
                 unknown_token_catcher,
                 missing_token_catcher,
                 bad_json_catcher
-        ])
-        .attach(AdHoc::on_attach("Config", |rocket| async {
-            let data = Database::load_or_create(parsed_config2)
-                .await
-                .expect("config is valid");
-
-            data.sending.start_handler(&data);
-
-            Ok(rocket.manage(data))
-        }));
+            ],
+        );
 
     (rocket, parsed_config)
 }
 
 #[rocket::main]
 async fn main() {
-    let (rocket, config) = setup_rocket();
+    let (rocket, config) = setup_rocket().await;
 
     if config.allow_jaeger {
         let (tracer, _uninstall) = opentelemetry_jaeger::new_pipeline()
@@ -213,11 +212,11 @@ async fn main() {
 
         rocket.launch().await.unwrap();
     } else {
-        std::env::set_var("CONDUIT_LOG", config.log);
-        pretty_env_logger::init_custom_env("CONDUIT_LOG");
+        //std::env::set_var("CONDUIT_LOG", config.log);
+        //pretty_env_logger::init_custom_env("CONDUIT_LOG");
 
-        let root = span!(tracing::Level::INFO, "app_start", work_units = 2);
-        let _enter = root.enter();
+        //let root = span!(tracing::Level::INFO, "app_start", work_units = 2);
+        //let _enter = root.enter();
 
         rocket.launch().await.unwrap();
     }

src/ruma_wrapper.rs

@@ -11,10 +11,7 @@ use {
     crate::{server_server, utils},
     log::{debug, warn},
     rocket::{
-        data::{
-            ByteUnit, Data, FromDataFuture, FromTransformedData, Transform, TransformFuture,
-            Transformed,
-        },
+        data::{self, ByteUnit, Data, FromData},
         http::Status,
         outcome::Outcome::*,
         response::{self, Responder},
@@ -42,29 +39,15 @@ pub struct Ruma<T: Outgoing> {
 }
 
 #[cfg(feature = "conduit_bin")]
-impl<'a, T: Outgoing> FromTransformedData<'a> for Ruma<T>
+#[rocket::async_trait]
+impl<'a, T: Outgoing> FromData<'a> for Ruma<T>
 where
     T::Incoming: IncomingRequest,
 {
     type Error = ();
-    type Owned = Data;
-    type Borrowed = Self::Owned;
-
-    fn transform<'r>(
-        _req: &'r Request<'_>,
-        data: Data,
-    ) -> TransformFuture<'r, Self::Owned, Self::Error> {
-        Box::pin(async move { Transform::Owned(Success(data)) })
-    }
 
-    fn from_data(
-        request: &'a Request<'_>,
-        outcome: Transformed<'a, Self>,
-    ) -> FromDataFuture<'a, Self, Self::Error> {
+    async fn from_data(request: &'a Request<'_>, data: Data) -> data::Outcome<Self, Self::Error> {
         let metadata = T::Incoming::METADATA;
-
-        Box::pin(async move {
-            let data = rocket::try_outcome!(outcome.owned());
         let db = request
             .guard::<State<'_, crate::Database>>()
             .await
@@ -75,15 +58,15 @@ where
             .headers()
             .get_one("Authorization")
             .map(|s| s[7..].to_owned()) // Split off "Bearer "
-                .or_else(|| request.get_query_value("access_token").and_then(|r| r.ok()));
+            .or_else(|| request.query_value("access_token").and_then(|r| r.ok()));
 
         let limit = db.globals.max_request_size();
         let mut handle = data.open(ByteUnit::Byte(limit.into()));
         let mut body = Vec::new();
         handle.read_to_end(&mut body).await.unwrap();
 
-            let (sender_user, sender_device, from_appservice) = if let Some((_id, registration)) =
-                db.appservice
+        let (sender_user, sender_device, from_appservice) = if let Some((_id, registration)) = db
+            .appservice
             .iter_all()
             .filter_map(|r| r.ok())
             .find(|(_id, registration)| {
@@ -94,7 +77,7 @@ where
             }) {
             match metadata.authentication {
                 AuthScheme::AccessToken | AuthScheme::QueryOnlyAccessToken => {
-                        let user_id = request.get_query_value::<String>("user_id").map_or_else(
+                    let user_id = request.query_value::<String>("user_id").map_or_else(
                         || {
                             UserId::parse_with_server_name(
                                 registration
@@ -217,15 +200,16 @@ where
                         "uri".to_owned(),
                         CanonicalJsonValue::String(request.uri().to_string()),
                     );
+
+                    println!("{}: {:?}", origin, request.uri().to_string());
+
                     request_map.insert(
                         "origin".to_owned(),
                         CanonicalJsonValue::String(origin.as_str().to_owned()),
                     );
                     request_map.insert(
                         "destination".to_owned(),
-                            CanonicalJsonValue::String(
-                                db.globals.server_name().as_str().to_owned(),
-                            ),
+                        CanonicalJsonValue::String(db.globals.server_name().as_str().to_owned()),
                     );
 
                     let mut origin_signatures = BTreeMap::new();
@@ -265,10 +249,7 @@ where
                     match ruma::signatures::verify_json(&pub_key_map, &request_map) {
                         Ok(()) => (None, None, false),
                         Err(e) => {
-                                warn!(
-                                    "Failed to verify json request: {}: {:?} {:?}",
-                                    e, pub_key_map, request_map
-                                );
+                            warn!("Failed to verify json request from {}: {}", origin, e,);
 
                             // Forbidden
                             return Failure((Status::raw(580), ()));
@@ -304,7 +285,6 @@ where
                 Failure((Status::raw(583), ()))
             }
         }
-        })
     }
 }