plugin

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

type CustomLib struct {
	EnvOptions        []cel.EnvOption
	ProgramOptionList []cel.ProgramOption
}

type GoPlugin interface {
	Init() Plugin
	Check(netloc string, meta TaskMeta) bool
	GetResult() []Plugin
}

type JSONPlugin struct {
	Target  string `json:"target"` // 插件对应资产类型
	Meta    Plugin `json:"meta,omitempty"`
	Request struct {
		Path     string            `json:"path"`
		PostData string            `json:"postdata,omitempty"`
		Headers  map[string]string `json:"headers,omitempty"`
	} `json:"request"`
	Verify struct {
		Type  string `json:"type"`
		Match string `json:"match"`
	} `json:"verify"`
	Extra bool
}

type ListItem struct {
	Key   string
	Value []string
}

type ListMap []ListItem

type Plugin struct {
	PluginType string     `json:"plugintype,omitempty"` //插件类型
	Name       string     `json:"name"`
	Remarks    string     `json:"remarks"`
	Level      int        `json:"level"` //0:严重 1:高危 2:中危 3:低危 4:提示
	Type       string     `json:"type"`  //漏洞类型
	Author     string     `json:"author"`
	References References `json:"references"`
	Target     string     `json:"target,omitempty"`
	// json plugin
	Request  string `json:"request,omitempty"`
	Response string `json:"response,omitempty"`
	// yaml plugin
	Set        string `json:"set,omitempty"`
	Sets       string `json:"sets,omitempty"`
	Rules      string `json:"rules,omitempty"`
	Expression string `json:"expression,omitempty"`

	Reviewer      string    `bson:"reviewer" json:"reviewer"`
	ReviewStatus  uint      `bson:"reviewstatus" json:"reviewstatus"` //1：未审核 2: 审核中 3: 已审核
	ReviewOpinion string    `bson:"reviewopinion" json:"reviewopinion"`
	ApplyTime     time.Time `bson:"applytime" json:"applytime"`
	ReviewTime    time.Time `bson:"reviewtime" json:"reviewtime"`
	UpdateTime    time.Time `bson:"updatetime" json:"updatetime"`
}

type PocDetail struct {
	Author        string    `yaml:"author"`
	Reference     []string  `yaml:"reference,omitempty"`
	Severity      string    `yaml:"severity"` //critical:严重 high:高危 medium:中危 low:低危 warn: 提示
	Description   string    `yaml:"description"`
	VulType       string    `yaml:"vultype"`
	KPID          string    `yaml:"kpid"`
	CVE           string    `yaml:"cve,omitempty"`
	Reviewer      string    `yaml:"reviewer"`
	ReviewStatus  uint      `yaml:"reviewstatus"`
	ReviewOpinion string    `yaml:"reviewopinion"`
	ApplyTime     time.Time `yaml:"applytime"`
	ReviewTime    time.Time `yaml:"reviewtime"`
	UpdateTime    time.Time `yaml:"updatetime"`
}

type References struct {
	URL  string `json:"url,omitempty"` //参考链接
	CVE  string `json:"cve,omitempty"`
	KPID string `json:"kpid"`
}

type Rule struct {
	Request    RuleRequest   `yaml:"request"`
	Expression string        `yaml:"expression"`
	Output     yaml.MapSlice `yaml:"output,omitempty"`
	// contains filtered or unexported fields
}

type RuleMap struct {
	Key   string
	Value Rule
}

type RuleMapSlice []RuleMap

type RuleRequest struct {
	Method          string            `yaml:"method,omitempty"`
	Path            string            `yaml:"path,omitempty"`
	Body            string            `yaml:"body,omitempty"`
	FollowRedirects bool              `yaml:"follow_redirects,omitempty"`
	Headers         map[string]string `yaml:"headers,omitempty"`
	// 以下字段用于tcp类漏洞验证
	Type     string `yaml:"type,omitempty"`
	Host     string `yaml:"host,omitempty"`
	Data     string `yaml:"data,omitempty"`
	DataType string `yaml:"data-type,omitempty"`
}

type Task struct {
	Type   string   `json:"type"`
	Netloc string   `json:"netloc"`
	Target string   `json:"target"` //扫描插件ID
	Meta   TaskMeta `json:"meta"`
}

type TaskMeta struct {
	System           string   `json:"system"`
	PathList         []string `json:"pathlist"`
	FileList         []string `json:"filelist"`
	PassList         []string `json:"passlist"`
	AppUser          string   `json:"appuser"`
	WeakPassInterval int      `json:"weakpassinterval"`
}

type TmpRule struct {
	Request    RuleRequest   `yaml:"request"`
	Expression string        `yaml:"expression"`
	Output     yaml.MapSlice `yaml:"output"`
}

type YAMLPlugin struct {
	Target     string        `yaml:"target"`
	Name       string        `yaml:"name"`
	Set        yaml.MapSlice `yaml:"set,omitempty"`
	Sets       ListMap       `yaml:"sets"`
	Rules      RuleMapSlice  `yaml:"rules"`
	Expression string        `yaml:"expression"`
	Detail     PocDetail     `yaml:"detail"`
}