mirror of https://github.com/daffainfo/nuclei.git
120 lines
3.4 KiB
Go
120 lines
3.4 KiB
Go
package workflows
|
|
|
|
import (
|
|
"github.com/projectdiscovery/gologger"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/output"
|
|
"github.com/remeh/sizedwaitgroup"
|
|
"go.uber.org/atomic"
|
|
)
|
|
|
|
// RunWorkflow runs a workflow on an input and returns true or false
|
|
func (w *Workflow) RunWorkflow(input string) bool {
|
|
results := &atomic.Bool{}
|
|
|
|
swg := sizedwaitgroup.New(w.Options.Options.TemplateThreads)
|
|
for _, template := range w.Workflows {
|
|
swg.Add()
|
|
func(template *WorkflowTemplate) {
|
|
err := w.runWorkflowStep(template, input, results, &swg)
|
|
if err != nil {
|
|
gologger.Warning().Msgf("[%s] Could not execute workflow step: %s\n", template.Template, err)
|
|
}
|
|
swg.Done()
|
|
}(template)
|
|
}
|
|
swg.Wait()
|
|
return results.Load()
|
|
}
|
|
|
|
// runWorkflowStep runs a workflow step for the workflow. It executes the workflow
|
|
// in a recursive manner running all subtemplates and matchers.
|
|
func (w *Workflow) runWorkflowStep(template *WorkflowTemplate, input string, results *atomic.Bool, swg *sizedwaitgroup.SizedWaitGroup) error {
|
|
var firstMatched bool
|
|
var err error
|
|
var mainErr error
|
|
|
|
if len(template.Matchers) == 0 {
|
|
for _, executer := range template.Executers {
|
|
executer.Options.Progress.AddToTotal(int64(executer.Executer.Requests()))
|
|
|
|
// Don't print results with subtemplates, only print results on template.
|
|
if len(template.Subtemplates) > 0 {
|
|
err = executer.Executer.ExecuteWithResults(input, func(result *output.InternalWrappedEvent) {
|
|
if result.OperatorsResult == nil {
|
|
return
|
|
}
|
|
if len(result.Results) > 0 {
|
|
firstMatched = true
|
|
}
|
|
})
|
|
} else {
|
|
firstMatched, err = executer.Executer.Execute(input)
|
|
}
|
|
if err != nil {
|
|
if len(template.Executers) == 1 {
|
|
mainErr = err
|
|
} else {
|
|
gologger.Warning().Msgf("[%s] Could not execute workflow step: %s\n", template.Template, err)
|
|
}
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
if len(template.Subtemplates) == 0 {
|
|
results.CAS(false, firstMatched)
|
|
}
|
|
if len(template.Matchers) > 0 {
|
|
for _, executer := range template.Executers {
|
|
executer.Options.Progress.AddToTotal(int64(executer.Executer.Requests()))
|
|
|
|
err := executer.Executer.ExecuteWithResults(input, func(event *output.InternalWrappedEvent) {
|
|
if event.OperatorsResult == nil {
|
|
return
|
|
}
|
|
|
|
for _, matcher := range template.Matchers {
|
|
_, matchOK := event.OperatorsResult.Matches[matcher.Name]
|
|
_, extractOK := event.OperatorsResult.Extracts[matcher.Name]
|
|
if !matchOK && !extractOK {
|
|
continue
|
|
}
|
|
|
|
for _, subtemplate := range matcher.Subtemplates {
|
|
swg.Add()
|
|
|
|
go func(subtemplate *WorkflowTemplate) {
|
|
if err := w.runWorkflowStep(subtemplate, input, results, swg); err != nil {
|
|
gologger.Warning().Msgf("[%s] Could not execute workflow step: %s\n", subtemplate.Template, err)
|
|
}
|
|
swg.Done()
|
|
}(subtemplate)
|
|
}
|
|
}
|
|
})
|
|
if err != nil {
|
|
if len(template.Executers) == 1 {
|
|
mainErr = err
|
|
} else {
|
|
gologger.Warning().Msgf("[%s] Could not execute workflow step: %s\n", template.Template, err)
|
|
}
|
|
continue
|
|
}
|
|
}
|
|
return mainErr
|
|
}
|
|
if len(template.Subtemplates) > 0 && firstMatched {
|
|
for _, subtemplate := range template.Subtemplates {
|
|
swg.Add()
|
|
|
|
go func(template *WorkflowTemplate) {
|
|
err := w.runWorkflowStep(template, input, results, swg)
|
|
if err != nil {
|
|
gologger.Warning().Msgf("[%s] Could not execute workflow step: %s\n", template.Template, err)
|
|
}
|
|
swg.Done()
|
|
}(subtemplate)
|
|
}
|
|
}
|
|
return mainErr
|
|
}
|