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

Fixing stats printing with invalid templates

dev
mzack 2022-03-23 14:13:30 +01:00
parent d2fbf875cd
commit 7d75bd9248
1 changed files with 43 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
v2/pkg/progress/progress.go
View File

@ -136,45 +136,56 @@ func (p *StatsTicker) IncrementFailedRequestsBy(count int64) {
func printCallback(stats clistats.StatisticsClient) {
builder := &strings.Builder{}
builder.WriteRune('[')
startedAt, _ := stats.GetStatic("startedAt")
duration := time.Since(startedAt.(time.Time))
builder.WriteString(fmtDuration(duration))
builder.WriteRune(']')
templates, _ := stats.GetStatic("templates")
builder.WriteString(" | Templates: ")
builder.WriteString(clistats.String(templates))
hosts, _ := stats.GetStatic("hosts")
builder.WriteString(" | Hosts: ")
builder.WriteString(clistats.String(hosts))
var duration time.Duration
if startedAt, ok := stats.GetStatic("startedAt"); ok {
if startedAtTime, ok := startedAt.(time.Time); ok {
duration = time.Since(startedAtTime)
builder.WriteString(fmt.Sprintf("[%s]", fmtDuration(duration)))
}
}
requests, _ := stats.GetCounter("requests")
total, _ := stats.GetCounter("total")
if templates, ok := stats.GetStatic("templates"); ok {
builder.WriteString(" | Templates: ")
builder.WriteString(clistats.String(templates))
}
builder.WriteString(" | RPS: ")
builder.WriteString(clistats.String(uint64(float64(requests) / duration.Seconds())))
if hosts, ok := stats.GetStatic("hosts"); ok {
builder.WriteString(" | Hosts: ")
builder.WriteString(clistats.String(hosts))
}
matched, _ := stats.GetCounter("matched")
requests, okRequests := stats.GetCounter("requests")
total, okTotal := stats.GetCounter("total")
builder.WriteString(" | Matched: ")
builder.WriteString(clistats.String(matched))
if okRequests && okTotal && duration > 0 {
builder.WriteString(" | RPS: ")
builder.WriteString(clistats.String(uint64(float64(requests) / duration.Seconds())))
}
errors, _ := stats.GetCounter("errors")
builder.WriteString(" | Errors: ")
builder.WriteString(clistats.String(errors))
if matched, ok := stats.GetCounter("matched"); ok {
builder.WriteString(" | Matched: ")
builder.WriteString(clistats.String(matched))
}
builder.WriteString(" | Requests: ")
builder.WriteString(clistats.String(requests))
builder.WriteRune('/')
builder.WriteString(clistats.String(total))
builder.WriteRune(' ')
builder.WriteRune('(')
//nolint:gomnd // this is not a magic number
builder.WriteString(clistats.String(uint64(float64(requests) / float64(total) * 100.0)))
builder.WriteRune('%')
builder.WriteRune(')')
builder.WriteRune('\n')
if errors, ok := stats.GetCounter("errors"); ok {
builder.WriteString(" | Errors: ")
builder.WriteString(clistats.String(errors))
}
if okRequests && okTotal {
builder.WriteString(" | Requests: ")
builder.WriteString(clistats.String(requests))
builder.WriteRune('/')
builder.WriteString(clistats.String(total))
builder.WriteRune(' ')
builder.WriteRune('(')
//nolint:gomnd // this is not a magic number
builder.WriteString(clistats.String(uint64(float64(requests) / float64(total) * 100.0)))
builder.WriteRune('%')
builder.WriteRune(')')
builder.WriteRune('\n')
}
fmt.Fprintf(os.Stderr, "%s", builder.String())
}