mirror of
https://github.com/space-wizards/RobustToolbox.git
synced 2026-02-15 03:30:53 +01:00
faff0797bf
* make lidgren use spans everywhere where it can convert custom pooling to shared array pool impl add unit tests for read/write add native socket extensions to socket so we can legit pass spans for SendTo/ReceiveFrom bump version in lidgren csproj replace some random "% 8" w/ "& 7" more minor nullability hacks to fix static analysis complaints made receiving packets use span minor native sockets refactor to use pinvoke add read/write constrained/prealloc'd bit stream impl to lidgren and update usages fixed missing stream cleanup remove outstanding stream cleanup since it refs buffer thru the class, can't read some other buf apply suggestions from code review remove unsafe cruft * add tests to gh actions * make stats use interpolation in tostring and remove m_bytesAllocated since it's all in the shared pool now * this pr still open so fuck it stats, human readability, faster BitsToHold methods * add api compatible version of ReadBytes * rename ReadOnlyStreamWrapper -> ReadOnlyWrapperStream rename WriteOnlyStreamWrapper -> WriteOnlyWrapperStream add AppendViaStream, AppenderStream impl add and update documentation on read/write bytes methods also fix some goofs
98 lines
2.9 KiB
C#
98 lines
2.9 KiB
C#
using System;
|
|
using System.Buffers;
|
|
using System.Collections.Generic;
|
|
|
|
namespace Lidgren.Network
|
|
{
|
|
/// <summary>
|
|
/// Base for a non-threadsafe encryption class
|
|
/// </summary>
|
|
public abstract class NetBlockEncryptionBase : NetEncryption
|
|
{
|
|
|
|
/// <summary>
|
|
/// Block size in bytes for this cipher
|
|
/// </summary>
|
|
public abstract int BlockSize { get; }
|
|
|
|
/// <summary>
|
|
/// NetBlockEncryptionBase constructor
|
|
/// </summary>
|
|
public NetBlockEncryptionBase(NetPeer peer)
|
|
: base(peer)
|
|
{
|
|
}
|
|
|
|
/// <summary>
|
|
/// Encrypt am outgoing message with this algorithm; no writing can be done to the message after encryption, or message will be corrupted
|
|
/// </summary>
|
|
public override bool Encrypt(NetOutgoingMessage msg)
|
|
{
|
|
int payloadBitLength = msg.LengthBits;
|
|
int numBytes = msg.LengthBytes;
|
|
int blockSize = BlockSize;
|
|
Span<byte> tmp = stackalloc byte[BlockSize];
|
|
int numBlocks = (int)Math.Ceiling(numBytes / (double)blockSize);
|
|
int dstSize = numBlocks * blockSize;
|
|
|
|
msg.EnsureBufferSize(dstSize * 8 + (4 * 8)); // add 4 bytes for payload length at end
|
|
msg.LengthBits = dstSize * 8; // length will automatically adjust +4 bytes when payload length is written
|
|
|
|
var dataSpan = msg.m_data;
|
|
for(int i=0;i<numBlocks;i++)
|
|
{
|
|
// TODO: honestly is tmp even necessary here?
|
|
EncryptBlock(dataSpan, (i * blockSize), tmp);
|
|
//Buffer.BlockCopy(m_tmp, 0, dataSpan, (i * blockSize), m_tmp.Length);
|
|
tmp.CopyTo(dataSpan.Slice(i * blockSize, tmp.Length));
|
|
}
|
|
|
|
// add true payload length last
|
|
msg.Write((UInt32)payloadBitLength);
|
|
|
|
return true;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Decrypt an incoming message encrypted with corresponding Encrypt
|
|
/// </summary>
|
|
/// <param name="msg">message to decrypt</param>
|
|
/// <returns>true if successful; false if failed</returns>
|
|
public override bool Decrypt(NetIncomingMessage msg)
|
|
{
|
|
int numEncryptedBytes = msg.LengthBytes - 4; // last 4 bytes is true bit length
|
|
int blockSize = BlockSize;
|
|
Span<byte> tmp = stackalloc byte[BlockSize];
|
|
int numBlocks = numEncryptedBytes / blockSize;
|
|
if (numBlocks * blockSize != numEncryptedBytes)
|
|
return false;
|
|
|
|
var dataSpan = msg.m_data;
|
|
for (int i = 0; i < numBlocks; i++)
|
|
{
|
|
// TODO: honestly is tmp even necessary here?
|
|
DecryptBlock(dataSpan, (i * blockSize), tmp);
|
|
//Buffer.BlockCopy(m_tmp, 0, dataSpan, (i * blockSize), m_tmp.Length);
|
|
tmp.CopyTo(dataSpan.Slice(i * blockSize, tmp.Length));
|
|
}
|
|
|
|
// read 32 bits of true payload length
|
|
uint realSize = NetBitWriter.ReadUInt32(dataSpan, 32, (numEncryptedBytes * 8));
|
|
msg.m_bitLength = (int)realSize;
|
|
return true;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Encrypt a block of bytes
|
|
/// </summary>
|
|
protected abstract void EncryptBlock(ReadOnlySpan<byte> source, int sourceOffset, Span<byte> destination);
|
|
|
|
/// <summary>
|
|
/// Decrypt a block of bytes
|
|
/// </summary>
|
|
protected abstract void DecryptBlock(ReadOnlySpan<byte> source, int sourceOffset, Span<byte> destination);
|
|
|
|
|
|
}
|
|
}
|