45 lines
1.3 KiB
Java
45 lines
1.3 KiB
Java
package net.minecraft.network;
|
|
|
|
import io.netty.buffer.ByteBuf;
|
|
import io.netty.channel.ChannelHandlerContext;
|
|
import javax.crypto.Cipher;
|
|
import javax.crypto.ShortBufferException;
|
|
|
|
public class CipherBase {
|
|
private final Cipher cipher;
|
|
private byte[] heapIn = new byte[0];
|
|
private byte[] heapOut = new byte[0];
|
|
|
|
protected CipherBase(Cipher cipher) {
|
|
this.cipher = cipher;
|
|
}
|
|
|
|
private byte[] bufToByte(ByteBuf buffer) {
|
|
int i = buffer.readableBytes();
|
|
if (this.heapIn.length < i) {
|
|
this.heapIn = new byte[i];
|
|
}
|
|
|
|
buffer.readBytes(this.heapIn, 0, i);
|
|
return this.heapIn;
|
|
}
|
|
|
|
protected ByteBuf decipher(ChannelHandlerContext ctx, ByteBuf buffer) throws ShortBufferException {
|
|
int i = buffer.readableBytes();
|
|
byte[] bs = this.bufToByte(buffer);
|
|
ByteBuf byteBuf = ctx.alloc().heapBuffer(this.cipher.getOutputSize(i));
|
|
byteBuf.writerIndex(this.cipher.update(bs, 0, i, byteBuf.array(), byteBuf.arrayOffset()));
|
|
return byteBuf;
|
|
}
|
|
|
|
protected void encipher(ByteBuf input, ByteBuf out) throws ShortBufferException {
|
|
int i = input.readableBytes();
|
|
byte[] bs = this.bufToByte(input);
|
|
int j = this.cipher.getOutputSize(i);
|
|
if (this.heapOut.length < j) {
|
|
this.heapOut = new byte[j];
|
|
}
|
|
|
|
out.writeBytes(this.heapOut, 0, this.cipher.update(bs, 0, i, this.heapOut));
|
|
}
|
|
}
|