fix an offset issue in OTA

phrnet · Jul 13, 2016 · fc69c77 · fc69c77
1 parent d602025
commit fc69c77
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/proxy/shadowsocks/ota.go b/proxy/shadowsocks/ota.go
@@ -96,7 +96,7 @@ func (this *ChunkReader) Read() (*alloc.Buffer, error) {
 		log.Debug("AuthenticationReader: Unexpected auth: ", authBytes)
 		return nil, transport.ErrCorruptedPacket
 	}
-	buffer.Value = payload
+	buffer.SliceFrom(AuthSize)
 
 	return buffer, nil
 }
diff --git a/proxy/shadowsocks/ota_test.go b/proxy/shadowsocks/ota_test.go
@@ -18,4 +18,7 @@ func TestNormalChunkReading(t *testing.T) {
 	payload, err := reader.Read()
 	assert.Error(err).IsNil()
 	assert.Bytes(payload.Value).Equals([]byte{11, 12, 13, 14, 15, 16, 17, 18})
+
+	payload.PrependBytes(3, 4)
+	assert.Bytes(payload.Value).Equals([]byte{3, 4, 11, 12, 13, 14, 15, 16, 17, 18})
 }