mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-21 18:35:30 +00:00
4801cc65ef
* feat: fixed max-host-error blocking wrong port for template with error * feat: log total results with time taken at end of execution * bugfix: skip non-executed requests with progress in flow protocol * feat: fixed request calculation in http protocol for progress * misc adjustments --------- Co-authored-by: Ice3man <nizamulrana@gmail.com>
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package output
|
|
|
|
import (
|
|
"github.com/logrusorgru/aurora"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/output/stats"
|
|
)
|
|
|
|
// StatsOutputWriter implements writer interface for stats observation
|
|
type StatsOutputWriter struct {
|
|
colorizer aurora.Aurora
|
|
Tracker *stats.Tracker
|
|
}
|
|
|
|
var _ Writer = &StatsOutputWriter{}
|
|
|
|
// NewStatsOutputWriter returns a new StatsOutputWriter instance.
|
|
func NewTrackerWriter(t *stats.Tracker) *StatsOutputWriter {
|
|
return &StatsOutputWriter{
|
|
colorizer: aurora.NewAurora(true),
|
|
Tracker: t,
|
|
}
|
|
}
|
|
|
|
func (tw *StatsOutputWriter) Close() {}
|
|
|
|
func (tw *StatsOutputWriter) Colorizer() aurora.Aurora {
|
|
return tw.colorizer
|
|
}
|
|
|
|
func (tw *StatsOutputWriter) Write(event *ResultEvent) error {
|
|
return nil
|
|
}
|
|
|
|
func (tw *StatsOutputWriter) WriteFailure(event *InternalWrappedEvent) error {
|
|
return nil
|
|
}
|
|
|
|
func (tw *StatsOutputWriter) Request(templateID, url, requestType string, err error) {
|
|
if err == nil {
|
|
return
|
|
}
|
|
jsonReq := getJSONLogRequestFromError(templateID, url, requestType, err)
|
|
tw.Tracker.TrackErrorKind(jsonReq.Error)
|
|
}
|
|
|
|
func (tw *StatsOutputWriter) WriteStoreDebugData(host, templateID, eventType string, data string) {}
|
|
|
|
func (tw *StatsOutputWriter) RequestStatsLog(statusCode, response string) {
|
|
tw.Tracker.TrackStatusCode(statusCode)
|
|
tw.Tracker.TrackWAFDetected(response)
|
|
}
|
|
func (tw *StatsOutputWriter) ResultCount() int {
|
|
return 0
|
|
}
|