Pass through ureq::Error wrapped as io::Error

Author: algesten

CHANGELOG.md

@@ -1,6 +1,7 @@
 # Unreleased
 
- * send_json should set content-length header (#983)
+  * ureq::Error wrapped as io::Error should pass through body chain (#984)
+  * send_json should set content-length header (#983)
 
 # 3.0.4
 

src/body/brotli.rs

@@ -2,6 +2,7 @@ use std::io;
 
 use brotli_decompressor::Decompressor;
 
+use crate::error::is_wrapped_ureq_error;
 use crate::Error;
 
 pub(crate) struct BrotliDecoder<R: io::Read>(Decompressor<R>);
@@ -14,8 +15,13 @@ impl<R: io::Read> BrotliDecoder<R> {
 
 impl<R: io::Read> io::Read for BrotliDecoder<R> {
     fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
-        self.0
-            .read(buf)
-            .map_err(|e| Error::Decompress("brotli", e).into_io())
+        self.0.read(buf).map_err(|e| {
+            if is_wrapped_ureq_error(&e) {
+                // If this already is a ureq::Error, like Timeout, pass it along.
+                e
+            } else {
+                Error::Decompress("brotli", e).into_io()
+            }
+        })
     }
 }

src/body/gzip.rs

@@ -2,6 +2,7 @@ use std::io;
 
 use flate2::read::MultiGzDecoder;
 
+use crate::error::is_wrapped_ureq_error;
 use crate::Error;
 
 pub(crate) struct GzipDecoder<R>(MultiGzDecoder<R>);
@@ -14,9 +15,14 @@ impl<R: io::Read> GzipDecoder<R> {
 
 impl<R: io::Read> io::Read for GzipDecoder<R> {
     fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
-        self.0
-            .read(buf)
-            .map_err(|e| Error::Decompress("gzip", e).into_io())
+        self.0.read(buf).map_err(|e| {
+            if is_wrapped_ureq_error(&e) {
+                // If this already is a ureq::Error, like Timeout, pass it along.
+                e
+            } else {
+                Error::Decompress("gzip", e).into_io()
+            }
+        })
     }
 }
 

src/error.rs

@@ -169,11 +169,13 @@ impl Error {
     }
 }
 
+pub(crate) fn is_wrapped_ureq_error(e: &io::Error) -> bool {
+    e.get_ref().map(|x| x.is::<Error>()).unwrap_or(false)
+}
+
 impl From<io::Error> for Error {
     fn from(e: io::Error) -> Self {
-        let is_wrapped_ureq_error = e.get_ref().map(|x| x.is::<Error>()).unwrap_or(false);
-
-        if is_wrapped_ureq_error {
+        if is_wrapped_ureq_error(&e) {
             // unwraps are ok, see above.
             let boxed = e.into_inner().unwrap();
             let ureq = boxed.downcast::<Error>().unwrap();