Override #ensureCanReadBytes in Netty3 ChannelBufferStreamInput

Author: s1monw

modules/transport-netty3/src/main/java/org/elasticsearch/transport/netty3/ChannelBufferStreamInput.java

@@ -24,6 +24,7 @@
 import org.elasticsearch.common.io.stream.StreamInput;
 import org.jboss.netty.buffer.ChannelBuffer;
 
+import java.io.EOFException;
 import java.io.IOException;
 
 /**
@@ -67,6 +68,14 @@ public int available() throws IOException {
         return endIndex - buffer.readerIndex();
     }
 
+    @Override
+    protected void ensureCanReadBytes(int length) throws EOFException {
+        int bytesAvailable = endIndex - buffer.readerIndex();
+        if (bytesAvailable < length) {
+            throw new EOFException("tried to read: " + length + " bytes but only " + bytesAvailable + " remaining");
+        }
+    }
+
     @Override
     public void mark(int readlimit) {
         buffer.markReaderIndex();