From 1895496e0bfb1dc42f6c562144c5c9308bb4c421 Mon Sep 17 00:00:00 2001
From: M09Ic <m09ician@gmail.com>
Date: Thu, 17 Nov 2022 16:33:56 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=9F=90=E4=BA=9B=E6=83=85?=
 =?UTF-8?q?=E5=86=B5=E4=B8=8Bstatuscode=E6=97=A0=E6=B3=95=E6=AD=A3?=
 =?UTF-8?q?=E7=A1=AE=E8=AF=BB=E5=8F=96=E7=9A=84bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/types.go | 2 +-
 pkg/baseline.go   | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/internal/types.go b/internal/types.go
index 3b04e8d..9e0b58e 100644
--- a/internal/types.go
+++ b/internal/types.go
@@ -21,7 +21,7 @@ func (e ErrorType) Error() string {
 	case ErrRequestFailed:
 		return "request failed"
 	case ErrWaf:
-		return "maybe ban of waf"
+		return "maybe banned by waf"
 	case ErrRedirect:
 		return "duplicate redirect url"
 	case ErrCompareFailed:
diff --git a/pkg/baseline.go b/pkg/baseline.go
index d1d63d3..8376387 100644
--- a/pkg/baseline.go
+++ b/pkg/baseline.go
@@ -24,7 +24,7 @@ func NewBaseline(u, host string, resp *ihttp.Response) *Baseline {
 	}
 
 	bl.Body = resp.Body()
-	bl.BodyLength = resp.ContentLength()
+	bl.BodyLength = len(resp.Body())
 	bl.Header = resp.Header()
 	bl.HeaderLength = len(bl.Header)
 	bl.RedirectURL = resp.GetHeader("Location")
@@ -50,7 +50,7 @@ func NewInvalidBaseline(u, host string, resp *ihttp.Response, reason string) *Ba
 	}
 
 	bl.Body = resp.Body()
-	bl.BodyLength = resp.ContentLength()
+	bl.BodyLength = len(bl.Body)
 	bl.RedirectURL = string(resp.GetHeader("Location"))
 
 	return bl