/* * Copyright (C) 2020 Nan1t * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ package ru.nanit.limbo.protocol; import io.netty.buffer.*; import io.netty.handler.codec.DecoderException; import io.netty.handler.codec.EncoderException; import io.netty.util.ByteProcessor; import net.kyori.adventure.nbt.BinaryTagIO; import net.kyori.adventure.nbt.CompoundBinaryTag; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.nio.ByteBuffer; import java.nio.ByteOrder; import java.nio.channels.FileChannel; import java.nio.channels.GatheringByteChannel; import java.nio.channels.ScatteringByteChannel; import java.nio.charset.Charset; import java.nio.charset.StandardCharsets; import java.util.UUID; public class ByteMessage extends ByteBuf { private final ByteBuf buf; public ByteMessage(ByteBuf buf) { this.buf = buf; } public byte[] toByteArray() { byte[] bytes = new byte[buf.readableBytes()]; buf.readBytes(bytes); return bytes; } /* Minecraft's protocol methods */ public int readVarInt() { int i = 0; int maxRead = Math.min(5, buf.readableBytes()); for (int j = 0; j < maxRead; j++) { int k = buf.readByte(); i |= (k & 0x7F) << j * 7; if ((k & 0x80) != 128) { return i; } } buf.readBytes(maxRead); throw new IllegalArgumentException("Cannot read VarInt"); } public void writeVarInt(int value) { while (true) { if ((value & 0xFFFFFF80) == 0) { buf.writeByte(value); return; } buf.writeByte(value & 0x7F | 0x80); value >>>= 7; } } public String readString() { return readString(readVarInt()); } public String readString(int length) { String str = buf.toString(buf.readerIndex(), length, StandardCharsets.UTF_8); buf.skipBytes(length); return str; } public void writeString(CharSequence str) { int size = ByteBufUtil.utf8Bytes(str); writeVarInt(size); buf.writeCharSequence(str, StandardCharsets.UTF_8); } public byte[] readBytesArray() { int length = readVarInt(); byte[] array = new byte[length]; buf.readBytes(array); return array; } public void writeBytesArray(byte[] array) { writeVarInt(array.length); buf.writeBytes(array); } public int[] readIntArray() { int len = readVarInt(); int[] array = new int[len]; for (int i = 0; i < len; i++) { array[i] = readVarInt(); } return array; } public UUID readUuid() { long msb = buf.readLong(); long lsb = buf.readLong(); return new UUID(msb, lsb); } public void writeUuid(UUID uuid) { buf.writeLong(uuid.getMostSignificantBits()); buf.writeLong(uuid.getLeastSignificantBits()); } public String[] readStringsArray() { int length = readVarInt(); String[] ret = new String[length]; for (int i = 0; i < length; i++) { ret[i] = readString(); } return ret; } public void writeStringsArray(String[] stringArray) { writeVarInt(stringArray.length); for (String str : stringArray) { writeString(str); } } public void writeVarIntArray(int[] array) { writeVarInt(array.length); for (int i : array) { writeVarInt(i); } } public void writeLongArray(long[] array) { writeVarInt(array.length); for (long i : array) { writeLong(i); } } public void writeCompoundTagArray(CompoundBinaryTag[] compoundTags) { try (ByteBufOutputStream stream = new ByteBufOutputStream(buf)) { writeVarInt(compoundTags.length); for (CompoundBinaryTag tag : compoundTags) { BinaryTagIO.writer().write(tag, (OutputStream) stream); } } catch (IOException e) { throw new EncoderException("Cannot write NBT CompoundTag"); } } public CompoundBinaryTag readCompoundTag() { try (ByteBufInputStream stream = new ByteBufInputStream(buf)) { return BinaryTagIO.reader().read((InputStream) stream); } catch (IOException thrown) { throw new DecoderException("Cannot read NBT CompoundTag"); } } public void writeCompoundTag(CompoundBinaryTag compoundTag) { try (ByteBufOutputStream stream = new ByteBufOutputStream(buf)) { BinaryTagIO.writer().write(compoundTag, (OutputStream) stream); } catch (IOException e) { throw new EncoderException("Cannot write NBT CompoundTag"); } } /* Delegated methods */ @Override public int capacity() { return buf.capacity(); } @Override public ByteBuf capacity(int newCapacity) { return buf.capacity(newCapacity); } @Override public int maxCapacity() { return buf.maxCapacity(); } @Override public ByteBufAllocator alloc() { return buf.alloc(); } @Override @Deprecated public ByteOrder order() { return buf.order(); } @Override @Deprecated public ByteBuf order(ByteOrder endianness) { return buf.order(endianness); } @Override public ByteBuf unwrap() { return buf.unwrap(); } @Override public boolean isDirect() { return buf.isDirect(); } @Override public boolean isReadOnly() { return buf.isReadOnly(); } @Override public ByteBuf asReadOnly() { return buf.asReadOnly(); } @Override public int readerIndex() { return buf.readerIndex(); } @Override public ByteBuf readerIndex(int readerIndex) { return buf.readerIndex(readerIndex); } @Override public int writerIndex() { return buf.writerIndex(); } @Override public ByteBuf writerIndex(int writerIndex) { return buf.writerIndex(writerIndex); } @Override public ByteBuf setIndex(int readerIndex, int writerIndex) { return buf.setIndex(readerIndex, writerIndex); } @Override public int readableBytes() { return buf.readableBytes(); } @Override public int writableBytes() { return buf.writableBytes(); } @Override public int maxWritableBytes() { return buf.maxWritableBytes(); } @Override public int maxFastWritableBytes() { return buf.maxFastWritableBytes(); } @Override public boolean isReadable() { return buf.isReadable(); } @Override public boolean isReadable(int size) { return buf.isReadable(size); } @Override public boolean isWritable() { return buf.isWritable(); } @Override public boolean isWritable(int size) { return buf.isWritable(size); } @Override public ByteBuf clear() { return buf.clear(); } @Override public ByteBuf markReaderIndex() { return buf.markReaderIndex(); } @Override public ByteBuf resetReaderIndex() { return buf.resetReaderIndex(); } @Override public ByteBuf markWriterIndex() { return buf.markWriterIndex(); } @Override public ByteBuf resetWriterIndex() { return buf.resetWriterIndex(); } @Override public ByteBuf discardReadBytes() { return buf.discardReadBytes(); } @Override public ByteBuf discardSomeReadBytes() { return buf.discardSomeReadBytes(); } @Override public ByteBuf ensureWritable(int minWritableBytes) { return buf.ensureWritable(minWritableBytes); } @Override public int ensureWritable(int minWritableBytes, boolean force) { return buf.ensureWritable(minWritableBytes, force); } @Override public boolean getBoolean(int index) { return buf.getBoolean(index); } @Override public byte getByte(int index) { return buf.getByte(index); } @Override public short getUnsignedByte(int index) { return buf.getUnsignedByte(index); } @Override public short getShort(int index) { return buf.getShort(index); } @Override public short getShortLE(int index) { return buf.getShortLE(index); } @Override public int getUnsignedShort(int index) { return buf.getUnsignedShort(index); } @Override public int getUnsignedShortLE(int index) { return buf.getUnsignedShortLE(index); } @Override public int getMedium(int index) { return buf.getMedium(index); } @Override public int getMediumLE(int index) { return buf.getMediumLE(index); } @Override public int getUnsignedMedium(int index) { return buf.getUnsignedMedium(index); } @Override public int getUnsignedMediumLE(int index) { return buf.getUnsignedMediumLE(index); } @Override public int getInt(int index) { return buf.getInt(index); } @Override public int getIntLE(int index) { return buf.getIntLE(index); } @Override public long getUnsignedInt(int index) { return buf.getUnsignedInt(index); } @Override public long getUnsignedIntLE(int index) { return buf.getUnsignedIntLE(index); } @Override public long getLong(int index) { return buf.getLong(index); } @Override public long getLongLE(int index) { return buf.getLongLE(index); } @Override public char getChar(int index) { return buf.getChar(index); } @Override public float getFloat(int index) { return buf.getFloat(index); } @Override public float getFloatLE(int index) { return buf.getFloatLE(index); } @Override public double getDouble(int index) { return buf.getDouble(index); } @Override public double getDoubleLE(int index) { return buf.getDoubleLE(index); } @Override public ByteBuf getBytes(int index, ByteBuf dst) { return buf.getBytes(index, dst); } @Override public ByteBuf getBytes(int index, ByteBuf dst, int length) { return buf.getBytes(index, dst, length); } @Override public ByteBuf getBytes(int index, ByteBuf dst, int dstIndex, int length) { return buf.getBytes(index, dst, dstIndex, length); } @Override public ByteBuf getBytes(int index, byte[] dst) { return buf.getBytes(index, dst); } @Override public ByteBuf getBytes(int index, byte[] dst, int dstIndex, int length) { return buf.getBytes(index, dst, dstIndex, length); } @Override public ByteBuf getBytes(int index, ByteBuffer dst) { return buf.getBytes(index, dst); } @Override public ByteBuf getBytes(int index, OutputStream out, int length) throws IOException { return buf.getBytes(index, out, length); } @Override public int getBytes(int index, GatheringByteChannel out, int length) throws IOException { return buf.getBytes(index, out, length); } @Override public int getBytes(int index, FileChannel out, long position, int length) throws IOException { return buf.getBytes(index, out, position, length); } @Override public CharSequence getCharSequence(int index, int length, Charset charset) { return buf.getCharSequence(index, length, charset); } @Override public ByteBuf setBoolean(int index, boolean value) { return buf.setBoolean(index, value); } @Override public ByteBuf setByte(int index, int value) { return buf.setByte(index, value); } @Override public ByteBuf setShort(int index, int value) { return buf.setShort(index, value); } @Override public ByteBuf setShortLE(int index, int value) { return buf.setShortLE(index, value); } @Override public ByteBuf setMedium(int index, int value) { return buf.setMedium(index, value); } @Override public ByteBuf setMediumLE(int index, int value) { return buf.setMediumLE(index, value); } @Override public ByteBuf setInt(int index, int value) { return buf.setInt(index, value); } @Override public ByteBuf setIntLE(int index, int value) { return buf.setIntLE(index, value); } @Override public ByteBuf setLong(int index, long value) { return buf.setLong(index, value); } @Override public ByteBuf setLongLE(int index, long value) { return buf.setLongLE(index, value); } @Override public ByteBuf setChar(int index, int value) { return buf.setChar(index, value); } @Override public ByteBuf setFloat(int index, float value) { return buf.setFloat(index, value); } @Override public ByteBuf setFloatLE(int index, float value) { return buf.setFloatLE(index, value); } @Override public ByteBuf setDouble(int index, double value) { return buf.setDouble(index, value); } @Override public ByteBuf setDoubleLE(int index, double value) { return buf.setDoubleLE(index, value); } @Override public ByteBuf setBytes(int index, ByteBuf src) { return buf.setBytes(index, src); } @Override public ByteBuf setBytes(int index, ByteBuf src, int length) { return buf.setBytes(index, src, length); } @Override public ByteBuf setBytes(int index, ByteBuf src, int srcIndex, int length) { return buf.setBytes(index, src, srcIndex, length); } @Override public ByteBuf setBytes(int index, byte[] src) { return buf.setBytes(index, src); } @Override public ByteBuf setBytes(int index, byte[] src, int srcIndex, int length) { return buf.setBytes(index, src, srcIndex, length); } @Override public ByteBuf setBytes(int index, ByteBuffer src) { return buf.setBytes(index, src); } @Override public int setBytes(int index, InputStream in, int length) throws IOException { return buf.setBytes(index, in, length); } @Override public int setBytes(int index, ScatteringByteChannel in, int length) throws IOException { return buf.setBytes(index, in, length); } @Override public int setBytes(int index, FileChannel in, long position, int length) throws IOException { return buf.setBytes(index, in, position, length); } @Override public ByteBuf setZero(int index, int length) { return buf.setZero(index, length); } @Override public int setCharSequence(int index, CharSequence sequence, Charset charset) { return buf.setCharSequence(index, sequence, charset); } @Override public boolean readBoolean() { return buf.readBoolean(); } @Override public byte readByte() { return buf.readByte(); } @Override public short readUnsignedByte() { return buf.readUnsignedByte(); } @Override public short readShort() { return buf.readShort(); } @Override public short readShortLE() { return buf.readShortLE(); } @Override public int readUnsignedShort() { return buf.readUnsignedShort(); } @Override public int readUnsignedShortLE() { return buf.readUnsignedShortLE(); } @Override public int readMedium() { return buf.readMedium(); } @Override public int readMediumLE() { return buf.readMediumLE(); } @Override public int readUnsignedMedium() { return buf.readUnsignedMedium(); } @Override public int readUnsignedMediumLE() { return buf.readUnsignedMediumLE(); } @Override public int readInt() { return buf.readInt(); } @Override public int readIntLE() { return buf.readIntLE(); } @Override public long readUnsignedInt() { return buf.readUnsignedInt(); } @Override public long readUnsignedIntLE() { return buf.readUnsignedIntLE(); } @Override public long readLong() { return buf.readLong(); } @Override public long readLongLE() { return buf.readLongLE(); } @Override public char readChar() { return buf.readChar(); } @Override public float readFloat() { return buf.readFloat(); } @Override public float readFloatLE() { return buf.readFloatLE(); } @Override public double readDouble() { return buf.readDouble(); } @Override public double readDoubleLE() { return buf.readDoubleLE(); } @Override public ByteBuf readBytes(int length) { return buf.readBytes(length); } @Override public ByteBuf readSlice(int length) { return buf.readSlice(length); } @Override public ByteBuf readRetainedSlice(int length) { return buf.readRetainedSlice(length); } @Override public ByteBuf readBytes(ByteBuf dst) { return buf.readBytes(dst); } @Override public ByteBuf readBytes(ByteBuf dst, int length) { return buf.readBytes(dst, length); } @Override public ByteBuf readBytes(ByteBuf dst, int dstIndex, int length) { return buf.readBytes(dst, dstIndex, length); } @Override public ByteBuf readBytes(byte[] dst) { return buf.readBytes(dst); } @Override public ByteBuf readBytes(byte[] dst, int dstIndex, int length) { return buf.readBytes(dst, dstIndex, length); } @Override public ByteBuf readBytes(ByteBuffer dst) { return buf.readBytes(dst); } @Override public ByteBuf readBytes(OutputStream out, int length) throws IOException { return buf.readBytes(out, length); } @Override public int readBytes(GatheringByteChannel out, int length) throws IOException { return buf.readBytes(out, length); } @Override public CharSequence readCharSequence(int length, Charset charset) { return buf.readCharSequence(length, charset); } @Override public int readBytes(FileChannel out, long position, int length) throws IOException { return buf.readBytes(out, position, length); } @Override public ByteBuf skipBytes(int length) { return buf.skipBytes(length); } @Override public ByteBuf writeBoolean(boolean value) { return buf.writeBoolean(value); } @Override public ByteBuf writeByte(int value) { return buf.writeByte(value); } @Override public ByteBuf writeShort(int value) { return buf.writeShort(value); } @Override public ByteBuf writeShortLE(int value) { return buf.writeShortLE(value); } @Override public ByteBuf writeMedium(int value) { return buf.writeMedium(value); } @Override public ByteBuf writeMediumLE(int value) { return buf.writeMediumLE(value); } @Override public ByteBuf writeInt(int value) { return buf.writeInt(value); } @Override public ByteBuf writeIntLE(int value) { return buf.writeIntLE(value); } @Override public ByteBuf writeLong(long value) { return buf.writeLong(value); } @Override public ByteBuf writeLongLE(long value) { return buf.writeLongLE(value); } @Override public ByteBuf writeChar(int value) { return buf.writeChar(value); } @Override public ByteBuf writeFloat(float value) { return buf.writeFloat(value); } @Override public ByteBuf writeFloatLE(float value) { return buf.writeFloatLE(value); } @Override public ByteBuf writeDouble(double value) { return buf.writeDouble(value); } @Override public ByteBuf writeDoubleLE(double value) { return buf.writeDoubleLE(value); } @Override public ByteBuf writeBytes(ByteBuf src) { return buf.writeBytes(src); } @Override public ByteBuf writeBytes(ByteBuf src, int length) { return buf.writeBytes(src, length); } @Override public ByteBuf writeBytes(ByteBuf src, int srcIndex, int length) { return buf.writeBytes(src, srcIndex, length); } @Override public ByteBuf writeBytes(byte[] src) { return buf.writeBytes(src); } @Override public ByteBuf writeBytes(byte[] src, int srcIndex, int length) { return buf.writeBytes(src, srcIndex, length); } @Override public ByteBuf writeBytes(ByteBuffer src) { return buf.writeBytes(src); } @Override public int writeBytes(InputStream in, int length) throws IOException { return buf.writeBytes(in, length); } @Override public int writeBytes(ScatteringByteChannel in, int length) throws IOException { return buf.writeBytes(in, length); } @Override public int writeBytes(FileChannel in, long position, int length) throws IOException { return buf.writeBytes(in, position, length); } @Override public ByteBuf writeZero(int length) { return buf.writeZero(length); } @Override public int writeCharSequence(CharSequence sequence, Charset charset) { return buf.writeCharSequence(sequence, charset); } @Override public int indexOf(int fromIndex, int toIndex, byte value) { return buf.indexOf(fromIndex, toIndex, value); } @Override public int bytesBefore(byte value) { return buf.bytesBefore(value); } @Override public int bytesBefore(int length, byte value) { return buf.bytesBefore(length, value); } @Override public int bytesBefore(int index, int length, byte value) { return buf.bytesBefore(index, length, value); } @Override public int forEachByte(ByteProcessor processor) { return buf.forEachByte(processor); } @Override public int forEachByte(int index, int length, ByteProcessor processor) { return buf.forEachByte(index, length, processor); } @Override public int forEachByteDesc(ByteProcessor processor) { return buf.forEachByteDesc(processor); } @Override public int forEachByteDesc(int index, int length, ByteProcessor processor) { return buf.forEachByteDesc(index, length, processor); } @Override public ByteBuf copy() { return buf.copy(); } @Override public ByteBuf copy(int index, int length) { return buf.copy(index, length); } @Override public ByteBuf slice() { return buf.slice(); } @Override public ByteBuf retainedSlice() { return buf.retainedSlice(); } @Override public ByteBuf slice(int index, int length) { return buf.slice(index, length); } @Override public ByteBuf retainedSlice(int index, int length) { return buf.retainedSlice(index, length); } @Override public ByteBuf duplicate() { return buf.duplicate(); } @Override public ByteBuf retainedDuplicate() { return buf.retainedDuplicate(); } @Override public int nioBufferCount() { return buf.nioBufferCount(); } @Override public ByteBuffer nioBuffer() { return buf.nioBuffer(); } @Override public ByteBuffer nioBuffer(int index, int length) { return buf.nioBuffer(index, length); } @Override public ByteBuffer internalNioBuffer(int index, int length) { return buf.internalNioBuffer(index, length); } @Override public ByteBuffer[] nioBuffers() { return buf.nioBuffers(); } @Override public ByteBuffer[] nioBuffers(int index, int length) { return buf.nioBuffers(index, length); } @Override public boolean hasArray() { return buf.hasArray(); } @Override public byte[] array() { return buf.array(); } @Override public int arrayOffset() { return buf.arrayOffset(); } @Override public boolean hasMemoryAddress() { return buf.hasMemoryAddress(); } @Override public long memoryAddress() { return buf.memoryAddress(); } @Override public boolean isContiguous() { return buf.isContiguous(); } @Override public String toString(Charset charset) { return buf.toString(charset); } @Override public String toString(int index, int length, Charset charset) { return buf.toString(index, length, charset); } @Override public int hashCode() { return buf.hashCode(); } @Override public boolean equals(Object obj) { return buf.equals(obj); } @Override public int compareTo(ByteBuf buffer) { return buf.compareTo(buffer); } @Override public String toString() { return buf.toString(); } @Override public ByteBuf retain(int increment) { return buf.retain(increment); } @Override public ByteBuf retain() { return buf.retain(); } @Override public ByteBuf touch() { return buf.touch(); } @Override public ByteBuf touch(Object hint) { return buf.touch(hint); } @Override public int refCnt() { return buf.refCnt(); } @Override public boolean release() { return buf.release(); } @Override public boolean release(int decrement) { return buf.release(decrement); } public static ByteMessage create() { return new ByteMessage(Unpooled.buffer()); } }