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