feat: added rebuildGenerators for misc rebuilding (#6037)

Co-authored-by: Ice3man <nizamulrana@gmail.com>
This commit is contained in:
Sandeep Singh 2025-02-08 21:05:23 +05:30 committed by GitHub
parent bfb1808a2d
commit 04db1bb21d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -505,6 +505,16 @@ func (request *Request) Compile(options *protocols.ExecutorOptions) error {
return nil
}
// RebuildGenerator rebuilds the generator for the request
func (request *Request) RebuildGenerator() error {
generator, err := generators.New(request.Payloads, request.AttackType.Value, request.options.TemplatePath, request.options.Catalog, request.options.Options.AttackType, request.options.Options)
if err != nil {
return errors.Wrap(err, "could not parse payloads")
}
request.generator = generator
return nil
}
// Requests returns the total number of requests the YAML rule will perform
func (request *Request) Requests() int {
if request.generator != nil {