mirror of https://github.com/daffainfo/nuclei.git
36 lines
782 B
Go
36 lines
782 B
Go
|
package extractors
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"regexp"
|
||
|
|
)
|
||
|
|
|
||
|
|
// CompileExtractors performs the initial setup operation on a extractor
|
||
|
|
func (e *Extractor) CompileExtractors() error {
|
||
|
|
// Setup the matcher type
|
||
|
|
_, ok := ExtractorTypes[e.Type]
|
||
|
|
if !ok {
|
||
|
|
return fmt.Errorf("unknown extractor type specified: %s", e.Type)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Compile the regexes
|
||
|
|
for _, regex := range e.Regex {
|
||
|
|
compiled, err := regexp.Compile(regex)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not compile regex: %s", regex)
|
||
|
|
}
|
||
|
|
e.regexCompiled = append(e.regexCompiled, compiled)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Setup the part of the request to match, if any.
|
||
|
|
if e.Part != "" {
|
||
|
|
e.part, ok = PartTypes[e.Part]
|
||
|
|
if !ok {
|
||
|
|
return fmt.Errorf("unknown matcher part specified: %s", e.Part)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
e.part = BodyPart
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|