micialware/mROA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rewrite reading length of message

Browse Source
This commit is contained in:
micialware
2025-07-01 15:41:48 +03:00
parent 1b0f867653
commit aa3606401b
1 changed files with 7 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mROA/Implementation/ChannelInteractionModule.cs
+7 -14
View File
@@ -166,7 +166,7 @@ namespace mROA.Implementation
private readonly Memory<byte> _buffer = new byte[BufferSize]; private readonly Memory<byte> _buffer = new byte[BufferSize];
private bool _manualConnectionState = true; private bool _manualConnectionState = true;
private readonly IEndPointContext _context; private readonly IEndPointContext _context;
private readonly byte[] _lenBuffer;
public StreamExtractor(Stream ioStream, IContextualSerializationToolKit serializationToolkit, public StreamExtractor(Stream ioStream, IContextualSerializationToolKit serializationToolkit,
IEndPointContext context) IEndPointContext context)
{ {
@@ -174,23 +174,16 @@ namespace mROA.Implementation
_serializationToolkit = serializationToolkit; _serializationToolkit = serializationToolkit;
_context = context; _context = context;
profiler.Start(); profiler.Start();
_lenBuffer = new byte[2];
} }
public Action<NetworkMessageHeader> MessageReceived = _ => { }; public Action<NetworkMessageHeader> MessageReceived = _ => { };
private ushort ReadMessageLength() private async Task<ushort> ReadMessageLength()
{ {
var firstBit = _ioStream.ReadByte(); await _ioStream.ReadAsync(_lenBuffer, 0, 2);
if (firstBit == -1)
{ var len = BitConverter.ToUInt16(_lenBuffer);
_manualConnectionState = false;
throw new EndOfStreamException();
}
_manualConnectionState = true;
var secondBit = (byte)_ioStream.ReadByte();
var len = BitConverter.ToUInt16(new[] { (byte)firstBit, secondBit });
return len; return len;
} }
@@ -198,7 +191,7 @@ namespace mROA.Implementation
public async Task SingleReceive(CancellationToken token = default) public async Task SingleReceive(CancellationToken token = default)
{ {
Console.WriteLine($"Receive start. Time: {profiler.ElapsedMilliseconds}ms"); Console.WriteLine($"Receive start. Time: {profiler.ElapsedMilliseconds}ms");
var len = ReadMessageLength(); var len = await ReadMessageLength();
Console.WriteLine($"Received len {len}. Time: {profiler.ElapsedMilliseconds}ms"); Console.WriteLine($"Received len {len}. Time: {profiler.ElapsedMilliseconds}ms");
var localSpan = _buffer[..len]; var localSpan = _buffer[..len];