let status = resp.status();
let mut resp = resp.into_body();
let max_body = c.ic.max_batch_down.sat() + MAX_OVERHEAD;
- let resp = read_limited_bytes(max_body, default(), &mut resp).await
+ let resp = read_limited_bytes(
+ max_body, default(), default(), &mut resp
+ ).await
.discard_data().context("fetching response body")?;
if ! status.is_success() {
let mut body = req.into_body();
let initial = match read_limited_bytes(
- METADATA_MAX_LEN, default(), &mut body
+ METADATA_MAX_LEN, default(), default(), &mut body
).await {
Ok(all) => all,
Err(ReadLimitedError::Truncated { sofar,.. }) => sofar,
#[throws(ReadLimitedError)]
pub async fn read_limited_bytes<S>(limit: usize, initial: Box<[u8]>,
+ capacity: usize,
stream: &mut S) -> Box<[u8]>
where S: futures::Stream<Item=Result<hyper::body::Bytes,hyper::Error>>
+ Debug + Unpin,
// we also require that the Stream is cancellation-safe
{
let mut accum = initial.into_vec();
+ let capacity = min(limit, capacity);
+ if capacity > accum.len() { accum.reserve(capacity - accum.len()); }
while let Some(item) = stream.next().await {
let b = item?;
accum.extend(b);