poc

type Classification struct {
	CvssMetrics string  `yaml:"cvss-metrics"`
	CvssScore   float64 `yaml:"cvss-score"`
	CveId       string  `yaml:"cve-id"`
	CweId       string  `yaml:"cwe-id"`
}

type EmbedPocFinderFunc func(pocId string) ([]byte, error)

type Extractors struct {
	Type      string        `yaml:"type"`      // regex,str
	Extractor yaml.MapSlice `yaml:"extractor"` //
}

type Info struct {
	Name           string         `yaml:"name"`
	Author         string         `yaml:"author"`
	Severity       string         `yaml:"severity"`
	Verified       bool           `yaml:"verified"`
	Description    string         `yaml:"description"`
	Reference      []string       `yaml:"reference"`
	Affected       string         `yaml:"affected"`  // 影响版本
	Solutions      string         `yaml:"solutions"` // 解决方案
	Tags           string         `yaml:"tags"`      // 标签
	Requires       []string       `yaml:"requires"`
	RequiresMode   string         `yaml:"requires-mode"`
	Classification Classification `yaml:"classification"`
	Created        string         `yaml:"created"` // create time
}

type MigrateReport struct {
	FilesSeen    int
	FilesChanged int
	Changes      int
}

type NetReadStep struct {
	ReadSize    int    `yaml:"read-size"`
	ReadTimeout int    `yaml:"read-timeout"`
	ReadUntil   string `yaml:"read-until,omitempty"`
	ReadType    string `yaml:"read-type,omitempty"`
	SaveAs      string `yaml:"save-as"`
}

type NetStep struct {
	Read  *NetReadStep  `yaml:"read,omitempty"`
	Write *NetWriteStep `yaml:"write,omitempty"`
}

type NetWriteStep struct {
	Data     string `yaml:"data"`
	DataType string `yaml:"data-type"`
}

type Payloads struct {
	Continue bool          `yaml:"continue"`
	Payloads yaml.MapSlice `yaml:"payloads"`
}

type Poc struct {
	Id         string        `yaml:"id"`        //  脚本名称
	Transport  string        `yaml:"transport"` // 传输方式，该字段用于指定发送数据包的协议，该字段用于指定发送数据包的协议:①tcp ②udp ③http
	Set        yaml.MapSlice `yaml:"set"`       // 全局变量定义，该字段用于定义全局变量。比如随机数，反连平台等
	Payloads   Payloads      `yaml:"payloads"`
	Rules      RuleMapSlice  `yaml:"rules"`
	Expression string        `yaml:"expression"`
	Info       Info          `yaml:"info"`
	Gopoc      string        `yaml:"gopoc"` // Gopoc 脚本名称
	Extractors []Extractors  `yaml:"extractors"`

	EstimatedTaskTimeoutSec    int    `yaml:"-"`
	EstimatedTaskTimeoutReason string `yaml:"-"`
}

type PocMeta struct {
	Id   string `yaml:"id"`
	Info Info   `yaml:"info"`
}

type Rule struct {
	Brute          yaml.MapSlice `yaml:"brute"`
	Request        RuleRequest   `yaml:"request"`
	Expression     string        `yaml:"expression"`
	Expressions    []string      `yaml:"expressions"`
	Output         yaml.MapSlice `yaml:"output"`
	Extractors     []Extractors  `yaml:"extractors"`
	StopIfMatch    bool          `yaml:"stop_if_match"`
	StopIfMismatch bool          `yaml:"stop_if_mismatch"`
	BeforeSleep    int           `yaml:"before_sleep"`
	// contains filtered or unexported fields
}

type RuleMap struct {
	Key   string
	Value Rule
}

type RuleMapSlice []RuleMap

type RuleRequest struct {
	Type            string            `yaml:"type"`         // 传输方式，默认 http，可选：tcp,udp,ssl,go 等任意扩展
	Host            string            `yaml:"host"`         // tcp/udp 请求的主机名
	Port            int               `yaml:"port"`         // tcp/udp 端口（可选，优先级低于 host 中显式端口）
	Data            string            `yaml:"data"`         // tcp/udp 发送的内容
	DataType        string            `yaml:"data-type"`    // tcp/udp 发送的数据类型，默认字符串
	ReadSize        int               `yaml:"read-size"`    // tcp/udp 读取内容的长度
	ReadTimeout     int               `yaml:"read-timeout"` // tcp/udp专用
	Steps           []NetStep         `yaml:"steps"`
	Raw             string            `yaml:"raw"` // raw 专用
	Method          string            `yaml:"method"`
	Path            string            `yaml:"path"`
	Headers         map[string]string `yaml:"headers"`
	Body            string            `yaml:"body"`
	FollowRedirects bool              `yaml:"follow_redirects"`
}

type TaskTimeoutEstimate struct {
	TimeoutSec int
	Reason     string
}

type TaskTimeoutPolicy struct {
	VisibleCapSec int
	NetCapSec     int
	GoCapSec      int
}

type WaitGroupTask struct {
	Key   int
	Value any
}