From 63c6762c15ec790f54391a71794315599ae0ced8 Mon Sep 17 00:00:00 2001
From: Kevin Butler <haqkrs@gmail.com>
Date: Thu, 29 Oct 2015 23:51:59 +0000
Subject: [PATCH] fix(response): respond with a 500 if a handler panics

---
 src/server/response.rs | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/src/server/response.rs b/src/server/response.rs
index 18c6168933..3523d47f12 100644
--- a/src/server/response.rs
+++ b/src/server/response.rs
@@ -7,6 +7,7 @@ use std::marker::PhantomData;
 use std::mem;
 use std::io::{self, Write};
 use std::ptr;
+use std::thread;
 
 use time::now_utc;
 
@@ -237,6 +238,10 @@ enum Body {
 impl<'a, T: Any> Drop for Response<'a, T> {
     fn drop(&mut self) {
         if TypeId::of::<T>() == TypeId::of::<Fresh>() {
+            if thread::panicking() {
+                self.status = status::StatusCode::InternalServerError;
+            }
+
             let mut body = match self.write_head() {
                 Ok(Body::Chunked) => ChunkedWriter(self.body.get_mut()),
                 Ok(Body::Sized(len)) => SizedWriter(self.body.get_mut(), len),
@@ -343,6 +348,43 @@ mod tests {
         }
     }
 
+    #[test]
+    fn test_fresh_drop_panicing() {
+        use std::thread;
+        use std::sync::{Arc, Mutex};
+
+        use status::StatusCode;
+
+        let stream = MockStream::new();
+        let stream = Arc::new(Mutex::new(stream));
+        let inner_stream = stream.clone();
+        let join_handle = thread::spawn(move || {
+            let mut headers = Headers::new();
+            let mut stream = inner_stream.lock().unwrap();
+            let mut res = Response::new(&mut *stream, &mut headers);
+            *res.status_mut() = StatusCode::NotFound;
+
+            panic!("inside")
+        });
+
+        assert!(join_handle.join().is_err());
+
+        let stream = match stream.lock() {
+            Err(poisoned) => poisoned.into_inner().clone(),
+            Ok(_) => unreachable!()
+        };
+
+        lines! { stream =
+            "HTTP/1.1 500 Internal Server Error",
+            _date,
+            _transfer_encoding,
+            "",
+            "0",
+            "" // empty zero body
+        }
+    }
+
+
     #[test]
     fn test_streaming_drop() {
         use std::io::Write;