KenSchae
/
StabilityMatrix
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Fix missing inputs when APC follows another APC without line-ending

pull/240/head
Ionite 1 year ago
parent
f84bc04e7e
commit
1c5b26333e
No known key found for this signature in database
1 changed files with 63 additions and 42 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 67
      StabilityMatrix.Core/Processes/AsyncStreamReader.cs

67
StabilityMatrix.Core/Processes/AsyncStreamReader.cs
Unescape View File

@ -23,7 +23,7 @@ namespace StabilityMatrix.Core.Processes;
[SuppressMessage("ReSharper", "InconsistentNaming")]
internal sealed class AsyncStreamReader : IDisposable
{
private const int DefaultBufferSize = 1024; // Byte buffer size
private const int DefaultBufferSize = 1024; // Byte buffer size
private readonly Stream _stream;
private readonly Decoder _decoder;
@ -96,14 +96,18 @@ internal sealed class AsyncStreamReader : IDisposable
{
try
{
var bytesRead = await _stream.ReadAsync(new Memory<byte>(_byteBuffer), _cts.Token).ConfigureAwait(false);
var bytesRead = await _stream
.ReadAsync(new Memory<byte>(_byteBuffer), _cts.Token)
.ConfigureAwait(false);
if (bytesRead == 0)
break;
var charLen = _decoder.GetChars(_byteBuffer, 0, bytesRead, _charBuffer, 0);
Debug.WriteLine($"AsyncStreamReader - Read {charLen} chars: " +
$"{new string(_charBuffer, 0, charLen).ToRepr()}");
Debug.WriteLine(
$"AsyncStreamReader - Read {charLen} chars: "
+ $"{new string(_charBuffer, 0, charLen).ToRepr()}"
);
_sb!.Append(_charBuffer, 0, charLen);
MoveLinesFromStringBuilderToMessageQueue();
@ -140,7 +144,9 @@ internal sealed class AsyncStreamReader : IDisposable
var remaining = _sb.ToString();
_messageQueue.Enqueue(remaining);
_sb.Length = 0;
Debug.WriteLine($"AsyncStreamReader - Reached EOF, sent remaining buffer: {remaining}");
Debug.WriteLine(
$"AsyncStreamReader - Reached EOF, sent remaining buffer: {remaining}"
);
}
_messageQueue.Enqueue(null);
}
@ -154,7 +160,8 @@ internal sealed class AsyncStreamReader : IDisposable
{
lock (_messageQueue)
{
if (_sb!.Length == 0) return;
if (_sb!.Length == 0)
return;
_messageQueue.Enqueue(_sb.ToString());
_sb.Length = 0;
@ -167,7 +174,8 @@ internal sealed class AsyncStreamReader : IDisposable
{
lock (_messageQueue)
{
if (_sb!.Length == 0) return;
if (_sb!.Length == 0)
return;
_messageQueue.Enqueue(_sb.ToString(startIndex, _sb.Length - startIndex));
_sb.Length = 0;
@ -177,7 +185,8 @@ internal sealed class AsyncStreamReader : IDisposable
// Sends a message to the queue if not null or empty
private void SendToQueue(string? message)
{
if (string.IsNullOrEmpty(message)) return;
if (string.IsNullOrEmpty(message))
return;
lock (_messageQueue)
{
_messageQueue.Enqueue(message);
@ -196,14 +205,6 @@ internal sealed class AsyncStreamReader : IDisposable
var lineStart = 0;
var len = _sb!.Length;
// If flagged, send next buffer immediately
if (_sendNextBufferImmediately)
{
SendRemainingBuffer();
_sendNextBufferImmediately = false;
return;
}
// skip a beginning '\n' character of new block if last block ended
// with '\r'
if (_bLastCarriageReturn && len > 0 && _sb[0] == '\n')
@ -255,7 +256,8 @@ internal sealed class AsyncStreamReader : IDisposable
{
// Send buffer up to this point, not including \r
// But skip if there's no content
if (currentIndex == lineStart) break;
if (currentIndex == lineStart)
break;
var line = _sb.ToString(lineStart, currentIndex - lineStart);
lock (_messageQueue)
@ -294,9 +296,13 @@ internal sealed class AsyncStreamReader : IDisposable
// lineStart = searchIndex + 1;
currentIndex = searchIndex;
var remainingStart = currentIndex + 1;
var remainingStr =
_sb.ToString(remainingStart, _sb.Length - remainingStart);
Debug.WriteLine($"AsyncStreamReader - Sending remaining buffer: '{remainingStr}'");
var remainingStr = _sb.ToString(
remainingStart,
_sb.Length - remainingStart
);
Debug.WriteLine(
$"AsyncStreamReader - Sending remaining buffer: '{remainingStr}'"
);
// Send the rest of the buffer immediately
SendRemainingBuffer(currentIndex + 1);
return;
@ -324,7 +330,9 @@ internal sealed class AsyncStreamReader : IDisposable
var escapeSequence = result.Value;
SendToQueue(escapeSequence);
Debug.WriteLine($"AsyncStreamReader - Sent Ansi escape sequence: {escapeSequence.ToRepr()}");
Debug.WriteLine(
$"AsyncStreamReader - Sent Ansi escape sequence: {escapeSequence.ToRepr()}"
);
// Advance currentIndex and lineStart to end of escape sequence
// minus 1 since we will increment currentIndex at the end of the loop
@ -333,7 +341,9 @@ internal sealed class AsyncStreamReader : IDisposable
}
else
{
Debug.WriteLine($"AsyncStreamReader - No match for Ansi escape sequence: {remaining.ToRepr()}");
Debug.WriteLine(
$"AsyncStreamReader - No match for Ansi escape sequence: {remaining.ToRepr()}"
);
}
break;
@ -346,6 +356,14 @@ internal sealed class AsyncStreamReader : IDisposable
_bLastCarriageReturn = true;
}
// If flagged, send remaining buffer immediately
if (_sendNextBufferImmediately)
{
SendRemainingBuffer();
_sendNextBufferImmediately = false;
return;
}
// Keep the rest characters which can't form a new line in string builder.
if (lineStart < len)
{
@ -404,7 +422,10 @@ internal sealed class AsyncStreamReader : IDisposable
// Otherwise, let the exception propagate.
if (rethrowInNewThread)
{
ThreadPool.QueueUserWorkItem(edi => ((ExceptionDispatchInfo)edi!).Throw(), ExceptionDispatchInfo.Capture(e));
ThreadPool.QueueUserWorkItem(
edi => ((ExceptionDispatchInfo)edi!).Throw(),
ExceptionDispatchInfo.Capture(e)
);
return true;
}
throw;

Write Preview
Loading…
Cancel
Save