Send reason in the Websocket close handshake

1 files changed, 4 insertions, 1 deletions

diff --git a/components/script/dom/websocket.rs b/components/script/dom/websocket.rs
index 279447070d0..9bc8b7bda88 100644
--- a/components/script/dom/websocket.rs
+++ b/components/script/dom/websocket.rs
@@ -40,6 +40,7 @@ use websocket::client::receiver::Receiver;
 use websocket::client::request::Url;
 use websocket::client::sender::Sender;
 use websocket::header::Origin;
+use websocket::message::CloseData;
 use websocket::result::WebSocketResult;
 use websocket::stream::WebSocketStream;
 use websocket::ws::receiver::Receiver as WSReceiver;
@@ -292,7 +293,9 @@ impl WebSocketMethods for WebSocket {
             let mut sender = this.sender.borrow_mut();
             //TODO: Also check if the buffer is full
             if let Some(sender) = sender.as_mut() {
-                let _ = sender.lock().unwrap().send_message(Message::Close(None));
+                let code: u16 = this.code.get();
+                let reason = this.reason.borrow().clone();
+                let _ = sender.lock().unwrap().send_message(Message::Close(Some(CloseData::new(code, reason))));
             }
         }