From 94b496354bbc02e4d0e6b8a750888755960472cb Mon Sep 17 00:00:00 2001
From: Chocobo1 <Chocobo1@users.noreply.github.com>
Date: Mon, 17 Apr 2017 14:12:20 +0800
Subject: [PATCH] Rewrite rules for gzipping http response content

---
 src/base/http/responsegenerator.cpp | 42 +++++++++++++++++++++--------
 src/base/http/responsegenerator.h   |  1 +
 2 files changed, 32 insertions(+), 11 deletions(-)

diff --git a/src/base/http/responsegenerator.cpp b/src/base/http/responsegenerator.cpp
index cee8c3df7..324bfd552 100644
--- a/src/base/http/responsegenerator.cpp
+++ b/src/base/http/responsegenerator.cpp
@@ -37,17 +37,7 @@
 
 QByteArray Http::toByteArray(Response response)
 {
-    if (response.headers.value(HEADER_CONTENT_ENCODING) == "gzip") {
-        // A gzip seems to have 23 bytes overhead.
-        // Also "Content-Encoding: gzip\r\n" is 26 bytes long
-        // So we only benefit from gzip if the message is bigger than 23+26 = 49
-        // If the message is smaller than 49 bytes we actually send MORE data if we gzip
-        QByteArray destBuf;
-        if ((response.content.size() > 49) && (Utils::Gzip::compress(response.content, destBuf)))
-            response.content = destBuf;
-        else
-            response.headers.remove(HEADER_CONTENT_ENCODING);
-    }
+    compressContent(response);
 
     response.headers[HEADER_CONTENT_LENGTH] = QString::number(response.content.length());
     response.headers[HEADER_DATE] = httpDate();
@@ -84,3 +74,33 @@ QString Http::httpDate()
     return QLocale::c().toString(QDateTime::currentDateTimeUtc(), QLatin1String("ddd, dd MMM yyyy HH:mm:ss"))
         .append(QLatin1String(" GMT"));
 }
+
+void Http::compressContent(Response &response)
+{
+    if (response.headers.value(HEADER_CONTENT_ENCODING) != QLatin1String("gzip"))
+        return;
+
+    response.headers.remove(HEADER_CONTENT_ENCODING);
+
+    // for very small files, compressing them only wastes cpu cycles
+    const int contentSize = response.content.size();
+    if (contentSize <= 1024)  // 1 kb
+        return;
+
+    // filter out known hard-to-compress types
+    const QString contentType = response.headers[HEADER_CONTENT_TYPE];
+    if ((contentType == CONTENT_TYPE_GIF) || (contentType == CONTENT_TYPE_PNG))
+        return;
+
+    // try compressing
+    QByteArray buf;
+    if (!Utils::Gzip::compress(response.content, buf))
+        return;
+
+    // "Content-Encoding: gzip\r\n" is 24 bytes long
+    if ((buf.size() + 24) >= contentSize)
+        return;
+
+    response.content = buf;
+    response.headers[HEADER_CONTENT_ENCODING] = QLatin1String("gzip");
+}
diff --git a/src/base/http/responsegenerator.h b/src/base/http/responsegenerator.h
index a2fbb2ce2..7c714e050 100644
--- a/src/base/http/responsegenerator.h
+++ b/src/base/http/responsegenerator.h
@@ -39,6 +39,7 @@ namespace Http
 {
     QByteArray toByteArray(Response response);
     QString httpDate();
+    void compressContent(Response &response);
 }
 
 #endif // HTTP_RESPONSEGENERATOR_H