nuclei/pkg/extractors/extractors.go

package extractors

import "regexp"

// Extractor is used to extract part of response using a regex.
type Extractor struct {
	// Regex are the regex pattern required to be present in the response
	Regex string `yaml:"regex"`
	// regexCompiled is the compiled variant
	regexCompiled *regexp.Regexp

	// Part is the part of the request to match
	//
	// By default, matching is performed in request body.
	Part string `yaml:"part,omitempty"`
	// part is the part of the request to match
	part Part
}

// Part is the part of the request to match
type Part int

const (
	// BodyPart matches body of the response.
	BodyPart Part = iota + 1
	// HeaderPart matches headers of the response.
	HeaderPart
	// AllPart matches both response body and headers of the response.
	AllPart
)

// PartTypes is an table for conversion of part type from string.
var PartTypes = map[string]Part{
	"body":   BodyPart,
	"header": HeaderPart,
	"all":    AllPart,
}

// GetPart returns the part of the matcher
func (e *Extractor) GetPart() Part {
	return e.part
}