starred/sharpcompress
1
0
Fork 0
mirror of https://github.com/adamhathcock/sharpcompress.git synced 2026-09-24 16:05:27 +00:00
Code Issues 975 Packages Projects Releases 25 Wiki Activity

Explicit exception for read shortcut

Browse Source
This commit is contained in:
Adam Hathcock
2021-02-20 12:52:29 +00:00
parent 30da0b91ed
commit 0c35abdebe
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/SharpCompress/Utility.cs
View File

@@ -21,8 +21,13 @@ namespace SharpCompress
public static async ValueTask<T> ReadPrimitive<T>(this Stream stream, int bytes, Func<ReadOnlyMemory<byte>, T> func, CancellationToken cancellationToken)
{
using var buffer = MemoryPool<byte>.Shared.Rent(bytes);
await stream.ReadAsync(buffer.Memory.Slice(0, bytes), cancellationToken);
return func(buffer.Memory.Slice(0, bytes));
var memory = buffer.Memory.Slice(0, bytes);
var n = await stream.ReadAsync(memory, cancellationToken);
if (n != memory.Length)
{
throw new InvalidOperationException("Unexpected length");
}
return func(memory);
}
public static ValueTask<byte> ReadByteAsync(this Stream stream, CancellationToken cancellationToken)