From 59faf95885c61818cd9a0289de973d6964889ace Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= <iwind.liu@gmail.com>
Date: Thu, 31 Mar 2022 16:30:15 +0800
Subject: [PATCH] =?UTF-8?q?=E9=99=90=E5=88=B6WebP=E5=8F=AF=E8=BD=AC?=
 =?UTF-8?q?=E6=8D=A2=E7=9A=84=E6=9C=80=E5=A4=A7=E9=95=BF=E5=BA=A6=E4=B8=BA?=
 =?UTF-8?q?128M=EF=BC=88=E9=9D=9EChunkEncoding=E4=B8=8B=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/nodes/http_writer.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/internal/nodes/http_writer.go b/internal/nodes/http_writer.go
index ba06d6d..aa9f818 100644
--- a/internal/nodes/http_writer.go
+++ b/internal/nodes/http_writer.go
@@ -12,6 +12,7 @@ import (
 	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
 	"github.com/TeaOSLab/EdgeNode/internal/utils"
 	"github.com/TeaOSLab/EdgeNode/internal/utils/readers"
+	"github.com/TeaOSLab/EdgeNode/internal/utils/sizes"
 	"github.com/TeaOSLab/EdgeNode/internal/utils/writers"
 	_ "github.com/biessek/golang-ico"
 	"github.com/iwind/TeaGo/lists"
@@ -448,6 +449,16 @@ func (this *HTTPWriter) PrepareWebP(resp *http.Response, size int64) {
 		return
 	}
 
+	// 限制最小和最大尺寸
+	// TODO 需要可以在集群里WebP选项里设置
+	// TODO 需要将reader修改为LimitReader
+	if resp.ContentLength == 0 {
+		return
+	}
+	if resp.ContentLength > 128*sizes.M {
+		return
+	}
+
 	var contentType = this.GetHeader("Content-Type")
 
 	if this.req.web != nil &&