public ByteBuffer encode() {
int length = 2;
for (Packet packet : content) {
int pLength = packet.encode().array().length;
if (pLength >= 255) {
length += 3;
} else {
length += 1;
}
length += pLength;
}
ByteBuffer data = ByteBuffer.allocate(length);
data.order(ByteOrder.BIG_ENDIAN).putShort((short) 3);
for (Packet packet : content) {
byte[] pData = packet.encode().array();
if (pData.length >= 255) {
data.put((byte) 255);
addShort(data, (short) pData.length);
} else {
data.put((byte) pData.length);
}
data.put(pData);
}
return data;
}
public void decode(ByteBuffer data) {
data.position(2);
int length = data.array().length;
int pLength = 0;
for (int i = 2; i < length; i += pLength) {
if (data.remaining() < 1) return;
pLength = getByte(data) & 0xFF;
if (pLength == 255) {
if (data.remaining() < 2) return;
pLength = data.order(ByteOrder.LITTLE_ENDIAN).getShort();
}
if (pLength > data.remaining()) {
data.position(data.position() - 1);
return;
}
byte[] pData = new byte[pLength];
data.get(pData);
Packet p = new Packet();
p.decode(ByteBuffer.wrap(pData));
content.add(p);
}
}
private int getPacketLength(Packet packet) {
if (packet instanceof DataChannelA) return ((DataChannelA) packet).getLength();
else if (packet instanceof Acknowledge || packet instanceof OutOfOrder) return 4;
else if (packet instanceof Disconnect) return 8;
return packet.encode().array().length;
}
