From d35c2010c48661e7562b0173fdb8ce1c512afcbc Mon Sep 17 00:00:00 2001
From: Trustin Lee <t@motd.kr>
Date: Wed, 18 Dec 2013 20:29:13 +0900
Subject: [PATCH] Stop decoding after protocol upgrade / Do not use
 singleDecode option

---
 .../handler/codec/http/HttpObjectDecoder.java | 21 +++++++++++++++++--
 .../websocketx/WebSocketClientHandshaker.java |  4 +---
 2 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/codec-http/src/main/java/io/netty/handler/codec/http/HttpObjectDecoder.java b/codec-http/src/main/java/io/netty/handler/codec/http/HttpObjectDecoder.java
index c3532b6b48..88585cd965 100644
--- a/codec-http/src/main/java/io/netty/handler/codec/http/HttpObjectDecoder.java
+++ b/codec-http/src/main/java/io/netty/handler/codec/http/HttpObjectDecoder.java
@@ -126,7 +126,8 @@ public abstract class HttpObjectDecoder extends ReplayingDecoder<HttpObjectDecod
         READ_CHUNKED_CONTENT,
         READ_CHUNK_DELIMITER,
         READ_CHUNK_FOOTER,
-        BAD_MESSAGE
+        BAD_MESSAGE,
+        UPGRADED
     }
 
     /**
@@ -354,6 +355,12 @@ public abstract class HttpObjectDecoder extends ReplayingDecoder<HttpObjectDecod
         case BAD_MESSAGE: {
             // Keep discarding until disconnection.
             buffer.skipBytes(actualReadableBytes());
+            break;
+        }
+        case UPGRADED: {
+            // Do not touch anything read - other handler will replace this codec with the upgraded protocol codec to
+            // take the trafic over.
+            break;
         }
         }
     }
@@ -410,7 +417,17 @@ public abstract class HttpObjectDecoder extends ReplayingDecoder<HttpObjectDecod
     }
 
     private void reset() {
-        message = null;
+        HttpMessage message = this.message;
+        this.message = null;
+
+        if (!isDecodingRequest()) {
+            HttpResponse res = (HttpResponse) message;
+            if (res != null && res.getStatus().code() == 101) {
+                checkpoint(State.UPGRADED);
+                return;
+            }
+        }
+
         checkpoint(State.SKIP_CONTROL_CHARS);
     }
 
diff --git a/codec-http/src/main/java/io/netty/handler/codec/http/websocketx/WebSocketClientHandshaker.java b/codec-http/src/main/java/io/netty/handler/codec/http/websocketx/WebSocketClientHandshaker.java
index 127a1eeda0..ecd9323f71 100644
--- a/codec-http/src/main/java/io/netty/handler/codec/http/websocketx/WebSocketClientHandshaker.java
+++ b/codec-http/src/main/java/io/netty/handler/codec/http/websocketx/WebSocketClientHandshaker.java
@@ -157,10 +157,8 @@ public abstract class WebSocketClientHandshaker {
                        "a HttpResponseDecoder or HttpClientCodec"));
                return promise;
             }
-            codec.setSingleDecode(true);
-        } else {
-            decoder.setSingleDecode(true);
         }
+
         channel.writeAndFlush(request).addListener(new ChannelFutureListener() {
             @Override
             public void operationComplete(ChannelFuture future) {