Add limit to Gzip to 4Mb

unosquare · Sep 12, 2016 · 4f7257a · 4f7257a
1 parent 8ca03e3
commit 4f7257a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/Unosquare.Labs.EmbedIO.Samples/StaticFilesSample.cs b/Unosquare.Labs.EmbedIO.Samples/StaticFilesSample.cs
@@ -4,7 +4,7 @@
     using Unosquare.Labs.EmbedIO.Modules;
 
     /// <summary>
-    /// 
+    /// Sample helper
     /// </summary>
     public static class StaticFilesSample
     {
@@ -23,7 +23,7 @@ public static string HtmlRootPath
                 // This lets you edit the files without restarting the server.
                 return Path.GetFullPath(Path.Combine(assemblyPath, "..\\..\\html"));
 #else
-                // This is when you have deployed ythe server.
+                // This is when you have deployed the server.
                 return Path.Combine(assemblyPath, "html");
 #endif
             }
@@ -38,7 +38,6 @@ public static void Setup(WebServer server)
             server.RegisterModule(new StaticFilesModule(HtmlRootPath));
             // The static files module will cache small files in ram until it detects they have been modified.
             server.Module<StaticFilesModule>().UseRamCache = false;
-            server.Module<StaticFilesModule>().UseGzip = false;
             server.Module<StaticFilesModule>().DefaultExtension = ".html"; 
             // We don't need to add the line below. The default document is always index.html.
             //server.Module<StaticFilesWebModule>().DefaultDocument = "index.html";

diff --git a/Unosquare.Labs.EmbedIO/Modules/StaticFilesModule.cs b/Unosquare.Labs.EmbedIO/Modules/StaticFilesModule.cs
@@ -19,6 +19,11 @@ public class StaticFilesModule : WebModuleBase
         /// </summary>
         private const int ChuckSize = 256*1024;
 
+        /// <summary>
+        /// The maximum gzip input length
+        /// </summary>
+        private const int MaxGzipInputLength = 4*1024*1024;
+
         private readonly Dictionary<string, string> m_VirtualPaths =
             new Dictionary<string, string>(StringComparer.InvariantCultureIgnoreCase);
 
@@ -298,7 +303,9 @@ private bool HandleGet(HttpListenerContext context, WebServer server, bool sendB
             }
             else
             {
-                if (UseGzip && context.RequestHeader(Constants.HeaderAcceptEncoding).Contains(Constants.HeaderCompressionGzip))
+                if (UseGzip &&
+                    context.RequestHeader(Constants.HeaderAcceptEncoding).Contains(Constants.HeaderCompressionGzip) &&
+                    buffer.Length < MaxGzipInputLength)
                 {
                     // Perform compression if available
                     buffer = buffer.Compress();