| | | 1 | | // Licensed to the .NET Foundation under one or more agreements. |
| | | 2 | | // The .NET Foundation licenses this file to you under the MIT license. |
| | | 3 | | |
| | | 4 | | using System.Buffers; |
| | | 5 | | using System.Linq; |
| | | 6 | | using System.IO.Compression; |
| | | 7 | | using System.Reflection; |
| | | 8 | | using System.Threading.Tasks; |
| | | 9 | | |
| | | 10 | | namespace DotnetFuzzing.Fuzzers; |
| | | 11 | | |
| | | 12 | | internal sealed class Deflate64Fuzzer : IFuzzer |
| | | 13 | | { |
| | 1 | 14 | | public string[] TargetAssemblies { get; } = ["System.IO.Compression"]; |
| | 0 | 15 | | public string[] TargetCoreLibPrefixes => []; |
| | 0 | 16 | | public string Corpus => "deflate64"; |
| | | 17 | | |
| | | 18 | | public void FuzzTarget(ReadOnlySpan<byte> bytes) |
| | 3251 | 19 | | { |
| | 3251 | 20 | | if (bytes.IsEmpty) |
| | 0 | 21 | | { |
| | 0 | 22 | | return; |
| | | 23 | | } |
| | | 24 | | |
| | 3251 | 25 | | TestArchive(CopyToRentedArray(bytes), bytes.Length, async: false).GetAwaiter().GetResult(); |
| | 3251 | 26 | | TestArchive(CopyToRentedArray(bytes), bytes.Length, async: true).GetAwaiter().GetResult(); |
| | 3251 | 27 | | } |
| | | 28 | | |
| | | 29 | | #pragma warning disable IL2026 // RequiresUnreferencedCode |
| | 1 | 30 | | private static readonly Type _deflateStreamType = typeof(ZipArchive).Assembly.GetType("System.IO.Compression.Deflate |
| | | 31 | | #pragma warning restore IL2026 |
| | | 32 | | |
| | | 33 | | private static Stream CreateStream(byte[] bytes, int length) |
| | 6502 | 34 | | { |
| | | 35 | | #pragma warning disable IL2077 // dynamic access to non-public ctors |
| | 6502 | 36 | | return (Stream)Activator.CreateInstance( |
| | 6502 | 37 | | _deflateStreamType, |
| | 6502 | 38 | | bindingAttr: BindingFlags.NonPublic | BindingFlags.Instance, |
| | 6502 | 39 | | binder: null, |
| | 6502 | 40 | | args: new object[] { new MemoryStream(bytes, 0, length), ZipCompressionMethod.Deflate64, -1L }, |
| | 6502 | 41 | | culture: null)!; |
| | | 42 | | #pragma warning restore IL2077 |
| | 6502 | 43 | | } |
| | | 44 | | |
| | | 45 | | private byte[] CopyToRentedArray(ReadOnlySpan<byte> bytes) |
| | 6502 | 46 | | { |
| | 6502 | 47 | | byte[] buffer = ArrayPool<byte>.Shared.Rent(bytes.Length); |
| | | 48 | | try |
| | 6502 | 49 | | { |
| | 6502 | 50 | | bytes.CopyTo(buffer); |
| | 6502 | 51 | | return buffer; |
| | | 52 | | } |
| | 0 | 53 | | catch |
| | 0 | 54 | | { |
| | 0 | 55 | | ArrayPool<byte>.Shared.Return(buffer); |
| | 0 | 56 | | throw; |
| | | 57 | | } |
| | 6502 | 58 | | } |
| | | 59 | | |
| | | 60 | | private async Task TestArchive(byte[] buffer, int length, bool async) |
| | 6502 | 61 | | { |
| | | 62 | | try |
| | 6502 | 63 | | { |
| | 6502 | 64 | | using var stream = CreateStream(buffer, length); |
| | 6502 | 65 | | if (async) |
| | 3251 | 66 | | { |
| | 3251 | 67 | | await stream.CopyToAsync(Stream.Null); |
| | 2088 | 68 | | } |
| | | 69 | | else |
| | 3251 | 70 | | { |
| | 3251 | 71 | | stream.CopyTo(Stream.Null); |
| | 2088 | 72 | | } |
| | 4176 | 73 | | } |
| | 2326 | 74 | | catch (InvalidDataException) |
| | 2326 | 75 | | { |
| | | 76 | | // ignore, this exception is expected to be thrown for invalid/corrupted archives. |
| | 2326 | 77 | | } |
| | | 78 | | finally |
| | 6502 | 79 | | { |
| | 6502 | 80 | | ArrayPool<byte>.Shared.Return(buffer); |
| | 6502 | 81 | | } |
| | 6502 | 82 | | } |
| | | 83 | | } |