daffainfo
/
nuclei
mirror of https://github.com/daffainfo/nuclei.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixing resume skipping on-going scans

dev
mzack 2021-12-10 12:27:34 +01:00
parent 2140ffa382
commit 2f4b131f26
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
v2/internal/runner/runner.go
View File

@ -477,6 +477,15 @@ func (r *Runner) countNewTemplates() int {
func (r *Runner) SaveResumeConfig() error {
resumeCfg := types.NewResumeCfg()
resumeCfg.TemplatesResumeFrom = r.resumeCfg.TemplatesCurrent
// Since the maximum number of concurrent targets is determined by bulk-size, calculate the current progress as
// CurrentIndex - BulkSize so that the starting point will not skip on-going scans
for templateId, currentIndex := range r.resumeCfg.TemplatesCurrentIndex {
var diff uint32
if d := int(currentIndex) - r.options.BulkSize; d > 0 {
diff = uint32(d)
}
resumeCfg.TemplatesResumeFromIndex[templateId] = diff
}
resumeCfg.TemplatesResumeFromIndex = r.resumeCfg.TemplatesCurrentIndex
file, _ := json.MarshalIndent(resumeCfg, "", "\t")