Austin/BililiveRecorder
1
0
Fork 0
mirror of https://github.com/BililiveRecorder/BililiveRecorder.git synced 2024-11-16 11:42:22 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Core: Fix DanmakuClient buffer length

Browse Source
This commit is contained in:
genteure 2022-06-04 14:04:28 +08:00
parent fadb90f487
commit 9d0df74159
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
BililiveRecorder.Core/Api/Danmaku/DanmakuClient.cs
View File

@ -225,18 +225,19 @@ namespace BililiveRecorder.Core.Api.Danmaku
if (stream is null) if (stream is null)
throw new ArgumentNullException(nameof(stream)); throw new ArgumentNullException(nameof(stream));
var playload = Encoding.UTF8.GetBytes(body); var playload = ((body?.Length ?? 0) > 0) ? Encoding.UTF8.GetBytes(body) : Array.Empty<byte>();
var size = playload.Length + 16; const int headerLength = 16;
var buffer = ArrayPool<byte>.Shared.Rent(16); var size = playload.Length + headerLength;
var buffer = ArrayPool<byte>.Shared.Rent(headerLength);
try try
{ {
BinaryPrimitives.WriteUInt32BigEndian(new Span<byte>(buffer, 0, 4), (uint)size); BinaryPrimitives.WriteUInt32BigEndian(new Span<byte>(buffer, 0, 4), (uint)size);
BinaryPrimitives.WriteUInt16BigEndian(new Span<byte>(buffer, 4, 2), 16); BinaryPrimitives.WriteUInt16BigEndian(new Span<byte>(buffer, 4, 2), headerLength);
BinaryPrimitives.WriteUInt16BigEndian(new Span<byte>(buffer, 6, 2), 1); BinaryPrimitives.WriteUInt16BigEndian(new Span<byte>(buffer, 6, 2), 1);
BinaryPrimitives.WriteUInt32BigEndian(new Span<byte>(buffer, 8, 4), (uint)action); BinaryPrimitives.WriteUInt32BigEndian(new Span<byte>(buffer, 8, 4), (uint)action);
BinaryPrimitives.WriteUInt32BigEndian(new Span<byte>(buffer, 12, 4), 1); BinaryPrimitives.WriteUInt32BigEndian(new Span<byte>(buffer, 12, 4), 1);
await stream.WriteAsync(buffer, 0, buffer.Length).ConfigureAwait(false); await stream.WriteAsync(buffer, 0, headerLength).ConfigureAwait(false);
if (playload.Length > 0) if (playload.Length > 0)
await stream.WriteAsync(playload, 0, playload.Length).ConfigureAwait(false); await stream.WriteAsync(playload, 0, playload.Length).ConfigureAwait(false);
await stream.FlushAsync().ConfigureAwait(false); await stream.FlushAsync().ConfigureAwait(false);