From 8eccf044b02b5dcaefd090a34b6e64bc703c7b85 Mon Sep 17 00:00:00 2001
From: James Coglan <jcoglan@gmail.com>
Date: Sat, 4 Jul 2015 21:06:50 +0100
Subject: [PATCH] Change stage in the Base.shutdown() method to that all
 parsers stop processing input.

---
 lib/websocket/driver.rb      | 1 +
 lib/websocket/driver/hybi.rb | 1 -
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/websocket/driver.rb b/lib/websocket/driver.rb
index a5d1059..577cf7c 100644
--- a/lib/websocket/driver.rb
+++ b/lib/websocket/driver.rb
@@ -130,6 +130,7 @@ module WebSocket
 
     def shutdown(code, reason)
       @ready_state = 3
+      @stage = 5
       emit(:close, CloseEvent.new(code, reason))
     end
 
diff --git a/lib/websocket/driver/hybi.rb b/lib/websocket/driver/hybi.rb
index de851f3..a54304f 100644
--- a/lib/websocket/driver/hybi.rb
+++ b/lib/websocket/driver/hybi.rb
@@ -258,7 +258,6 @@ module WebSocket
       def shutdown(code, reason)
         frame(reason, :close, code) if @ready_state < 2
         @frame = @message = nil
-        @stage = 5
         @extensions.close
         super
       end