| Index: third_party/protobuf/java/src/main/java/com/google/protobuf/NioByteString.java
|
| diff --git a/third_party/protobuf/java/src/main/java/com/google/protobuf/NioByteString.java b/third_party/protobuf/java/src/main/java/com/google/protobuf/NioByteString.java
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f71e41b2643dc0084b0055f3146d4b7cca1dea3a
|
| --- /dev/null
|
| +++ b/third_party/protobuf/java/src/main/java/com/google/protobuf/NioByteString.java
|
| @@ -0,0 +1,309 @@
|
| +// 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 com.google.protobuf;
|
| +
|
| +import java.io.FileOutputStream;
|
| +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.InvalidMarkException;
|
| +import java.nio.channels.Channels;
|
| +import java.nio.charset.Charset;
|
| +import java.util.Collections;
|
| +import java.util.List;
|
| +
|
| +/**
|
| + * A {@link ByteString} that wraps around a {@link ByteBuffer}.
|
| + */
|
| +final class NioByteString extends ByteString.LeafByteString {
|
| + private final ByteBuffer buffer;
|
| +
|
| + NioByteString(ByteBuffer buffer) {
|
| + if (buffer == null) {
|
| + throw new NullPointerException("buffer");
|
| + }
|
| +
|
| + this.buffer = buffer.slice();
|
| + }
|
| +
|
| + // =================================================================
|
| + // Serializable
|
| +
|
| + /**
|
| + * Magic method that lets us override serialization behavior.
|
| + */
|
| + private Object writeReplace() {
|
| + return ByteString.copyFrom(buffer.slice());
|
| + }
|
| +
|
| + /**
|
| + * Magic method that lets us override deserialization behavior.
|
| + */
|
| + private void readObject(@SuppressWarnings("unused") ObjectInputStream in) throws IOException {
|
| + throw new InvalidObjectException("NioByteString instances are not to be serialized directly");
|
| + }
|
| +
|
| + // =================================================================
|
| +
|
| + @Override
|
| + public byte byteAt(int index) {
|
| + try {
|
| + return buffer.get(index);
|
| + } catch (ArrayIndexOutOfBoundsException e) {
|
| + throw e;
|
| + } catch (IndexOutOfBoundsException e) {
|
| + throw new ArrayIndexOutOfBoundsException(e.getMessage());
|
| + }
|
| + }
|
| +
|
| + @Override
|
| + public int size() {
|
| + return buffer.remaining();
|
| + }
|
| +
|
| + @Override
|
| + public ByteString substring(int beginIndex, int endIndex) {
|
| + try {
|
| + ByteBuffer slice = slice(beginIndex, endIndex);
|
| + return new NioByteString(slice);
|
| + } catch (ArrayIndexOutOfBoundsException e) {
|
| + throw e;
|
| + } catch (IndexOutOfBoundsException e) {
|
| + throw new ArrayIndexOutOfBoundsException(e.getMessage());
|
| + }
|
| + }
|
| +
|
| + @Override
|
| + protected void copyToInternal(
|
| + byte[] target, int sourceOffset, int targetOffset, int numberToCopy) {
|
| + ByteBuffer slice = buffer.slice();
|
| + slice.position(sourceOffset);
|
| + slice.get(target, targetOffset, numberToCopy);
|
| + }
|
| +
|
| + @Override
|
| + public void copyTo(ByteBuffer target) {
|
| + target.put(buffer.slice());
|
| + }
|
| +
|
| + @Override
|
| + public void writeTo(OutputStream out) throws IOException {
|
| + writeToInternal(out, buffer.position(), buffer.remaining());
|
| + }
|
| +
|
| + @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 {
|
| + if (buffer.hasArray()) {
|
| + // Optimized write for array-backed buffers.
|
| + // Note that we're taking the risk that a malicious OutputStream could modify the array.
|
| + int bufferOffset = buffer.arrayOffset() + buffer.position() + sourceOffset;
|
| + out.write(buffer.array(), bufferOffset, numberToWrite);
|
| + return;
|
| + }
|
| +
|
| + // Slow path
|
| + if (out instanceof FileOutputStream || numberToWrite >= 8192) {
|
| + // Use a channel to write out the ByteBuffer.
|
| + Channels.newChannel(out).write(slice(sourceOffset, sourceOffset + numberToWrite));
|
| + } else {
|
| + // Just copy the data to an array and write it.
|
| + out.write(toByteArray());
|
| + }
|
| + }
|
| +
|
| + @Override
|
| + public ByteBuffer asReadOnlyByteBuffer() {
|
| + return buffer.asReadOnlyBuffer();
|
| + }
|
| +
|
| + @Override
|
| + public List<ByteBuffer> asReadOnlyByteBufferList() {
|
| + return Collections.singletonList(asReadOnlyByteBuffer());
|
| + }
|
| +
|
| + @Override
|
| + protected String toStringInternal(Charset charset) {
|
| + byte[] bytes;
|
| + int offset;
|
| + if (buffer.hasArray()) {
|
| + bytes = buffer.array();
|
| + offset = buffer.arrayOffset() + buffer.position();
|
| + } else {
|
| + bytes = toByteArray();
|
| + offset = 0;
|
| + }
|
| + return new String(bytes, offset, size(), charset);
|
| + }
|
| +
|
| + @Override
|
| + public boolean isValidUtf8() {
|
| + // TODO(nathanmittler): add a ByteBuffer fork for Utf8.isValidUtf8 to avoid the copy
|
| + byte[] bytes;
|
| + int startIndex;
|
| + if (buffer.hasArray()) {
|
| + bytes = buffer.array();
|
| + startIndex = buffer.arrayOffset() + buffer.position();
|
| + } else {
|
| + bytes = toByteArray();
|
| + startIndex = 0;
|
| + }
|
| + return Utf8.isValidUtf8(bytes, startIndex, startIndex + size());
|
| + }
|
| +
|
| + @Override
|
| + protected int partialIsValidUtf8(int state, int offset, int length) {
|
| + // TODO(nathanmittler): TODO add a ByteBuffer fork for Utf8.partialIsValidUtf8 to avoid the copy
|
| + byte[] bytes;
|
| + int startIndex;
|
| + if (buffer.hasArray()) {
|
| + bytes = buffer.array();
|
| + startIndex = buffer.arrayOffset() + buffer.position();
|
| + } else {
|
| + bytes = toByteArray();
|
| + startIndex = 0;
|
| + }
|
| + return Utf8.partialIsValidUtf8(state, bytes, startIndex, startIndex + size());
|
| + }
|
| +
|
| + @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 NioByteString) {
|
| + return buffer.equals(((NioByteString) other).buffer);
|
| + }
|
| + if (other instanceof RopeByteString) {
|
| + return other.equals(this);
|
| + }
|
| + return buffer.equals(otherString.asReadOnlyByteBuffer());
|
| + }
|
| +
|
| + @Override
|
| + protected int partialHash(int h, int offset, int length) {
|
| + for (int i = offset; i < offset + length; i++) {
|
| + h = h * 31 + buffer.get(i);
|
| + }
|
| + return h;
|
| + }
|
| +
|
| + @Override
|
| + public InputStream newInput() {
|
| + return new InputStream() {
|
| + private final ByteBuffer buf = buffer.slice();
|
| +
|
| + @Override
|
| + public void mark(int readlimit) {
|
| + buf.mark();
|
| + }
|
| +
|
| + @Override
|
| + public boolean markSupported() {
|
| + return true;
|
| + }
|
| +
|
| + @Override
|
| + public void reset() throws IOException {
|
| + try {
|
| + buf.reset();
|
| + } catch (InvalidMarkException e) {
|
| + throw new IOException(e);
|
| + }
|
| + }
|
| +
|
| + @Override
|
| + public int available() throws IOException {
|
| + return buf.remaining();
|
| + }
|
| +
|
| + @Override
|
| + public int read() throws IOException {
|
| + if (!buf.hasRemaining()) {
|
| + return -1;
|
| + }
|
| + return buf.get() & 0xFF;
|
| + }
|
| +
|
| + @Override
|
| + public int read(byte[] bytes, int off, int len) throws IOException {
|
| + if (!buf.hasRemaining()) {
|
| + return -1;
|
| + }
|
| +
|
| + len = Math.min(len, buf.remaining());
|
| + buf.get(bytes, off, len);
|
| + return len;
|
| + }
|
| + };
|
| + }
|
| +
|
| + @Override
|
| + public CodedInputStream newCodedInput() {
|
| + return CodedInputStream.newInstance(buffer);
|
| + }
|
| +
|
| + /**
|
| + * Creates a slice of a range of this buffer.
|
| + *
|
| + * @param beginIndex the beginning index of the slice (inclusive).
|
| + * @param endIndex the end index of the slice (exclusive).
|
| + * @return the requested slice.
|
| + */
|
| + private ByteBuffer slice(int beginIndex, int endIndex) {
|
| + if (beginIndex < buffer.position() || endIndex > buffer.limit() || beginIndex > endIndex) {
|
| + throw new IllegalArgumentException(
|
| + String.format("Invalid indices [%d, %d]", beginIndex, endIndex));
|
| + }
|
| +
|
| + ByteBuffer slice = buffer.slice();
|
| + slice.position(beginIndex - buffer.position());
|
| + slice.limit(endIndex - buffer.position());
|
| + return slice;
|
| + }
|
| +}
|
|
|