From 4f880c6b75e6b02072fa5dd37ee475538533393a Mon Sep 17 00:00:00 2001
From: Valentine Wallace <vwallace@protonmail.com>
Date: Tue, 11 May 2021 19:59:55 -0400
Subject: [PATCH] Return more error details on http's read_response

Otherwise helpful error information gets swallowed
---
 lightning-block-sync/src/http.rs | 51 ++++++++++++++++++++++++--------
 lightning-block-sync/src/rest.rs |  2 +-
 lightning-block-sync/src/rpc.rs  |  2 +-
 3 files changed, 40 insertions(+), 15 deletions(-)

diff --git a/lightning-block-sync/src/http.rs b/lightning-block-sync/src/http.rs
index 2cfb8e505..154afa488 100644
--- a/lightning-block-sync/src/http.rs
+++ b/lightning-block-sync/src/http.rs
@@ -279,32 +279,27 @@ impl HttpClient {
 			}
 		}
 
-		if !status.is_ok() {
-			// TODO: Handle 3xx redirection responses.
-			return Err(std::io::Error::new(std::io::ErrorKind::NotFound, "not found"));
-		}
-
 		// Read message body
 		let read_limit = MAX_HTTP_MESSAGE_BODY_SIZE - reader.buffer().len();
 		reader.get_mut().set_limit(read_limit as u64);
-		match message_length {
-			HttpMessageLength::Empty => { Ok(Vec::new()) },
+		let contents = match message_length {
+			HttpMessageLength::Empty => { Vec::new() },
 			HttpMessageLength::ContentLength(length) => {
 				if length == 0 || length > MAX_HTTP_MESSAGE_BODY_SIZE {
-					Err(std::io::Error::new(std::io::ErrorKind::InvalidData, "out of range"))
+					return Err(std::io::Error::new(std::io::ErrorKind::InvalidData, "out of range"))
 				} else {
 					let mut content = vec![0; length];
 					#[cfg(feature = "tokio")]
 					reader.read_exact(&mut content[..]).await?;
 					#[cfg(not(feature = "tokio"))]
 					reader.read_exact(&mut content[..])?;
-					Ok(content)
+					content
 				}
 			},
 			HttpMessageLength::TransferEncoding(coding) => {
 				if !coding.eq_ignore_ascii_case("chunked") {
-					Err(std::io::Error::new(
-							std::io::ErrorKind::InvalidInput, "unsupported transfer coding"))
+					return Err(std::io::Error::new(
+						std::io::ErrorKind::InvalidInput, "unsupported transfer coding"))
 				} else {
 					let mut content = Vec::new();
 					#[cfg(feature = "tokio")]
@@ -339,17 +334,30 @@ impl HttpClient {
 							reader.read_exact(&mut content[chunk_offset..]).await?;
 							content.resize(chunk_offset + chunk_size, 0);
 						}
-						Ok(content)
+						content
 					}
 					#[cfg(not(feature = "tokio"))]
 					{
 						let mut decoder = chunked_transfer::Decoder::new(reader);
 						decoder.read_to_end(&mut content)?;
-						Ok(content)
+						content
 					}
 				}
 			},
+		};
+
+		if !status.is_ok() {
+			// TODO: Handle 3xx redirection responses.
+			let error_details = match contents.is_ascii() {
+				true => String::from_utf8_lossy(&contents).to_string(),
+				false => "binary".to_string()
+			};
+			let error_msg = format!("Errored with status: {} and contents: {}",
+			                        status.code, error_details);
+			return Err(std::io::Error::new(std::io::ErrorKind::Other, error_msg));
 		}
+
+		Ok(contents)
 	}
 }
 
@@ -720,6 +728,23 @@ pub(crate) mod client_tests {
 		}
 	}
 
+	#[tokio::test]
+	async fn read_error() {
+		let response = String::from(
+			"HTTP/1.1 500 Internal Server Error\r\n\
+			 Content-Length: 10\r\n\r\ntest error\r\n");
+		let server = HttpServer::responding_with(response);
+
+		let mut client = HttpClient::connect(&server.endpoint()).unwrap();
+		match client.get::<JsonResponse>("/foo", "foo.com").await {
+			Err(e) => {
+				assert_eq!(e.get_ref().unwrap().to_string(), "Errored with status: 500 and contents: test error");
+				assert_eq!(e.kind(), std::io::ErrorKind::Other);
+			},
+			Ok(_) => panic!("Expected error"),
+		}
+	}
+
 	#[tokio::test]
 	async fn read_empty_message_body() {
 		let server = HttpServer::responding_with_ok::<String>(MessageBody::Empty);
diff --git a/lightning-block-sync/src/rest.rs b/lightning-block-sync/src/rest.rs
index 4c1938827..e04bb86d4 100644
--- a/lightning-block-sync/src/rest.rs
+++ b/lightning-block-sync/src/rest.rs
@@ -84,7 +84,7 @@ mod tests {
 		let mut client = RestClient::new(server.endpoint()).unwrap();
 
 		match client.request_resource::<BinaryResponse, u32>("/").await {
-			Err(e) => assert_eq!(e.kind(), std::io::ErrorKind::NotFound),
+			Err(e) => assert_eq!(e.kind(), std::io::ErrorKind::Other),
 			Ok(_) => panic!("Expected error"),
 		}
 	}
diff --git a/lightning-block-sync/src/rpc.rs b/lightning-block-sync/src/rpc.rs
index cc3bc4525..66570bcb5 100644
--- a/lightning-block-sync/src/rpc.rs
+++ b/lightning-block-sync/src/rpc.rs
@@ -118,7 +118,7 @@ mod tests {
 		let mut client = RpcClient::new(CREDENTIALS, server.endpoint()).unwrap();
 
 		match client.call_method::<u64>("getblockcount", &[]).await {
-			Err(e) => assert_eq!(e.kind(), std::io::ErrorKind::NotFound),
+			Err(e) => assert_eq!(e.kind(), std::io::ErrorKind::Other),
 			Ok(_) => panic!("Expected error"),
 		}
 	}
-- 
2.39.5