From 4f7257af66479944943c99ba0aeb8d999ee56ee9 Mon Sep 17 00:00:00 2001
From: Geo Perez <geovanni.perez@gmail.com>
Date: Mon, 12 Sep 2016 12:39:41 -0500
Subject: [PATCH] Add limit to Gzip to 4Mb

---
 Unosquare.Labs.EmbedIO.Samples/StaticFilesSample.cs | 5 ++---
 Unosquare.Labs.EmbedIO/Modules/StaticFilesModule.cs | 9 ++++++++-
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/Unosquare.Labs.EmbedIO.Samples/StaticFilesSample.cs b/Unosquare.Labs.EmbedIO.Samples/StaticFilesSample.cs
index 006a25a31..b30b03a45 100644
--- 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
index 7981befcc..ae4bca20b 100644
--- 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();