mirror of https://github.com/daffainfo/nuclei.git
96 lines
2.1 KiB
Go
96 lines
2.1 KiB
Go
package extractors
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// Extract extracts response from the parts of request using a regex
|
|
func (e *Extractor) Extract(resp *http.Response, body, headers string) map[string]struct{} {
|
|
switch e.extractorType {
|
|
case RegexExtractor:
|
|
if e.part == BodyPart {
|
|
return e.extractRegex(body)
|
|
} else if e.part == HeaderPart {
|
|
return e.extractRegex(headers)
|
|
} else {
|
|
matches := e.extractRegex(headers)
|
|
if len(matches) > 0 {
|
|
return matches
|
|
}
|
|
return e.extractRegex(body)
|
|
}
|
|
case KValExtractor:
|
|
if e.part == HeaderPart {
|
|
return e.extractKVal(resp)
|
|
}
|
|
|
|
matches := e.extractKVal(resp)
|
|
|
|
if len(matches) > 0 {
|
|
return matches
|
|
}
|
|
|
|
return e.extractCookieKVal(resp)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ExtractDNS extracts response from dns message using a regex
|
|
// nolint:interfacer // dns.Msg is out of current scope
|
|
func (e *Extractor) ExtractDNS(msg *dns.Msg) map[string]struct{} {
|
|
switch e.extractorType {
|
|
case RegexExtractor:
|
|
return e.extractRegex(msg.String())
|
|
case KValExtractor:
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// extractRegex extracts text from a corpus and returns it
|
|
func (e *Extractor) extractRegex(corpus string) map[string]struct{} {
|
|
results := make(map[string]struct{})
|
|
|
|
groupPlusOne := e.RegexGroup + 1
|
|
for _, regex := range e.regexCompiled {
|
|
matches := regex.FindAllStringSubmatch(corpus, -1)
|
|
for _, match := range matches {
|
|
if len(match) >= groupPlusOne {
|
|
results[match[e.RegexGroup]] = struct{}{}
|
|
}
|
|
}
|
|
}
|
|
return results
|
|
}
|
|
|
|
// extractKVal extracts text from http response
|
|
func (e *Extractor) extractKVal(r *http.Response) map[string]struct{} {
|
|
results := make(map[string]struct{})
|
|
|
|
for _, k := range e.KVal {
|
|
for _, v := range r.Header.Values(k) {
|
|
results[v] = struct{}{}
|
|
}
|
|
}
|
|
|
|
return results
|
|
}
|
|
|
|
// extractCookieKVal extracts text from cookies
|
|
func (e *Extractor) extractCookieKVal(r *http.Response) map[string]struct{} {
|
|
results := make(map[string]struct{})
|
|
|
|
for _, k := range e.KVal {
|
|
for _, cookie := range r.Cookies() {
|
|
if cookie.Name == k {
|
|
results[cookie.Value] = struct{}{}
|
|
}
|
|
}
|
|
}
|
|
|
|
return results
|
|
}
|