External/nuclei
1
0
Fork 0
mirror of https://github.com/projectdiscovery/nuclei.git synced 2025-12-22 21:05:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix can't set user agent in templates

Browse Source
This commit is contained in:
Gia. Bui Dai 2021-04-13 13:27:36 +07:00
parent 781a71e3d5
commit 7f82270ea7
No known key found for this signature in database
GPG Key ID: 56FDE34966629045
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
v2/pkg/protocols/http/build_request.go
View File

@ -11,6 +11,7 @@ import (
"strings" "strings"
"time" "time"
"github.com/corpix/uarand"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/expressions" "github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/expressions"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/generators" "github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/generators"
@ -182,7 +183,7 @@ func (r *requestGenerator) fillRequest(req *http.Request, values map[string]inte
if r.request.Body != "" { if r.request.Body != "" {
req.Body = ioutil.NopCloser(strings.NewReader(r.request.Body)) req.Body = ioutil.NopCloser(strings.NewReader(r.request.Body))
} }
setHeader(req, "User-Agent", "Nuclei - Open-source project (github.com/projectdiscovery/nuclei)") setHeader(req, "User-Agent", uarand.GetRandom())
// Only set these headers on non raw requests // Only set these headers on non raw requests
if len(r.request.Raw) == 0 { if len(r.request.Raw) == 0 {

5
v2/pkg/protocols/http/http.go
View File

@ -3,7 +3,6 @@ package http
import ( import (
"strings" "strings"
"github.com/corpix/uarand"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/projectdiscovery/nuclei/v2/pkg/operators" "github.com/projectdiscovery/nuclei/v2/pkg/operators"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols" "github.com/projectdiscovery/nuclei/v2/pkg/protocols"
@ -102,10 +101,6 @@ func (r *Request) Compile(options *protocols.ExecuterOptions) error {
} }
r.customHeaders[parts[0]] = strings.TrimSpace(parts[1]) r.customHeaders[parts[0]] = strings.TrimSpace(parts[1])
} }
// Add User-Agent value randomly to the customHeaders slice if `random-agent` flag is given
if _, ok := r.customHeaders["User-Agent"]; !ok {
r.customHeaders["User-Agent"] = uarand.GetRandom()
}
if r.Body != "" && !strings.Contains(r.Body, "\r\n") { if r.Body != "" && !strings.Contains(r.Body, "\r\n") {
r.Body = strings.ReplaceAll(r.Body, "\n", "\r\n") r.Body = strings.ReplaceAll(r.Body, "\n", "\r\n")