Fixed reading unsigned byte

Author: pontusmelke

driver/src/main/java/org/neo4j/driver/internal/connector/socket/BufferingChunkedInput.java

@@ -96,6 +96,7 @@ int remainingChunkSize()
         return remainingChunkSize;
     }
 
+
     /**
      * Internal state machine used for reading data from the channel into the buffer.
      */
@@ -126,7 +127,7 @@ else if ( ctx.buffer.remaining() >= 2 )
                         {
                             //only 1 byte in buffer, read that and continue
                             //to read header
-                            byte partialChunkSize = ctx.buffer.get();
+                            int partialChunkSize = getUnsignedByteFromBuffer( ctx.buffer );
                             ctx.remainingChunkSize = partialChunkSize << 8;
                             return IN_HEADER.readChunkSize( ctx );
                         }
@@ -391,6 +392,12 @@ public byte peekByte() throws IOException
         return buffer.get( buffer.position() );
     }
 
+    static int getUnsignedByteFromBuffer( ByteBuffer buffer )
+    {
+        return buffer.get() & 0xFF;
+    }
+
+
     private boolean hasMoreDataUnreadInCurrentChunk()
     {
         return remainingChunkSize > 0;

driver/src/test/java/org/neo4j/driver/internal/connector/socket/BufferingChunkedInputTest.java

@@ -104,13 +104,35 @@ public void shouldReadChunkWithSplitHeaderForBigMessages() throws IOException
         assertThat( input.readByte(), equalTo( (byte) 1 ) );
         assertThat( input.remainingChunkSize(), equalTo( packetSize - 1 ) );
 
-        for ( int i = 1; i < 384; i++ )
+        for ( int i = 1; i < packetSize; i++ )
         {
             assertThat( input.readByte(), equalTo( (byte) 1 ) );
         }
         assertThat( input.remainingChunkSize(), equalTo( 0 ) );
     }
 
+    @Test
+    public void shouldReadChunkWithSplitHeaderForBigMessagesWhenInternalBufferHasOneByte() throws IOException
+    {
+        // Given
+        int packetSize = 32780;
+        BufferingChunkedInput input =
+                new BufferingChunkedInput( packets( packet( -128 ), packet( 12 ), fillPacket( packetSize, 1 ) ), 1);
+
+        // Then
+        assertThat( input.readByte(), equalTo( (byte) 1 ) );
+        assertThat( input.remainingChunkSize(), equalTo( packetSize - 1 ) );
+    }
+
+    @Test
+    public void shouldReadUnsignedByteFromBuffer() throws IOException
+    {
+        ByteBuffer buffer = ByteBuffer.allocate( 1 );
+        buffer.put( (byte) -1 );
+        buffer.flip();
+        assertThat(BufferingChunkedInput.getUnsignedByteFromBuffer( buffer ), equalTo( 255 ));
+    }
+
     @Test
     public void shouldReadOneByteInOneChunkWhenBustingBuffer() throws IOException
     {

driver/src/test/java/org/neo4j/driver/internal/connector/socket/ChunkedInputTest.java

@@ -18,6 +18,11 @@
  */
 package org.neo4j.driver.internal.connector.socket;
 
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.mockito.Matchers;
+
 import java.io.ByteArrayInputStream;
 import java.io.IOException;
 import java.nio.ByteBuffer;
@@ -26,11 +31,6 @@
 import java.nio.channels.ReadableByteChannel;
 import java.util.Arrays;
 
-import org.junit.Rule;
-import org.junit.Test;
-import org.junit.rules.ExpectedException;
-import org.mockito.Matchers;
-
 import org.neo4j.driver.v1.exceptions.ClientException;
 import org.neo4j.driver.v1.util.RecordingByteChannel;