lib

type AfrogPoc struct {
	ID   string `yaml:"id"`
	Info struct {
		Name        string         `yaml:"name"`
		Author      yamlStringList `yaml:"author"`
		Severity    string         `yaml:"severity"`
		Verified    bool           `yaml:"verified"`
		Description string         `yaml:"description"`
		Reference   yamlStringList `yaml:"reference"`
		Tags        string         `yaml:"tags"`
		Created     string         `yaml:"created"`
	} `yaml:"info"`
	Set        map[string]interface{} `yaml:"set"`
	Rules      map[string]XrayRule    `yaml:"rules"` // 复用 xray 的 rule 结构
	Expression string                 `yaml:"expression"`
}

type AfrogPocAdapter struct {
	*AfrogPoc
}

type CelProgCache map[string]cel.Program

type CustomLib struct {
	// contains filtered or unexported fields
}

type Detail struct {
	Author      string   `yaml:"author"`      // POC作者
	Links       []string `yaml:"links"`       // 相关链接
	Description string   `yaml:"description"` // POC描述
	Version     string   `yaml:"version"`     // POC版本
}

type FscanPocAdapter struct {
	*Poc
}

type ListItem struct {
	Key   string   // 键名
	Value []string // 值列表
}

type ListMap []ListItem

type MapSlice = yaml.MapSlice

type NucleiMatcher struct {
	Type      string   `yaml:"type"`
	Words     []string `yaml:"words"`
	Status    []int    `yaml:"status"`
	Regex     []string `yaml:"regex"`
	Condition string   `yaml:"condition"`
	Part      string   `yaml:"part"`
	Negative  bool     `yaml:"negative"`
}

type NucleiPoc struct {
	ID   string `yaml:"id"`
	Info struct {
		Name        string         `yaml:"name"`
		Author      yamlStringList `yaml:"author"`
		Severity    string         `yaml:"severity"`
		Description string         `yaml:"description"`
		Reference   yamlStringList `yaml:"reference"`
	} `yaml:"info"`
	HTTP []struct {
		Method            string            `yaml:"method"`
		Path              []string          `yaml:"path"`
		Headers           map[string]string `yaml:"headers"`
		Body              string            `yaml:"body"`
		Matchers          []NucleiMatcher   `yaml:"matchers"`
		MatchersCondition string            `yaml:"matchers-condition"`
	} `yaml:"http"`
}

type NucleiPocAdapter struct {
	*NucleiPoc
}

type POCContext struct {
	DNSLog  bool // 是否启用DNSLog检测
	POCFull bool // 是否完整POC扫描
	Session *common.ScanSession
}

type Poc struct {
	Name   string  `yaml:"name"`   // POC名称
	Set    StrMap  `yaml:"set"`    // 单值配置映射
	Sets   ListMap `yaml:"sets"`   // 列表值配置映射
	Rules  []Rules `yaml:"rules"`  // 检测规则列表
	Groups RuleMap `yaml:"groups"` // 规则组映射
	Detail Detail  `yaml:"detail"` // 漏洞详情
}

type PocFormat string

type Request struct {
	URL         *UrlType          `protobuf:"bytes,1,opt,name=url,proto3" json:"url,omitempty"`
	Method      string            `protobuf:"bytes,2,opt,name=method,proto3" json:"method,omitempty"`
	Headers     map[string]string `` /* 155-byte string literal not displayed */
	ContentType string            `protobuf:"bytes,4,opt,name=content_type,json=contentType,proto3" json:"content_type,omitempty"`
	Body        []byte            `protobuf:"bytes,5,opt,name=body,proto3" json:"body,omitempty"`
	// contains filtered or unexported fields
}

type Response struct {
	URL         *UrlType          `protobuf:"bytes,1,opt,name=url,proto3" json:"url,omitempty"`
	Status      int32             `protobuf:"varint,2,opt,name=status,proto3" json:"status,omitempty"`
	Headers     map[string]string `` /* 155-byte string literal not displayed */
	ContentType string            `protobuf:"bytes,4,opt,name=content_type,json=contentType,proto3" json:"content_type,omitempty"`
	Body        []byte            `protobuf:"bytes,5,opt,name=body,proto3" json:"body,omitempty"`
	Duration    float64           `protobuf:"fixed64,6,opt,name=duration,proto3" json:"duration,omitempty"`
	// contains filtered or unexported fields
}

type Reverse struct {
	URL                string `protobuf:"bytes,1,opt,name=url,proto3" json:"url,omitempty"`
	Domain             string `protobuf:"bytes,2,opt,name=domain,proto3" json:"domain,omitempty"`
	Ip                 string `protobuf:"bytes,3,opt,name=ip,proto3" json:"ip,omitempty"`
	IsDomainNameServer bool   `protobuf:"varint,4,opt,name=is_domain_name_server,json=isDomainNameServer,proto3" json:"is_domain_name_server,omitempty"`
	// contains filtered or unexported fields
}

type RuleItem struct {
	Key   string  // 键名
	Value []Rules // 规则列表
}

type RuleMap []RuleItem

type Rules struct {
	Method          string            `yaml:"method"`           // HTTP请求方法
	Path            string            `yaml:"path"`             // 请求路径
	Headers         map[string]string `yaml:"headers"`          // 请求头
	Body            string            `yaml:"body"`             // 请求体
	Search          string            `yaml:"search"`           // 搜索模式
	FollowRedirects bool              `yaml:"follow_redirects"` // 是否跟随重定向
	Expression      string            `yaml:"expression"`       // 匹配表达式
	Continue        bool              `yaml:"continue"`         // 是否继续执行
}

type StrItem struct {
	Key   string // 键名
	Value string // 值
}

type StrMap []StrItem

type Task struct {
	Req *http.Request // HTTP请求对象
	Poc *Poc          // POC检测脚本
}

type UniversalPoc interface {
	GetName() string           // 获取POC名称
	GetFormat() PocFormat      // 获取格式类型
	ToFscanPoc() (*Poc, error) // 转换为fscan内部格式
}

type UrlType struct {
	Scheme   string `protobuf:"bytes,1,opt,name=scheme,proto3" json:"scheme,omitempty"`
	Domain   string `protobuf:"bytes,2,opt,name=domain,proto3" json:"domain,omitempty"`
	Host     string `protobuf:"bytes,3,opt,name=host,proto3" json:"host,omitempty"`
	Port     string `protobuf:"bytes,4,opt,name=port,proto3" json:"port,omitempty"`
	Path     string `protobuf:"bytes,5,opt,name=path,proto3" json:"path,omitempty"`
	Query    string `protobuf:"bytes,6,opt,name=query,proto3" json:"query,omitempty"`
	Fragment string `protobuf:"bytes,7,opt,name=fragment,proto3" json:"fragment,omitempty"`
	// contains filtered or unexported fields
}

type VulnResult struct {
	Poc     *Poc                   // POC脚本
	VulName string                 // 漏洞名称
	Target  string                 // 目标URL
	Details map[string]interface{} // 详细信息
}

type XrayPoc struct {
	Name       string                 `yaml:"name"`
	Transport  string                 `yaml:"transport"`
	Set        map[string]interface{} `yaml:"set"`
	Rules      map[string]XrayRule    `yaml:"rules"`
	Expression string                 `yaml:"expression"`
	Detail     Detail                 `yaml:"detail"`
}

type XrayPocAdapter struct {
	*XrayPoc
}

type XrayRule struct {
	Request struct {
		Cache           bool              `yaml:"cache"`
		Method          string            `yaml:"method"`
		Path            string            `yaml:"path"`
		Headers         map[string]string `yaml:"headers"`
		Body            string            `yaml:"body"`
		FollowRedirects bool              `yaml:"follow_redirects"`
	} `yaml:"request"`
	Expression string                 `yaml:"expression"`
	Output     map[string]interface{} `yaml:"output"`
}