From acdf893362196125729cd5ec7a6feb7fe5668e9a Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Tue, 25 Jan 2022 11:29:21 +0100 Subject: [PATCH] Fix partnumber --- src/api/s3_get.rs | 208 ++++++++++++++++++++++++++-------------------- 1 file changed, 117 insertions(+), 91 deletions(-) diff --git a/src/api/s3_get.rs b/src/api/s3_get.rs index 86b58d54..7f647e15 100644 --- a/src/api/s3_get.rs +++ b/src/api/s3_get.rs @@ -19,6 +19,8 @@ use garage_model::version_table::*; use crate::error::*; +const X_AMZ_MP_PARTS_COUNT: &str = "x-amz-mp-parts-count"; + fn object_headers( version: &ObjectVersion, version_meta: &ObjectVersionMeta, @@ -120,39 +122,49 @@ pub async fn handle_head( } if let Some(pn) = part_number { - if let ObjectVersionData::Inline(_, _) = version_data { - // Not a multipart upload - return Err(Error::BadRequest( - "Cannot process part_number argument: not a multipart upload".into(), - )); + match version_data { + ObjectVersionData::Inline(_, bytes) => { + if pn != 1 { + return Err(Error::InvalidPart); + } + Ok(object_headers(object_version, version_meta) + .header(CONTENT_LENGTH, format!("{}", bytes.len())) + .header( + CONTENT_RANGE, + format!("bytes 0-{}/{}", bytes.len() - 1, bytes.len()), + ) + .header(X_AMZ_MP_PARTS_COUNT, "1") + .status(StatusCode::PARTIAL_CONTENT) + .body(Body::empty())?) + } + ObjectVersionData::FirstBlock(_, _) => { + let version = garage + .version_table + .get(&object_version.uuid, &EmptyKey) + .await? + .ok_or(Error::NoSuchKey)?; + + let (part_offset, part_end) = + calculate_part_bounds(&version, pn).ok_or(Error::InvalidPart)?; + let n_parts = version.parts_etags.items().len(); + + Ok(object_headers(object_version, version_meta) + .header(CONTENT_LENGTH, format!("{}", part_end - part_offset)) + .header( + CONTENT_RANGE, + format!( + "bytes {}-{}/{}", + part_offset, + part_end - 1, + version_meta.size + ), + ) + .header(X_AMZ_MP_PARTS_COUNT, format!("{}", n_parts)) + .status(StatusCode::PARTIAL_CONTENT) + .body(Body::empty())?) + } + _ => unreachable!(), } - - let version = garage - .version_table - .get(&object_version.uuid, &EmptyKey) - .await? - .ok_or(Error::NoSuchKey)?; - if !version.has_part_number(pn) { - return Err(Error::BadRequest(format!( - "Part number {} does not exist", - pn - ))); - } - - let part_size: u64 = version - .blocks - .items() - .iter() - .filter(|(k, _)| k.part_number == pn) - .map(|(_, b)| b.size) - .sum(); - let n_parts = version.parts_etags.items().len(); - - Ok(object_headers(object_version, version_meta) - .header(CONTENT_LENGTH, format!("{}", part_size)) - .header("x-amz-mp-parts-count", format!("{}", n_parts)) - .status(StatusCode::OK) - .body(Body::empty())?) } else { Ok(object_headers(object_version, version_meta) .header(CONTENT_LENGTH, format!("{}", version_meta.size)) @@ -196,22 +208,27 @@ pub async fn handle_get( return Ok(cached); } - if let Some(pn) = part_number { - return handle_get_part(garage, req, last_v, last_v_data, last_v_meta, pn).await; - } - - // No part_number specified, it's a normal get object - - if let Some(range) = parse_range_header(req, last_v_meta.size)? { - return handle_get_range( - garage, - last_v, - last_v_data, - last_v_meta, - range.start, - range.start + range.length, - ) - .await; + match (part_number, parse_range_header(req, last_v_meta.size)?) { + (Some(_), Some(_)) => { + return Err(Error::BadRequest( + "Cannot specify both partNumber and Range header".into(), + )); + } + (Some(pn), None) => { + return handle_get_part(garage, last_v, last_v_data, last_v_meta, pn).await; + } + (None, Some(range)) => { + return handle_get_range( + garage, + last_v, + last_v_data, + last_v_meta, + range.start, + range.start + range.length, + ) + .await; + } + (None, None) => (), } let resp_builder = object_headers(last_v, last_v_meta) @@ -305,58 +322,51 @@ async fn handle_get_range( async fn handle_get_part( garage: Arc, - req: &Request, object_version: &ObjectVersion, version_data: &ObjectVersionData, version_meta: &ObjectVersionMeta, part_number: u64, ) -> Result, Error> { - let version = if let ObjectVersionData::FirstBlock(_, _) = version_data { - garage - .version_table - .get(&object_version.uuid, &EmptyKey) - .await? - .ok_or(Error::NoSuchKey)? - } else { - return Err(Error::BadRequest( - "Cannot handle part_number: not a multipart upload.".into(), - )); - }; + let resp_builder = + object_headers(object_version, version_meta).status(StatusCode::PARTIAL_CONTENT); - let blocks = version - .blocks - .items() - .iter() - .filter(|(k, _)| k.part_number == part_number) - .cloned() - .collect::>(); + match version_data { + ObjectVersionData::Inline(_, bytes) => { + if part_number != 1 { + return Err(Error::InvalidPart); + } + Ok(resp_builder + .header(CONTENT_LENGTH, format!("{}", bytes.len())) + .header( + CONTENT_RANGE, + format!("bytes {}-{}/{}", 0, bytes.len() - 1, bytes.len()), + ) + .header(X_AMZ_MP_PARTS_COUNT, "1") + .body(Body::from(bytes.to_vec()))?) + } + ObjectVersionData::FirstBlock(_, _) => { + let version = garage + .version_table + .get(&object_version.uuid, &EmptyKey) + .await? + .ok_or(Error::NoSuchKey)?; - if blocks.is_empty() { - return Err(Error::BadRequest(format!("No such part: {}", part_number))); - } + let (begin, end) = + calculate_part_bounds(&version, part_number).ok_or(Error::InvalidPart)?; + let n_parts = version.parts_etags.items().len(); - let part_size = blocks.iter().map(|(_, b)| b.size).sum(); + let body = body_from_blocks_range(garage, version.blocks.items(), begin, end); - if let Some(r) = parse_range_header(req, part_size)? { - let range_begin = r.start; - let range_end = r.start + r.length; - let body = body_from_blocks_range(garage, &blocks[..], range_begin, range_end); - - Ok(object_headers(object_version, version_meta) - .header(CONTENT_LENGTH, format!("{}", range_end - range_begin)) - .header( - CONTENT_RANGE, - format!("bytes {}-{}/{}", range_begin, range_end - 1, part_size), - ) - .status(StatusCode::PARTIAL_CONTENT) - .body(body)?) - } else { - let body = body_from_blocks_range(garage, &blocks[..], 0, part_size); - - Ok(object_headers(object_version, version_meta) - .header(CONTENT_LENGTH, format!("{}", part_size)) - .status(StatusCode::OK) - .body(body)?) + Ok(resp_builder + .header(CONTENT_LENGTH, format!("{}", end - begin)) + .header( + CONTENT_RANGE, + format!("bytes {}-{}/{}", begin, end - 1, version_meta.size), + ) + .header(X_AMZ_MP_PARTS_COUNT, format!("{}", n_parts)) + .body(body)?) + } + _ => unreachable!(), } } @@ -382,6 +392,22 @@ fn parse_range_header( Ok(range) } +fn calculate_part_bounds(v: &Version, part_number: u64) -> Option<(u64, u64)> { + let mut offset = 0; + for (i, (bk, bv)) in v.blocks.items().iter().enumerate() { + if bk.part_number == part_number { + let size: u64 = v.blocks.items()[i..] + .iter() + .take_while(|(k, _)| k.part_number == part_number) + .map(|(_, v)| v.size) + .sum(); + return Some((offset, offset + size)); + } + offset += bv.size; + } + None +} + fn body_from_blocks_range( garage: Arc, all_blocks: &[(VersionBlockKey, VersionBlock)],