From 4533bb63cf24b13ba3bdb00cb2aa65f1e42cf1a8 Mon Sep 17 00:00:00 2001 From: Michael Stack Date: Mon, 24 Oct 2016 14:36:42 -0700 Subject: [PATCH] Revert "Revert "HBASE-15789 PB related changes to work with offheap"" Restore this change but with a clean of the generated dirs first so that we avoid trying to apply a patch on top of an already patched src. This reverts commit 0f384158fc3aec9672f4aae4c18fa4d47965567e. --- hbase-assembly/pom.xml | 4 - hbase-protocol-shaded/README.txt | 27 +- hbase-protocol-shaded/pom.xml | 107 +- .../com/google/protobuf/ByteBufferWriter.java | 2 +- .../shaded/com/google/protobuf/ByteInput.java | 81 ++ .../google/protobuf/ByteInputByteString.java | 249 ++++ .../com/google/protobuf/ByteString.java | 7 + .../com/google/protobuf/CodedInputStream.java | 657 +++++++++ .../shaded/com/google/protobuf/Utf8.java | 191 +++ .../src/main/patches/HBASE-15789_V2.patch | 1262 +++++++++++++++++ hbase-protocol/pom.xml | 8 + hbase-server/pom.xml | 4 - 12 files changed, 2550 insertions(+), 49 deletions(-) create mode 100644 hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInput.java create mode 100644 hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInputByteString.java create mode 100644 hbase-protocol-shaded/src/main/patches/HBASE-15789_V2.patch diff --git a/hbase-assembly/pom.xml b/hbase-assembly/pom.xml index d430e3a03e3..185e6813e5c 100644 --- a/hbase-assembly/pom.xml +++ b/hbase-assembly/pom.xml @@ -161,10 +161,6 @@ org.apache.hbase hbase-server - - org.apache.hbase - hbase-thrift - org.apache.hbase diff --git a/hbase-protocol-shaded/README.txt b/hbase-protocol-shaded/README.txt index d6f6ae23549..cbd0eea593d 100644 --- a/hbase-protocol-shaded/README.txt +++ b/hbase-protocol-shaded/README.txt @@ -16,11 +16,16 @@ protobuf Message class is at org.apache.hadoop.hbase.shaded.com.google.protobuf.Message rather than at com.google.protobuf.Message. -Below we describe how to generate the java files for this -module. Run this step any time you change the proto files -in this module or if you change the protobuf version. If you -add a new file, be sure to add mention of the proto in the -pom.xml (scroll till you see the listing of protos to consider). +Finally, this module also includes patches applied on top of +protobuf to add functionality not yet in protobuf that we +need now. + +The shaded generated java files, including the patched protobuf +source files are all checked in. + +If you make changes to protos, to the protobuf version or to +the patches you want to apply to protobuf, you must rerun this +step. First ensure that the appropriate protobuf protoc tool is in your $PATH as in: @@ -33,16 +38,22 @@ build protoc first. Run: - $ mvn install -Dgenerate-shaded-classes + $ mvn install -Dcompile-protobuf or - $ mvn install -Pgenerate-shaded-classes + $ mvn install -Pcompille-protobuf to build and trigger the special generate-shaded-classes profile. When finished, the content of src/main/java/org/apache/hadoop/hbase/shaded will have -been updated. Check in the changes. +been updated. Make sure all builds and then carefully +check in the changes. Files may have been added or removed +by the steps above. + +If you have patches for the protobuf, add them to +src/main/patches directory. They will be applied after +protobuf is shaded and unbundled into src/main/java. See the pom.xml under the generate-shaded-classes profile for more info on how this step works. diff --git a/hbase-protocol-shaded/pom.xml b/hbase-protocol-shaded/pom.xml index 071b8df2b97..01845ae2fd0 100644 --- a/hbase-protocol-shaded/pom.xml +++ b/hbase-protocol-shaded/pom.xml @@ -151,37 +151,32 @@ true - - generate-shaded-classes + compile-protobuf + - generate-shaded-classes + compile-protobuf @@ -191,10 +186,35 @@ - ${profile.id}.${artifactId}-${project.version} + ${profile.id}.${project.artifactId}-${project.version} + + maven-clean-plugin + + + pre-compile-protoc + generate-sources + + clean + + + + + ${basedir}/src/main/java/org/apache/hadoop/hbase/shaded + + ipc/protobuf/generated/**/*.java + protobuf/generated/**/*.java + com/google/protobuf/**/*.java + + false + + + + + + org.apache.hadoop hadoop-maven-plugins @@ -337,9 +357,32 @@ - + + org.apache.maven.plugins + maven-patch-plugin + 1.2 + + + ${basedir}/.. + false + + + + patch + + 1 + src/main/patches + ${project.build.directory}/patches-applied.txt + true + + package + + + apply + + + + diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteBufferWriter.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteBufferWriter.java index c0ed636b3c3..906c216bd70 100644 --- a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteBufferWriter.java +++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteBufferWriter.java @@ -112,7 +112,7 @@ final class ByteBufferWriter { } } - private static byte[] getOrCreateBuffer(int requestedSize) { + static byte[] getOrCreateBuffer(int requestedSize) { requestedSize = max(requestedSize, MIN_CACHED_BUFFER_SIZE); byte[] buffer = getBuffer(); diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInput.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInput.java new file mode 100644 index 00000000000..a745d37991f --- /dev/null +++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInput.java @@ -0,0 +1,81 @@ +// Protocol Buffers - Google's data interchange format +// Copyright 2008 Google Inc. All rights reserved. +// https://developers.google.com/protocol-buffers/ +// +// Redistribution and use in source and binary forms, with or without +// modification, are permitted provided that the following conditions are +// met: +// +// * Redistributions of source code must retain the above copyright +// notice, this list of conditions and the following disclaimer. +// * Redistributions in binary form must reproduce the above +// copyright notice, this list of conditions and the following disclaimer +// in the documentation and/or other materials provided with the +// distribution. +// * Neither the name of Google Inc. nor the names of its +// contributors may be used to endorse or promote products derived from +// this software without specific prior written permission. +// +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +package org.apache.hadoop.hbase.shaded.com.google.protobuf; + +import java.io.IOException; +import java.nio.ByteBuffer; + +/** + * An input for raw bytes. This is similar to an InputStream but it is offset addressable. All the + * read APIs are relative. + */ +@ExperimentalApi +public abstract class ByteInput { + + /** + * Reads a single byte from the given offset. + * @param offset The offset from where byte to be read + * @return The byte of data at given offset + */ + public abstract byte read(int offset); + + /** + * Reads bytes of data from the given offset into an array of bytes. + * @param offset The src offset within this ByteInput from where data to be read. + * @param out Destination byte array to read data into. + * @return The number of bytes read from ByteInput + */ + public int read(int offset, byte b[]) throws IOException { + return read(offset, b, 0, b.length); + } + + /** + * Reads up to len bytes of data from the given offset into an array of bytes. + * @param offset The src offset within this ByteInput from where data to be read. + * @param out Destination byte array to read data into. + * @param outOffset Offset within the the out byte[] where data to be read into. + * @param len The number of bytes to read. + * @return The number of bytes read from ByteInput + */ + public abstract int read(int offset, byte[] out, int outOffset, int len); + + /** + * Reads bytes of data from the given offset into given {@link ByteBuffer}. + * @param offset he src offset within this ByteInput from where data to be read. + * @param out Destination {@link ByteBuffer} to read data into. + * @return The number of bytes read from ByteInput + */ + public abstract int read(int offset, ByteBuffer out); + + /** + * @return Total number of bytes in this ByteInput. + */ + public abstract int size(); +} diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInputByteString.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInputByteString.java new file mode 100644 index 00000000000..1949602500c --- /dev/null +++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInputByteString.java @@ -0,0 +1,249 @@ +// Protocol Buffers - Google's data interchange format +// Copyright 2008 Google Inc. All rights reserved. +// https://developers.google.com/protocol-buffers/ +// +// Redistribution and use in source and binary forms, with or without +// modification, are permitted provided that the following conditions are +// met: +// +// * Redistributions of source code must retain the above copyright +// notice, this list of conditions and the following disclaimer. +// * Redistributions in binary form must reproduce the above +// copyright notice, this list of conditions and the following disclaimer +// in the documentation and/or other materials provided with the +// distribution. +// * Neither the name of Google Inc. nor the names of its +// contributors may be used to endorse or promote products derived from +// this software without specific prior written permission. +// +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + +package org.apache.hadoop.hbase.shaded.com.google.protobuf; + +import java.io.IOException; +import java.io.InputStream; +import java.io.InvalidObjectException; +import java.io.ObjectInputStream; +import java.io.OutputStream; +import java.nio.ByteBuffer; +import java.nio.charset.Charset; +import java.util.Arrays; +import java.util.Collections; +import java.util.List; + +/** + * A {@link ByteString} that wraps around a {@link ByteInput}. + */ +final class ByteInputByteString extends ByteString.LeafByteString { + private final ByteInput buffer; + private final int offset, length; + + ByteInputByteString(ByteInput buffer, int offset, int length) { + if (buffer == null) { + throw new NullPointerException("buffer"); + } + this.buffer = buffer; + this.offset = offset; + this.length = length; + } + + // ================================================================= + // Serializable + + /** + * Magic method that lets us override serialization behavior. + */ + private Object writeReplace() { + return ByteString.wrap(toByteArray()); + } + + /** + * Magic method that lets us override deserialization behavior. + */ + private void readObject(@SuppressWarnings("unused") ObjectInputStream in) throws IOException { + throw new InvalidObjectException("ByteInputByteString instances are not to be serialized directly");// TODO check here + } + + // ================================================================= + + @Override + public byte byteAt(int index) { + return buffer.read(getAbsoluteOffset(index)); + } + + private int getAbsoluteOffset(int relativeOffset) { + return this.offset + relativeOffset; + } + + @Override + public int size() { + return length; + } + + @Override + public ByteString substring(int beginIndex, int endIndex) { + if (beginIndex < 0 || beginIndex >= size() || endIndex < beginIndex || endIndex >= size()) { + throw new IllegalArgumentException( + String.format("Invalid indices [%d, %d]", beginIndex, endIndex)); + } + return new ByteInputByteString(this.buffer, getAbsoluteOffset(beginIndex), endIndex - beginIndex); + } + + @Override + protected void copyToInternal( + byte[] target, int sourceOffset, int targetOffset, int numberToCopy) { + this.buffer.read(getAbsoluteOffset(sourceOffset), target, targetOffset, numberToCopy); + } + + @Override + public void copyTo(ByteBuffer target) { + this.buffer.read(this.offset, target); + } + + @Override + public void writeTo(OutputStream out) throws IOException { + out.write(toByteArray());// TODO + } + + @Override + boolean equalsRange(ByteString other, int offset, int length) { + return substring(0, length).equals(other.substring(offset, offset + length)); + } + + @Override + void writeToInternal(OutputStream out, int sourceOffset, int numberToWrite) throws IOException { + byte[] buf = ByteBufferWriter.getOrCreateBuffer(numberToWrite); + this.buffer.read(getAbsoluteOffset(sourceOffset), buf, 0, numberToWrite); + out.write(buf, 0, numberToWrite); + } + + @Override + void writeTo(ByteOutput output) throws IOException { + output.writeLazy(toByteArray(), 0, length); + } + + @Override + public ByteBuffer asReadOnlyByteBuffer() { + return ByteBuffer.wrap(toByteArray()).asReadOnlyBuffer(); + } + + @Override + public List asReadOnlyByteBufferList() { + return Collections.singletonList(asReadOnlyByteBuffer()); + } + + @Override + protected String toStringInternal(Charset charset) { + byte[] bytes = toByteArray(); + return new String(bytes, 0, bytes.length, charset); + } + + @Override + public boolean isValidUtf8() { + return Utf8.isValidUtf8(buffer, offset, offset + length); + } + + @Override + protected int partialIsValidUtf8(int state, int offset, int length) { + int off = getAbsoluteOffset(offset); + return Utf8.partialIsValidUtf8(state, buffer, off, off + length); + } + + @Override + public boolean equals(Object other) { + if (other == this) { + return true; + } + if (!(other instanceof ByteString)) { + return false; + } + ByteString otherString = ((ByteString) other); + if (size() != otherString.size()) { + return false; + } + if (size() == 0) { + return true; + } + if (other instanceof RopeByteString) { + return other.equals(this); + } + return Arrays.equals(this.toByteArray(), otherString.toByteArray()); + } + + @Override + protected int partialHash(int h, int offset, int length) { + offset = getAbsoluteOffset(offset); + int end = offset + length; + for (int i = offset; i < end; i++) { + h = h * 31 + buffer.read(i); + } + return h; + } + + @Override + public InputStream newInput() { + return new InputStream() { + private final ByteInput buf = buffer; + private int pos = offset; + private int limit = pos + length; + private int mark = pos; + + @Override + public void mark(int readlimit) { + this.mark = readlimit; + } + + @Override + public boolean markSupported() { + return true; + } + + @Override + public void reset() throws IOException { + this.pos = this.mark; + } + + @Override + public int available() throws IOException { + return this.limit - this.pos; + } + + @Override + public int read() throws IOException { + if (available() <= 0) { + return -1; + } + return this.buf.read(pos++) & 0xFF; + } + + @Override + public int read(byte[] bytes, int off, int len) throws IOException { + int remain = available(); + if (remain <= 0) { + return -1; + } + len = Math.min(len, remain); + buf.read(pos, bytes, off, len); + pos += len; + return len; + } + }; + } + + @Override + public CodedInputStream newCodedInput() { + // We trust CodedInputStream not to modify the bytes, or to give anyone + // else access to them. + return CodedInputStream.newInstance(buffer, offset, length, true); + } +} diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteString.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteString.java index 8cae888f861..ebfa7fa01da 100644 --- a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteString.java +++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteString.java @@ -322,6 +322,13 @@ public abstract class ByteString implements Iterable, Serializable { } } + /** + * Wraps the given bytes into a {@code ByteString}. Intended for internal only usage. + */ + static ByteString wrap(ByteInput buffer, int offset, int length) { + return new ByteInputByteString(buffer, offset, length); + } + /** * Wraps the given bytes into a {@code ByteString}. Intended for internal only * usage to force a classload of ByteString before LiteralByteString. diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/CodedInputStream.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/CodedInputStream.java index 12d70ce9b32..0ad20e5fd11 100644 --- a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/CodedInputStream.java +++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/CodedInputStream.java @@ -149,6 +149,15 @@ public abstract class CodedInputStream { return newInstance(buffer, 0, buffer.length, true); } + /** Create a new CodedInputStream wrapping the given {@link ByteInput}. */ + public static CodedInputStream newInstance(ByteInput buf, boolean bufferIsImmutable) { + return new ByteInputDecoder(buf, bufferIsImmutable); + } + + public static CodedInputStream newInstance(ByteInput buf, int off, int len, boolean bufferIsImmutable) { + return new ByteInputDecoder(buf, off, len, bufferIsImmutable); + } + /** Disable construction/inheritance outside of this class. */ private CodedInputStream() {} @@ -2892,4 +2901,652 @@ public abstract class CodedInputStream { pos = size - tempPos; } } + + private static final class ByteInputDecoder extends CodedInputStream { + + private final ByteInput buffer; + private final boolean immutable; + private int limit; + private int bufferSizeAfterLimit; + private int pos; + private int startPos; + private int lastTag; + private boolean enableAliasing; + + /** The absolute position of the end of the current message. */ + private int currentLimit = Integer.MAX_VALUE; + + private ByteInputDecoder(ByteInput buffer, boolean immutable) { + this(buffer, 0, buffer.size(), immutable); + } + + private ByteInputDecoder(ByteInput buffer, int off, int len, boolean immutable) { + this.buffer = buffer; + pos = off; + limit = off + len; + startPos = pos; + this.immutable = immutable; + } + + @Override + public int readTag() throws IOException { + if (isAtEnd()) { + lastTag = 0; + return 0; + } + + lastTag = readRawVarint32(); + if (WireFormat.getTagFieldNumber(lastTag) == 0) { + // If we actually read zero (or any tag number corresponding to field + // number zero), that's not a valid tag. + throw InvalidProtocolBufferException.invalidTag(); + } + return lastTag; + } + + @Override + public void checkLastTagWas(int value) throws InvalidProtocolBufferException { + if (lastTag != value) { + throw InvalidProtocolBufferException.invalidEndTag(); + } + } + + @Override + public int getLastTag() { + return lastTag; + } + + @Override + public boolean skipField(int tag) throws IOException { + switch (WireFormat.getTagWireType(tag)) { + case WireFormat.WIRETYPE_VARINT: + skipRawVarint(); + return true; + case WireFormat.WIRETYPE_FIXED64: + skipRawBytes(FIXED_64_SIZE); + return true; + case WireFormat.WIRETYPE_LENGTH_DELIMITED: + skipRawBytes(readRawVarint32()); + return true; + case WireFormat.WIRETYPE_START_GROUP: + skipMessage(); + checkLastTagWas( + WireFormat.makeTag(WireFormat.getTagFieldNumber(tag), WireFormat.WIRETYPE_END_GROUP)); + return true; + case WireFormat.WIRETYPE_END_GROUP: + return false; + case WireFormat.WIRETYPE_FIXED32: + skipRawBytes(FIXED_32_SIZE); + return true; + default: + throw InvalidProtocolBufferException.invalidWireType(); + } + } + + @Override + public boolean skipField(int tag, CodedOutputStream output) throws IOException { + switch (WireFormat.getTagWireType(tag)) { + case WireFormat.WIRETYPE_VARINT: + { + long value = readInt64(); + output.writeRawVarint32(tag); + output.writeUInt64NoTag(value); + return true; + } + case WireFormat.WIRETYPE_FIXED64: + { + long value = readRawLittleEndian64(); + output.writeRawVarint32(tag); + output.writeFixed64NoTag(value); + return true; + } + case WireFormat.WIRETYPE_LENGTH_DELIMITED: + { + ByteString value = readBytes(); + output.writeRawVarint32(tag); + output.writeBytesNoTag(value); + return true; + } + case WireFormat.WIRETYPE_START_GROUP: + { + output.writeRawVarint32(tag); + skipMessage(output); + int endtag = + WireFormat.makeTag( + WireFormat.getTagFieldNumber(tag), WireFormat.WIRETYPE_END_GROUP); + checkLastTagWas(endtag); + output.writeRawVarint32(endtag); + return true; + } + case WireFormat.WIRETYPE_END_GROUP: + { + return false; + } + case WireFormat.WIRETYPE_FIXED32: + { + int value = readRawLittleEndian32(); + output.writeRawVarint32(tag); + output.writeFixed32NoTag(value); + return true; + } + default: + throw InvalidProtocolBufferException.invalidWireType(); + } + } + + @Override + public void skipMessage() throws IOException { + while (true) { + final int tag = readTag(); + if (tag == 0 || !skipField(tag)) { + return; + } + } + } + + @Override + public void skipMessage(CodedOutputStream output) throws IOException { + while (true) { + final int tag = readTag(); + if (tag == 0 || !skipField(tag, output)) { + return; + } + } + } + + public double readDouble() throws IOException { + return Double.longBitsToDouble(readRawLittleEndian64()); + } + + @Override + public float readFloat() throws IOException { + return Float.intBitsToFloat(readRawLittleEndian32()); + } + + @Override + public long readUInt64() throws IOException { + return readRawVarint64(); + } + + @Override + public long readInt64() throws IOException { + return readRawVarint64(); + } + + @Override + public int readInt32() throws IOException { + return readRawVarint32(); + } + + @Override + public long readFixed64() throws IOException { + return readRawLittleEndian64(); + } + + @Override + public int readFixed32() throws IOException { + return readRawLittleEndian32(); + } + + @Override + public boolean readBool() throws IOException { + return readRawVarint64() != 0; + } + + @Override + public String readString() throws IOException { + final int size = readRawVarint32(); + if (size > 0 && size <= remaining()) { + byte[] bytes = copyToArray(pos, size); + pos += size; + return new String(bytes, UTF_8); + } + + if (size == 0) { + return ""; + } + if (size < 0) { + throw InvalidProtocolBufferException.negativeSize(); + } + throw InvalidProtocolBufferException.truncatedMessage(); + } + + @Override + public String readStringRequireUtf8() throws IOException { + final int size = readRawVarint32(); + if (size > 0 && size <= remaining()) { + if (!Utf8.isValidUtf8(buffer, pos, pos + size)) { + throw InvalidProtocolBufferException.invalidUtf8(); + } + byte[] bytes = copyToArray(pos, size); + pos += size; + return new String(bytes, UTF_8); + } + + if (size == 0) { + return ""; + } + if (size <= 0) { + throw InvalidProtocolBufferException.negativeSize(); + } + throw InvalidProtocolBufferException.truncatedMessage(); + } + + @Override + public void readGroup(int fieldNumber, MessageLite.Builder builder, + ExtensionRegistryLite extensionRegistry) throws IOException { + if (recursionDepth >= recursionLimit) { + throw InvalidProtocolBufferException.recursionLimitExceeded(); + } + ++recursionDepth; + builder.mergeFrom(this, extensionRegistry); + checkLastTagWas(WireFormat.makeTag(fieldNumber, WireFormat.WIRETYPE_END_GROUP)); + --recursionDepth; + } + + @Override + public T readGroup(int fieldNumber, Parser parser, + ExtensionRegistryLite extensionRegistry) throws IOException { + if (recursionDepth >= recursionLimit) { + throw InvalidProtocolBufferException.recursionLimitExceeded(); + } + ++recursionDepth; + T result = parser.parsePartialFrom(this, extensionRegistry); + checkLastTagWas(WireFormat.makeTag(fieldNumber, WireFormat.WIRETYPE_END_GROUP)); + --recursionDepth; + return result; + } + + @Deprecated + @Override + public void readUnknownGroup(int fieldNumber, MessageLite.Builder builder) throws IOException { + readGroup(fieldNumber, builder, ExtensionRegistryLite.getEmptyRegistry()); + } + + @Override + public void readMessage(MessageLite.Builder builder, ExtensionRegistryLite extensionRegistry) + throws IOException { + final int length = readRawVarint32(); + if (recursionDepth >= recursionLimit) { + throw InvalidProtocolBufferException.recursionLimitExceeded(); + } + final int oldLimit = pushLimit(length); + ++recursionDepth; + builder.mergeFrom(this, extensionRegistry); + checkLastTagWas(0); + --recursionDepth; + popLimit(oldLimit); + } + + @Override + public T readMessage(Parser parser, + ExtensionRegistryLite extensionRegistry) throws IOException { + int length = readRawVarint32(); + if (recursionDepth >= recursionLimit) { + throw InvalidProtocolBufferException.recursionLimitExceeded(); + } + final int oldLimit = pushLimit(length); + ++recursionDepth; + T result = parser.parsePartialFrom(this, extensionRegistry); + checkLastTagWas(0); + --recursionDepth; + popLimit(oldLimit); + return result; + } + + @Override + public ByteString readBytes() throws IOException { + final int size = readRawVarint32(); + if (size > 0 && size <= (limit - pos)) { + // Fast path: We already have the bytes in a contiguous buffer, so + // just copy directly from it. + + final ByteString result = + immutable && enableAliasing + ? ByteString.wrap(buffer, pos, size) + : ByteString.wrap(copyToArray(pos, size)); + pos += size; + return result; + } + if (size == 0) { + return ByteString.EMPTY; + } + // Slow path: Build a byte array first then copy it. + return ByteString.wrap(readRawBytes(size)); + } + + @Override + public byte[] readByteArray() throws IOException { + return readRawBytes(readRawVarint32()); + } + + @Override + public ByteBuffer readByteBuffer() throws IOException { + return ByteBuffer.wrap(readByteArray()); + } + + @Override + public int readUInt32() throws IOException { + return readRawVarint32(); + } + + @Override + public int readEnum() throws IOException { + return readRawVarint32(); + } + + @Override + public int readSFixed32() throws IOException { + return readRawLittleEndian32(); + } + + @Override + public long readSFixed64() throws IOException { + return readRawLittleEndian64(); + } + + @Override + public int readSInt32() throws IOException { + return decodeZigZag32(readRawVarint32()); + } + + @Override + public long readSInt64() throws IOException { + return decodeZigZag64(readRawVarint64()); + } + + @Override + public int readRawVarint32() throws IOException { + // See implementation notes for readRawVarint64 + fastpath: + { + int tempPos = pos; + + if (limit == tempPos) { + break fastpath; + } + + int x; + if ((x = buffer.read(tempPos++)) >= 0) { + pos = tempPos; + return x; + } else if (limit - tempPos < 9) { + break fastpath; + } else if ((x ^= (buffer.read(tempPos++) << 7)) < 0) { + x ^= (~0 << 7); + } else if ((x ^= (buffer.read(tempPos++) << 14)) >= 0) { + x ^= (~0 << 7) ^ (~0 << 14); + } else if ((x ^= (buffer.read(tempPos++) << 21)) < 0) { + x ^= (~0 << 7) ^ (~0 << 14) ^ (~0 << 21); + } else { + int y = buffer.read(tempPos++); + x ^= y << 28; + x ^= (~0 << 7) ^ (~0 << 14) ^ (~0 << 21) ^ (~0 << 28); + if (y < 0 + && buffer.read(tempPos++) < 0 + && buffer.read(tempPos++) < 0 + && buffer.read(tempPos++) < 0 + && buffer.read(tempPos++) < 0 + && buffer.read(tempPos++) < 0) { + break fastpath; // Will throw malformedVarint() + } + } + pos = tempPos; + return x; + } + return (int) readRawVarint64SlowPath(); + } + + @Override + public long readRawVarint64() throws IOException { + fastpath: + { + int tempPos = pos; + + if (limit == tempPos) { + break fastpath; + } + + long x; + int y; + if ((y = buffer.read(tempPos++)) >= 0) { + pos = tempPos; + return y; + } else if (limit - tempPos < 9) { + break fastpath; + } else if ((y ^= (buffer.read(tempPos++) << 7)) < 0) { + x = y ^ (~0 << 7); + } else if ((y ^= (buffer.read(tempPos++) << 14)) >= 0) { + x = y ^ ((~0 << 7) ^ (~0 << 14)); + } else if ((y ^= (buffer.read(tempPos++) << 21)) < 0) { + x = y ^ ((~0 << 7) ^ (~0 << 14) ^ (~0 << 21)); + } else if ((x = y ^ ((long) buffer.read(tempPos++) << 28)) >= 0L) { + x ^= (~0L << 7) ^ (~0L << 14) ^ (~0L << 21) ^ (~0L << 28); + } else if ((x ^= ((long) buffer.read(tempPos++) << 35)) < 0L) { + x ^= (~0L << 7) ^ (~0L << 14) ^ (~0L << 21) ^ (~0L << 28) ^ (~0L << 35); + } else if ((x ^= ((long) buffer.read(tempPos++) << 42)) >= 0L) { + x ^= (~0L << 7) ^ (~0L << 14) ^ (~0L << 21) ^ (~0L << 28) ^ (~0L << 35) ^ (~0L << 42); + } else if ((x ^= ((long) buffer.read(tempPos++) << 49)) < 0L) { + x ^= + (~0L << 7) + ^ (~0L << 14) + ^ (~0L << 21) + ^ (~0L << 28) + ^ (~0L << 35) + ^ (~0L << 42) + ^ (~0L << 49); + } else { + x ^= ((long) buffer.read(tempPos++) << 56); + x ^= + (~0L << 7) + ^ (~0L << 14) + ^ (~0L << 21) + ^ (~0L << 28) + ^ (~0L << 35) + ^ (~0L << 42) + ^ (~0L << 49) + ^ (~0L << 56); + if (x < 0L) { + if (buffer.read(tempPos++) < 0L) { + break fastpath; // Will throw malformedVarint() + } + } + } + pos = tempPos; + return x; + } + return readRawVarint64SlowPath(); + } + + @Override + long readRawVarint64SlowPath() throws IOException { + long result = 0; + for (int shift = 0; shift < 64; shift += 7) { + final byte b = readRawByte(); + result |= (long) (b & 0x7F) << shift; + if ((b & 0x80) == 0) { + return result; + } + } + throw InvalidProtocolBufferException.malformedVarint(); + } + + @Override + public int readRawLittleEndian32() throws IOException { + int tempPos = pos; + + if (limit - tempPos < FIXED_32_SIZE) { + throw InvalidProtocolBufferException.truncatedMessage(); + } + + pos = tempPos + FIXED_32_SIZE; + return (((buffer.read(tempPos) & 0xff)) + | ((buffer.read(tempPos + 1) & 0xff) << 8) + | ((buffer.read(tempPos + 2) & 0xff) << 16) + | ((buffer.read(tempPos + 3) & 0xff) << 24)); + } + + @Override + public long readRawLittleEndian64() throws IOException { + int tempPos = pos; + + if (limit - tempPos < FIXED_64_SIZE) { + throw InvalidProtocolBufferException.truncatedMessage(); + } + + pos = tempPos + FIXED_64_SIZE; + return (((buffer.read(tempPos) & 0xffL)) + | ((buffer.read(tempPos + 1) & 0xffL) << 8) + | ((buffer.read(tempPos + 2) & 0xffL) << 16) + | ((buffer.read(tempPos + 3) & 0xffL) << 24) + | ((buffer.read(tempPos + 4) & 0xffL) << 32) + | ((buffer.read(tempPos + 5) & 0xffL) << 40) + | ((buffer.read(tempPos + 6) & 0xffL) << 48) + | ((buffer.read(tempPos + 7) & 0xffL) << 56)); + } + + @Override + public void enableAliasing(boolean enabled) { + this.enableAliasing = enabled; + } + + @Override + public void resetSizeCounter() { + startPos = pos; + } + + @Override + public int pushLimit(int byteLimit) throws InvalidProtocolBufferException { + if (byteLimit < 0) { + throw InvalidProtocolBufferException.negativeSize(); + } + byteLimit += getTotalBytesRead(); + final int oldLimit = currentLimit; + if (byteLimit > oldLimit) { + throw InvalidProtocolBufferException.truncatedMessage(); + } + currentLimit = byteLimit; + + recomputeBufferSizeAfterLimit(); + + return oldLimit; + } + + @Override + public void popLimit(int oldLimit) { + currentLimit = oldLimit; + recomputeBufferSizeAfterLimit(); + } + + @Override + public int getBytesUntilLimit() { + if (currentLimit == Integer.MAX_VALUE) { + return -1; + } + + return currentLimit - getTotalBytesRead(); + } + + @Override + public boolean isAtEnd() throws IOException { + return pos == limit; + } + + @Override + public int getTotalBytesRead() { + return pos - startPos; + } + + @Override + public byte readRawByte() throws IOException { + if (pos == limit) { + throw InvalidProtocolBufferException.truncatedMessage(); + } + return buffer.read(pos++); + } + + @Override + public byte[] readRawBytes(int length) throws IOException { + if (length > 0 && length <= (limit - pos)) { + byte[] bytes = copyToArray(pos, length); + pos += length; + return bytes; + } + + if (length <= 0) { + if (length == 0) { + return Internal.EMPTY_BYTE_ARRAY; + } else { + throw InvalidProtocolBufferException.negativeSize(); + } + } + throw InvalidProtocolBufferException.truncatedMessage(); + } + + @Override + public void skipRawBytes(int length) throws IOException { + if (length >= 0 && length <= (limit - pos)) { + // We have all the bytes we need already. + pos += length; + return; + } + + if (length < 0) { + throw InvalidProtocolBufferException.negativeSize(); + } + throw InvalidProtocolBufferException.truncatedMessage(); + } + + private void recomputeBufferSizeAfterLimit() { + limit += bufferSizeAfterLimit; + final int bufferEnd = limit - startPos; + if (bufferEnd > currentLimit) { + // Limit is in current buffer. + bufferSizeAfterLimit = bufferEnd - currentLimit; + limit -= bufferSizeAfterLimit; + } else { + bufferSizeAfterLimit = 0; + } + } + + private int remaining() { + return (int) (limit - pos); + } + + private byte[] copyToArray(int begin, int size) throws IOException { + try { + byte[] bytes = new byte[size]; + buffer.read(begin, bytes); + return bytes; + } catch (IOException e) { + throw InvalidProtocolBufferException.truncatedMessage(); + } + } + + private void skipRawVarint() throws IOException { + if (limit - pos >= MAX_VARINT_SIZE) { + skipRawVarintFastPath(); + } else { + skipRawVarintSlowPath(); + } + } + + private void skipRawVarintFastPath() throws IOException { + for (int i = 0; i < MAX_VARINT_SIZE; i++) { + if (buffer.read(pos++) >= 0) { + return; + } + } + throw InvalidProtocolBufferException.malformedVarint(); + } + + private void skipRawVarintSlowPath() throws IOException { + for (int i = 0; i < MAX_VARINT_SIZE; i++) { + if (readRawByte() >= 0) { + return; + } + } + throw InvalidProtocolBufferException.malformedVarint(); + } + } } diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/Utf8.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/Utf8.java index e6a497de3ef..b84efd68b32 100644 --- a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/Utf8.java +++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/Utf8.java @@ -229,6 +229,16 @@ final class Utf8 { } } + private static int incompleteStateFor(ByteInput bytes, int index, int limit) { + int byte1 = bytes.read(index - 1); + switch (limit - index) { + case 0: return incompleteStateFor(byte1); + case 1: return incompleteStateFor(byte1, bytes.read(index)); + case 2: return incompleteStateFor(byte1, bytes.read(index), bytes.read(index + 1)); + default: throw new AssertionError(); + } + } + // These UTF-8 handling methods are copied from Guava's Utf8 class with a modification to throw // a protocol buffer local exception. This exception is then caught in CodedOutputStream so it can // fallback to more lenient behavior. @@ -331,6 +341,24 @@ final class Utf8 { return processor.partialIsValidUtf8(state, buffer, index, limit); } + /** + * Determines if the given {@link ByteInput} is a valid UTF-8 string. + * + * @param buffer the buffer to check. + */ + static boolean isValidUtf8(ByteInput buffer, int index, int limit) { + return processor.isValidUtf8(buffer, index, limit); + } + + /** + * Determines if the given {@link ByteInput} is a partially valid UTF-8 string. + * + * @param buffer the buffer to check. + */ + static int partialIsValidUtf8(int state, ByteInput buffer, int index, int limit) { + return processor.partialIsValidUtf8(state, buffer, index, limit); + } + /** * Encodes the given characters to the target {@link ByteBuffer} using UTF-8 encoding. * @@ -610,6 +638,169 @@ final class Utf8 { } } + public boolean isValidUtf8(ByteInput buffer, int index, int limit) { + return partialIsValidUtf8(COMPLETE, buffer, index, limit) == COMPLETE; + } + + int partialIsValidUtf8(int state, ByteInput bytes, int index, int limit) { + if (state != COMPLETE) { + // The previous decoding operation was incomplete (or malformed). + // We look for a well-formed sequence consisting of bytes from + // the previous decoding operation (stored in state) together + // with bytes from the array slice. + // + // We expect such "straddler characters" to be rare. + + if (index >= limit) { // No bytes? No progress. + return state; + } + int byte1 = (byte) state; + // byte1 is never ASCII. + if (byte1 < (byte) 0xE0) { + // two-byte form + + // Simultaneously checks for illegal trailing-byte in + // leading position and overlong 2-byte form. + if (byte1 < (byte) 0xC2 + // byte2 trailing-byte test + || bytes.read(index++) > (byte) 0xBF) { + return MALFORMED; + } + } else if (byte1 < (byte) 0xF0) { + // three-byte form + + // Get byte2 from saved state or array + int byte2 = (byte) ~(state >> 8); + if (byte2 == 0) { + byte2 = bytes.read(index++); + if (index >= limit) { + return incompleteStateFor(byte1, byte2); + } + } + if (byte2 > (byte) 0xBF + // overlong? 5 most significant bits must not all be zero + || (byte1 == (byte) 0xE0 && byte2 < (byte) 0xA0) + // illegal surrogate codepoint? + || (byte1 == (byte) 0xED && byte2 >= (byte) 0xA0) + // byte3 trailing-byte test + || bytes.read(index++) > (byte) 0xBF) { + return MALFORMED; + } + } else { + // four-byte form + + // Get byte2 and byte3 from saved state or array + int byte2 = (byte) ~(state >> 8); + int byte3 = 0; + if (byte2 == 0) { + byte2 = bytes.read(index++); + if (index >= limit) { + return incompleteStateFor(byte1, byte2); + } + } else { + byte3 = (byte) (state >> 16); + } + if (byte3 == 0) { + byte3 = bytes.read(index++); + if (index >= limit) { + return incompleteStateFor(byte1, byte2, byte3); + } + } + + // If we were called with state == MALFORMED, then byte1 is 0xFF, + // which never occurs in well-formed UTF-8, and so we will return + // MALFORMED again below. + + if (byte2 > (byte) 0xBF + // Check that 1 <= plane <= 16. Tricky optimized form of: + // if (byte1 > (byte) 0xF4 || + // byte1 == (byte) 0xF0 && byte2 < (byte) 0x90 || + // byte1 == (byte) 0xF4 && byte2 > (byte) 0x8F) + || (((byte1 << 28) + (byte2 - (byte) 0x90)) >> 30) != 0 + // byte3 trailing-byte test + || byte3 > (byte) 0xBF + // byte4 trailing-byte test + || bytes.read(index++) > (byte) 0xBF) { + return MALFORMED; + } + } + } + + return partialIsValidUtf8(bytes, index, limit); + } + + private static int partialIsValidUtf8(ByteInput bytes, int index, int limit) { + // Optimize for 100% ASCII (Hotspot loves small simple top-level loops like this). + // This simple loop stops when we encounter a byte >= 0x80 (i.e. non-ASCII). + while (index < limit && bytes.read(index) >= 0) { + index++; + } + + return (index >= limit) ? COMPLETE : partialIsValidUtf8NonAscii(bytes, index, limit); + } + + private static int partialIsValidUtf8NonAscii(ByteInput bytes, int index, int limit) { + for (;;) { + int byte1, byte2; + + // Optimize for interior runs of ASCII bytes. + do { + if (index >= limit) { + return COMPLETE; + } + } while ((byte1 = bytes.read(index++)) >= 0); + + if (byte1 < (byte) 0xE0) { + // two-byte form + + if (index >= limit) { + // Incomplete sequence + return byte1; + } + + // Simultaneously checks for illegal trailing-byte in + // leading position and overlong 2-byte form. + if (byte1 < (byte) 0xC2 + || bytes.read(index++) > (byte) 0xBF) { + return MALFORMED; + } + } else if (byte1 < (byte) 0xF0) { + // three-byte form + + if (index >= limit - 1) { // incomplete sequence + return incompleteStateFor(bytes, index, limit); + } + if ((byte2 = bytes.read(index++)) > (byte) 0xBF + // overlong? 5 most significant bits must not all be zero + || (byte1 == (byte) 0xE0 && byte2 < (byte) 0xA0) + // check for illegal surrogate codepoints + || (byte1 == (byte) 0xED && byte2 >= (byte) 0xA0) + // byte3 trailing-byte test + || bytes.read(index++) > (byte) 0xBF) { + return MALFORMED; + } + } else { + // four-byte form + + if (index >= limit - 2) { // incomplete sequence + return incompleteStateFor(bytes, index, limit); + } + if ((byte2 = bytes.read(index++)) > (byte) 0xBF + // Check that 1 <= plane <= 16. Tricky optimized form of: + // if (byte1 > (byte) 0xF4 || + // byte1 == (byte) 0xF0 && byte2 < (byte) 0x90 || + // byte1 == (byte) 0xF4 && byte2 > (byte) 0x8F) + || (((byte1 << 28) + (byte2 - (byte) 0x90)) >> 30) != 0 + // byte3 trailing-byte test + || bytes.read(index++) > (byte) 0xBF + // byte4 trailing-byte test + || bytes.read(index++) > (byte) 0xBF) { + return MALFORMED; + } + } + } + } + /** * Encodes an input character sequence ({@code in}) to UTF-8 in the target array ({@code out}). * For a string, this method is similar to diff --git a/hbase-protocol-shaded/src/main/patches/HBASE-15789_V2.patch b/hbase-protocol-shaded/src/main/patches/HBASE-15789_V2.patch new file mode 100644 index 00000000000..0c0be0f4611 --- /dev/null +++ b/hbase-protocol-shaded/src/main/patches/HBASE-15789_V2.patch @@ -0,0 +1,1262 @@ +diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteBufferWriter.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteBufferWriter.java +index c0ed636..906c216 100644 +--- a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteBufferWriter.java ++++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteBufferWriter.java +@@ -112,7 +112,7 @@ final class ByteBufferWriter { + } + } + +- private static byte[] getOrCreateBuffer(int requestedSize) { ++ static byte[] getOrCreateBuffer(int requestedSize) { + requestedSize = max(requestedSize, MIN_CACHED_BUFFER_SIZE); + + byte[] buffer = getBuffer(); +diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInput.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInput.java +new file mode 100644 +index 0000000..a745d37 +--- /dev/null ++++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInput.java +@@ -0,0 +1,81 @@ ++// Protocol Buffers - Google's data interchange format ++// Copyright 2008 Google Inc. All rights reserved. ++// https://developers.google.com/protocol-buffers/ ++// ++// Redistribution and use in source and binary forms, with or without ++// modification, are permitted provided that the following conditions are ++// met: ++// ++// * Redistributions of source code must retain the above copyright ++// notice, this list of conditions and the following disclaimer. ++// * Redistributions in binary form must reproduce the above ++// copyright notice, this list of conditions and the following disclaimer ++// in the documentation and/or other materials provided with the ++// distribution. ++// * Neither the name of Google Inc. nor the names of its ++// contributors may be used to endorse or promote products derived from ++// this software without specific prior written permission. ++// ++// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ++// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT ++// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR ++// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT ++// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, ++// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT ++// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, ++// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY ++// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT ++// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE ++// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. ++package org.apache.hadoop.hbase.shaded.com.google.protobuf; ++ ++import java.io.IOException; ++import java.nio.ByteBuffer; ++ ++/** ++ * An input for raw bytes. This is similar to an InputStream but it is offset addressable. All the ++ * read APIs are relative. ++ */ ++@ExperimentalApi ++public abstract class ByteInput { ++ ++ /** ++ * Reads a single byte from the given offset. ++ * @param offset The offset from where byte to be read ++ * @return The byte of data at given offset ++ */ ++ public abstract byte read(int offset); ++ ++ /** ++ * Reads bytes of data from the given offset into an array of bytes. ++ * @param offset The src offset within this ByteInput from where data to be read. ++ * @param out Destination byte array to read data into. ++ * @return The number of bytes read from ByteInput ++ */ ++ public int read(int offset, byte b[]) throws IOException { ++ return read(offset, b, 0, b.length); ++ } ++ ++ /** ++ * Reads up to len bytes of data from the given offset into an array of bytes. ++ * @param offset The src offset within this ByteInput from where data to be read. ++ * @param out Destination byte array to read data into. ++ * @param outOffset Offset within the the out byte[] where data to be read into. ++ * @param len The number of bytes to read. ++ * @return The number of bytes read from ByteInput ++ */ ++ public abstract int read(int offset, byte[] out, int outOffset, int len); ++ ++ /** ++ * Reads bytes of data from the given offset into given {@link ByteBuffer}. ++ * @param offset he src offset within this ByteInput from where data to be read. ++ * @param out Destination {@link ByteBuffer} to read data into. ++ * @return The number of bytes read from ByteInput ++ */ ++ public abstract int read(int offset, ByteBuffer out); ++ ++ /** ++ * @return Total number of bytes in this ByteInput. ++ */ ++ public abstract int size(); ++} +diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInputByteString.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInputByteString.java +new file mode 100644 +index 0000000..1949602 +--- /dev/null ++++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteInputByteString.java +@@ -0,0 +1,249 @@ ++// Protocol Buffers - Google's data interchange format ++// Copyright 2008 Google Inc. All rights reserved. ++// https://developers.google.com/protocol-buffers/ ++// ++// Redistribution and use in source and binary forms, with or without ++// modification, are permitted provided that the following conditions are ++// met: ++// ++// * Redistributions of source code must retain the above copyright ++// notice, this list of conditions and the following disclaimer. ++// * Redistributions in binary form must reproduce the above ++// copyright notice, this list of conditions and the following disclaimer ++// in the documentation and/or other materials provided with the ++// distribution. ++// * Neither the name of Google Inc. nor the names of its ++// contributors may be used to endorse or promote products derived from ++// this software without specific prior written permission. ++// ++// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ++// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT ++// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR ++// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT ++// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, ++// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT ++// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, ++// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY ++// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT ++// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE ++// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. ++ ++package org.apache.hadoop.hbase.shaded.com.google.protobuf; ++ ++import java.io.IOException; ++import java.io.InputStream; ++import java.io.InvalidObjectException; ++import java.io.ObjectInputStream; ++import java.io.OutputStream; ++import java.nio.ByteBuffer; ++import java.nio.charset.Charset; ++import java.util.Arrays; ++import java.util.Collections; ++import java.util.List; ++ ++/** ++ * A {@link ByteString} that wraps around a {@link ByteInput}. ++ */ ++final class ByteInputByteString extends ByteString.LeafByteString { ++ private final ByteInput buffer; ++ private final int offset, length; ++ ++ ByteInputByteString(ByteInput buffer, int offset, int length) { ++ if (buffer == null) { ++ throw new NullPointerException("buffer"); ++ } ++ this.buffer = buffer; ++ this.offset = offset; ++ this.length = length; ++ } ++ ++ // ================================================================= ++ // Serializable ++ ++ /** ++ * Magic method that lets us override serialization behavior. ++ */ ++ private Object writeReplace() { ++ return ByteString.wrap(toByteArray()); ++ } ++ ++ /** ++ * Magic method that lets us override deserialization behavior. ++ */ ++ private void readObject(@SuppressWarnings("unused") ObjectInputStream in) throws IOException { ++ throw new InvalidObjectException("ByteInputByteString instances are not to be serialized directly");// TODO check here ++ } ++ ++ // ================================================================= ++ ++ @Override ++ public byte byteAt(int index) { ++ return buffer.read(getAbsoluteOffset(index)); ++ } ++ ++ private int getAbsoluteOffset(int relativeOffset) { ++ return this.offset + relativeOffset; ++ } ++ ++ @Override ++ public int size() { ++ return length; ++ } ++ ++ @Override ++ public ByteString substring(int beginIndex, int endIndex) { ++ if (beginIndex < 0 || beginIndex >= size() || endIndex < beginIndex || endIndex >= size()) { ++ throw new IllegalArgumentException( ++ String.format("Invalid indices [%d, %d]", beginIndex, endIndex)); ++ } ++ return new ByteInputByteString(this.buffer, getAbsoluteOffset(beginIndex), endIndex - beginIndex); ++ } ++ ++ @Override ++ protected void copyToInternal( ++ byte[] target, int sourceOffset, int targetOffset, int numberToCopy) { ++ this.buffer.read(getAbsoluteOffset(sourceOffset), target, targetOffset, numberToCopy); ++ } ++ ++ @Override ++ public void copyTo(ByteBuffer target) { ++ this.buffer.read(this.offset, target); ++ } ++ ++ @Override ++ public void writeTo(OutputStream out) throws IOException { ++ out.write(toByteArray());// TODO ++ } ++ ++ @Override ++ boolean equalsRange(ByteString other, int offset, int length) { ++ return substring(0, length).equals(other.substring(offset, offset + length)); ++ } ++ ++ @Override ++ void writeToInternal(OutputStream out, int sourceOffset, int numberToWrite) throws IOException { ++ byte[] buf = ByteBufferWriter.getOrCreateBuffer(numberToWrite); ++ this.buffer.read(getAbsoluteOffset(sourceOffset), buf, 0, numberToWrite); ++ out.write(buf, 0, numberToWrite); ++ } ++ ++ @Override ++ void writeTo(ByteOutput output) throws IOException { ++ output.writeLazy(toByteArray(), 0, length); ++ } ++ ++ @Override ++ public ByteBuffer asReadOnlyByteBuffer() { ++ return ByteBuffer.wrap(toByteArray()).asReadOnlyBuffer(); ++ } ++ ++ @Override ++ public List asReadOnlyByteBufferList() { ++ return Collections.singletonList(asReadOnlyByteBuffer()); ++ } ++ ++ @Override ++ protected String toStringInternal(Charset charset) { ++ byte[] bytes = toByteArray(); ++ return new String(bytes, 0, bytes.length, charset); ++ } ++ ++ @Override ++ public boolean isValidUtf8() { ++ return Utf8.isValidUtf8(buffer, offset, offset + length); ++ } ++ ++ @Override ++ protected int partialIsValidUtf8(int state, int offset, int length) { ++ int off = getAbsoluteOffset(offset); ++ return Utf8.partialIsValidUtf8(state, buffer, off, off + length); ++ } ++ ++ @Override ++ public boolean equals(Object other) { ++ if (other == this) { ++ return true; ++ } ++ if (!(other instanceof ByteString)) { ++ return false; ++ } ++ ByteString otherString = ((ByteString) other); ++ if (size() != otherString.size()) { ++ return false; ++ } ++ if (size() == 0) { ++ return true; ++ } ++ if (other instanceof RopeByteString) { ++ return other.equals(this); ++ } ++ return Arrays.equals(this.toByteArray(), otherString.toByteArray()); ++ } ++ ++ @Override ++ protected int partialHash(int h, int offset, int length) { ++ offset = getAbsoluteOffset(offset); ++ int end = offset + length; ++ for (int i = offset; i < end; i++) { ++ h = h * 31 + buffer.read(i); ++ } ++ return h; ++ } ++ ++ @Override ++ public InputStream newInput() { ++ return new InputStream() { ++ private final ByteInput buf = buffer; ++ private int pos = offset; ++ private int limit = pos + length; ++ private int mark = pos; ++ ++ @Override ++ public void mark(int readlimit) { ++ this.mark = readlimit; ++ } ++ ++ @Override ++ public boolean markSupported() { ++ return true; ++ } ++ ++ @Override ++ public void reset() throws IOException { ++ this.pos = this.mark; ++ } ++ ++ @Override ++ public int available() throws IOException { ++ return this.limit - this.pos; ++ } ++ ++ @Override ++ public int read() throws IOException { ++ if (available() <= 0) { ++ return -1; ++ } ++ return this.buf.read(pos++) & 0xFF; ++ } ++ ++ @Override ++ public int read(byte[] bytes, int off, int len) throws IOException { ++ int remain = available(); ++ if (remain <= 0) { ++ return -1; ++ } ++ len = Math.min(len, remain); ++ buf.read(pos, bytes, off, len); ++ pos += len; ++ return len; ++ } ++ }; ++ } ++ ++ @Override ++ public CodedInputStream newCodedInput() { ++ // We trust CodedInputStream not to modify the bytes, or to give anyone ++ // else access to them. ++ return CodedInputStream.newInstance(buffer, offset, length, true); ++ } ++} +diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteString.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteString.java +index 8cae888..ebfa7fa 100644 +--- a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteString.java ++++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/ByteString.java +@@ -323,6 +323,13 @@ public abstract class ByteString implements Iterable, Serializable { + } + + /** ++ * Wraps the given bytes into a {@code ByteString}. Intended for internal only usage. ++ */ ++ static ByteString wrap(ByteInput buffer, int offset, int length) { ++ return new ByteInputByteString(buffer, offset, length); ++ } ++ ++ /** + * Wraps the given bytes into a {@code ByteString}. Intended for internal only + * usage to force a classload of ByteString before LiteralByteString. + */ +diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/CodedInputStream.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/CodedInputStream.java +index 12d70ce..e4c8e05 100644 +--- a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/CodedInputStream.java ++++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/CodedInputStream.java +@@ -149,6 +149,15 @@ public abstract class CodedInputStream { + return newInstance(buffer, 0, buffer.length, true); + } + ++ /** Create a new CodedInputStream wrapping the given {@link ByteInput}. */ ++ public static CodedInputStream newInstance(ByteInput buf, boolean bufferIsImmutable) { ++ return new ByteInputDecoder(buf, bufferIsImmutable); ++ } ++ ++ public static CodedInputStream newInstance(ByteInput buf, int off, int len, boolean bufferIsImmutable) { ++ return new ByteInputDecoder(buf, off, len, bufferIsImmutable); ++ } ++ + /** Disable construction/inheritance outside of this class. */ + private CodedInputStream() {} + +@@ -2892,4 +2901,652 @@ public abstract class CodedInputStream { + pos = size - tempPos; + } + } ++ ++ private static final class ByteInputDecoder extends CodedInputStream { ++ ++ private final ByteInput buffer; ++ private final boolean immutable; ++ private int limit; ++ private int bufferSizeAfterLimit; ++ private int pos; ++ private int startPos; ++ private int lastTag; ++ private boolean enableAliasing; ++ ++ /** The absolute position of the end of the current message. */ ++ private int currentLimit = Integer.MAX_VALUE; ++ ++ private ByteInputDecoder(ByteInput buffer, boolean immutable) { ++ this(buffer, 0, buffer.size(), immutable); ++ } ++ ++ private ByteInputDecoder(ByteInput buffer, int off, int len, boolean immutable) { ++ this.buffer = buffer; ++ pos = off; ++ limit = off + len; ++ startPos = pos; ++ this.immutable = immutable; ++ } ++ ++ @Override ++ public int readTag() throws IOException { ++ if (isAtEnd()) { ++ lastTag = 0; ++ return 0; ++ } ++ ++ lastTag = readRawVarint32(); ++ if (WireFormat.getTagFieldNumber(lastTag) == 0) { ++ // If we actually read zero (or any tag number corresponding to field ++ // number zero), that's not a valid tag. ++ throw InvalidProtocolBufferException.invalidTag(); ++ } ++ return lastTag; ++ } ++ ++ @Override ++ public void checkLastTagWas(int value) throws InvalidProtocolBufferException { ++ if (lastTag != value) { ++ throw InvalidProtocolBufferException.invalidEndTag(); ++ } ++ } ++ ++ @Override ++ public int getLastTag() { ++ return lastTag; ++ } ++ ++ @Override ++ public boolean skipField(int tag) throws IOException { ++ switch (WireFormat.getTagWireType(tag)) { ++ case WireFormat.WIRETYPE_VARINT: ++ skipRawVarint(); ++ return true; ++ case WireFormat.WIRETYPE_FIXED64: ++ skipRawBytes(FIXED_64_SIZE); ++ return true; ++ case WireFormat.WIRETYPE_LENGTH_DELIMITED: ++ skipRawBytes(readRawVarint32()); ++ return true; ++ case WireFormat.WIRETYPE_START_GROUP: ++ skipMessage(); ++ checkLastTagWas( ++ WireFormat.makeTag(WireFormat.getTagFieldNumber(tag), WireFormat.WIRETYPE_END_GROUP)); ++ return true; ++ case WireFormat.WIRETYPE_END_GROUP: ++ return false; ++ case WireFormat.WIRETYPE_FIXED32: ++ skipRawBytes(FIXED_32_SIZE); ++ return true; ++ default: ++ throw InvalidProtocolBufferException.invalidWireType(); ++ } ++ } ++ ++ @Override ++ public boolean skipField(int tag, CodedOutputStream output) throws IOException { ++ switch (WireFormat.getTagWireType(tag)) { ++ case WireFormat.WIRETYPE_VARINT: ++ { ++ long value = readInt64(); ++ output.writeRawVarint32(tag); ++ output.writeUInt64NoTag(value); ++ return true; ++ } ++ case WireFormat.WIRETYPE_FIXED64: ++ { ++ long value = readRawLittleEndian64(); ++ output.writeRawVarint32(tag); ++ output.writeFixed64NoTag(value); ++ return true; ++ } ++ case WireFormat.WIRETYPE_LENGTH_DELIMITED: ++ { ++ ByteString value = readBytes(); ++ output.writeRawVarint32(tag); ++ output.writeBytesNoTag(value); ++ return true; ++ } ++ case WireFormat.WIRETYPE_START_GROUP: ++ { ++ output.writeRawVarint32(tag); ++ skipMessage(output); ++ int endtag = ++ WireFormat.makeTag( ++ WireFormat.getTagFieldNumber(tag), WireFormat.WIRETYPE_END_GROUP); ++ checkLastTagWas(endtag); ++ output.writeRawVarint32(endtag); ++ return true; ++ } ++ case WireFormat.WIRETYPE_END_GROUP: ++ { ++ return false; ++ } ++ case WireFormat.WIRETYPE_FIXED32: ++ { ++ int value = readRawLittleEndian32(); ++ output.writeRawVarint32(tag); ++ output.writeFixed32NoTag(value); ++ return true; ++ } ++ default: ++ throw InvalidProtocolBufferException.invalidWireType(); ++ } ++ } ++ ++ @Override ++ public void skipMessage() throws IOException { ++ while (true) { ++ final int tag = readTag(); ++ if (tag == 0 || !skipField(tag)) { ++ return; ++ } ++ } ++ } ++ ++ @Override ++ public void skipMessage(CodedOutputStream output) throws IOException { ++ while (true) { ++ final int tag = readTag(); ++ if (tag == 0 || !skipField(tag, output)) { ++ return; ++ } ++ } ++ } ++ ++ public double readDouble() throws IOException { ++ return Double.longBitsToDouble(readRawLittleEndian64()); ++ } ++ ++ @Override ++ public float readFloat() throws IOException { ++ return Float.intBitsToFloat(readRawLittleEndian32()); ++ } ++ ++ @Override ++ public long readUInt64() throws IOException { ++ return readRawVarint64(); ++ } ++ ++ @Override ++ public long readInt64() throws IOException { ++ return readRawVarint64(); ++ } ++ ++ @Override ++ public int readInt32() throws IOException { ++ return readRawVarint32(); ++ } ++ ++ @Override ++ public long readFixed64() throws IOException { ++ return readRawLittleEndian64(); ++ } ++ ++ @Override ++ public int readFixed32() throws IOException { ++ return readRawLittleEndian32(); ++ } ++ ++ @Override ++ public boolean readBool() throws IOException { ++ return readRawVarint64() != 0; ++ } ++ ++ @Override ++ public String readString() throws IOException { ++ final int size = readRawVarint32(); ++ if (size > 0 && size <= remaining()) { ++ byte[] bytes = copyToArray(pos, size); ++ pos += size; ++ return new String(bytes, UTF_8); ++ } ++ ++ if (size == 0) { ++ return ""; ++ } ++ if (size < 0) { ++ throw InvalidProtocolBufferException.negativeSize(); ++ } ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ ++ @Override ++ public String readStringRequireUtf8() throws IOException { ++ final int size = readRawVarint32(); ++ if (size > 0 && size <= remaining()) { ++ if (!Utf8.isValidUtf8(buffer, pos, pos + size)) { ++ throw InvalidProtocolBufferException.invalidUtf8(); ++ } ++ byte[] bytes = copyToArray(pos, size); ++ pos += size; ++ return new String(bytes, UTF_8); ++ } ++ ++ if (size == 0) { ++ return ""; ++ } ++ if (size <= 0) { ++ throw InvalidProtocolBufferException.negativeSize(); ++ } ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ ++ @Override ++ public void readGroup(int fieldNumber, MessageLite.Builder builder, ++ ExtensionRegistryLite extensionRegistry) throws IOException { ++ if (recursionDepth >= recursionLimit) { ++ throw InvalidProtocolBufferException.recursionLimitExceeded(); ++ } ++ ++recursionDepth; ++ builder.mergeFrom(this, extensionRegistry); ++ checkLastTagWas(WireFormat.makeTag(fieldNumber, WireFormat.WIRETYPE_END_GROUP)); ++ --recursionDepth; ++ } ++ ++ @Override ++ public T readGroup(int fieldNumber, Parser parser, ++ ExtensionRegistryLite extensionRegistry) throws IOException { ++ if (recursionDepth >= recursionLimit) { ++ throw InvalidProtocolBufferException.recursionLimitExceeded(); ++ } ++ ++recursionDepth; ++ T result = parser.parsePartialFrom(this, extensionRegistry); ++ checkLastTagWas(WireFormat.makeTag(fieldNumber, WireFormat.WIRETYPE_END_GROUP)); ++ --recursionDepth; ++ return result; ++ } ++ ++ @Deprecated ++ @Override ++ public void readUnknownGroup(int fieldNumber, MessageLite.Builder builder) throws IOException { ++ readGroup(fieldNumber, builder, ExtensionRegistryLite.getEmptyRegistry()); ++ } ++ ++ @Override ++ public void readMessage(MessageLite.Builder builder, ExtensionRegistryLite extensionRegistry) ++ throws IOException { ++ final int length = readRawVarint32(); ++ if (recursionDepth >= recursionLimit) { ++ throw InvalidProtocolBufferException.recursionLimitExceeded(); ++ } ++ final int oldLimit = pushLimit(length); ++ ++recursionDepth; ++ builder.mergeFrom(this, extensionRegistry); ++ checkLastTagWas(0); ++ --recursionDepth; ++ popLimit(oldLimit); ++ } ++ ++ @Override ++ public T readMessage(Parser parser, ++ ExtensionRegistryLite extensionRegistry) throws IOException { ++ int length = readRawVarint32(); ++ if (recursionDepth >= recursionLimit) { ++ throw InvalidProtocolBufferException.recursionLimitExceeded(); ++ } ++ final int oldLimit = pushLimit(length); ++ ++recursionDepth; ++ T result = parser.parsePartialFrom(this, extensionRegistry); ++ checkLastTagWas(0); ++ --recursionDepth; ++ popLimit(oldLimit); ++ return result; ++ } ++ ++ @Override ++ public ByteString readBytes() throws IOException { ++ final int size = readRawVarint32(); ++ if (size > 0 && size <= (limit - pos)) { ++ // Fast path: We already have the bytes in a contiguous buffer, so ++ // just copy directly from it. ++ ++ final ByteString result = ++ immutable && enableAliasing ++ ? ByteString.wrap(buffer, pos, size) ++ : ByteString.wrap(copyToArray(pos, size)); ++ pos += size; ++ return result; ++ } ++ if (size == 0) { ++ return ByteString.EMPTY; ++ } ++ // Slow path: Build a byte array first then copy it. ++ return ByteString.wrap(readRawBytes(size)); ++ } ++ ++ @Override ++ public byte[] readByteArray() throws IOException { ++ return readRawBytes(readRawVarint32()); ++ } ++ ++ @Override ++ public ByteBuffer readByteBuffer() throws IOException { ++ return ByteBuffer.wrap(readByteArray()); ++ } ++ ++ @Override ++ public int readUInt32() throws IOException { ++ return readRawVarint32(); ++ } ++ ++ @Override ++ public int readEnum() throws IOException { ++ return readRawVarint32(); ++ } ++ ++ @Override ++ public int readSFixed32() throws IOException { ++ return readRawLittleEndian32(); ++ } ++ ++ @Override ++ public long readSFixed64() throws IOException { ++ return readRawLittleEndian64(); ++ } ++ ++ @Override ++ public int readSInt32() throws IOException { ++ return decodeZigZag32(readRawVarint32()); ++ } ++ ++ @Override ++ public long readSInt64() throws IOException { ++ return decodeZigZag64(readRawVarint64()); ++ } ++ ++ @Override ++ public int readRawVarint32() throws IOException { ++ // See implementation notes for readRawVarint64 ++ fastpath: ++ { ++ int tempPos = pos; ++ ++ if (limit == tempPos) { ++ break fastpath; ++ } ++ ++ int x; ++ if ((x = buffer.read(tempPos++)) >= 0) { ++ pos = tempPos; ++ return x; ++ } else if (limit - tempPos < 9) { ++ break fastpath; ++ } else if ((x ^= (buffer.read(tempPos++) << 7)) < 0) { ++ x ^= (~0 << 7); ++ } else if ((x ^= (buffer.read(tempPos++) << 14)) >= 0) { ++ x ^= (~0 << 7) ^ (~0 << 14); ++ } else if ((x ^= (buffer.read(tempPos++) << 21)) < 0) { ++ x ^= (~0 << 7) ^ (~0 << 14) ^ (~0 << 21); ++ } else { ++ int y = buffer.read(tempPos++); ++ x ^= y << 28; ++ x ^= (~0 << 7) ^ (~0 << 14) ^ (~0 << 21) ^ (~0 << 28); ++ if (y < 0 ++ && buffer.read(tempPos++) < 0 ++ && buffer.read(tempPos++) < 0 ++ && buffer.read(tempPos++) < 0 ++ && buffer.read(tempPos++) < 0 ++ && buffer.read(tempPos++) < 0) { ++ break fastpath; // Will throw malformedVarint() ++ } ++ } ++ pos = tempPos; ++ return x; ++ } ++ return (int) readRawVarint64SlowPath(); ++ } ++ ++ @Override ++ public long readRawVarint64() throws IOException { ++ fastpath: ++ { ++ int tempPos = pos; ++ ++ if (limit == tempPos) { ++ break fastpath; ++ } ++ ++ long x; ++ int y; ++ if ((y = buffer.read(tempPos++)) >= 0) { ++ pos = tempPos; ++ return y; ++ } else if (limit - tempPos < 9) { ++ break fastpath; ++ } else if ((y ^= (buffer.read(tempPos++) << 7)) < 0) { ++ x = y ^ (~0 << 7); ++ } else if ((y ^= (buffer.read(tempPos++) << 14)) >= 0) { ++ x = y ^ ((~0 << 7) ^ (~0 << 14)); ++ } else if ((y ^= (buffer.read(tempPos++) << 21)) < 0) { ++ x = y ^ ((~0 << 7) ^ (~0 << 14) ^ (~0 << 21)); ++ } else if ((x = y ^ ((long) buffer.read(tempPos++) << 28)) >= 0L) { ++ x ^= (~0L << 7) ^ (~0L << 14) ^ (~0L << 21) ^ (~0L << 28); ++ } else if ((x ^= ((long) buffer.read(tempPos++) << 35)) < 0L) { ++ x ^= (~0L << 7) ^ (~0L << 14) ^ (~0L << 21) ^ (~0L << 28) ^ (~0L << 35); ++ } else if ((x ^= ((long) buffer.read(tempPos++) << 42)) >= 0L) { ++ x ^= (~0L << 7) ^ (~0L << 14) ^ (~0L << 21) ^ (~0L << 28) ^ (~0L << 35) ^ (~0L << 42); ++ } else if ((x ^= ((long) buffer.read(tempPos++) << 49)) < 0L) { ++ x ^= ++ (~0L << 7) ++ ^ (~0L << 14) ++ ^ (~0L << 21) ++ ^ (~0L << 28) ++ ^ (~0L << 35) ++ ^ (~0L << 42) ++ ^ (~0L << 49); ++ } else { ++ x ^= ((long) buffer.read(tempPos++) << 56); ++ x ^= ++ (~0L << 7) ++ ^ (~0L << 14) ++ ^ (~0L << 21) ++ ^ (~0L << 28) ++ ^ (~0L << 35) ++ ^ (~0L << 42) ++ ^ (~0L << 49) ++ ^ (~0L << 56); ++ if (x < 0L) { ++ if (buffer.read(tempPos++) < 0L) { ++ break fastpath; // Will throw malformedVarint() ++ } ++ } ++ } ++ pos = tempPos; ++ return x; ++ } ++ return readRawVarint64SlowPath(); ++ } ++ ++ @Override ++ long readRawVarint64SlowPath() throws IOException { ++ long result = 0; ++ for (int shift = 0; shift < 64; shift += 7) { ++ final byte b = readRawByte(); ++ result |= (long) (b & 0x7F) << shift; ++ if ((b & 0x80) == 0) { ++ return result; ++ } ++ } ++ throw InvalidProtocolBufferException.malformedVarint(); ++ } ++ ++ @Override ++ public int readRawLittleEndian32() throws IOException { ++ int tempPos = pos; ++ ++ if (limit - tempPos < FIXED_32_SIZE) { ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ ++ pos = tempPos + FIXED_32_SIZE; ++ return (((buffer.read(tempPos) & 0xff)) ++ | ((buffer.read(tempPos + 1) & 0xff) << 8) ++ | ((buffer.read(tempPos + 2) & 0xff) << 16) ++ | ((buffer.read(tempPos + 3) & 0xff) << 24)); ++ } ++ ++ @Override ++ public long readRawLittleEndian64() throws IOException { ++ int tempPos = pos; ++ ++ if (limit - tempPos < FIXED_64_SIZE) { ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ ++ pos = tempPos + FIXED_64_SIZE; ++ return (((buffer.read(tempPos) & 0xffL)) ++ | ((buffer.read(tempPos + 1) & 0xffL) << 8) ++ | ((buffer.read(tempPos + 2) & 0xffL) << 16) ++ | ((buffer.read(tempPos + 3) & 0xffL) << 24) ++ | ((buffer.read(tempPos + 4) & 0xffL) << 32) ++ | ((buffer.read(tempPos + 5) & 0xffL) << 40) ++ | ((buffer.read(tempPos + 6) & 0xffL) << 48) ++ | ((buffer.read(tempPos + 7) & 0xffL) << 56)); ++ } ++ ++ @Override ++ public void enableAliasing(boolean enabled) { ++ this.enableAliasing = enabled; ++ } ++ ++ @Override ++ public void resetSizeCounter() { ++ startPos = pos; ++ } ++ ++ @Override ++ public int pushLimit(int byteLimit) throws InvalidProtocolBufferException { ++ if (byteLimit < 0) { ++ throw InvalidProtocolBufferException.negativeSize(); ++ } ++ byteLimit += getTotalBytesRead(); ++ final int oldLimit = currentLimit; ++ if (byteLimit > oldLimit) { ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ currentLimit = byteLimit; ++ ++ recomputeBufferSizeAfterLimit(); ++ ++ return oldLimit; ++ } ++ ++ @Override ++ public void popLimit(int oldLimit) { ++ currentLimit = oldLimit; ++ recomputeBufferSizeAfterLimit(); ++ } ++ ++ @Override ++ public int getBytesUntilLimit() { ++ if (currentLimit == Integer.MAX_VALUE) { ++ return -1; ++ } ++ ++ return currentLimit - getTotalBytesRead(); ++ } ++ ++ @Override ++ public boolean isAtEnd() throws IOException { ++ return pos == limit; ++ } ++ ++ @Override ++ public int getTotalBytesRead() { ++ return pos - startPos; ++ } ++ ++ @Override ++ public byte readRawByte() throws IOException { ++ if (pos == limit) { ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ return buffer.read(pos++); ++ } ++ ++ @Override ++ public byte[] readRawBytes(int length) throws IOException { ++ if (length > 0 && length <= (limit - pos)) { ++ byte[] bytes = copyToArray(pos, length); ++ pos += length; ++ return bytes; ++ } ++ ++ if (length <= 0) { ++ if (length == 0) { ++ return Internal.EMPTY_BYTE_ARRAY; ++ } else { ++ throw InvalidProtocolBufferException.negativeSize(); ++ } ++ } ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ ++ @Override ++ public void skipRawBytes(int length) throws IOException { ++ if (length >= 0 && length <= (limit - pos)) { ++ // We have all the bytes we need already. ++ pos += length; ++ return; ++ } ++ ++ if (length < 0) { ++ throw InvalidProtocolBufferException.negativeSize(); ++ } ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ ++ private void recomputeBufferSizeAfterLimit() { ++ limit += bufferSizeAfterLimit; ++ final int bufferEnd = limit - startPos; ++ if (bufferEnd > currentLimit) { ++ // Limit is in current buffer. ++ bufferSizeAfterLimit = bufferEnd - currentLimit; ++ limit -= bufferSizeAfterLimit; ++ } else { ++ bufferSizeAfterLimit = 0; ++ } ++ } ++ ++ private int remaining() { ++ return (int) (limit - pos); ++ } ++ ++ private byte[] copyToArray(int begin, int size) throws IOException { ++ try { ++ byte[] bytes = new byte[size]; ++ buffer.read(begin, bytes); ++ return bytes; ++ } catch (IOException e) { ++ throw InvalidProtocolBufferException.truncatedMessage(); ++ } ++ } ++ ++ private void skipRawVarint() throws IOException { ++ if (limit - pos >= MAX_VARINT_SIZE) { ++ skipRawVarintFastPath(); ++ } else { ++ skipRawVarintSlowPath(); ++ } ++ } ++ ++ private void skipRawVarintFastPath() throws IOException { ++ for (int i = 0; i < MAX_VARINT_SIZE; i++) { ++ if (buffer.read(pos++) >= 0) { ++ return; ++ } ++ } ++ throw InvalidProtocolBufferException.malformedVarint(); ++ } ++ ++ private void skipRawVarintSlowPath() throws IOException { ++ for (int i = 0; i < MAX_VARINT_SIZE; i++) { ++ if (readRawByte() >= 0) { ++ return; ++ } ++ } ++ throw InvalidProtocolBufferException.malformedVarint(); ++ } ++ } + } +diff --git a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/Utf8.java b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/Utf8.java +index e6a497d..b84efd6 100644 +--- a/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/Utf8.java ++++ b/hbase-protocol-shaded/src/main/java/org/apache/hadoop/hbase/shaded/com/google/protobuf/Utf8.java +@@ -229,6 +229,16 @@ final class Utf8 { + } + } + ++ private static int incompleteStateFor(ByteInput bytes, int index, int limit) { ++ int byte1 = bytes.read(index - 1); ++ switch (limit - index) { ++ case 0: return incompleteStateFor(byte1); ++ case 1: return incompleteStateFor(byte1, bytes.read(index)); ++ case 2: return incompleteStateFor(byte1, bytes.read(index), bytes.read(index + 1)); ++ default: throw new AssertionError(); ++ } ++ } ++ + // These UTF-8 handling methods are copied from Guava's Utf8 class with a modification to throw + // a protocol buffer local exception. This exception is then caught in CodedOutputStream so it can + // fallback to more lenient behavior. +@@ -332,6 +342,24 @@ final class Utf8 { + } + + /** ++ * Determines if the given {@link ByteInput} is a valid UTF-8 string. ++ * ++ * @param buffer the buffer to check. ++ */ ++ static boolean isValidUtf8(ByteInput buffer, int index, int limit) { ++ return processor.isValidUtf8(buffer, index, limit); ++ } ++ ++ /** ++ * Determines if the given {@link ByteInput} is a partially valid UTF-8 string. ++ * ++ * @param buffer the buffer to check. ++ */ ++ static int partialIsValidUtf8(int state, ByteInput buffer, int index, int limit) { ++ return processor.partialIsValidUtf8(state, buffer, index, limit); ++ } ++ ++ /** + * Encodes the given characters to the target {@link ByteBuffer} using UTF-8 encoding. + * + *

Selects an optimal algorithm based on the type of {@link ByteBuffer} (i.e. heap or direct) +@@ -610,6 +638,169 @@ final class Utf8 { + } + } + ++ public boolean isValidUtf8(ByteInput buffer, int index, int limit) { ++ return partialIsValidUtf8(COMPLETE, buffer, index, limit) == COMPLETE; ++ } ++ ++ int partialIsValidUtf8(int state, ByteInput bytes, int index, int limit) { ++ if (state != COMPLETE) { ++ // The previous decoding operation was incomplete (or malformed). ++ // We look for a well-formed sequence consisting of bytes from ++ // the previous decoding operation (stored in state) together ++ // with bytes from the array slice. ++ // ++ // We expect such "straddler characters" to be rare. ++ ++ if (index >= limit) { // No bytes? No progress. ++ return state; ++ } ++ int byte1 = (byte) state; ++ // byte1 is never ASCII. ++ if (byte1 < (byte) 0xE0) { ++ // two-byte form ++ ++ // Simultaneously checks for illegal trailing-byte in ++ // leading position and overlong 2-byte form. ++ if (byte1 < (byte) 0xC2 ++ // byte2 trailing-byte test ++ || bytes.read(index++) > (byte) 0xBF) { ++ return MALFORMED; ++ } ++ } else if (byte1 < (byte) 0xF0) { ++ // three-byte form ++ ++ // Get byte2 from saved state or array ++ int byte2 = (byte) ~(state >> 8); ++ if (byte2 == 0) { ++ byte2 = bytes.read(index++); ++ if (index >= limit) { ++ return incompleteStateFor(byte1, byte2); ++ } ++ } ++ if (byte2 > (byte) 0xBF ++ // overlong? 5 most significant bits must not all be zero ++ || (byte1 == (byte) 0xE0 && byte2 < (byte) 0xA0) ++ // illegal surrogate codepoint? ++ || (byte1 == (byte) 0xED && byte2 >= (byte) 0xA0) ++ // byte3 trailing-byte test ++ || bytes.read(index++) > (byte) 0xBF) { ++ return MALFORMED; ++ } ++ } else { ++ // four-byte form ++ ++ // Get byte2 and byte3 from saved state or array ++ int byte2 = (byte) ~(state >> 8); ++ int byte3 = 0; ++ if (byte2 == 0) { ++ byte2 = bytes.read(index++); ++ if (index >= limit) { ++ return incompleteStateFor(byte1, byte2); ++ } ++ } else { ++ byte3 = (byte) (state >> 16); ++ } ++ if (byte3 == 0) { ++ byte3 = bytes.read(index++); ++ if (index >= limit) { ++ return incompleteStateFor(byte1, byte2, byte3); ++ } ++ } ++ ++ // If we were called with state == MALFORMED, then byte1 is 0xFF, ++ // which never occurs in well-formed UTF-8, and so we will return ++ // MALFORMED again below. ++ ++ if (byte2 > (byte) 0xBF ++ // Check that 1 <= plane <= 16. Tricky optimized form of: ++ // if (byte1 > (byte) 0xF4 || ++ // byte1 == (byte) 0xF0 && byte2 < (byte) 0x90 || ++ // byte1 == (byte) 0xF4 && byte2 > (byte) 0x8F) ++ || (((byte1 << 28) + (byte2 - (byte) 0x90)) >> 30) != 0 ++ // byte3 trailing-byte test ++ || byte3 > (byte) 0xBF ++ // byte4 trailing-byte test ++ || bytes.read(index++) > (byte) 0xBF) { ++ return MALFORMED; ++ } ++ } ++ } ++ ++ return partialIsValidUtf8(bytes, index, limit); ++ } ++ ++ private static int partialIsValidUtf8(ByteInput bytes, int index, int limit) { ++ // Optimize for 100% ASCII (Hotspot loves small simple top-level loops like this). ++ // This simple loop stops when we encounter a byte >= 0x80 (i.e. non-ASCII). ++ while (index < limit && bytes.read(index) >= 0) { ++ index++; ++ } ++ ++ return (index >= limit) ? COMPLETE : partialIsValidUtf8NonAscii(bytes, index, limit); ++ } ++ ++ private static int partialIsValidUtf8NonAscii(ByteInput bytes, int index, int limit) { ++ for (;;) { ++ int byte1, byte2; ++ ++ // Optimize for interior runs of ASCII bytes. ++ do { ++ if (index >= limit) { ++ return COMPLETE; ++ } ++ } while ((byte1 = bytes.read(index++)) >= 0); ++ ++ if (byte1 < (byte) 0xE0) { ++ // two-byte form ++ ++ if (index >= limit) { ++ // Incomplete sequence ++ return byte1; ++ } ++ ++ // Simultaneously checks for illegal trailing-byte in ++ // leading position and overlong 2-byte form. ++ if (byte1 < (byte) 0xC2 ++ || bytes.read(index++) > (byte) 0xBF) { ++ return MALFORMED; ++ } ++ } else if (byte1 < (byte) 0xF0) { ++ // three-byte form ++ ++ if (index >= limit - 1) { // incomplete sequence ++ return incompleteStateFor(bytes, index, limit); ++ } ++ if ((byte2 = bytes.read(index++)) > (byte) 0xBF ++ // overlong? 5 most significant bits must not all be zero ++ || (byte1 == (byte) 0xE0 && byte2 < (byte) 0xA0) ++ // check for illegal surrogate codepoints ++ || (byte1 == (byte) 0xED && byte2 >= (byte) 0xA0) ++ // byte3 trailing-byte test ++ || bytes.read(index++) > (byte) 0xBF) { ++ return MALFORMED; ++ } ++ } else { ++ // four-byte form ++ ++ if (index >= limit - 2) { // incomplete sequence ++ return incompleteStateFor(bytes, index, limit); ++ } ++ if ((byte2 = bytes.read(index++)) > (byte) 0xBF ++ // Check that 1 <= plane <= 16. Tricky optimized form of: ++ // if (byte1 > (byte) 0xF4 || ++ // byte1 == (byte) 0xF0 && byte2 < (byte) 0x90 || ++ // byte1 == (byte) 0xF4 && byte2 > (byte) 0x8F) ++ || (((byte1 << 28) + (byte2 - (byte) 0x90)) >> 30) != 0 ++ // byte3 trailing-byte test ++ || bytes.read(index++) > (byte) 0xBF ++ // byte4 trailing-byte test ++ || bytes.read(index++) > (byte) 0xBF) { ++ return MALFORMED; ++ } ++ } ++ } ++ } ++ + /** + * Encodes an input character sequence ({@code in}) to UTF-8 in the target array ({@code out}). + * For a string, this method is similar to diff --git a/hbase-protocol/pom.xml b/hbase-protocol/pom.xml index 3ada92d5909..e21617bffcd 100644 --- a/hbase-protocol/pom.xml +++ b/hbase-protocol/pom.xml @@ -143,6 +143,14 @@ compile-protobuf + compile-protobuf diff --git a/hbase-server/pom.xml b/hbase-server/pom.xml index a43100634eb..0bdee40692e 100644 --- a/hbase-server/pom.xml +++ b/hbase-server/pom.xml @@ -393,10 +393,6 @@ hbase-resource-bundle ${project.version} true - - - org.apache.httpcomponents - httpclient commons-codec