| Index: third_party/protobuf/java/src/main/java/com/google/protobuf/ByteString.java
|
| diff --git a/third_party/protobuf/java/src/main/java/com/google/protobuf/ByteString.java b/third_party/protobuf/java/src/main/java/com/google/protobuf/ByteString.java
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..68f20d5199b61ecd085a16d153e4a2b21fa1a9ca
|
| --- /dev/null
|
| +++ b/third_party/protobuf/java/src/main/java/com/google/protobuf/ByteString.java
|
| @@ -0,0 +1,1152 @@
|
| +// 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.ByteArrayOutputStream;
|
| +import java.io.IOException;
|
| +import java.io.InputStream;
|
| +import java.io.OutputStream;
|
| +import java.io.Serializable;
|
| +import java.io.UnsupportedEncodingException;
|
| +import java.nio.ByteBuffer;
|
| +import java.nio.charset.Charset;
|
| +import java.nio.charset.UnsupportedCharsetException;
|
| +import java.util.ArrayList;
|
| +import java.util.Collection;
|
| +import java.util.Iterator;
|
| +import java.util.List;
|
| +import java.util.NoSuchElementException;
|
| +
|
| +/**
|
| + * Immutable sequence of bytes. Substring is supported by sharing the reference
|
| + * to the immutable underlying bytes, as with {@link String}. Concatenation is
|
| + * likewise supported without copying (long strings) by building a tree of
|
| + * pieces in {@link RopeByteString}.
|
| + * <p>
|
| + * Like {@link String}, the contents of a {@link ByteString} can never be
|
| + * observed to change, not even in the presence of a data race or incorrect
|
| + * API usage in the client code.
|
| + *
|
| + * @author crazybob@google.com Bob Lee
|
| + * @author kenton@google.com Kenton Varda
|
| + * @author carlanton@google.com Carl Haverl
|
| + * @author martinrb@google.com Martin Buchholz
|
| + */
|
| +public abstract class ByteString implements Iterable<Byte>, Serializable {
|
| +
|
| + /**
|
| + * When two strings to be concatenated have a combined length shorter than
|
| + * this, we just copy their bytes on {@link #concat(ByteString)}.
|
| + * The trade-off is copy size versus the overhead of creating tree nodes
|
| + * in {@link RopeByteString}.
|
| + */
|
| + static final int CONCATENATE_BY_COPY_SIZE = 128;
|
| +
|
| + /**
|
| + * When copying an InputStream into a ByteString with .readFrom(),
|
| + * the chunks in the underlying rope start at 256 bytes, but double
|
| + * each iteration up to 8192 bytes.
|
| + */
|
| + static final int MIN_READ_FROM_CHUNK_SIZE = 0x100; // 256b
|
| + static final int MAX_READ_FROM_CHUNK_SIZE = 0x2000; // 8k
|
| +
|
| + /**
|
| + * Empty {@code ByteString}.
|
| + */
|
| + public static final ByteString EMPTY = new LiteralByteString(new byte[0]);
|
| +
|
| + /**
|
| + * Cached hash value. Intentionally accessed via a data race, which
|
| + * is safe because of the Java Memory Model's "no out-of-thin-air values"
|
| + * guarantees for ints. A value of 0 implies that the hash has not been set.
|
| + */
|
| + private int hash = 0;
|
| +
|
| + // This constructor is here to prevent subclassing outside of this package,
|
| + ByteString() {}
|
| +
|
| + /**
|
| + * Gets the byte at the given index. This method should be used only for
|
| + * random access to individual bytes. To access bytes sequentially, use the
|
| + * {@link ByteIterator} returned by {@link #iterator()}, and call {@link
|
| + * #substring(int, int)} first if necessary.
|
| + *
|
| + * @param index index of byte
|
| + * @return the value
|
| + * @throws ArrayIndexOutOfBoundsException {@code index < 0 or index >= size}
|
| + */
|
| + public abstract byte byteAt(int index);
|
| +
|
| + /**
|
| + * Return a {@link ByteString.ByteIterator} over the bytes in the ByteString.
|
| + * To avoid auto-boxing, you may get the iterator manually and call
|
| + * {@link ByteIterator#nextByte()}.
|
| + *
|
| + * @return the iterator
|
| + */
|
| + @Override
|
| + public final ByteIterator iterator() {
|
| + return new ByteIterator() {
|
| + private int position = 0;
|
| + private final int limit = size();
|
| +
|
| + @Override
|
| + public boolean hasNext() {
|
| + return position < limit;
|
| + }
|
| +
|
| + @Override
|
| + public Byte next() {
|
| + // Boxing calls Byte.valueOf(byte), which does not instantiate.
|
| + return nextByte();
|
| + }
|
| +
|
| + @Override
|
| + public byte nextByte() {
|
| + try {
|
| + return byteAt(position++);
|
| + } catch (ArrayIndexOutOfBoundsException e) {
|
| + throw new NoSuchElementException(e.getMessage());
|
| + }
|
| + }
|
| +
|
| + @Override
|
| + public void remove() {
|
| + throw new UnsupportedOperationException();
|
| + }
|
| + };
|
| + }
|
| +
|
| + /**
|
| + * This interface extends {@code Iterator<Byte>}, so that we can return an
|
| + * unboxed {@code byte}.
|
| + */
|
| + public interface ByteIterator extends Iterator<Byte> {
|
| + /**
|
| + * An alternative to {@link Iterator#next()} that returns an
|
| + * unboxed primitive {@code byte}.
|
| + *
|
| + * @return the next {@code byte} in the iteration
|
| + * @throws NoSuchElementException if the iteration has no more elements
|
| + */
|
| + byte nextByte();
|
| + }
|
| +
|
| + /**
|
| + * Gets the number of bytes.
|
| + *
|
| + * @return size in bytes
|
| + */
|
| + public abstract int size();
|
| +
|
| + /**
|
| + * Returns {@code true} if the size is {@code 0}, {@code false} otherwise.
|
| + *
|
| + * @return true if this is zero bytes long
|
| + */
|
| + public final boolean isEmpty() {
|
| + return size() == 0;
|
| + }
|
| +
|
| + // =================================================================
|
| + // ByteString -> substring
|
| +
|
| + /**
|
| + * Return the substring from {@code beginIndex}, inclusive, to the end of the
|
| + * string.
|
| + *
|
| + * @param beginIndex start at this index
|
| + * @return substring sharing underlying data
|
| + * @throws IndexOutOfBoundsException if {@code beginIndex < 0} or
|
| + * {@code beginIndex > size()}.
|
| + */
|
| + public final ByteString substring(int beginIndex) {
|
| + return substring(beginIndex, size());
|
| + }
|
| +
|
| + /**
|
| + * Return the substring from {@code beginIndex}, inclusive, to {@code
|
| + * endIndex}, exclusive.
|
| + *
|
| + * @param beginIndex start at this index
|
| + * @param endIndex the last character is the one before this index
|
| + * @return substring sharing underlying data
|
| + * @throws IndexOutOfBoundsException if {@code beginIndex < 0},
|
| + * {@code endIndex > size()}, or {@code beginIndex > endIndex}.
|
| + */
|
| + public abstract ByteString substring(int beginIndex, int endIndex);
|
| +
|
| + /**
|
| + * Tests if this bytestring starts with the specified prefix.
|
| + * Similar to {@link String#startsWith(String)}
|
| + *
|
| + * @param prefix the prefix.
|
| + * @return <code>true</code> if the byte sequence represented by the
|
| + * argument is a prefix of the byte sequence represented by
|
| + * this string; <code>false</code> otherwise.
|
| + */
|
| + public final boolean startsWith(ByteString prefix) {
|
| + return size() >= prefix.size() &&
|
| + substring(0, prefix.size()).equals(prefix);
|
| + }
|
| +
|
| + /**
|
| + * Tests if this bytestring ends with the specified suffix.
|
| + * Similar to {@link String#endsWith(String)}
|
| + *
|
| + * @param suffix the suffix.
|
| + * @return <code>true</code> if the byte sequence represented by the
|
| + * argument is a suffix of the byte sequence represented by
|
| + * this string; <code>false</code> otherwise.
|
| + */
|
| + public final boolean endsWith(ByteString suffix) {
|
| + return size() >= suffix.size() &&
|
| + substring(size() - suffix.size()).equals(suffix);
|
| + }
|
| +
|
| + // =================================================================
|
| + // byte[] -> ByteString
|
| +
|
| + /**
|
| + * Copies the given bytes into a {@code ByteString}.
|
| + *
|
| + * @param bytes source array
|
| + * @param offset offset in source array
|
| + * @param size number of bytes to copy
|
| + * @return new {@code ByteString}
|
| + */
|
| + public static ByteString copyFrom(byte[] bytes, int offset, int size) {
|
| + byte[] copy = new byte[size];
|
| + System.arraycopy(bytes, offset, copy, 0, size);
|
| + return new LiteralByteString(copy);
|
| + }
|
| +
|
| + /**
|
| + * Copies the given bytes into a {@code ByteString}.
|
| + *
|
| + * @param bytes to copy
|
| + * @return new {@code ByteString}
|
| + */
|
| + public static ByteString copyFrom(byte[] bytes) {
|
| + return copyFrom(bytes, 0, bytes.length);
|
| + }
|
| +
|
| + /**
|
| + * Copies the next {@code size} bytes from a {@code java.nio.ByteBuffer} into
|
| + * a {@code ByteString}.
|
| + *
|
| + * @param bytes source buffer
|
| + * @param size number of bytes to copy
|
| + * @return new {@code ByteString}
|
| + */
|
| + public static ByteString copyFrom(ByteBuffer bytes, int size) {
|
| + byte[] copy = new byte[size];
|
| + bytes.get(copy);
|
| + return new LiteralByteString(copy);
|
| + }
|
| +
|
| + /**
|
| + * Copies the remaining bytes from a {@code java.nio.ByteBuffer} into
|
| + * a {@code ByteString}.
|
| + *
|
| + * @param bytes sourceBuffer
|
| + * @return new {@code ByteString}
|
| + */
|
| + public static ByteString copyFrom(ByteBuffer bytes) {
|
| + return copyFrom(bytes, bytes.remaining());
|
| + }
|
| +
|
| + /**
|
| + * Encodes {@code text} into a sequence of bytes using the named charset
|
| + * and returns the result as a {@code ByteString}.
|
| + *
|
| + * @param text source string
|
| + * @param charsetName encoding to use
|
| + * @return new {@code ByteString}
|
| + * @throws UnsupportedEncodingException if the encoding isn't found
|
| + */
|
| + public static ByteString copyFrom(String text, String charsetName)
|
| + throws UnsupportedEncodingException {
|
| + return new LiteralByteString(text.getBytes(charsetName));
|
| + }
|
| +
|
| + /**
|
| + * Encodes {@code text} into a sequence of bytes using the named charset
|
| + * and returns the result as a {@code ByteString}.
|
| + *
|
| + * @param text source string
|
| + * @param charset encode using this charset
|
| + * @return new {@code ByteString}
|
| + */
|
| + public static ByteString copyFrom(String text, Charset charset) {
|
| + return new LiteralByteString(text.getBytes(charset));
|
| + }
|
| +
|
| + /**
|
| + * Encodes {@code text} into a sequence of UTF-8 bytes and returns the
|
| + * result as a {@code ByteString}.
|
| + *
|
| + * @param text source string
|
| + * @return new {@code ByteString}
|
| + */
|
| + public static ByteString copyFromUtf8(String text) {
|
| + return new LiteralByteString(text.getBytes(Internal.UTF_8));
|
| + }
|
| +
|
| + // =================================================================
|
| + // InputStream -> ByteString
|
| +
|
| + /**
|
| + * Completely reads the given stream's bytes into a
|
| + * {@code ByteString}, blocking if necessary until all bytes are
|
| + * read through to the end of the stream.
|
| + *
|
| + * <b>Performance notes:</b> The returned {@code ByteString} is an
|
| + * immutable tree of byte arrays ("chunks") of the stream data. The
|
| + * first chunk is small, with subsequent chunks each being double
|
| + * the size, up to 8K.
|
| + *
|
| + * <p>Each byte read from the input stream will be copied twice to ensure
|
| + * that the resulting ByteString is truly immutable.
|
| + *
|
| + * @param streamToDrain The source stream, which is read completely
|
| + * but not closed.
|
| + * @return A new {@code ByteString} which is made up of chunks of
|
| + * various sizes, depending on the behavior of the underlying
|
| + * stream.
|
| + * @throws IOException IOException is thrown if there is a problem
|
| + * reading the underlying stream.
|
| + */
|
| + public static ByteString readFrom(InputStream streamToDrain)
|
| + throws IOException {
|
| + return readFrom(streamToDrain, MIN_READ_FROM_CHUNK_SIZE, MAX_READ_FROM_CHUNK_SIZE);
|
| + }
|
| +
|
| + /**
|
| + * Completely reads the given stream's bytes into a
|
| + * {@code ByteString}, blocking if necessary until all bytes are
|
| + * read through to the end of the stream.
|
| + *
|
| + * <b>Performance notes:</b> The returned {@code ByteString} is an
|
| + * immutable tree of byte arrays ("chunks") of the stream data. The
|
| + * chunkSize parameter sets the size of these byte arrays.
|
| + *
|
| + * <p>Each byte read from the input stream will be copied twice to ensure
|
| + * that the resulting ByteString is truly immutable.
|
| + *
|
| + * @param streamToDrain The source stream, which is read completely
|
| + * but not closed.
|
| + * @param chunkSize The size of the chunks in which to read the
|
| + * stream.
|
| + * @return A new {@code ByteString} which is made up of chunks of
|
| + * the given size.
|
| + * @throws IOException IOException is thrown if there is a problem
|
| + * reading the underlying stream.
|
| + */
|
| + public static ByteString readFrom(InputStream streamToDrain, int chunkSize)
|
| + throws IOException {
|
| + return readFrom(streamToDrain, chunkSize, chunkSize);
|
| + }
|
| +
|
| + // Helper method that takes the chunk size range as a parameter.
|
| + public static ByteString readFrom(InputStream streamToDrain, int minChunkSize,
|
| + int maxChunkSize) throws IOException {
|
| + Collection<ByteString> results = new ArrayList<ByteString>();
|
| +
|
| + // copy the inbound bytes into a list of chunks; the chunk size
|
| + // grows exponentially to support both short and long streams.
|
| + int chunkSize = minChunkSize;
|
| + while (true) {
|
| + ByteString chunk = readChunk(streamToDrain, chunkSize);
|
| + if (chunk == null) {
|
| + break;
|
| + }
|
| + results.add(chunk);
|
| + chunkSize = Math.min(chunkSize * 2, maxChunkSize);
|
| + }
|
| +
|
| + return ByteString.copyFrom(results);
|
| + }
|
| +
|
| + /**
|
| + * Blocks until a chunk of the given size can be made from the
|
| + * stream, or EOF is reached. Calls read() repeatedly in case the
|
| + * given stream implementation doesn't completely fill the given
|
| + * buffer in one read() call.
|
| + *
|
| + * @return A chunk of the desired size, or else a chunk as large as
|
| + * was available when end of stream was reached. Returns null if the
|
| + * given stream had no more data in it.
|
| + */
|
| + private static ByteString readChunk(InputStream in, final int chunkSize)
|
| + throws IOException {
|
| + final byte[] buf = new byte[chunkSize];
|
| + int bytesRead = 0;
|
| + while (bytesRead < chunkSize) {
|
| + final int count = in.read(buf, bytesRead, chunkSize - bytesRead);
|
| + if (count == -1) {
|
| + break;
|
| + }
|
| + bytesRead += count;
|
| + }
|
| +
|
| + if (bytesRead == 0) {
|
| + return null;
|
| + }
|
| +
|
| + // Always make a copy since InputStream could steal a reference to buf.
|
| + return ByteString.copyFrom(buf, 0, bytesRead);
|
| + }
|
| +
|
| + // =================================================================
|
| + // Multiple ByteStrings -> One ByteString
|
| +
|
| + /**
|
| + * Concatenate the given {@code ByteString} to this one. Short concatenations,
|
| + * of total size smaller than {@link ByteString#CONCATENATE_BY_COPY_SIZE}, are
|
| + * produced by copying the underlying bytes (as per Rope.java, <a
|
| + * href="http://www.cs.ubc.ca/local/reading/proceedings/spe91-95/spe/vol25/issue12/spe986.pdf">
|
| + * BAP95 </a>. In general, the concatenate involves no copying.
|
| + *
|
| + * @param other string to concatenate
|
| + * @return a new {@code ByteString} instance
|
| + */
|
| + public final ByteString concat(ByteString other) {
|
| + if (Integer.MAX_VALUE - size() < other.size()) {
|
| + throw new IllegalArgumentException("ByteString would be too long: " +
|
| + size() + "+" + other.size());
|
| + }
|
| +
|
| + return RopeByteString.concatenate(this, other);
|
| + }
|
| +
|
| + /**
|
| + * Concatenates all byte strings in the iterable and returns the result.
|
| + * This is designed to run in O(list size), not O(total bytes).
|
| + *
|
| + * <p>The returned {@code ByteString} is not necessarily a unique object.
|
| + * If the list is empty, the returned object is the singleton empty
|
| + * {@code ByteString}. If the list has only one element, that
|
| + * {@code ByteString} will be returned without copying.
|
| + *
|
| + * @param byteStrings strings to be concatenated
|
| + * @return new {@code ByteString}
|
| + */
|
| + public static ByteString copyFrom(Iterable<ByteString> byteStrings) {
|
| + // Determine the size;
|
| + final int size;
|
| + if (!(byteStrings instanceof Collection)) {
|
| + int tempSize = 0;
|
| + for (Iterator<ByteString> iter = byteStrings.iterator(); iter.hasNext();
|
| + iter.next(), ++tempSize) {
|
| + }
|
| + size = tempSize;
|
| + } else {
|
| + size = ((Collection<ByteString>) byteStrings).size();
|
| + }
|
| +
|
| + if (size == 0) {
|
| + return EMPTY;
|
| + }
|
| +
|
| + return balancedConcat(byteStrings.iterator(), size);
|
| + }
|
| +
|
| + // Internal function used by copyFrom(Iterable<ByteString>).
|
| + // Create a balanced concatenation of the next "length" elements from the
|
| + // iterable.
|
| + private static ByteString balancedConcat(Iterator<ByteString> iterator, int length) {
|
| + assert length >= 1;
|
| + ByteString result;
|
| + if (length == 1) {
|
| + result = iterator.next();
|
| + } else {
|
| + int halfLength = length >>> 1;
|
| + ByteString left = balancedConcat(iterator, halfLength);
|
| + ByteString right = balancedConcat(iterator, length - halfLength);
|
| + result = left.concat(right);
|
| + }
|
| + return result;
|
| + }
|
| +
|
| + // =================================================================
|
| + // ByteString -> byte[]
|
| +
|
| + /**
|
| + * Copies bytes into a buffer at the given offset.
|
| + *
|
| + * @param target buffer to copy into
|
| + * @param offset in the target buffer
|
| + * @throws IndexOutOfBoundsException if the offset is negative or too large
|
| + */
|
| + public void copyTo(byte[] target, int offset) {
|
| + copyTo(target, 0, offset, size());
|
| + }
|
| +
|
| + /**
|
| + * Copies bytes into a buffer.
|
| + *
|
| + * @param target buffer to copy into
|
| + * @param sourceOffset offset within these bytes
|
| + * @param targetOffset offset within the target buffer
|
| + * @param numberToCopy number of bytes to copy
|
| + * @throws IndexOutOfBoundsException if an offset or size is negative or too
|
| + * large
|
| + */
|
| + public final void copyTo(byte[] target, int sourceOffset, int targetOffset,
|
| + int numberToCopy) {
|
| + checkRange(sourceOffset, sourceOffset + numberToCopy, size());
|
| + checkRange(targetOffset, targetOffset + numberToCopy, target.length);
|
| + if (numberToCopy > 0) {
|
| + copyToInternal(target, sourceOffset, targetOffset, numberToCopy);
|
| + }
|
| + }
|
| +
|
| + /**
|
| + * Internal (package private) implementation of
|
| + * {@link #copyTo(byte[],int,int,int)}.
|
| + * It assumes that all error checking has already been performed and that
|
| + * {@code numberToCopy > 0}.
|
| + */
|
| + protected abstract void copyToInternal(byte[] target, int sourceOffset,
|
| + int targetOffset, int numberToCopy);
|
| +
|
| + /**
|
| + * Copies bytes into a ByteBuffer.
|
| + *
|
| + * @param target ByteBuffer to copy into.
|
| + * @throws java.nio.ReadOnlyBufferException if the {@code target} is read-only
|
| + * @throws java.nio.BufferOverflowException if the {@code target}'s
|
| + * remaining() space is not large enough to hold the data.
|
| + */
|
| + public abstract void copyTo(ByteBuffer target);
|
| +
|
| + /**
|
| + * Copies bytes to a {@code byte[]}.
|
| + *
|
| + * @return copied bytes
|
| + */
|
| + public final byte[] toByteArray() {
|
| + final int size = size();
|
| + if (size == 0) {
|
| + return Internal.EMPTY_BYTE_ARRAY;
|
| + }
|
| + byte[] result = new byte[size];
|
| + copyToInternal(result, 0, 0, size);
|
| + return result;
|
| + }
|
| +
|
| + /**
|
| + * Writes the complete contents of this byte string to
|
| + * the specified output stream argument.
|
| + *
|
| + * <p>It is assumed that the {@link OutputStream} will not modify the contents passed it
|
| + * it. It may be possible for a malicious {@link OutputStream} to corrupt
|
| + * the data underlying the {@link ByteString}.
|
| + *
|
| + * @param out the output stream to which to write the data.
|
| + * @throws IOException if an I/O error occurs.
|
| + */
|
| + public abstract void writeTo(OutputStream out) throws IOException;
|
| +
|
| + /**
|
| + * Writes a specified part of this byte string to an output stream.
|
| + *
|
| + * @param out the output stream to which to write the data.
|
| + * @param sourceOffset offset within these bytes
|
| + * @param numberToWrite number of bytes to write
|
| + * @throws IOException if an I/O error occurs.
|
| + * @throws IndexOutOfBoundsException if an offset or size is negative or too
|
| + * large
|
| + */
|
| + final void writeTo(OutputStream out, int sourceOffset, int numberToWrite)
|
| + throws IOException {
|
| + checkRange(sourceOffset, sourceOffset + numberToWrite, size());
|
| + if (numberToWrite > 0) {
|
| + writeToInternal(out, sourceOffset, numberToWrite);
|
| + }
|
| + }
|
| +
|
| + /**
|
| + * Internal version of {@link #writeTo(OutputStream,int,int)} that assumes
|
| + * all error checking has already been done.
|
| + */
|
| + abstract void writeToInternal(OutputStream out, int sourceOffset, int numberToWrite)
|
| + throws IOException;
|
| +
|
| + /**
|
| + * Constructs a read-only {@code java.nio.ByteBuffer} whose content
|
| + * is equal to the contents of this byte string.
|
| + * The result uses the same backing array as the byte string, if possible.
|
| + *
|
| + * @return wrapped bytes
|
| + */
|
| + public abstract ByteBuffer asReadOnlyByteBuffer();
|
| +
|
| + /**
|
| + * Constructs a list of read-only {@code java.nio.ByteBuffer} objects
|
| + * such that the concatenation of their contents is equal to the contents
|
| + * of this byte string. The result uses the same backing arrays as the
|
| + * byte string.
|
| + * <p>
|
| + * By returning a list, implementations of this method may be able to avoid
|
| + * copying even when there are multiple backing arrays.
|
| + *
|
| + * @return a list of wrapped bytes
|
| + */
|
| + public abstract List<ByteBuffer> asReadOnlyByteBufferList();
|
| +
|
| + /**
|
| + * Constructs a new {@code String} by decoding the bytes using the
|
| + * specified charset.
|
| + *
|
| + * @param charsetName encode using this charset
|
| + * @return new string
|
| + * @throws UnsupportedEncodingException if charset isn't recognized
|
| + */
|
| + public final String toString(String charsetName)
|
| + throws UnsupportedEncodingException {
|
| + try {
|
| + return toString(Charset.forName(charsetName));
|
| + } catch (UnsupportedCharsetException e) {
|
| + UnsupportedEncodingException exception = new UnsupportedEncodingException(charsetName);
|
| + exception.initCause(e);
|
| + throw exception;
|
| + }
|
| + }
|
| +
|
| + /**
|
| + * Constructs a new {@code String} by decoding the bytes using the
|
| + * specified charset. Returns the same empty String if empty.
|
| + *
|
| + * @param charset encode using this charset
|
| + * @return new string
|
| + */
|
| + public final String toString(Charset charset) {
|
| + return size() == 0 ? "" : toStringInternal(charset);
|
| + }
|
| +
|
| + /**
|
| + * Constructs a new {@code String} by decoding the bytes using the
|
| + * specified charset.
|
| + *
|
| + * @param charset encode using this charset
|
| + * @return new string
|
| + */
|
| + protected abstract String toStringInternal(Charset charset);
|
| +
|
| + // =================================================================
|
| + // UTF-8 decoding
|
| +
|
| + /**
|
| + * Constructs a new {@code String} by decoding the bytes as UTF-8.
|
| + *
|
| + * @return new string using UTF-8 encoding
|
| + */
|
| + public final String toStringUtf8() {
|
| + return toString(Internal.UTF_8);
|
| + }
|
| +
|
| + /**
|
| + * Tells whether this {@code ByteString} represents a well-formed UTF-8
|
| + * byte sequence, such that the original bytes can be converted to a
|
| + * String object and then round tripped back to bytes without loss.
|
| + *
|
| + * <p>More precisely, returns {@code true} whenever: <pre> {@code
|
| + * Arrays.equals(byteString.toByteArray(),
|
| + * new String(byteString.toByteArray(), "UTF-8").getBytes("UTF-8"))
|
| + * }</pre>
|
| + *
|
| + * <p>This method returns {@code false} for "overlong" byte sequences,
|
| + * as well as for 3-byte sequences that would map to a surrogate
|
| + * character, in accordance with the restricted definition of UTF-8
|
| + * introduced in Unicode 3.1. Note that the UTF-8 decoder included in
|
| + * Oracle's JDK has been modified to also reject "overlong" byte
|
| + * sequences, but (as of 2011) still accepts 3-byte surrogate
|
| + * character byte sequences.
|
| + *
|
| + * <p>See the Unicode Standard,<br>
|
| + * Table 3-6. <em>UTF-8 Bit Distribution</em>,<br>
|
| + * Table 3-7. <em>Well Formed UTF-8 Byte Sequences</em>.
|
| + *
|
| + * @return whether the bytes in this {@code ByteString} are a
|
| + * well-formed UTF-8 byte sequence
|
| + */
|
| + public abstract boolean isValidUtf8();
|
| +
|
| + /**
|
| + * Tells whether the given byte sequence is a well-formed, malformed, or
|
| + * incomplete UTF-8 byte sequence. This method accepts and returns a partial
|
| + * state result, allowing the bytes for a complete UTF-8 byte sequence to be
|
| + * composed from multiple {@code ByteString} segments.
|
| + *
|
| + * @param state either {@code 0} (if this is the initial decoding operation)
|
| + * or the value returned from a call to a partial decoding method for the
|
| + * previous bytes
|
| + * @param offset offset of the first byte to check
|
| + * @param length number of bytes to check
|
| + *
|
| + * @return {@code -1} if the partial byte sequence is definitely malformed,
|
| + * {@code 0} if it is well-formed (no additional input needed), or, if the
|
| + * byte sequence is "incomplete", i.e. apparently terminated in the middle of
|
| + * a character, an opaque integer "state" value containing enough information
|
| + * to decode the character when passed to a subsequent invocation of a
|
| + * partial decoding method.
|
| + */
|
| + protected abstract int partialIsValidUtf8(int state, int offset, int length);
|
| +
|
| + // =================================================================
|
| + // equals() and hashCode()
|
| +
|
| + @Override
|
| + public abstract boolean equals(Object o);
|
| +
|
| + /**
|
| + * Base class for leaf {@link ByteString}s (i.e. non-ropes).
|
| + */
|
| + abstract static class LeafByteString extends ByteString {
|
| + @Override
|
| + protected final int getTreeDepth() {
|
| + return 0;
|
| + }
|
| +
|
| + @Override
|
| + protected final boolean isBalanced() {
|
| + return true;
|
| + }
|
| +
|
| + /**
|
| + * Check equality of the substring of given length of this object starting at
|
| + * zero with another {@code ByteString} substring starting at offset.
|
| + *
|
| + * @param other what to compare a substring in
|
| + * @param offset offset into other
|
| + * @param length number of bytes to compare
|
| + * @return true for equality of substrings, else false.
|
| + */
|
| + abstract boolean equalsRange(ByteString other, int offset, int length);
|
| + }
|
| +
|
| + /**
|
| + * Compute the hashCode using the traditional algorithm from {@link
|
| + * ByteString}.
|
| + *
|
| + * @return hashCode value
|
| + */
|
| + @Override
|
| + public final int hashCode() {
|
| + int h = hash;
|
| +
|
| + if (h == 0) {
|
| + int size = size();
|
| + h = partialHash(size, 0, size);
|
| + if (h == 0) {
|
| + h = 1;
|
| + }
|
| + hash = h;
|
| + }
|
| + return h;
|
| + }
|
| +
|
| + // =================================================================
|
| + // Input stream
|
| +
|
| + /**
|
| + * Creates an {@code InputStream} which can be used to read the bytes.
|
| + * <p>
|
| + * The {@link InputStream} returned by this method is guaranteed to be
|
| + * completely non-blocking. The method {@link InputStream#available()}
|
| + * returns the number of bytes remaining in the stream. The methods
|
| + * {@link InputStream#read(byte[])}, {@link InputStream#read(byte[],int,int)}
|
| + * and {@link InputStream#skip(long)} will read/skip as many bytes as are
|
| + * available. The method {@link InputStream#markSupported()} returns
|
| + * {@code true}.
|
| + * <p>
|
| + * The methods in the returned {@link InputStream} might <b>not</b> be
|
| + * thread safe.
|
| + *
|
| + * @return an input stream that returns the bytes of this byte string.
|
| + */
|
| + public abstract InputStream newInput();
|
| +
|
| + /**
|
| + * Creates a {@link CodedInputStream} which can be used to read the bytes.
|
| + * Using this is often more efficient than creating a {@link CodedInputStream}
|
| + * that wraps the result of {@link #newInput()}.
|
| + *
|
| + * @return stream based on wrapped data
|
| + */
|
| + public abstract CodedInputStream newCodedInput();
|
| +
|
| + // =================================================================
|
| + // Output stream
|
| +
|
| + /**
|
| + * Creates a new {@link Output} with the given initial capacity. Call {@link
|
| + * Output#toByteString()} to create the {@code ByteString} instance.
|
| + * <p>
|
| + * A {@link ByteString.Output} offers the same functionality as a
|
| + * {@link ByteArrayOutputStream}, except that it returns a {@link ByteString}
|
| + * rather than a {@code byte} array.
|
| + *
|
| + * @param initialCapacity estimate of number of bytes to be written
|
| + * @return {@code OutputStream} for building a {@code ByteString}
|
| + */
|
| + public static Output newOutput(int initialCapacity) {
|
| + return new Output(initialCapacity);
|
| + }
|
| +
|
| + /**
|
| + * Creates a new {@link Output}. Call {@link Output#toByteString()} to create
|
| + * the {@code ByteString} instance.
|
| + * <p>
|
| + * A {@link ByteString.Output} offers the same functionality as a
|
| + * {@link ByteArrayOutputStream}, except that it returns a {@link ByteString}
|
| + * rather than a {@code byte array}.
|
| + *
|
| + * @return {@code OutputStream} for building a {@code ByteString}
|
| + */
|
| + public static Output newOutput() {
|
| + return new Output(CONCATENATE_BY_COPY_SIZE);
|
| + }
|
| +
|
| + /**
|
| + * Outputs to a {@code ByteString} instance. Call {@link #toByteString()} to
|
| + * create the {@code ByteString} instance.
|
| + */
|
| + public static final class Output extends OutputStream {
|
| + // Implementation note.
|
| + // The public methods of this class must be synchronized. ByteStrings
|
| + // are guaranteed to be immutable. Without some sort of locking, it could
|
| + // be possible for one thread to call toByteSring(), while another thread
|
| + // is still modifying the underlying byte array.
|
| +
|
| + private static final byte[] EMPTY_BYTE_ARRAY = new byte[0];
|
| + // argument passed by user, indicating initial capacity.
|
| + private final int initialCapacity;
|
| + // ByteStrings to be concatenated to create the result
|
| + private final ArrayList<ByteString> flushedBuffers;
|
| + // Total number of bytes in the ByteStrings of flushedBuffers
|
| + private int flushedBuffersTotalBytes;
|
| + // Current buffer to which we are writing
|
| + private byte[] buffer;
|
| + // Location in buffer[] to which we write the next byte.
|
| + private int bufferPos;
|
| +
|
| + /**
|
| + * Creates a new ByteString output stream with the specified
|
| + * initial capacity.
|
| + *
|
| + * @param initialCapacity the initial capacity of the output stream.
|
| + */
|
| + Output(int initialCapacity) {
|
| + if (initialCapacity < 0) {
|
| + throw new IllegalArgumentException("Buffer size < 0");
|
| + }
|
| + this.initialCapacity = initialCapacity;
|
| + this.flushedBuffers = new ArrayList<ByteString>();
|
| + this.buffer = new byte[initialCapacity];
|
| + }
|
| +
|
| + @Override
|
| + public synchronized void write(int b) {
|
| + if (bufferPos == buffer.length) {
|
| + flushFullBuffer(1);
|
| + }
|
| + buffer[bufferPos++] = (byte)b;
|
| + }
|
| +
|
| + @Override
|
| + public synchronized void write(byte[] b, int offset, int length) {
|
| + if (length <= buffer.length - bufferPos) {
|
| + // The bytes can fit into the current buffer.
|
| + System.arraycopy(b, offset, buffer, bufferPos, length);
|
| + bufferPos += length;
|
| + } else {
|
| + // Use up the current buffer
|
| + int copySize = buffer.length - bufferPos;
|
| + System.arraycopy(b, offset, buffer, bufferPos, copySize);
|
| + offset += copySize;
|
| + length -= copySize;
|
| + // Flush the buffer, and get a new buffer at least big enough to cover
|
| + // what we still need to output
|
| + flushFullBuffer(length);
|
| + System.arraycopy(b, offset, buffer, 0 /* count */, length);
|
| + bufferPos = length;
|
| + }
|
| + }
|
| +
|
| + /**
|
| + * Creates a byte string. Its size is the current size of this output
|
| + * stream and its output has been copied to it.
|
| + *
|
| + * @return the current contents of this output stream, as a byte string.
|
| + */
|
| + public synchronized ByteString toByteString() {
|
| + flushLastBuffer();
|
| + return ByteString.copyFrom(flushedBuffers);
|
| + }
|
| +
|
| + /**
|
| + * Implement java.util.Arrays.copyOf() for jdk 1.5.
|
| + */
|
| + private byte[] copyArray(byte[] buffer, int length) {
|
| + byte[] result = new byte[length];
|
| + System.arraycopy(buffer, 0, result, 0, Math.min(buffer.length, length));
|
| + return result;
|
| + }
|
| +
|
| + /**
|
| + * Writes the complete contents of this byte array output stream to
|
| + * the specified output stream argument.
|
| + *
|
| + * @param out the output stream to which to write the data.
|
| + * @throws IOException if an I/O error occurs.
|
| + */
|
| + public void writeTo(OutputStream out) throws IOException {
|
| + ByteString[] cachedFlushBuffers;
|
| + byte[] cachedBuffer;
|
| + int cachedBufferPos;
|
| + synchronized (this) {
|
| + // Copy the information we need into local variables so as to hold
|
| + // the lock for as short a time as possible.
|
| + cachedFlushBuffers =
|
| + flushedBuffers.toArray(new ByteString[flushedBuffers.size()]);
|
| + cachedBuffer = buffer;
|
| + cachedBufferPos = bufferPos;
|
| + }
|
| + for (ByteString byteString : cachedFlushBuffers) {
|
| + byteString.writeTo(out);
|
| + }
|
| +
|
| + out.write(copyArray(cachedBuffer, cachedBufferPos));
|
| + }
|
| +
|
| + /**
|
| + * Returns the current size of the output stream.
|
| + *
|
| + * @return the current size of the output stream
|
| + */
|
| + public synchronized int size() {
|
| + return flushedBuffersTotalBytes + bufferPos;
|
| + }
|
| +
|
| + /**
|
| + * Resets this stream, so that all currently accumulated output in the
|
| + * output stream is discarded. The output stream can be used again,
|
| + * reusing the already allocated buffer space.
|
| + */
|
| + public synchronized void reset() {
|
| + flushedBuffers.clear();
|
| + flushedBuffersTotalBytes = 0;
|
| + bufferPos = 0;
|
| + }
|
| +
|
| + @Override
|
| + public String toString() {
|
| + return String.format("<ByteString.Output@%s size=%d>",
|
| + Integer.toHexString(System.identityHashCode(this)), size());
|
| + }
|
| +
|
| + /**
|
| + * Internal function used by writers. The current buffer is full, and the
|
| + * writer needs a new buffer whose size is at least the specified minimum
|
| + * size.
|
| + */
|
| + private void flushFullBuffer(int minSize) {
|
| + flushedBuffers.add(new LiteralByteString(buffer));
|
| + flushedBuffersTotalBytes += buffer.length;
|
| + // We want to increase our total capacity by 50%, but as a minimum,
|
| + // the new buffer should also at least be >= minSize and
|
| + // >= initial Capacity.
|
| + int newSize = Math.max(initialCapacity,
|
| + Math.max(minSize, flushedBuffersTotalBytes >>> 1));
|
| + buffer = new byte[newSize];
|
| + bufferPos = 0;
|
| + }
|
| +
|
| + /**
|
| + * Internal function used by {@link #toByteString()}. The current buffer may
|
| + * or may not be full, but it needs to be flushed.
|
| + */
|
| + private void flushLastBuffer() {
|
| + if (bufferPos < buffer.length) {
|
| + if (bufferPos > 0) {
|
| + byte[] bufferCopy = copyArray(buffer, bufferPos);
|
| + flushedBuffers.add(new LiteralByteString(bufferCopy));
|
| + }
|
| + // We reuse this buffer for further writes.
|
| + } else {
|
| + // Buffer is completely full. Huzzah.
|
| + flushedBuffers.add(new LiteralByteString(buffer));
|
| + // 99% of the time, we're not going to use this OutputStream again.
|
| + // We set buffer to an empty byte stream so that we're handling this
|
| + // case without wasting space. In the rare case that more writes
|
| + // *do* occur, this empty buffer will be flushed and an appropriately
|
| + // sized new buffer will be created.
|
| + buffer = EMPTY_BYTE_ARRAY;
|
| + }
|
| + flushedBuffersTotalBytes += bufferPos;
|
| + bufferPos = 0;
|
| + }
|
| + }
|
| +
|
| + /**
|
| + * Constructs a new {@code ByteString} builder, which allows you to
|
| + * efficiently construct a {@code ByteString} by writing to a {@link
|
| + * CodedOutputStream}. Using this is much more efficient than calling {@code
|
| + * newOutput()} and wrapping that in a {@code CodedOutputStream}.
|
| + *
|
| + * <p>This is package-private because it's a somewhat confusing interface.
|
| + * Users can call {@link Message#toByteString()} instead of calling this
|
| + * directly.
|
| + *
|
| + * @param size The target byte size of the {@code ByteString}. You must write
|
| + * exactly this many bytes before building the result.
|
| + * @return the builder
|
| + */
|
| + static CodedBuilder newCodedBuilder(int size) {
|
| + return new CodedBuilder(size);
|
| + }
|
| +
|
| + /** See {@link ByteString#newCodedBuilder(int)}. */
|
| + static final class CodedBuilder {
|
| + private final CodedOutputStream output;
|
| + private final byte[] buffer;
|
| +
|
| + private CodedBuilder(int size) {
|
| + buffer = new byte[size];
|
| + output = CodedOutputStream.newInstance(buffer);
|
| + }
|
| +
|
| + public ByteString build() {
|
| + output.checkNoSpaceLeft();
|
| +
|
| + // We can be confident that the CodedOutputStream will not modify the
|
| + // underlying bytes anymore because it already wrote all of them. So,
|
| + // no need to make a copy.
|
| + return new LiteralByteString(buffer);
|
| + }
|
| +
|
| + public CodedOutputStream getCodedOutput() {
|
| + return output;
|
| + }
|
| + }
|
| +
|
| + // =================================================================
|
| + // Methods {@link RopeByteString} needs on instances, which aren't part of the
|
| + // public API.
|
| +
|
| + /**
|
| + * Return the depth of the tree representing this {@code ByteString}, if any,
|
| + * whose root is this node. If this is a leaf node, return 0.
|
| + *
|
| + * @return tree depth or zero
|
| + */
|
| + protected abstract int getTreeDepth();
|
| +
|
| + /**
|
| + * Return {@code true} if this ByteString is literal (a leaf node) or a
|
| + * flat-enough tree in the sense of {@link RopeByteString}.
|
| + *
|
| + * @return true if the tree is flat enough
|
| + */
|
| + protected abstract boolean isBalanced();
|
| +
|
| + /**
|
| + * Return the cached hash code if available.
|
| + *
|
| + * @return value of cached hash code or 0 if not computed yet
|
| + */
|
| + protected final int peekCachedHashCode() {
|
| + return hash;
|
| + }
|
| +
|
| + /**
|
| + * Compute the hash across the value bytes starting with the given hash, and
|
| + * return the result. This is used to compute the hash across strings
|
| + * represented as a set of pieces by allowing the hash computation to be
|
| + * continued from piece to piece.
|
| + *
|
| + * @param h starting hash value
|
| + * @param offset offset into this value to start looking at data values
|
| + * @param length number of data values to include in the hash computation
|
| + * @return ending hash value
|
| + */
|
| + protected abstract int partialHash(int h, int offset, int length);
|
| +
|
| + /**
|
| + * Checks that the given index falls within the specified array size.
|
| + *
|
| + * @param index the index position to be tested
|
| + * @param size the length of the array
|
| + * @throws ArrayIndexOutOfBoundsException if the index does not fall within the array.
|
| + */
|
| + static void checkIndex(int index, int size) {
|
| + if ((index | (size - (index + 1))) < 0) {
|
| + if (index < 0) {
|
| + throw new ArrayIndexOutOfBoundsException("Index < 0: " + index);
|
| + }
|
| + throw new ArrayIndexOutOfBoundsException("Index > length: " + index + ", " + size);
|
| + }
|
| + }
|
| +
|
| + /**
|
| + * Checks that the given range falls within the bounds of an array
|
| + *
|
| + * @param startIndex the start index of the range (inclusive)
|
| + * @param endIndex the end index of the range (exclusive)
|
| + * @param size the size of the array.
|
| + * @return the length of the range.
|
| + * @throws ArrayIndexOutOfBoundsException some or all of the range falls outside of the array.
|
| + */
|
| + static int checkRange(int startIndex, int endIndex, int size) {
|
| + final int length = endIndex - startIndex;
|
| + if ((startIndex | endIndex | length | (size - endIndex)) < 0) {
|
| + if (startIndex < 0) {
|
| + throw new IndexOutOfBoundsException("Beginning index: " + startIndex + " < 0");
|
| + }
|
| + if (endIndex < startIndex) {
|
| + throw new IndexOutOfBoundsException(
|
| + "Beginning index larger than ending index: " + startIndex + ", " + endIndex);
|
| + }
|
| + // endIndex >= size
|
| + throw new IndexOutOfBoundsException("End index: " + endIndex + " >= " + size);
|
| + }
|
| + return length;
|
| + }
|
| +
|
| + @Override
|
| + public final String toString() {
|
| + return String.format("<ByteString@%s size=%d>",
|
| + Integer.toHexString(System.identityHashCode(this)), size());
|
| + }
|
| +}
|
|
|