fix: better error handling

vectordotdev · Dec 31, 2024 · 8631757 · 8631757
1 parent 59679e0
commit 8631757
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/sources/util/http_client.rs b/src/sources/util/http_client.rs
@@ -219,11 +219,13 @@ pub(crate) async fn call<
                         if status != 401 {
                             return Ok(Err(crate::http::HttpError::DigestAuthExpectation))
                         }
-                        let parts = response_headers
-                            .get("www-authenticate")
-                            .unwrap()
-                            .to_str()
-                            .unwrap();
+                        let parts = match response_headers.get("www-authenticate") {
+                            Some(header_value) => match header_value.to_str() {
+                                Ok(value) => value,
+                                Err(_) => return Ok(Err(crate::http::HttpError::DigestAuthExpectation)),
+                            },
+                            None => return Ok(Err(crate::http::HttpError::DigestAuthExpectation)),
+                        };
                         let parts: Vec<&str> = parts.split(",").collect();
                         let mut realm = "";
                         let mut nonce = "";