small refactor

projectdiscovery · Oct 31, 2023 · 19db3cf · 19db3cf
1 parent 6d0c3ca
commit 19db3cf
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/pkg/engine/common/base.go b/pkg/engine/common/base.go
@@ -239,8 +239,7 @@ func (s *Shared) Do(crawlSession *CrawlSession, doRequest DoRequestFunc) error {
 			if resp.Resp == nil || resp.Reader == nil {
 				return
 			}
-			isRedirectResponse := resp.StatusCode >= 300 && resp.StatusCode < 400
-			if s.Options.Options.DisableRedirects && isRedirectResponse {
+			if s.Options.Options.DisableRedirects && resp.IsRedirect() {
 				return
 			}
 

diff --git a/pkg/engine/hybrid/crawl.go b/pkg/engine/hybrid/crawl.go
@@ -130,8 +130,7 @@ func (c *Crawler) navigateRequest(s *common.CrawlSession, request *navigation.Re
 		c.Enqueue(s.Queue, navigationRequests...)
 
 		// do not continue following the request if it's a redirect and redirects are disabled
-		isRedirectResponse := statusCode >= 300 && statusCode < 400
-		if c.Options.Options.DisableRedirects && isRedirectResponse {
+		if c.Options.Options.DisableRedirects && resp.IsRedirect() {
 			return nil
 		}
 		return FetchContinueRequest(page, e)

diff --git a/pkg/navigation/response.go b/pkg/navigation/response.go
@@ -58,3 +58,7 @@ func (n Response) AbsoluteURL(path string) string {
 	final := absURL.String()
 	return final
 }
+
+func (n Response) IsRedirect() bool {
+	return n.StatusCode >= 300 && n.StatusCode <= 399
+}