public void Encrypt(IByteBufferAllocator allocator, EncryptMode mode, Stream src, Stream dst, bool reliable)
{
using (var data = new BufferWrapper(allocator.Buffer().WithOrder(ByteOrder.LittleEndian)))
using (var encryptor = GetAlgorithm(mode).CreateEncryptor())
using (var cs = new CryptoStream(new NonClosingStream(dst), encryptor, CryptoStreamMode.Write))
using (var w = cs.ToBinaryWriter(false))
{
var blockSize = AES.BlockSize / 8;
var padding = blockSize - (src.Length + 1 + 4) % blockSize;
if (reliable)
{
padding = blockSize - (src.Length + 1 + 4 + 2) % blockSize;
}
if (reliable)
{
var counter = (ushort)(Interlocked.Increment(ref _encryptCounter) - 1);
data.Buffer.WriteShort(counter);
}
using (var dataStream = new WriteOnlyByteBufferStream(data.Buffer, false))
src.CopyTo(dataStream);
w.Write((byte)padding);
using (var dataStream = new ReadOnlyByteBufferStream(data.Buffer, false))
{
w.Write(Hash.GetUInt32 <CRC32>(dataStream));
dataStream.Position = 0;
dataStream.CopyTo(cs);
}
w.Fill((int)padding);
}
}
public void Decrypt(IByteBufferAllocator allocator, EncryptMode mode, Stream src, Stream dst, bool reliable)
{
if (RC4 == null || AES == null)
{
return;
}
//throw new ObjectDisposedException(GetType().FullName);
using (var data = new BufferWrapper(allocator.Buffer().WithOrder(ByteOrder.LittleEndian)))
using (var decryptor = GetAlgorithm(mode).CreateDecryptor())
using (var cs = new CryptoStream(src, decryptor, CryptoStreamMode.Read))
{
var padding = cs.ReadByte();
var checksum = cs.ReadByte() | (cs.ReadByte() << 8) | (cs.ReadByte() << 16) | (cs.ReadByte() << 24);
using (var dataStream = new WriteOnlyByteBufferStream(data.Buffer, false))
{
cs.CopyTo(dataStream);
}
if (reliable)
{
var counter = (ushort)(Interlocked.Increment(ref _decryptCounter) - 1);
var messageCounter = data.Buffer.GetShort(data.Buffer.ReaderIndex);
if (counter != messageCounter)
{
throw new ProudException($"Invalid decrypt counter! Remote: {messageCounter} Local: {counter}");
}
}
var slice = data.Buffer.ReadSlice(data.Buffer.ReadableBytes - padding);
using (var dataStream = new ReadOnlyByteBufferStream(slice, false))
{
if (Hash.GetUInt32 <CRC32>(dataStream) != (uint)checksum)
{
throw new ProudException("Invalid checksum");
}
dataStream.Position = reliable ? 2 : 0;
dataStream.CopyTo(dst);
}
}
}
public void Encrypt(IByteBufferAllocator allocator, EncryptMode mode, Stream src, Stream dst, bool reliable)
{
if (RC4 == null || AES == null)
{
return;
}
//throw new ObjectDisposedException(GetType().FullName);
using (var data = new BufferWrapper(allocator.Buffer()))
using (var encryptor = GetAlgorithm(mode).CreateEncryptor())
using (var cs = new CryptoStream(new NonClosingStream(dst), encryptor, CryptoStreamMode.Write))
using (var w = cs.ToBinaryWriter(false))
{
var blockSize = AES.BlockSize / 8;
var padding = blockSize - ((src.Length + 1 + 4) % blockSize);
if (reliable)
{
padding = blockSize - ((src.Length + 1 + 4 + 2) % blockSize);
}
if (reliable)
{
var counter = (ushort)(Interlocked.Increment(ref _encryptCounter) - 1);
data.Buffer.WriteShortLE(counter);
}
using (var dataStream = new WriteOnlyByteBufferStream(data.Buffer, false))
{
src.CopyTo(dataStream);
}
w.Write((byte)padding);
using (var dataStream = new ReadOnlyByteBufferStream(data.Buffer, false))
{
w.Write(Hash.GetUInt32 <CRC32>(dataStream));
dataStream.Position = 0;
dataStream.CopyTo(cs);
}
w.Fill((int)padding);
}
}
