nuclei/v2/pkg/templates/compile.go

64 lines
1.8 KiB
Go
Raw Normal View History

2020-04-03 21:20:32 +00:00
package templates
import (
"fmt"
2020-04-03 21:20:32 +00:00
"os"
"github.com/pkg/errors"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols"
2020-12-29 11:03:25 +00:00
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/dns"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/http"
2020-04-03 21:20:32 +00:00
"gopkg.in/yaml.v2"
)
2020-06-29 12:13:08 +00:00
// Parse parses a yaml request template file
func Parse(file string, options *protocols.ExecuterOptions) (*Template, error) {
2020-04-03 21:20:32 +00:00
template := &Template{}
f, err := os.Open(file)
if err != nil {
return nil, err
}
err = yaml.NewDecoder(f).Decode(template)
if err != nil {
return nil, err
}
2020-06-26 12:37:55 +00:00
defer f.Close()
// Setting up variables regarding template metadata
2020-07-31 15:13:51 +00:00
template.path = file
options.TemplateID = template.ID
options.TemplateInfo = template.Info
options.TemplatePath = file
2020-07-31 15:13:51 +00:00
2020-12-29 11:03:25 +00:00
// We don't support both http and dns in a single template
if len(template.RequestsDNS) > 0 && len(template.RequestsHTTP) > 0 {
return nil, fmt.Errorf("both http and dns requests for %s", template.ID)
}
2020-06-29 12:13:08 +00:00
// If no requests, and it is also not a workflow, return error.
2020-12-29 11:03:25 +00:00
if len(template.RequestsDNS)+len(template.RequestsDNS)+len(template.Workflows) == 0 {
return nil, fmt.Errorf("no requests defined for %s", template.ID)
2020-06-26 12:37:55 +00:00
}
2020-04-03 21:20:32 +00:00
// Compile the requests found
for _, request := range template.RequestsDNS {
template.totalRequests += request.Requests()
}
for _, request := range template.RequestsHTTP {
template.totalRequests += request.Requests()
2020-04-03 21:20:32 +00:00
}
2020-12-29 11:03:25 +00:00
if len(template.RequestsDNS) > 0 {
template.executer = dns.NewExecuter(template.RequestsDNS, options)
err = template.executer.Compile()
}
if len(template.RequestsHTTP) > 0 {
template.executer = http.NewExecuter(template.RequestsHTTP, options)
err = template.executer.Compile()
}
if err != nil {
return nil, errors.Wrap(err, "could not compile request")
}
2020-04-03 21:20:32 +00:00
return template, nil
}