7775460984
This supports more use cases. The ensureWritable method can now amortise its allocation cost by allocating more than what is strictly necessary to satisfy the immediate call. The bifurcate method can now split at a given offset.
1824 lines
60 KiB
Java
1824 lines
60 KiB
Java
/*
|
|
* Copyright 2020 The Netty Project
|
|
*
|
|
* The Netty Project licenses this file to you under the Apache License,
|
|
* version 2.0 (the "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at:
|
|
*
|
|
* https://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
* License for the specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
package io.netty.buffer.api;
|
|
|
|
import java.nio.ByteBuffer;
|
|
import java.nio.ByteOrder;
|
|
import java.util.Arrays;
|
|
import java.util.Collections;
|
|
import java.util.IdentityHashMap;
|
|
import java.util.Objects;
|
|
import java.util.Set;
|
|
import java.util.stream.Stream;
|
|
|
|
final class CompositeBuffer extends RcSupport<Buffer, CompositeBuffer> implements Buffer {
|
|
/**
|
|
* The max array size is JVM implementation dependant, but most seem to settle on {@code Integer.MAX_VALUE - 8}.
|
|
* We set the max composite buffer capacity to the same, since it would otherwise be impossible to create a
|
|
* non-composite copy of the buffer.
|
|
*/
|
|
private static final int MAX_CAPACITY = Integer.MAX_VALUE - 8;
|
|
private static final Drop<CompositeBuffer> COMPOSITE_DROP = new Drop<CompositeBuffer>() {
|
|
@Override
|
|
public void drop(CompositeBuffer buf) {
|
|
for (Buffer b : buf.bufs) {
|
|
b.close();
|
|
}
|
|
buf.makeInaccessible();
|
|
}
|
|
|
|
@Override
|
|
public String toString() {
|
|
return "COMPOSITE_DROP";
|
|
}
|
|
};
|
|
|
|
private final BufferAllocator allocator;
|
|
private final TornBufferAccessors tornBufAccessors;
|
|
private Buffer[] bufs;
|
|
private int[] offsets; // The offset, for the composite buffer, where each constituent buffer starts.
|
|
private int capacity;
|
|
private int roff;
|
|
private int woff;
|
|
private int subOffset; // The next offset *within* a consituent buffer to read from or write to.
|
|
private ByteOrder order;
|
|
private boolean closed;
|
|
private boolean readOnly;
|
|
|
|
CompositeBuffer(BufferAllocator allocator, Deref<Buffer>[] refs) {
|
|
this(allocator, filterExternalBufs(refs), COMPOSITE_DROP, false);
|
|
}
|
|
|
|
private static Buffer[] filterExternalBufs(Deref<Buffer>[] refs) {
|
|
// We filter out all zero-capacity buffers because they wouldn't contribute to the composite buffer anyway,
|
|
// and also, by ensuring that all constituent buffers contribute to the size of the composite buffer,
|
|
// we make sure that the number of composite buffers will never become greater than the number of bytes in
|
|
// the composite buffer.
|
|
// This restriction guarantees that methods like countComponents, forEachReadable and forEachWritable,
|
|
// will never overflow their component counts.
|
|
// Allocating a new array unconditionally also prevents external modification of the array.
|
|
Buffer[] bufs = Arrays.stream(refs)
|
|
.map(r -> r.get()) // Increments reference counts.
|
|
.filter(CompositeBuffer::discardEmpty)
|
|
.flatMap(CompositeBuffer::flattenBuffer)
|
|
.toArray(Buffer[]::new);
|
|
// Make sure there are no duplicates among the buffers.
|
|
Set<Buffer> duplicatesCheck = Collections.newSetFromMap(new IdentityHashMap<>());
|
|
duplicatesCheck.addAll(Arrays.asList(bufs));
|
|
if (duplicatesCheck.size() < bufs.length) {
|
|
for (Buffer buf : bufs) {
|
|
buf.close(); // Undo the increment we did with Deref.get().
|
|
}
|
|
throw new IllegalArgumentException(
|
|
"Cannot create composite buffer with duplicate constituent buffer components.");
|
|
}
|
|
return bufs;
|
|
}
|
|
|
|
private static boolean discardEmpty(Buffer buf) {
|
|
if (buf.capacity() > 0) {
|
|
return true;
|
|
} else {
|
|
// If we filter a buffer out, then we must make sure to close it since we incremented the reference count
|
|
// with Deref.get() earlier.
|
|
buf.close();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
private static Stream<Buffer> flattenBuffer(Buffer buf) {
|
|
if (buf instanceof CompositeBuffer) {
|
|
// Extract components and move our reference count from the composite onto the components.
|
|
var composite = (CompositeBuffer) buf;
|
|
var bufs = composite.bufs;
|
|
for (Buffer b : bufs) {
|
|
b.acquire();
|
|
}
|
|
buf.close(); // Important: acquire on components *before* closing composite.
|
|
return Stream.of(bufs);
|
|
}
|
|
return Stream.of(buf);
|
|
}
|
|
|
|
private CompositeBuffer(BufferAllocator allocator, Buffer[] bufs, Drop<CompositeBuffer> drop,
|
|
boolean acquireBufs) {
|
|
super(drop);
|
|
this.allocator = allocator;
|
|
if (acquireBufs) {
|
|
for (Buffer buf : bufs) {
|
|
buf.acquire();
|
|
}
|
|
}
|
|
try {
|
|
if (bufs.length > 0) {
|
|
ByteOrder targetOrder = bufs[0].order();
|
|
for (Buffer buf : bufs) {
|
|
if (buf.order() != targetOrder) {
|
|
throw new IllegalArgumentException("Constituent buffers have inconsistent byte order.");
|
|
}
|
|
}
|
|
order = bufs[0].order();
|
|
|
|
boolean targetReadOnly = bufs[0].readOnly();
|
|
for (Buffer buf : bufs) {
|
|
if (buf.readOnly() != targetReadOnly) {
|
|
throw new IllegalArgumentException("Constituent buffers have inconsistent read-only state.");
|
|
}
|
|
}
|
|
readOnly = targetReadOnly;
|
|
} else {
|
|
order = ByteOrder.nativeOrder();
|
|
}
|
|
this.bufs = bufs;
|
|
computeBufferOffsets();
|
|
tornBufAccessors = new TornBufferAccessors(this);
|
|
} catch (Exception e) {
|
|
// Always close bufs on exception, regardless of acquireBufs value.
|
|
// If acquireBufs is false, it just means the ref count increments happened prior to this constructor call.
|
|
for (Buffer buf : bufs) {
|
|
buf.close();
|
|
}
|
|
throw e;
|
|
}
|
|
}
|
|
|
|
private void computeBufferOffsets() {
|
|
if (bufs.length > 0) {
|
|
int woff = 0;
|
|
int roff = 0;
|
|
boolean woffMidpoint = false;
|
|
for (Buffer buf : bufs) {
|
|
if (buf.writableBytes() == 0) {
|
|
woff += buf.capacity();
|
|
} else if (!woffMidpoint) {
|
|
woff += buf.writerOffset();
|
|
woffMidpoint = true;
|
|
} else if (buf.writerOffset() != 0) {
|
|
throw new IllegalArgumentException(
|
|
"The given buffers cannot be composed because they leave an unwritten gap: " +
|
|
Arrays.toString(bufs) + '.');
|
|
}
|
|
}
|
|
boolean roffMidpoint = false;
|
|
for (Buffer buf : bufs) {
|
|
if (buf.readableBytes() == 0 && buf.writableBytes() == 0) {
|
|
roff += buf.capacity();
|
|
} else if (!roffMidpoint) {
|
|
roff += buf.readerOffset();
|
|
roffMidpoint = true;
|
|
} else if (buf.readerOffset() != 0) {
|
|
throw new IllegalArgumentException(
|
|
"The given buffers cannot be composed because they leave an unread gap: " +
|
|
Arrays.toString(bufs) + '.');
|
|
}
|
|
}
|
|
assert roff <= woff:
|
|
"The given buffers place the read offset ahead of the write offset: " + Arrays.toString(bufs) + '.';
|
|
// Commit computed offsets.
|
|
this.woff = woff;
|
|
this.roff = roff;
|
|
}
|
|
|
|
offsets = new int[bufs.length];
|
|
long cap = 0;
|
|
for (int i = 0; i < bufs.length; i++) {
|
|
offsets[i] = (int) cap;
|
|
cap += bufs[i].capacity();
|
|
}
|
|
if (cap > MAX_CAPACITY) {
|
|
throw new IllegalArgumentException(
|
|
"Combined size of the constituent buffers is too big. " +
|
|
"The maximum buffer capacity is " + MAX_CAPACITY + " (Interger.MAX_VALUE - 8), " +
|
|
"but the sum of the constituent buffer capacities was " + cap + '.');
|
|
}
|
|
capacity = (int) cap;
|
|
}
|
|
|
|
@Override
|
|
public String toString() {
|
|
return "Buffer[roff:" + roff + ", woff:" + woff + ", cap:" + capacity + ']';
|
|
}
|
|
|
|
@Override
|
|
public Buffer order(ByteOrder order) {
|
|
if (this.order != order) {
|
|
this.order = order;
|
|
for (Buffer buf : bufs) {
|
|
buf.order(order);
|
|
}
|
|
}
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public ByteOrder order() {
|
|
return order;
|
|
}
|
|
|
|
@Override
|
|
public int capacity() {
|
|
return capacity;
|
|
}
|
|
|
|
@Override
|
|
public int readerOffset() {
|
|
return roff;
|
|
}
|
|
|
|
@Override
|
|
public Buffer readerOffset(int index) {
|
|
prepRead(index, 0);
|
|
int indexLeft = index;
|
|
for (Buffer buf : bufs) {
|
|
buf.readerOffset(Math.min(indexLeft, buf.capacity()));
|
|
indexLeft = Math.max(0, indexLeft - buf.capacity());
|
|
}
|
|
roff = index;
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public int writerOffset() {
|
|
return woff;
|
|
}
|
|
|
|
@Override
|
|
public Buffer writerOffset(int index) {
|
|
checkWriteBounds(index, 0);
|
|
int indexLeft = index;
|
|
for (Buffer buf : bufs) {
|
|
buf.writerOffset(Math.min(indexLeft, buf.capacity()));
|
|
indexLeft = Math.max(0, indexLeft - buf.capacity());
|
|
}
|
|
woff = index;
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer fill(byte value) {
|
|
for (Buffer buf : bufs) {
|
|
buf.fill(value);
|
|
}
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public long nativeAddress() {
|
|
return 0;
|
|
}
|
|
|
|
@Override
|
|
public Buffer readOnly(boolean readOnly) {
|
|
for (Buffer buf : bufs) {
|
|
buf.readOnly(readOnly);
|
|
}
|
|
this.readOnly = readOnly;
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public boolean readOnly() {
|
|
return readOnly;
|
|
}
|
|
|
|
@Override
|
|
public Buffer slice(int offset, int length) {
|
|
checkWriteBounds(offset, length);
|
|
if (offset < 0 || length < 0) {
|
|
throw new IllegalArgumentException(
|
|
"Offset and length cannot be negative, but offset was " +
|
|
offset + ", and length was " + length + '.');
|
|
}
|
|
Buffer choice = (Buffer) chooseBuffer(offset, 0);
|
|
Buffer[] slices;
|
|
|
|
if (length > 0) {
|
|
slices = new Buffer[bufs.length];
|
|
int off = subOffset;
|
|
int cap = length;
|
|
int i;
|
|
for (i = searchOffsets(offset); cap > 0; i++) {
|
|
var buf = bufs[i];
|
|
int avail = buf.capacity() - off;
|
|
slices[i] = buf.slice(off, Math.min(cap, avail));
|
|
cap -= avail;
|
|
off = 0;
|
|
}
|
|
slices = Arrays.copyOf(slices, i);
|
|
} else {
|
|
// Specialize for length == 0, since we must slice from at least one constituent buffer.
|
|
slices = new Buffer[] { choice.slice(subOffset, 0) };
|
|
}
|
|
|
|
// Use the constructor that skips filtering out empty buffers, and skips acquiring on the buffers.
|
|
// This is important because 1) slice() already acquired the buffers, and 2) if this slice is empty
|
|
// then we need to keep holding on to it to prevent this originating composite buffer from getting
|
|
// ownership. If it did, its behaviour would be inconsistent with that of a non-composite buffer.
|
|
return new CompositeBuffer(allocator, slices, COMPOSITE_DROP, false);
|
|
}
|
|
|
|
@Override
|
|
public void copyInto(int srcPos, byte[] dest, int destPos, int length) {
|
|
copyInto(srcPos, (b, s, d, l) -> b.copyInto(s, dest, d, l), destPos, length);
|
|
}
|
|
|
|
@Override
|
|
public void copyInto(int srcPos, ByteBuffer dest, int destPos, int length) {
|
|
copyInto(srcPos, (b, s, d, l) -> b.copyInto(s, dest, d, l), destPos, length);
|
|
}
|
|
|
|
private void copyInto(int srcPos, CopyInto dest, int destPos, int length) {
|
|
if (length < 0) {
|
|
throw new IndexOutOfBoundsException("Length cannot be negative: " + length + '.');
|
|
}
|
|
if (srcPos < 0) {
|
|
throw indexOutOfBounds(srcPos, false);
|
|
}
|
|
if (srcPos + length > capacity) {
|
|
throw indexOutOfBounds(srcPos + length, false);
|
|
}
|
|
while (length > 0) {
|
|
var buf = (Buffer) chooseBuffer(srcPos, 0);
|
|
int toCopy = Math.min(buf.capacity() - subOffset, length);
|
|
dest.copyInto(buf, subOffset, destPos, toCopy);
|
|
srcPos += toCopy;
|
|
destPos += toCopy;
|
|
length -= toCopy;
|
|
}
|
|
}
|
|
|
|
@FunctionalInterface
|
|
private interface CopyInto {
|
|
void copyInto(Buffer src, int srcPos, int destPos, int length);
|
|
}
|
|
|
|
@Override
|
|
public void copyInto(int srcPos, Buffer dest, int destPos, int length) {
|
|
if (length < 0) {
|
|
throw new IndexOutOfBoundsException("Length cannot be negative: " + length + '.');
|
|
}
|
|
if (srcPos < 0) {
|
|
throw indexOutOfBounds(srcPos, false);
|
|
}
|
|
if (srcPos + length > capacity) {
|
|
throw indexOutOfBounds(srcPos + length, false);
|
|
}
|
|
|
|
// Iterate in reverse to account for src and dest buffer overlap.
|
|
// todo optimise by delegating to constituent buffers.
|
|
var cursor = openReverseCursor(srcPos + length - 1, length);
|
|
ByteOrder prevOrder = dest.order();
|
|
// We read longs in BE, in reverse, so they need to be flipped for writing.
|
|
dest.order(ByteOrder.LITTLE_ENDIAN);
|
|
try {
|
|
while (cursor.readLong()) {
|
|
length -= Long.BYTES;
|
|
dest.setLong(destPos + length, cursor.getLong());
|
|
}
|
|
while (cursor.readByte()) {
|
|
dest.setByte(destPos + --length, cursor.getByte());
|
|
}
|
|
} finally {
|
|
dest.order(prevOrder);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public ByteCursor openCursor() {
|
|
return openCursor(readerOffset(), readableBytes());
|
|
}
|
|
|
|
@Override
|
|
public ByteCursor openCursor(int fromOffset, int length) {
|
|
if (fromOffset < 0) {
|
|
throw new IllegalArgumentException("The fromOffset cannot be negative: " + fromOffset + '.');
|
|
}
|
|
if (length < 0) {
|
|
throw new IllegalArgumentException("The length cannot be negative: " + length + '.');
|
|
}
|
|
if (capacity < fromOffset + length) {
|
|
throw new IllegalArgumentException("The fromOffset+length is beyond the end of the buffer: " +
|
|
"fromOffset=" + fromOffset + ", length=" + length + '.');
|
|
}
|
|
int startBufferIndex = searchOffsets(fromOffset);
|
|
int off = fromOffset - offsets[startBufferIndex];
|
|
Buffer startBuf = bufs[startBufferIndex];
|
|
ByteCursor startCursor = startBuf.openCursor(off, Math.min(startBuf.capacity() - off, length));
|
|
return new ByteCursor() {
|
|
int index = fromOffset;
|
|
final int end = fromOffset + length;
|
|
int bufferIndex = startBufferIndex;
|
|
int initOffset = startCursor.currentOffset();
|
|
ByteCursor cursor = startCursor;
|
|
long longValue = -1;
|
|
byte byteValue = -1;
|
|
|
|
@Override
|
|
public boolean readLong() {
|
|
if (cursor.readLong()) {
|
|
longValue = cursor.getLong();
|
|
return true;
|
|
}
|
|
if (bytesLeft() >= Long.BYTES) {
|
|
longValue = nextLongFromBytes();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private long nextLongFromBytes() {
|
|
if (cursor.bytesLeft() == 0) {
|
|
nextCursor();
|
|
if (cursor.readLong()) {
|
|
return cursor.getLong();
|
|
}
|
|
}
|
|
long val = 0;
|
|
for (int i = 0; i < 8; i++) {
|
|
readByte();
|
|
val <<= 8;
|
|
val |= getByte();
|
|
}
|
|
return val;
|
|
}
|
|
|
|
@Override
|
|
public long getLong() {
|
|
return longValue;
|
|
}
|
|
|
|
@Override
|
|
public boolean readByte() {
|
|
if (cursor.readByte()) {
|
|
byteValue = cursor.getByte();
|
|
return true;
|
|
}
|
|
if (bytesLeft() > 0) {
|
|
nextCursor();
|
|
cursor.readByte();
|
|
byteValue = cursor.getByte();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private void nextCursor() {
|
|
bufferIndex++;
|
|
Buffer nextBuf = bufs[bufferIndex];
|
|
cursor = nextBuf.openCursor(0, Math.min(nextBuf.capacity(), bytesLeft()));
|
|
initOffset = 0;
|
|
}
|
|
|
|
@Override
|
|
public byte getByte() {
|
|
return byteValue;
|
|
}
|
|
|
|
@Override
|
|
public int currentOffset() {
|
|
int currOff = cursor.currentOffset();
|
|
index += currOff - initOffset;
|
|
initOffset = currOff;
|
|
return index;
|
|
}
|
|
|
|
@Override
|
|
public int bytesLeft() {
|
|
return end - currentOffset();
|
|
}
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public ByteCursor openReverseCursor(int fromOffset, int length) {
|
|
if (fromOffset < 0) {
|
|
throw new IllegalArgumentException("The fromOffset cannot be negative: " + fromOffset + '.');
|
|
}
|
|
if (length < 0) {
|
|
throw new IllegalArgumentException("The length cannot be negative: " + length + '.');
|
|
}
|
|
if (fromOffset - length < -1) {
|
|
throw new IllegalArgumentException("The fromOffset-length would underflow the buffer: " +
|
|
"fromOffset=" + fromOffset + ", length=" + length + '.');
|
|
}
|
|
int startBufferIndex = searchOffsets(fromOffset);
|
|
int off = fromOffset - offsets[startBufferIndex];
|
|
Buffer startBuf = bufs[startBufferIndex];
|
|
ByteCursor startCursor = startBuf.openReverseCursor(off, Math.min(off + 1, length));
|
|
return new ByteCursor() {
|
|
int index = fromOffset;
|
|
final int end = fromOffset - length;
|
|
int bufferIndex = startBufferIndex;
|
|
int initOffset = startCursor.currentOffset();
|
|
ByteCursor cursor = startCursor;
|
|
long longValue = -1;
|
|
byte byteValue = -1;
|
|
|
|
@Override
|
|
public boolean readLong() {
|
|
if (cursor.readLong()) {
|
|
longValue = cursor.getLong();
|
|
return true;
|
|
}
|
|
if (bytesLeft() >= Long.BYTES) {
|
|
longValue = nextLongFromBytes();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private long nextLongFromBytes() {
|
|
if (cursor.bytesLeft() == 0) {
|
|
nextCursor();
|
|
if (cursor.readLong()) {
|
|
return cursor.getLong();
|
|
}
|
|
}
|
|
long val = 0;
|
|
for (int i = 0; i < 8; i++) {
|
|
readByte();
|
|
val <<= 8;
|
|
val |= getByte();
|
|
}
|
|
return val;
|
|
}
|
|
|
|
@Override
|
|
public long getLong() {
|
|
return longValue;
|
|
}
|
|
|
|
@Override
|
|
public boolean readByte() {
|
|
if (cursor.readByte()) {
|
|
byteValue = cursor.getByte();
|
|
return true;
|
|
}
|
|
if (bytesLeft() > 0) {
|
|
nextCursor();
|
|
cursor.readByte();
|
|
byteValue = cursor.getByte();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private void nextCursor() {
|
|
bufferIndex--;
|
|
Buffer nextBuf = bufs[bufferIndex];
|
|
int length = Math.min(nextBuf.capacity(), bytesLeft());
|
|
int offset = nextBuf.capacity() - 1;
|
|
cursor = nextBuf.openReverseCursor(offset, length);
|
|
initOffset = offset;
|
|
}
|
|
|
|
@Override
|
|
public byte getByte() {
|
|
return byteValue;
|
|
}
|
|
|
|
@Override
|
|
public int currentOffset() {
|
|
int currOff = cursor.currentOffset();
|
|
index -= initOffset - currOff;
|
|
initOffset = currOff;
|
|
return index;
|
|
}
|
|
|
|
@Override
|
|
public int bytesLeft() {
|
|
return currentOffset() - end;
|
|
}
|
|
};
|
|
}
|
|
|
|
@Override
|
|
public void ensureWritable(int size, int minimumGrowth, boolean allowCompaction) {
|
|
if (!isOwned()) {
|
|
throw new IllegalStateException("Buffer is not owned. Only owned buffers can call ensureWritable.");
|
|
}
|
|
if (size < 0) {
|
|
throw new IllegalArgumentException("Cannot ensure writable for a negative size: " + size + '.');
|
|
}
|
|
if (minimumGrowth < 0) {
|
|
throw new IllegalArgumentException("The minimum growth cannot be negative: " + minimumGrowth + '.');
|
|
}
|
|
if (readOnly) {
|
|
throw bufferIsReadOnly();
|
|
}
|
|
if (writableBytes() >= size) {
|
|
// We already have enough space.
|
|
return;
|
|
}
|
|
|
|
if (allowCompaction && size <= roff) {
|
|
// Let's see if we can solve some or all of the requested size with compaction.
|
|
// We always compact as much as is possible, regardless of size. This amortizes our work.
|
|
int compactableBuffers = 0;
|
|
for (Buffer buf : bufs) {
|
|
if (buf.capacity() != buf.readerOffset()) {
|
|
break;
|
|
}
|
|
compactableBuffers++;
|
|
}
|
|
if (compactableBuffers > 0) {
|
|
Buffer[] compactable;
|
|
if (compactableBuffers < bufs.length) {
|
|
compactable = new Buffer[compactableBuffers];
|
|
System.arraycopy(bufs, 0, compactable, 0, compactable.length);
|
|
System.arraycopy(bufs, compactable.length, bufs, 0, bufs.length - compactable.length);
|
|
System.arraycopy(compactable, 0, bufs, bufs.length - compactable.length, compactable.length);
|
|
} else {
|
|
compactable = bufs;
|
|
}
|
|
for (Buffer buf : compactable) {
|
|
buf.reset();
|
|
}
|
|
computeBufferOffsets();
|
|
if (writableBytes() >= size) {
|
|
// Now we have enough space.
|
|
return;
|
|
}
|
|
} else if (bufs.length == 1) {
|
|
// If we only have a single component buffer, then we can safely compact that in-place.
|
|
bufs[0].compact();
|
|
computeBufferOffsets();
|
|
if (writableBytes() >= size) {
|
|
// Now we have enough space.
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
int growth = Math.max(size - writableBytes(), minimumGrowth);
|
|
BufferAllocator.checkSize(capacity() + (long) growth);
|
|
Buffer extension = allocator.allocate(growth, order());
|
|
unsafeExtendWith(extension);
|
|
}
|
|
|
|
void extendWith(Buffer extension) {
|
|
Objects.requireNonNull(extension, "Extension buffer cannot be null.");
|
|
if (!isOwned()) {
|
|
throw new IllegalStateException("This buffer cannot be extended because it is not in an owned state.");
|
|
}
|
|
if (bufs.length > 0 && extension.order() != order()) {
|
|
throw new IllegalArgumentException(
|
|
"This buffer uses " + order() + " byte order, and cannot be extended with " +
|
|
"a buffer that uses " + extension.order() + " byte order.");
|
|
}
|
|
if (bufs.length > 0 && extension.readOnly() != readOnly()) {
|
|
throw new IllegalArgumentException(
|
|
"This buffer is " + (readOnly? "read-only" : "writable") + ", " +
|
|
"and cannot be extended with a buffer that is " +
|
|
(extension.readOnly()? "read-only." : "writable."));
|
|
}
|
|
|
|
long extensionCapacity = extension.capacity();
|
|
if (extensionCapacity == 0) {
|
|
// Extending by a zero-sized buffer makes no difference. Especially since it's not allowed to change the
|
|
// capacity of buffers that are constiuents of composite buffers.
|
|
// This also ensures that methods like countComponents, and forEachReadable, do not have to worry about
|
|
// overflow in their component counters.
|
|
return;
|
|
}
|
|
|
|
long newSize = capacity() + extensionCapacity;
|
|
BufferAllocator.checkSize(newSize);
|
|
|
|
Buffer[] restoreTemp = bufs; // We need this to restore our buffer array, in case offset computations fail.
|
|
try {
|
|
if (extension instanceof CompositeBuffer) {
|
|
// If the extension is itself a composite buffer, then extend this one by all of the constituent
|
|
// component buffers.
|
|
CompositeBuffer compositeExtension = (CompositeBuffer) extension;
|
|
Buffer[] addedBuffers = compositeExtension.bufs;
|
|
Set<Buffer> duplicatesCheck = Collections.newSetFromMap(new IdentityHashMap<>());
|
|
duplicatesCheck.addAll(Arrays.asList(bufs));
|
|
duplicatesCheck.addAll(Arrays.asList(addedBuffers));
|
|
if (duplicatesCheck.size() < bufs.length + addedBuffers.length) {
|
|
throw extensionDuplicatesException();
|
|
}
|
|
for (Buffer addedBuffer : addedBuffers) {
|
|
addedBuffer.acquire();
|
|
}
|
|
int extendAtIndex = bufs.length;
|
|
bufs = Arrays.copyOf(bufs, extendAtIndex + addedBuffers.length);
|
|
System.arraycopy(addedBuffers, 0, bufs, extendAtIndex, addedBuffers.length);
|
|
computeBufferOffsets();
|
|
} else {
|
|
for (Buffer buf : restoreTemp) {
|
|
if (buf == extension) {
|
|
throw extensionDuplicatesException();
|
|
}
|
|
}
|
|
unsafeExtendWith(extension.acquire());
|
|
}
|
|
if (restoreTemp.length == 0) {
|
|
order = extension.order();
|
|
readOnly = extension.readOnly();
|
|
}
|
|
} catch (Exception e) {
|
|
bufs = restoreTemp;
|
|
throw e;
|
|
}
|
|
}
|
|
|
|
private static IllegalArgumentException extensionDuplicatesException() {
|
|
return new IllegalArgumentException(
|
|
"The composite buffer cannot be extended with the given extension," +
|
|
" as it would cause the buffer to have duplicate constituent buffers.");
|
|
}
|
|
|
|
private void unsafeExtendWith(Buffer extension) {
|
|
bufs = Arrays.copyOf(bufs, bufs.length + 1);
|
|
bufs[bufs.length - 1] = extension;
|
|
computeBufferOffsets();
|
|
}
|
|
|
|
@Override
|
|
public Buffer bifurcate(int splitOffset) {
|
|
if (splitOffset < 0) {
|
|
throw new IllegalArgumentException("The split offset cannot be negative: " + splitOffset + '.');
|
|
}
|
|
if (capacity() < splitOffset) {
|
|
throw new IllegalArgumentException("The split offset cannot be greater than the buffer capacity, " +
|
|
"but the split offset was " + splitOffset + ", and capacity is " + capacity() + '.');
|
|
}
|
|
if (!isOwned()) {
|
|
throw new IllegalStateException("Cannot bifurcate a buffer that is not owned.");
|
|
}
|
|
if (bufs.length == 0) {
|
|
// Bifurcating a zero-length buffer is trivial.
|
|
return new CompositeBuffer(allocator, bufs, unsafeGetDrop(), true).order(order);
|
|
}
|
|
|
|
int i = searchOffsets(splitOffset);
|
|
int off = splitOffset - offsets[i];
|
|
Buffer[] bifs = Arrays.copyOf(bufs, off == 0? i : 1 + i);
|
|
bufs = Arrays.copyOfRange(bufs, off == bufs[i].capacity()? 1 + i : i, bufs.length);
|
|
if (off > 0 && bifs.length > 0 && off < bifs[bifs.length - 1].capacity()) {
|
|
bifs[bifs.length - 1] = bufs[0].bifurcate(off);
|
|
}
|
|
computeBufferOffsets();
|
|
try {
|
|
var compositeBuf = new CompositeBuffer(allocator, bifs, unsafeGetDrop(), true);
|
|
compositeBuf.order = order; // Preserve byte order even if bifs array is empty.
|
|
return compositeBuf;
|
|
} finally {
|
|
// Drop our references to the buffers in the bifs array. They belong to the new composite buffer now.
|
|
for (Buffer bif : bifs) {
|
|
bif.close();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void compact() {
|
|
if (!isOwned()) {
|
|
throw new IllegalStateException("Buffer must be owned in order to compact.");
|
|
}
|
|
if (readOnly()) {
|
|
throw new IllegalStateException("Buffer must be writable in order to compact, but was read-only.");
|
|
}
|
|
int distance = roff;
|
|
if (distance == 0) {
|
|
return;
|
|
}
|
|
int pos = 0;
|
|
var oldOrder = order;
|
|
order = ByteOrder.BIG_ENDIAN;
|
|
try {
|
|
var cursor = openCursor();
|
|
while (cursor.readLong()) {
|
|
setLong(pos, cursor.getLong());
|
|
pos += Long.BYTES;
|
|
}
|
|
while (cursor.readByte()) {
|
|
setByte(pos, cursor.getByte());
|
|
pos++;
|
|
}
|
|
} finally {
|
|
order = oldOrder;
|
|
}
|
|
readerOffset(0);
|
|
writerOffset(woff - distance);
|
|
}
|
|
|
|
@Override
|
|
public int countComponents() {
|
|
int sum = 0;
|
|
for (Buffer buf : bufs) {
|
|
sum += buf.countComponents();
|
|
}
|
|
return sum;
|
|
}
|
|
|
|
@Override
|
|
public int countReadableComponents() {
|
|
int sum = 0;
|
|
for (Buffer buf : bufs) {
|
|
sum += buf.countReadableComponents();
|
|
}
|
|
return sum;
|
|
}
|
|
|
|
@Override
|
|
public int countWritableComponents() {
|
|
int sum = 0;
|
|
for (Buffer buf : bufs) {
|
|
sum += buf.countWritableComponents();
|
|
}
|
|
return sum;
|
|
}
|
|
|
|
@Override
|
|
public <E extends Exception> int forEachReadable(int initialIndex, ReadableComponentProcessor<E> processor)
|
|
throws E {
|
|
checkReadBounds(readerOffset(), Math.max(1, readableBytes()));
|
|
int visited = 0;
|
|
for (Buffer buf : bufs) {
|
|
if (buf.readableBytes() > 0) {
|
|
int count = buf.forEachReadable(visited + initialIndex, processor);
|
|
if (count > 0) {
|
|
visited += count;
|
|
} else {
|
|
visited = -visited + count;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return visited;
|
|
}
|
|
|
|
@Override
|
|
public <E extends Exception> int forEachWritable(int initialIndex, WritableComponentProcessor<E> processor)
|
|
throws E {
|
|
checkWriteBounds(writerOffset(), Math.max(1, writableBytes()));
|
|
int visited = 0;
|
|
for (Buffer buf : bufs) {
|
|
if (buf.writableBytes() > 0) {
|
|
int count = buf.forEachWritable(visited + initialIndex, processor);
|
|
if (count > 0) {
|
|
visited += count;
|
|
} else {
|
|
visited = -visited + count;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return visited;
|
|
}
|
|
|
|
// <editor-fold defaultstate="collapsed" desc="Primitive accessors.">
|
|
@Override
|
|
public byte readByte() {
|
|
return prepRead(Byte.BYTES).readByte();
|
|
}
|
|
|
|
@Override
|
|
public byte getByte(int roff) {
|
|
return prepGet(roff, Byte.BYTES).getByte(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public int readUnsignedByte() {
|
|
return prepRead(Byte.BYTES).readUnsignedByte();
|
|
}
|
|
|
|
@Override
|
|
public int getUnsignedByte(int roff) {
|
|
return prepGet(roff, Byte.BYTES).getUnsignedByte(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeByte(byte value) {
|
|
prepWrite(Byte.BYTES).writeByte(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setByte(int woff, byte value) {
|
|
prepWrite(woff, Byte.BYTES).setByte(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeUnsignedByte(int value) {
|
|
prepWrite(Byte.BYTES).writeUnsignedByte(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setUnsignedByte(int woff, int value) {
|
|
prepWrite(woff, Byte.BYTES).setUnsignedByte(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public char readChar() {
|
|
return prepRead(2).readChar();
|
|
}
|
|
|
|
@Override
|
|
public char getChar(int roff) {
|
|
return prepGet(roff, 2).getChar(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeChar(char value) {
|
|
prepWrite(2).writeChar(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setChar(int woff, char value) {
|
|
prepWrite(woff, 2).setChar(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public short readShort() {
|
|
return prepRead(Short.BYTES).readShort();
|
|
}
|
|
|
|
@Override
|
|
public short getShort(int roff) {
|
|
return prepGet(roff, Short.BYTES).getShort(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public int readUnsignedShort() {
|
|
return prepRead(Short.BYTES).readShort();
|
|
}
|
|
|
|
@Override
|
|
public int getUnsignedShort(int roff) {
|
|
return prepGet(roff, Short.BYTES).getUnsignedShort(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeShort(short value) {
|
|
prepWrite(Short.BYTES).writeShort(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setShort(int woff, short value) {
|
|
prepWrite(woff, Short.BYTES).setShort(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeUnsignedShort(int value) {
|
|
prepWrite(Short.BYTES).writeUnsignedShort(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setUnsignedShort(int woff, int value) {
|
|
prepWrite(woff, Short.BYTES).setUnsignedShort(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public int readMedium() {
|
|
return prepRead(3).readMedium();
|
|
}
|
|
|
|
@Override
|
|
public int getMedium(int roff) {
|
|
return prepGet(roff, 3).getMedium(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public int readUnsignedMedium() {
|
|
return prepRead(3).readMedium();
|
|
}
|
|
|
|
@Override
|
|
public int getUnsignedMedium(int roff) {
|
|
return prepGet(roff, 3).getMedium(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeMedium(int value) {
|
|
prepWrite(3).writeMedium(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setMedium(int woff, int value) {
|
|
prepWrite(woff, 3).setMedium(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeUnsignedMedium(int value) {
|
|
prepWrite(3).writeUnsignedMedium(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setUnsignedMedium(int woff, int value) {
|
|
prepWrite(woff, 3).setUnsignedMedium(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public int readInt() {
|
|
return prepRead(Integer.BYTES).readInt();
|
|
}
|
|
|
|
@Override
|
|
public int getInt(int roff) {
|
|
return prepGet(roff, Integer.BYTES).getInt(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public long readUnsignedInt() {
|
|
return prepRead(Integer.BYTES).readUnsignedInt();
|
|
}
|
|
|
|
@Override
|
|
public long getUnsignedInt(int roff) {
|
|
return prepGet(roff, Integer.BYTES).getUnsignedInt(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeInt(int value) {
|
|
prepWrite(Integer.BYTES).writeInt(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setInt(int woff, int value) {
|
|
prepWrite(woff, Integer.BYTES).setInt(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeUnsignedInt(long value) {
|
|
prepWrite(Integer.BYTES).writeUnsignedInt(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setUnsignedInt(int woff, long value) {
|
|
prepWrite(woff, Integer.BYTES).setUnsignedInt(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public float readFloat() {
|
|
return prepRead(Float.BYTES).readFloat();
|
|
}
|
|
|
|
@Override
|
|
public float getFloat(int roff) {
|
|
return prepGet(roff, Float.BYTES).getFloat(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeFloat(float value) {
|
|
prepWrite(Float.BYTES).writeFloat(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setFloat(int woff, float value) {
|
|
prepWrite(woff, Float.BYTES).setFloat(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public long readLong() {
|
|
return prepRead(Long.BYTES).readLong();
|
|
}
|
|
|
|
@Override
|
|
public long getLong(int roff) {
|
|
return prepGet(roff, Long.BYTES).getLong(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeLong(long value) {
|
|
prepWrite(Long.BYTES).writeLong(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setLong(int woff, long value) {
|
|
prepWrite(woff, Long.BYTES).setLong(subOffset, value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public double readDouble() {
|
|
return prepRead(Double.BYTES).readDouble();
|
|
}
|
|
|
|
@Override
|
|
public double getDouble(int roff) {
|
|
return prepGet(roff, Double.BYTES).getDouble(subOffset);
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeDouble(double value) {
|
|
prepWrite(Double.BYTES).writeDouble(value);
|
|
return this;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setDouble(int woff, double value) {
|
|
prepWrite(woff, Double.BYTES).setDouble(subOffset, value);
|
|
return this;
|
|
}
|
|
// </editor-fold>
|
|
|
|
@Override
|
|
protected Owned<CompositeBuffer> prepareSend() {
|
|
@SuppressWarnings("unchecked")
|
|
Send<Buffer>[] sends = new Send[bufs.length];
|
|
try {
|
|
for (int i = 0; i < bufs.length; i++) {
|
|
sends[i] = bufs[i].send();
|
|
}
|
|
} catch (Throwable throwable) {
|
|
// Repair our bufs array.
|
|
for (int i = 0; i < sends.length; i++) {
|
|
if (sends[i] != null) {
|
|
try {
|
|
bufs[i] = sends[i].receive();
|
|
} catch (Exception e) {
|
|
throwable.addSuppressed(e);
|
|
}
|
|
}
|
|
}
|
|
throw throwable;
|
|
}
|
|
boolean readOnly = this.readOnly;
|
|
makeInaccessible();
|
|
return new Owned<CompositeBuffer>() {
|
|
@Override
|
|
public CompositeBuffer transferOwnership(Drop<CompositeBuffer> drop) {
|
|
Buffer[] received = new Buffer[sends.length];
|
|
for (int i = 0; i < sends.length; i++) {
|
|
received[i] = sends[i].receive();
|
|
}
|
|
var composite = new CompositeBuffer(allocator, received, drop, true);
|
|
composite.readOnly = readOnly;
|
|
drop.attach(composite);
|
|
return composite;
|
|
}
|
|
};
|
|
}
|
|
|
|
void makeInaccessible() {
|
|
capacity = 0;
|
|
roff = 0;
|
|
woff = 0;
|
|
readOnly = false;
|
|
closed = true;
|
|
}
|
|
|
|
@Override
|
|
public boolean isOwned() {
|
|
return super.isOwned() && allConstituentsAreOwned();
|
|
}
|
|
|
|
private boolean allConstituentsAreOwned() {
|
|
boolean result = true;
|
|
for (Buffer buf : bufs) {
|
|
result &= buf.isOwned();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
long readPassThrough() {
|
|
var buf = choosePassThroughBuffer(subOffset++);
|
|
assert buf != tornBufAccessors: "Recursive call to torn buffer.";
|
|
return buf.readUnsignedByte();
|
|
}
|
|
|
|
void writePassThrough(int value) {
|
|
var buf = choosePassThroughBuffer(subOffset++);
|
|
assert buf != tornBufAccessors: "Recursive call to torn buffer.";
|
|
buf.writeUnsignedByte(value);
|
|
}
|
|
|
|
long getPassThrough(int roff) {
|
|
var buf = chooseBuffer(roff, 1);
|
|
assert buf != tornBufAccessors: "Recursive call to torn buffer.";
|
|
return buf.getUnsignedByte(subOffset);
|
|
}
|
|
|
|
void setPassThrough(int woff, int value) {
|
|
var buf = chooseBuffer(woff, 1);
|
|
assert buf != tornBufAccessors: "Recursive call to torn buffer.";
|
|
buf.setUnsignedByte(subOffset, value);
|
|
}
|
|
|
|
private BufferAccessors prepRead(int size) {
|
|
var buf = prepRead(roff, size);
|
|
roff += size;
|
|
return buf;
|
|
}
|
|
|
|
private BufferAccessors prepRead(int index, int size) {
|
|
checkReadBounds(index, size);
|
|
return chooseBuffer(index, size);
|
|
}
|
|
|
|
private void checkReadBounds(int index, int size) {
|
|
if (index < 0 || woff < index + size) {
|
|
throw indexOutOfBounds(index, false);
|
|
}
|
|
}
|
|
|
|
private BufferAccessors prepGet(int index, int size) {
|
|
checkGetBounds(index, size);
|
|
return chooseBuffer(index, size);
|
|
}
|
|
|
|
private void checkGetBounds(int index, int size) {
|
|
if (index < 0 || capacity < index + size) {
|
|
throw indexOutOfBounds(index, false);
|
|
}
|
|
}
|
|
|
|
private BufferAccessors prepWrite(int size) {
|
|
var buf = prepWrite(woff, size);
|
|
woff += size;
|
|
return buf;
|
|
}
|
|
|
|
private BufferAccessors prepWrite(int index, int size) {
|
|
checkWriteBounds(index, size);
|
|
return chooseBuffer(index, size);
|
|
}
|
|
|
|
private void checkWriteBounds(int index, int size) {
|
|
if (index < 0 || capacity < index + size) {
|
|
throw indexOutOfBounds(index, true);
|
|
}
|
|
}
|
|
|
|
private RuntimeException indexOutOfBounds(int index, boolean write) {
|
|
if (closed) {
|
|
return bufferIsClosed();
|
|
}
|
|
if (write && readOnly) {
|
|
return bufferIsReadOnly();
|
|
}
|
|
return new IndexOutOfBoundsException(
|
|
"Index " + index + " is out of bounds: [read 0 to " + woff + ", write 0 to " +
|
|
capacity + "].");
|
|
}
|
|
|
|
private static IllegalStateException bufferIsClosed() {
|
|
return new IllegalStateException("This buffer is closed.");
|
|
}
|
|
|
|
private static IllegalStateException bufferIsReadOnly() {
|
|
return new IllegalStateException("This buffer is read-only.");
|
|
}
|
|
|
|
private BufferAccessors chooseBuffer(int index, int size) {
|
|
int i = searchOffsets(index);
|
|
if (i == bufs.length) {
|
|
// This happens when the read/write offsets are parked 1 byte beyond the end of the buffer.
|
|
// In that case it should not matter what buffer is returned, because it shouldn't be used anyway.
|
|
return null;
|
|
}
|
|
int off = index - offsets[i];
|
|
Buffer candidate = bufs[i];
|
|
if (off + size <= candidate.capacity()) {
|
|
subOffset = off;
|
|
return candidate;
|
|
}
|
|
subOffset = index;
|
|
return tornBufAccessors;
|
|
}
|
|
|
|
private BufferAccessors choosePassThroughBuffer(int index) {
|
|
int i = searchOffsets(index);
|
|
return bufs[i];
|
|
}
|
|
|
|
private int searchOffsets(int index) {
|
|
int i = Arrays.binarySearch(offsets, index);
|
|
return i < 0? -(i + 2) : i;
|
|
}
|
|
|
|
// <editor-fold defaultstate="collapsed" desc="Torn buffer access.">
|
|
private static final class TornBufferAccessors implements BufferAccessors {
|
|
private final CompositeBuffer buf;
|
|
|
|
private TornBufferAccessors(CompositeBuffer buf) {
|
|
this.buf = buf;
|
|
}
|
|
|
|
@Override
|
|
public byte readByte() {
|
|
throw new AssertionError("Method should not be used.");
|
|
}
|
|
|
|
@Override
|
|
public byte getByte(int roff) {
|
|
throw new AssertionError("Method should not be used.");
|
|
}
|
|
|
|
@Override
|
|
public int readUnsignedByte() {
|
|
throw new AssertionError("Method should not be used.");
|
|
}
|
|
|
|
@Override
|
|
public int getUnsignedByte(int roff) {
|
|
throw new AssertionError("Method should not be used.");
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeByte(byte value) {
|
|
throw new AssertionError("Method should not be used.");
|
|
}
|
|
|
|
@Override
|
|
public Buffer setByte(int woff, byte value) {
|
|
throw new AssertionError("Method should not be used.");
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeUnsignedByte(int value) {
|
|
throw new AssertionError("Method should not be used.");
|
|
}
|
|
|
|
@Override
|
|
public Buffer setUnsignedByte(int woff, int value) {
|
|
throw new AssertionError("Method should not be used.");
|
|
}
|
|
|
|
@Override
|
|
public char readChar() {
|
|
if (bigEndian()) {
|
|
return (char) (read() << 8 | read());
|
|
} else {
|
|
return (char) (read() | read() << 8);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public char getChar(int roff) {
|
|
if (bigEndian()) {
|
|
return (char) (read(roff) << 8 | read(roff + 1));
|
|
} else {
|
|
return (char) (read(roff) | read(roff + 1) << 8);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeChar(char value) {
|
|
if (bigEndian()) {
|
|
write(value >>> 8);
|
|
write(value & 0xFF);
|
|
} else {
|
|
write(value & 0xFF);
|
|
write(value >>> 8);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setChar(int woff, char value) {
|
|
if (bigEndian()) {
|
|
write(woff, value >>> 8);
|
|
write(woff + 1, value & 0xFF);
|
|
} else {
|
|
write(woff, value & 0xFF);
|
|
write(woff + 1, value >>> 8);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public short readShort() {
|
|
if (bigEndian()) {
|
|
return (short) (read() << 8 | read());
|
|
} else {
|
|
return (short) (read() | read() << 8);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public short getShort(int roff) {
|
|
if (bigEndian()) {
|
|
return (short) (read(roff) << 8 | read(roff + 1));
|
|
} else {
|
|
return (short) (read(roff) | read(roff + 1) << 8);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public int readUnsignedShort() {
|
|
if (bigEndian()) {
|
|
return (int) (read() << 8 | read()) & 0xFFFF;
|
|
} else {
|
|
return (int) (read() | read() << 8) & 0xFFFF;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public int getUnsignedShort(int roff) {
|
|
if (bigEndian()) {
|
|
return (int) (read(roff) << 8 | read(roff + 1)) & 0xFFFF;
|
|
} else {
|
|
return (int) (read(roff) | read(roff + 1) << 8) & 0xFFFF;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeShort(short value) {
|
|
if (bigEndian()) {
|
|
write(value >>> 8);
|
|
write(value & 0xFF);
|
|
} else {
|
|
write(value & 0xFF);
|
|
write(value >>> 8);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setShort(int woff, short value) {
|
|
if (bigEndian()) {
|
|
write(woff, value >>> 8);
|
|
write(woff + 1, value & 0xFF);
|
|
} else {
|
|
write(woff, value & 0xFF);
|
|
write(woff + 1, value >>> 8);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeUnsignedShort(int value) {
|
|
if (bigEndian()) {
|
|
write(value >>> 8);
|
|
write(value & 0xFF);
|
|
} else {
|
|
write(value & 0xFF);
|
|
write(value >>> 8);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setUnsignedShort(int woff, int value) {
|
|
if (bigEndian()) {
|
|
write(woff, value >>> 8);
|
|
write(woff + 1, value & 0xFF);
|
|
} else {
|
|
write(woff, value & 0xFF);
|
|
write(woff + 1, value >>> 8);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public int readMedium() {
|
|
if (bigEndian()) {
|
|
return (int) (read() << 16 | read() << 8 | read());
|
|
} else {
|
|
return (int) (read() | read() << 8 | read() << 16);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public int getMedium(int roff) {
|
|
if (bigEndian()) {
|
|
return (int) (read(roff) << 16 | read(roff + 1) << 8 | read(roff + 2));
|
|
} else {
|
|
return (int) (read(roff) | read(roff + 1) << 8 | read(roff + 2) << 16);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public int readUnsignedMedium() {
|
|
if (bigEndian()) {
|
|
return (int) (read() << 16 | read() << 8 | read()) & 0xFFFFFF;
|
|
} else {
|
|
return (int) (read() | read() << 8 | read() << 16) & 0xFFFFFF;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public int getUnsignedMedium(int roff) {
|
|
if (bigEndian()) {
|
|
return (int) (read(roff) << 16 | read(roff + 1) << 8 | read(roff + 2)) & 0xFFFFFF;
|
|
} else {
|
|
return (int) (read(roff) | read(roff + 1) << 8 | read(roff + 2) << 16) & 0xFFFFFF;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeMedium(int value) {
|
|
if (bigEndian()) {
|
|
write(value >>> 16);
|
|
write(value >>> 8 & 0xFF);
|
|
write(value & 0xFF);
|
|
} else {
|
|
write(value & 0xFF);
|
|
write(value >>> 8 & 0xFF);
|
|
write(value >>> 16);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setMedium(int woff, int value) {
|
|
if (bigEndian()) {
|
|
write(woff, value >>> 16);
|
|
write(woff + 1, value >>> 8 & 0xFF);
|
|
write(woff + 2, value & 0xFF);
|
|
} else {
|
|
write(woff, value & 0xFF);
|
|
write(woff + 1, value >>> 8 & 0xFF);
|
|
write(woff + 2, value >>> 16);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeUnsignedMedium(int value) {
|
|
if (bigEndian()) {
|
|
write(value >>> 16);
|
|
write(value >>> 8 & 0xFF);
|
|
write(value & 0xFF);
|
|
} else {
|
|
write(value & 0xFF);
|
|
write(value >>> 8 & 0xFF);
|
|
write(value >>> 16);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setUnsignedMedium(int woff, int value) {
|
|
if (bigEndian()) {
|
|
write(woff, value >>> 16);
|
|
write(woff + 1, value >>> 8 & 0xFF);
|
|
write(woff + 2, value & 0xFF);
|
|
} else {
|
|
write(woff, value & 0xFF);
|
|
write(woff + 1, value >>> 8 & 0xFF);
|
|
write(woff + 2, value >>> 16);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public int readInt() {
|
|
if (bigEndian()) {
|
|
return (int) (read() << 24 | read() << 16 | read() << 8 | read());
|
|
} else {
|
|
return (int) (read() | read() << 8 | read() << 16 | read() << 24);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public int getInt(int roff) {
|
|
if (bigEndian()) {
|
|
return (int) (read(roff) << 24 | read(roff + 1) << 16 | read(roff + 2) << 8 | read(roff + 3));
|
|
} else {
|
|
return (int) (read(roff) | read(roff + 1) << 8 | read(roff + 2) << 16 | read(roff + 3) << 24);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public long readUnsignedInt() {
|
|
if (bigEndian()) {
|
|
return (read() << 24 | read() << 16 | read() << 8 | read()) & 0xFFFFFFFFL;
|
|
} else {
|
|
return (read() | read() << 8 | read() << 16 | read() << 24) & 0xFFFFFFFFL;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public long getUnsignedInt(int roff) {
|
|
if (bigEndian()) {
|
|
return (read(roff) << 24 | read(roff + 1) << 16 | read(roff + 2) << 8 | read(roff + 3)) & 0xFFFFFFFFL;
|
|
} else {
|
|
return (read(roff) | read(roff + 1) << 8 | read(roff + 2) << 16 | read(roff + 3) << 24) & 0xFFFFFFFFL;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeInt(int value) {
|
|
if (bigEndian()) {
|
|
write(value >>> 24);
|
|
write(value >>> 16 & 0xFF);
|
|
write(value >>> 8 & 0xFF);
|
|
write(value & 0xFF);
|
|
} else {
|
|
write(value & 0xFF);
|
|
write(value >>> 8 & 0xFF);
|
|
write(value >>> 16 & 0xFF);
|
|
write(value >>> 24);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setInt(int woff, int value) {
|
|
if (bigEndian()) {
|
|
write(woff, value >>> 24);
|
|
write(woff + 1, value >>> 16 & 0xFF);
|
|
write(woff + 2, value >>> 8 & 0xFF);
|
|
write(woff + 3, value & 0xFF);
|
|
} else {
|
|
write(woff, value & 0xFF);
|
|
write(woff + 1, value >>> 8 & 0xFF);
|
|
write(woff + 2, value >>> 16 & 0xFF);
|
|
write(woff + 3, value >>> 24);
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeUnsignedInt(long value) {
|
|
if (bigEndian()) {
|
|
write((int) (value >>> 24));
|
|
write((int) (value >>> 16 & 0xFF));
|
|
write((int) (value >>> 8 & 0xFF));
|
|
write((int) (value & 0xFF));
|
|
} else {
|
|
write((int) (value & 0xFF));
|
|
write((int) (value >>> 8 & 0xFF));
|
|
write((int) (value >>> 16 & 0xFF));
|
|
write((int) (value >>> 24));
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setUnsignedInt(int woff, long value) {
|
|
if (bigEndian()) {
|
|
write(woff, (int) (value >>> 24));
|
|
write(woff + 1, (int) (value >>> 16 & 0xFF));
|
|
write(woff + 2, (int) (value >>> 8 & 0xFF));
|
|
write(woff + 3, (int) (value & 0xFF));
|
|
} else {
|
|
write(woff, (int) (value & 0xFF));
|
|
write(woff + 1, (int) (value >>> 8 & 0xFF));
|
|
write(woff + 2, (int) (value >>> 16 & 0xFF));
|
|
write(woff + 3, (int) (value >>> 24));
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public float readFloat() {
|
|
return Float.intBitsToFloat(readInt());
|
|
}
|
|
|
|
@Override
|
|
public float getFloat(int roff) {
|
|
return Float.intBitsToFloat(getInt(roff));
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeFloat(float value) {
|
|
return writeUnsignedInt(Float.floatToRawIntBits(value));
|
|
}
|
|
|
|
@Override
|
|
public Buffer setFloat(int woff, float value) {
|
|
return setUnsignedInt(woff, Float.floatToRawIntBits(value));
|
|
}
|
|
|
|
@Override
|
|
public long readLong() {
|
|
if (bigEndian()) {
|
|
return read() << 56 | read() << 48 | read() << 40 | read() << 32 |
|
|
read() << 24 | read() << 16 | read() << 8 | read();
|
|
} else {
|
|
return read() | read() << 8 | read() << 16 | read() << 24 |
|
|
read() << 32 | read() << 40 | read() << 48 | read() << 56;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public long getLong(int roff) {
|
|
if (bigEndian()) {
|
|
return read(roff) << 56 | read(roff + 1) << 48 | read(roff + 2) << 40 | read(roff + 3) << 32 |
|
|
read(roff + 4) << 24 | read(roff + 5) << 16 | read(roff + 6) << 8 | read(roff + 7);
|
|
} else {
|
|
return read(roff) | read(roff + 1) << 8 | read(roff + 2) << 16 | read(roff + 3) << 24 |
|
|
read(roff + 4) << 32 | read(roff + 5) << 40 | read(roff + 6) << 48 | read(roff + 7) << 56;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeLong(long value) {
|
|
if (bigEndian()) {
|
|
write((int) (value >>> 56));
|
|
write((int) (value >>> 48 & 0xFF));
|
|
write((int) (value >>> 40 & 0xFF));
|
|
write((int) (value >>> 32 & 0xFF));
|
|
write((int) (value >>> 24 & 0xFF));
|
|
write((int) (value >>> 16 & 0xFF));
|
|
write((int) (value >>> 8 & 0xFF));
|
|
write((int) (value & 0xFF));
|
|
} else {
|
|
write((int) (value & 0xFF));
|
|
write((int) (value >>> 8 & 0xFF));
|
|
write((int) (value >>> 16 & 0xFF));
|
|
write((int) (value >>> 24));
|
|
write((int) (value >>> 32));
|
|
write((int) (value >>> 40));
|
|
write((int) (value >>> 48));
|
|
write((int) (value >>> 56));
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public Buffer setLong(int woff, long value) {
|
|
if (bigEndian()) {
|
|
write(woff, (int) (value >>> 56));
|
|
write(woff + 1, (int) (value >>> 48 & 0xFF));
|
|
write(woff + 2, (int) (value >>> 40 & 0xFF));
|
|
write(woff + 3, (int) (value >>> 32 & 0xFF));
|
|
write(woff + 4, (int) (value >>> 24 & 0xFF));
|
|
write(woff + 5, (int) (value >>> 16 & 0xFF));
|
|
write(woff + 6, (int) (value >>> 8 & 0xFF));
|
|
write(woff + 7, (int) (value & 0xFF));
|
|
} else {
|
|
write(woff, (int) (value & 0xFF));
|
|
write(woff + 1, (int) (value >>> 8 & 0xFF));
|
|
write(woff + 2, (int) (value >>> 16 & 0xFF));
|
|
write(woff + 3, (int) (value >>> 24));
|
|
write(woff + 4, (int) (value >>> 32));
|
|
write(woff + 5, (int) (value >>> 40));
|
|
write(woff + 6, (int) (value >>> 48));
|
|
write(woff + 7, (int) (value >>> 56));
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
@Override
|
|
public double readDouble() {
|
|
return Double.longBitsToDouble(readLong());
|
|
}
|
|
|
|
@Override
|
|
public double getDouble(int roff) {
|
|
return Double.longBitsToDouble(getLong(roff));
|
|
}
|
|
|
|
@Override
|
|
public Buffer writeDouble(double value) {
|
|
return writeLong(Double.doubleToRawLongBits(value));
|
|
}
|
|
|
|
@Override
|
|
public Buffer setDouble(int woff, double value) {
|
|
return setLong(woff, Double.doubleToRawLongBits(value));
|
|
}
|
|
|
|
private boolean bigEndian() {
|
|
return buf.order() == ByteOrder.BIG_ENDIAN;
|
|
}
|
|
|
|
private long read() {
|
|
return buf.readPassThrough();
|
|
}
|
|
|
|
private void write(int value) {
|
|
buf.writePassThrough(value);
|
|
}
|
|
|
|
private long read(int roff) {
|
|
return buf.getPassThrough(roff);
|
|
}
|
|
|
|
private void write(int woff, int value) {
|
|
buf.setPassThrough(woff, value);
|
|
}
|
|
}
|
|
// </editor-fold>
|
|
}
|