From e663cdca3df5471ffbf7bfc2d329e7780e5ccf6f Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Mon, 1 Jan 2018 09:48:16 +0100 Subject: 10M limit --- href.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/href.go b/href.go index bd4cec6..a4cc5c9 100644 --- a/href.go +++ b/href.go @@ -11,7 +11,10 @@ import ( "golang.org/x/net/html/charset" ) -var errNotHTML = errors.New("Not HTML") +var ( + errNotHTML = errors.New("Not HTML") + errTooBig = errors.New("Content too big") +) func title(n *html.Node) (s string) { if n.Type == html.ElementNode && n.Data == "title" { @@ -40,6 +43,10 @@ func getTitle(uri string) (string, error) { return "", errNotHTML } + if resp.ContentLength > 10*1024*1024 { + return "", errTooBig + } + r, err := charset.NewReader(resp.Body, ct) if err != nil { return "", err -- cgit v1.2.3