-
Notifications
You must be signed in to change notification settings - Fork 6.1k
Add Compression best practices guide #52968
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Changes from all commits
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,258 @@ | ||
| using System.Formats.Tar; | ||
| using System.IO.Compression; | ||
| // <SafeExtractEntry> | ||
| void SafeExtractEntry(ZipArchiveEntry entry, string destinationPath, long maxDecompressedSize) | ||
| { | ||
| // The runtime enforces that entry.Open() will never produce more than | ||
| // entry.Length bytes, so checking the declared size is sufficient. | ||
| if (entry.Length > maxDecompressedSize) | ||
| { | ||
| throw new InvalidOperationException( | ||
| $"Entry '{entry.FullName}' declares size {entry.Length}, exceeding limit {maxDecompressedSize}."); | ||
| } | ||
|
|
||
| entry.ExtractToFile(destinationPath, overwrite: false); | ||
| } | ||
| // </SafeExtractEntry> | ||
|
|
||
| // <SafeExtractArchive> | ||
| void SafeExtractArchive(ZipArchive archive, string destinationDir, | ||
| long maxTotalSize, int maxEntryCount) | ||
| { | ||
| // Some zip bombs contain millions of tiny entries (e.g., "42.zip"). | ||
| if (archive.Entries.Count > maxEntryCount) | ||
| { | ||
| throw new InvalidOperationException("Archive contains an excessive number of entries."); | ||
| } | ||
|
|
||
| long totalExtracted = 0; | ||
| foreach (ZipArchiveEntry entry in archive.Entries) | ||
| { | ||
| totalExtracted += entry.Length; | ||
| if (totalExtracted > maxTotalSize) | ||
| { | ||
| throw new InvalidOperationException( | ||
| $"Archive total decompressed size exceeds the allowed limit of {maxTotalSize} bytes."); | ||
| } | ||
| // ... extract each entry with per-entry limits too | ||
| } | ||
| } | ||
| // </SafeExtractArchive> | ||
|
|
||
| // <PathValidation> | ||
| void ValidatePaths(ZipArchive archive, string destinationDir) | ||
| { | ||
| string fullDestDir = Path.GetFullPath(destinationDir); | ||
| if (!fullDestDir.EndsWith(Path.DirectorySeparatorChar)) | ||
| fullDestDir += Path.DirectorySeparatorChar; | ||
|
|
||
| foreach (ZipArchiveEntry entry in archive.Entries) | ||
| { | ||
| string destPath = Path.GetFullPath(Path.Combine(fullDestDir, entry.FullName)); | ||
|
|
||
| if (!destPath.StartsWith(fullDestDir, StringComparison.Ordinal)) | ||
| throw new IOException( | ||
| $"Entry '{entry.FullName}' would extract outside the destination directory."); | ||
|
|
||
| // ... safe to extract | ||
| } | ||
| } | ||
| // </PathValidation> | ||
|
|
||
| // <VulnerablePattern> | ||
| void DangerousExtract(string extractDir) | ||
| { | ||
| // ⚠️ DANGEROUS: entry.FullName could contain "../" sequences | ||
| using ZipArchive archive = ZipFile.OpenRead("archive.zip"); | ||
| foreach (ZipArchiveEntry entry in archive.Entries) | ||
| { | ||
| string destinationPath = Path.Combine(extractDir, entry.FullName); | ||
| entry.ExtractToFile(destinationPath, overwrite: true); // May write outside of `extractDir` | ||
| } | ||
| } | ||
| // </VulnerablePattern> | ||
|
|
||
| // <SafeExtractZip> | ||
| void SafeExtractZip(string archivePath, string destinationDir, | ||
| long maxTotalSize, long maxEntrySize, int maxEntryCount) | ||
| { | ||
| // Resolve the destination to an absolute path and ensure it ends with a | ||
| // directory separator. This trailing separator is essential — without it, | ||
| // the StartsWith check below could be tricked by paths like | ||
| // "/safe-dir-evil/" matching "/safe-dir". | ||
| string fullDestDir = Path.GetFullPath(destinationDir); | ||
| if (!fullDestDir.EndsWith(Path.DirectorySeparatorChar)) | ||
| fullDestDir += Path.DirectorySeparatorChar; | ||
|
|
||
| Directory.CreateDirectory(fullDestDir); | ||
|
|
||
| using var archive = new ZipArchive(File.OpenRead(archivePath), ZipArchiveMode.Read); | ||
|
|
||
| // Check the entry count up front. ZIP central directory is read eagerly, | ||
| // so archive.Entries.Count is available immediately without iterating. | ||
| if (archive.Entries.Count > maxEntryCount) | ||
| throw new InvalidOperationException("Archive contains too many entries."); | ||
|
|
||
| long totalSize = 0; | ||
| foreach (ZipArchiveEntry entry in archive.Entries) | ||
| { | ||
| // Enforce per-entry and cumulative size limits using the declared | ||
| // uncompressed size. Note: this value is read from the archive header | ||
| // and could be spoofed by a malicious archive — for defense in depth, | ||
| // also monitor actual bytes read during decompression (see the zip | ||
| // bomb section for a streaming size check example). | ||
| totalSize += entry.Length; | ||
| if (entry.Length > maxEntrySize) | ||
| throw new InvalidOperationException( | ||
| $"Entry '{entry.FullName}' exceeds per-entry size limit."); | ||
| if (totalSize > maxTotalSize) | ||
| throw new InvalidOperationException("Archive exceeds total size limit."); | ||
|
|
||
| // Resolve the full destination path using Path.GetFullPath, which | ||
| // normalizes away any "../" segments. Then verify the result still | ||
| // starts with the destination directory. | ||
| string destPath = Path.GetFullPath(Path.Combine(fullDestDir, entry.FullName)); | ||
| if (!destPath.StartsWith(fullDestDir, StringComparison.Ordinal)) | ||
| throw new IOException( | ||
| $"Entry '{entry.FullName}' would extract outside the destination."); | ||
|
|
||
| // By convention, directory entries in ZIP archives have names ending | ||
| // in '/'. Path.GetFileName returns empty for these, so we use that | ||
| // to distinguish directories from files. | ||
| if (string.IsNullOrEmpty(Path.GetFileName(destPath))) | ||
| { | ||
| Directory.CreateDirectory(destPath); | ||
| } | ||
| else | ||
| { | ||
| // Create the parent directory and any missing intermediate directories. | ||
| Directory.CreateDirectory(Path.GetDirectoryName(destPath)!); | ||
| entry.ExtractToFile(destPath, overwrite: false); | ||
| } | ||
| } | ||
| } | ||
| // </SafeExtractZip> | ||
|
|
||
| // <SafeExtractTar> | ||
| void SafeExtractTar(Stream archiveStream, string destinationDir, | ||
| long maxTotalSize, long maxEntrySize, int maxEntryCount) | ||
| { | ||
| // Same trailing-separator technique as the ZIP example. | ||
| string fullDestDir = Path.GetFullPath(destinationDir); | ||
| if (!fullDestDir.EndsWith(Path.DirectorySeparatorChar)) | ||
| fullDestDir += Path.DirectorySeparatorChar; | ||
|
|
||
| Directory.CreateDirectory(fullDestDir); | ||
|
|
||
| using var reader = new TarReader(archiveStream); | ||
| TarEntry? entry; | ||
| long totalSize = 0; | ||
| int entryCount = 0; | ||
|
|
||
| // TAR has no central directory — entries are read one at a time. | ||
| // GetNextEntry() returns null when the archive is exhausted. | ||
| while ((entry = reader.GetNextEntry()) is not null) | ||
| { | ||
| if (++entryCount > maxEntryCount) | ||
| throw new InvalidOperationException("Archive contains too many entries."); | ||
|
|
||
| if (entry.Length > maxEntrySize) | ||
| throw new InvalidOperationException( | ||
| $"Entry '{entry.Name}' exceeds per-entry size limit."); | ||
| totalSize += entry.Length; | ||
| if (totalSize > maxTotalSize) | ||
| throw new InvalidOperationException("Archive exceeds total size limit."); | ||
|
|
||
| // Symbolic links and hard links can be used to write files outside the | ||
| // extraction directory or to overwrite sensitive files. The safest | ||
| // approach for untrusted input is to skip them entirely. | ||
| if (entry.EntryType is TarEntryType.SymbolicLink or TarEntryType.HardLink) | ||
| continue; | ||
alinpahontu2912 marked this conversation as resolved.
Show resolved
Hide resolved
|
||
|
|
||
| // Global extended attributes are PAX metadata entries that apply to all | ||
| // subsequent entries. They contain no file data and should be skipped. | ||
| if (entry.EntryType is TarEntryType.GlobalExtendedAttributes) | ||
| continue; | ||
|
|
||
| // Normalize and validate the path, same as the ZIP example. | ||
| string destPath = Path.GetFullPath(Path.Join(fullDestDir, entry.Name)); | ||
|
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Some samples use Path.Join, some Path.Combine, is there a meaningful difference between the two? |
||
| if (!destPath.StartsWith(fullDestDir, StringComparison.Ordinal)) | ||
| throw new IOException( | ||
| $"Entry '{entry.Name}' would extract outside the destination."); | ||
|
|
||
| if (entry.EntryType is TarEntryType.Directory) | ||
| { | ||
| Directory.CreateDirectory(destPath); | ||
| } | ||
| else if (entry.DataStream is not null) | ||
| { | ||
| // Create the parent directory and any missing intermediate directories. | ||
| Directory.CreateDirectory(Path.GetDirectoryName(destPath)!); | ||
| using var fileStream = File.Create(destPath); | ||
| entry.DataStream.CopyTo(fileStream); | ||
|
Comment on lines
+189
to
+192
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. What about |
||
| } | ||
| } | ||
| } | ||
| // </SafeExtractTar> | ||
|
|
||
| // <ValidateSymlink> | ||
| bool IsLinkTargetSafe(TarEntry entry, string fullDestDir) | ||
| { | ||
| string resolvedTarget; | ||
|
|
||
| if (entry.EntryType is TarEntryType.SymbolicLink) | ||
| { | ||
| // Symlink targets are relative to the symlink's own parent directory, or absolute. | ||
| string entryDir = Path.GetDirectoryName( | ||
| Path.GetFullPath(Path.Join(fullDestDir, entry.Name)))!; | ||
| resolvedTarget = Path.GetFullPath(Path.Join(entryDir, entry.LinkName)); | ||
| } | ||
| else | ||
| { | ||
| // Hard link targets are relative to the destination directory root. | ||
| resolvedTarget = Path.GetFullPath(Path.Join(fullDestDir, entry.LinkName)); | ||
| } | ||
|
|
||
| return resolvedTarget.StartsWith(fullDestDir, StringComparison.Ordinal); | ||
| } | ||
| // </ValidateSymlink> | ||
|
|
||
| // <StreamingApproach> | ||
| void StreamingModify() | ||
| { | ||
| // ✅ Streaming approach for large archives | ||
| using var input = new ZipArchive(File.OpenRead("large.zip"), ZipArchiveMode.Read); | ||
| using var output = new ZipArchive(File.Create("modified.zip"), ZipArchiveMode.Create); | ||
|
|
||
| foreach (var entry in input.Entries) | ||
| { | ||
| if (ShouldKeep(entry)) | ||
| { | ||
| var newEntry = output.CreateEntry(entry.FullName); | ||
| using var src = entry.Open(); | ||
| using var dst = newEntry.Open(); | ||
| src.CopyTo(dst); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| bool ShouldKeep(ZipArchiveEntry entry) => true; | ||
| // </StreamingApproach> | ||
|
|
||
| // <TarStreaming> | ||
| void TarStreamingRead(Stream archiveStream) | ||
| { | ||
| using var reader = new TarReader(archiveStream); | ||
| TarEntry? entry; | ||
| while ((entry = reader.GetNextEntry()) is not null) | ||
| { | ||
| if (entry.DataStream is not null) | ||
| { | ||
| string safePath = "output.bin"; | ||
| // Copy now — the stream becomes invalid after the next GetNextEntry() call | ||
| using var fileStream = File.Create(safePath); | ||
| entry.DataStream.CopyTo(fileStream); | ||
| } | ||
| } | ||
| } | ||
|
Comment on lines
+243
to
+257
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I am not sure what this code snippet is supposed to illustrate |
||
| // </TarStreaming> | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| <Project Sdk="Microsoft.NET.Sdk"> | ||
|
|
||
| <PropertyGroup> | ||
| <OutputType>Exe</OutputType> | ||
| <TargetFramework>net9.0</TargetFramework> | ||
| <ImplicitUsings>enable</ImplicitUsings> | ||
| <Nullable>enable</Nullable> | ||
| </PropertyGroup> | ||
|
|
||
|
|
||
| </Project> |
Uh oh!
There was an error while loading. Please reload this page.