From 025db41bba4eb54dc0a2ddba8f55f6913a200962 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Thu, 14 Apr 2022 15:02:49 +0200 Subject: [PATCH] cargo fmt --- src/api/k2v/api_server.rs | 16 ++++++++-------- src/api/k2v/router.rs | 21 ++++++++------------- src/api/router_macros.rs | 5 +---- src/api/s3/api_server.rs | 10 +++++++--- src/api/s3/router.rs | 2 +- src/api/signature/streaming.rs | 26 ++++++++++---------------- 6 files changed, 35 insertions(+), 45 deletions(-) diff --git a/src/api/k2v/api_server.rs b/src/api/k2v/api_server.rs index 2e347088..eb082c8d 100644 --- a/src/api/k2v/api_server.rs +++ b/src/api/k2v/api_server.rs @@ -12,16 +12,14 @@ use garage_util::error::Error as GarageError; use garage_model::garage::Garage; - use crate::error::*; use crate::generic_server::*; use crate::signature::payload::check_payload_signature; use crate::signature::streaming::*; - use crate::helpers::*; -use crate::k2v::router::{Endpoint}; +use crate::k2v::router::Endpoint; use crate::s3::cors::*; pub struct K2VApiServer { @@ -86,7 +84,12 @@ impl ApiHandler for K2VApiServer { Error::Forbidden("Garage does not support anonymous access yet".to_string()) })?; - let req = parse_streaming_body(&api_key, req, &mut content_sha256, &garage.config.s3_api.s3_region)?; + let req = parse_streaming_body( + &api_key, + req, + &mut content_sha256, + &garage.config.s3_api.s3_region, + )?; let bucket_id = resolve_bucket(&garage, &bucket_name, &api_key).await?; let bucket = garage @@ -141,9 +144,6 @@ impl ApiEndpoint for K2VApiEndpoint { } fn add_span_attributes(&self, span: SpanRef<'_>) { - span.set_attribute(KeyValue::new( - "bucket", - self.bucket_name.clone(), - )); + span.set_attribute(KeyValue::new("bucket", self.bucket_name.clone())); } } diff --git a/src/api/k2v/router.rs b/src/api/k2v/router.rs index 0a330785..54476910 100644 --- a/src/api/k2v/router.rs +++ b/src/api/k2v/router.rs @@ -2,11 +2,10 @@ use crate::error::*; use std::borrow::Cow; - use hyper::{Method, Request}; -use crate::router_macros::{router_match, generateQueryParameters}; use crate::helpers::Authorization; +use crate::router_macros::{generateQueryParameters, router_match}; router_match! {@func @@ -49,16 +48,15 @@ impl Endpoint { /// Determine which S3 endpoint a request is for using the request, and a bucket which was /// possibly extracted from the Host header. /// Returns Self plus bucket name, if endpoint is not Endpoint::ListBuckets - pub fn from_request( - req: &Request, - ) -> Result<(Self, String), Error> { + pub fn from_request(req: &Request) -> Result<(Self, String), Error> { let uri = req.uri(); let path = uri.path().trim_start_matches('/'); let query = uri.query(); - let (bucket, partition_key) = - path.split_once('/') .map(|(b, p)| (b.to_owned(), p.trim_start_matches('/'))) - .unwrap_or((path.to_owned(), "")); + let (bucket, partition_key) = path + .split_once('/') + .map(|(b, p)| (b.to_owned(), p.trim_start_matches('/'))) + .unwrap_or((path.to_owned(), "")); if bucket.is_empty() { return Err(Error::BadRequest("Missing bucket name".to_owned())); @@ -136,10 +134,7 @@ impl Endpoint { } /// Determine which endpoint a request is for, knowing it is a PUT. - fn from_put( - partition_key: String, - query: &mut QueryParameters<'_>, - ) -> Result { + fn from_put(partition_key: String, query: &mut QueryParameters<'_>) -> Result { router_match! { @gen_parser (query.keyword.take().unwrap_or_default().as_ref(), partition_key, query, None), @@ -227,7 +222,7 @@ generateQueryParameters! { } mod keywords { - //! This module contain all query parameters with no associated value + //! This module contain all query parameters with no associated value //! used to differentiate endpoints. pub const EMPTY: &str = ""; diff --git a/src/api/router_macros.rs b/src/api/router_macros.rs index fb7a8f6d..8471407c 100644 --- a/src/api/router_macros.rs +++ b/src/api/router_macros.rs @@ -1,5 +1,3 @@ - - /// This macro is used to generate very repetitive match {} blocks in this module /// It is _not_ made to be used anywhere else macro_rules! router_match { @@ -120,7 +118,6 @@ macro_rules! router_match { }; } - /// This macro is used to generate part of the code in this module. It must be called only one, and /// is useless outside of this module. macro_rules! generateQueryParameters { @@ -189,5 +186,5 @@ macro_rules! generateQueryParameters { } } -pub(crate) use router_match; pub(crate) use generateQueryParameters; +pub(crate) use router_match; diff --git a/src/api/s3/api_server.rs b/src/api/s3/api_server.rs index 0485737b..04e3727f 100644 --- a/src/api/s3/api_server.rs +++ b/src/api/s3/api_server.rs @@ -20,7 +20,6 @@ use crate::generic_server::*; use crate::signature::payload::check_payload_signature; use crate::signature::streaming::*; - use crate::helpers::*; use crate::s3::bucket::*; use crate::s3::copy::*; @@ -30,7 +29,7 @@ use crate::s3::get::*; use crate::s3::list::*; use crate::s3::post_object::handle_post_object; use crate::s3::put::*; -use crate::s3::router::{Endpoint}; +use crate::s3::router::Endpoint; use crate::s3::website::*; pub struct S3ApiServer { @@ -127,7 +126,12 @@ impl ApiHandler for S3ApiServer { Error::Forbidden("Garage does not support anonymous access yet".to_string()) })?; - let req = parse_streaming_body(&api_key, req, &mut content_sha256, &garage.config.s3_api.s3_region)?; + let req = parse_streaming_body( + &api_key, + req, + &mut content_sha256, + &garage.config.s3_api.s3_region, + )?; let bucket_name = match bucket_name { None => { diff --git a/src/api/s3/router.rs b/src/api/s3/router.rs index d3ef2eba..0525c649 100644 --- a/src/api/s3/router.rs +++ b/src/api/s3/router.rs @@ -5,8 +5,8 @@ use std::borrow::Cow; use hyper::header::HeaderValue; use hyper::{HeaderMap, Method, Request}; -use crate::router_macros::{router_match, generateQueryParameters}; use crate::helpers::Authorization; +use crate::router_macros::{generateQueryParameters, router_match}; router_match! {@func diff --git a/src/api/signature/streaming.rs b/src/api/signature/streaming.rs index c2067623..a46db706 100644 --- a/src/api/signature/streaming.rs +++ b/src/api/signature/streaming.rs @@ -3,23 +3,23 @@ use std::pin::Pin; use chrono::{DateTime, NaiveDateTime, Utc}; use futures::prelude::*; use futures::task; -use hyper::body::Bytes; -use hyper::{Body, Request}; use garage_model::key_table::Key; use hmac::Mac; +use hyper::body::Bytes; +use hyper::{Body, Request}; use garage_util::data::Hash; -use super::{sha256sum, HmacSha256, LONG_DATETIME, compute_scope}; +use super::{compute_scope, sha256sum, HmacSha256, LONG_DATETIME}; use crate::error::*; pub fn parse_streaming_body( - api_key: &Key, - req: Request, - content_sha256: &mut Option, - region: &str, - ) -> Result, Error> { + api_key: &Key, + req: Request, + content_sha256: &mut Option, + region: &str, +) -> Result, Error> { match req.headers().get("x-amz-content-sha256") { Some(header) if header == "STREAMING-AWS4-HMAC-SHA256-PAYLOAD" => { let signature = content_sha256 @@ -42,13 +42,8 @@ pub fn parse_streaming_body( let date: DateTime = DateTime::from_utc(date, Utc); let scope = compute_scope(&date, region); - let signing_hmac = crate::signature::signing_hmac( - &date, - secret_key, - region, - "s3", - ) - .ok_or_internal_error("Unable to build signing HMAC")?; + let signing_hmac = crate::signature::signing_hmac(&date, secret_key, region, "s3") + .ok_or_internal_error("Unable to build signing HMAC")?; Ok(req.map(move |body| { Body::wrap_stream( @@ -67,7 +62,6 @@ pub fn parse_streaming_body( } } - /// Result of `sha256("")` const EMPTY_STRING_HEX_DIGEST: &str = "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855";