From 680d18a5734648c8d3b30644288fe125b9f6ce4e Mon Sep 17 00:00:00 2001 From: M09Ic Date: Wed, 8 Feb 2023 19:18:33 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B7=B3=E8=BF=87=E4=BA=8C=E8=BF=9B=E5=88=B6?= =?UTF-8?q?=E7=9A=84=E8=BF=94=E5=9B=9E=E6=95=B0=E6=8D=AE=E8=AF=BB=E5=8F=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pkg/baseline.go | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/pkg/baseline.go b/pkg/baseline.go index c1924fa..1e825ab 100644 --- a/pkg/baseline.go +++ b/pkg/baseline.go @@ -18,12 +18,20 @@ func NewBaseline(u, host string, resp *ihttp.Response) *Baseline { Frameworks: make(parsers.Frameworks), }, } + + if t, ok := ContentTypeMap[resp.ContentType()]; ok { + bl.ContentType = t + bl.Title = t + " data" + } else { + bl.ContentType = "other" + } + header := resp.Header() bl.Header = make([]byte, len(header)) copy(bl.Header, header) bl.HeaderLength = len(bl.Header) - if i := resp.ContentLength(); i != 0 { + if i := resp.ContentLength(); i != 0 && bl.ContentType != "bin" { body := resp.Body() bl.Body = make([]byte, len(body)) copy(bl.Body, body) @@ -36,12 +44,6 @@ func NewBaseline(u, host string, resp *ihttp.Response) *Baseline { } } - if t, ok := ContentTypeMap[resp.ContentType()]; ok { - bl.ContentType = t - bl.Title = t + " data" - } else { - bl.ContentType = "other" - } bl.Raw = append(bl.Header, bl.Body...) if r := resp.GetHeader("Location"); r != "" { bl.RedirectURL = r