mirror of
https://github.com/maride/afl-prom.git
synced 2024-11-24 12:24:25 +00:00
Switch to label-based gauges
This commit is contained in:
parent
49d9be65b6
commit
220e28823c
137
fuzzer.go
137
fuzzer.go
@ -2,73 +2,25 @@ package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"github.com/prometheus/client_golang/prometheus/promauto"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"os"
|
||||
"path"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Represents a Fuzzer and its state, as reported in the "fuzzer_stats" file
|
||||
type Fuzzer struct {
|
||||
fuzzer_pid prometheus.Gauge
|
||||
cycles_done prometheus.Gauge
|
||||
execs_done prometheus.Gauge
|
||||
execs_per_sec prometheus.Gauge
|
||||
paths_total prometheus.Gauge
|
||||
paths_favored prometheus.Gauge
|
||||
paths_found prometheus.Gauge
|
||||
paths_imported prometheus.Gauge
|
||||
max_depth prometheus.Gauge
|
||||
cur_path prometheus.Gauge
|
||||
pending_favs prometheus.Gauge
|
||||
pending_total prometheus.Gauge
|
||||
variable_paths prometheus.Gauge
|
||||
stability prometheus.Gauge
|
||||
bitmap_cvg prometheus.Gauge
|
||||
unique_crashes prometheus.Gauge
|
||||
unique_hangs prometheus.Gauge
|
||||
last_path prometheus.Gauge
|
||||
last_crash prometheus.Gauge
|
||||
last_hang prometheus.Gauge
|
||||
execs_since_crash prometheus.Gauge
|
||||
exec_timeout prometheus.Gauge
|
||||
slowest_exec_ms prometheus.Gauge
|
||||
peak_rss_mb prometheus.Gauge
|
||||
afl_banner string
|
||||
afl_directory string
|
||||
// missing start_time, last_update, afl_version, target_mode and command_line, and to be honest, I don't see a reason to export these values via prometheus
|
||||
}
|
||||
|
||||
// Initialises all gauges. Needs afl_banner to be set properly before (and doesn't check that on its own)
|
||||
func (f *Fuzzer) initGauges() {
|
||||
log.Printf("%s %s", f.afl_directory, f.afl_banner)
|
||||
f.fuzzer_pid = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_fuzzer_pid"})
|
||||
f.cycles_done = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_cycles_done"})
|
||||
f.execs_done = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_execs_done"})
|
||||
f.execs_per_sec = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_execs_per_sec"})
|
||||
f.paths_total = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_paths_total"})
|
||||
f.paths_favored = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_paths_favored"})
|
||||
f.paths_found = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_paths_found"})
|
||||
f.paths_imported = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_paths_imported"})
|
||||
f.max_depth = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_max_depth"})
|
||||
f.cur_path = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_cur_path"})
|
||||
f.pending_favs = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_pending_favs"})
|
||||
f.pending_total = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_pending_total"})
|
||||
f.variable_paths = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_variable_paths"})
|
||||
f.stability = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_stability"})
|
||||
f.bitmap_cvg = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_bitmap_cvg"})
|
||||
f.unique_crashes = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_unique_crashes"})
|
||||
f.unique_hangs = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_unique_hangs"})
|
||||
f.last_path = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_last_path"})
|
||||
f.last_crash = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_last_crash"})
|
||||
f.last_hang = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_last_hang"})
|
||||
f.execs_since_crash = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_execs_since_crash"})
|
||||
f.exec_timeout = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_exec_timeout"})
|
||||
f.slowest_exec_ms = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_slowest_exec_ms"})
|
||||
f.peak_rss_mb = promauto.NewGauge(prometheus.GaugeOpts{Name: f.afl_banner + "_peak_rss_mb"})
|
||||
// Creates a fuzzer from the given target path
|
||||
func CreateFuzzer(target string) Fuzzer {
|
||||
return Fuzzer{
|
||||
afl_banner: path.Base(target),
|
||||
afl_directory: target,
|
||||
}
|
||||
}
|
||||
|
||||
// Parses the "fuzzer_stats" file present in the given directory, and updates the gauges of the Fuzzer instance accordingly
|
||||
@ -83,16 +35,9 @@ func (f *Fuzzer) ParseStatsFile() error {
|
||||
// Split file contents into lines
|
||||
fileLines := strings.Split(string(fileBytes), "\n")
|
||||
|
||||
// So we want to walk over the file two times.
|
||||
// 1) If it is unset: read "afl_banner", set it, and run initGauges()
|
||||
// 2) Set every other value to its corresponding gauge.
|
||||
bannerSet := f.afl_banner != "" // 1)
|
||||
gaugesSet := false // 2)
|
||||
for !(bannerSet && gaugesSet) {
|
||||
// Iterate over every line
|
||||
for _, l := range fileLines {
|
||||
// Skip empty lines
|
||||
if l == "" {
|
||||
// Skip unparseable files
|
||||
if !strings.Contains(l, ":") {
|
||||
continue
|
||||
}
|
||||
|
||||
@ -101,97 +46,81 @@ func (f *Fuzzer) ParseStatsFile() error {
|
||||
key := strings.TrimSpace(parts[0])
|
||||
val := strings.TrimSpace(parts[1])
|
||||
|
||||
// 1)
|
||||
if !bannerSet {
|
||||
if key == "afl_banner" {
|
||||
f.afl_banner = val
|
||||
bannerSet = true
|
||||
f.initGauges()
|
||||
break
|
||||
} else {
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
// 2)
|
||||
switch key {
|
||||
case "fuzzer_pid":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.fuzzer_pid.Set(float64(convVal))
|
||||
fuzzer_pid.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "cycles_done":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.cycles_done.Set(float64(convVal))
|
||||
cycles_done.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "execs_done":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.execs_done.Set(float64(convVal))
|
||||
execs_done.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "execs_per_sec":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.execs_per_sec.Set(float64(convVal))
|
||||
execs_per_sec.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "paths_total":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.paths_total.Set(float64(convVal))
|
||||
paths_total.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "paths_favored":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.paths_favored.Set(float64(convVal))
|
||||
paths_favored.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "paths_found":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.paths_found.Set(float64(convVal))
|
||||
paths_found.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "paths_imported":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.paths_imported.Set(float64(convVal))
|
||||
paths_imported.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "max_depth":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.max_depth.Set(float64(convVal))
|
||||
max_depth.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "cur_path":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.cur_path.Set(float64(convVal))
|
||||
cur_path.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "pending_favs":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.pending_favs.Set(float64(convVal))
|
||||
pending_favs.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "pending_total":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.pending_total.Set(float64(convVal))
|
||||
pending_total.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "variable_paths":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.variable_paths.Set(float64(convVal))
|
||||
variable_paths.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "stability":
|
||||
val = strings.Replace(val, "%", "", 1)
|
||||
convVal, _ := strconv.ParseFloat(val, 64)
|
||||
f.stability.Set(convVal)
|
||||
stability.WithLabelValues(f.afl_banner).Set(convVal)
|
||||
case "bitmap_cvg":
|
||||
val = strings.Replace(val, "%", "", 1)
|
||||
convVal, _ := strconv.ParseFloat(val, 64)
|
||||
f.bitmap_cvg.Set(convVal)
|
||||
bitmap_cvg.WithLabelValues(f.afl_banner).Set(convVal)
|
||||
case "unique_crashes":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.unique_crashes.Set(float64(convVal))
|
||||
unique_crashes.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "unique_hangs":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.unique_hangs.Set(float64(convVal))
|
||||
unique_hangs.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "last_path":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.last_path.Set(float64(convVal))
|
||||
last_path.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "last_crash":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.last_crash.Set(float64(convVal))
|
||||
last_crash.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "last_hang":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.last_hang.Set(float64(convVal))
|
||||
last_hang.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "execs_since_crash":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.execs_since_crash.Set(float64(convVal))
|
||||
execs_since_crash.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "exec_timeout":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.exec_timeout.Set(float64(convVal))
|
||||
exec_timeout.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "slowest_exec_ms":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.slowest_exec_ms.Set(float64(convVal))
|
||||
slowest_exec_ms.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
case "peak_rss_mb":
|
||||
convVal, _ := strconv.ParseInt(val, 10, 64)
|
||||
f.peak_rss_mb.Set(float64(convVal))
|
||||
}
|
||||
|
||||
gaugesSet = true
|
||||
peak_rss_mb.WithLabelValues(f.afl_banner).Set(float64(convVal))
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
86
gauges.go
Normal file
86
gauges.go
Normal file
@ -0,0 +1,86 @@
|
||||
package main
|
||||
|
||||
import "github.com/prometheus/client_golang/prometheus"
|
||||
|
||||
var (
|
||||
fuzzer_pid *prometheus.GaugeVec
|
||||
cycles_done *prometheus.GaugeVec
|
||||
execs_done *prometheus.GaugeVec
|
||||
execs_per_sec *prometheus.GaugeVec
|
||||
paths_total *prometheus.GaugeVec
|
||||
paths_favored *prometheus.GaugeVec
|
||||
paths_found *prometheus.GaugeVec
|
||||
paths_imported *prometheus.GaugeVec
|
||||
max_depth *prometheus.GaugeVec
|
||||
cur_path *prometheus.GaugeVec
|
||||
pending_favs *prometheus.GaugeVec
|
||||
pending_total *prometheus.GaugeVec
|
||||
variable_paths *prometheus.GaugeVec
|
||||
stability *prometheus.GaugeVec
|
||||
bitmap_cvg *prometheus.GaugeVec
|
||||
unique_crashes *prometheus.GaugeVec
|
||||
unique_hangs *prometheus.GaugeVec
|
||||
last_path *prometheus.GaugeVec
|
||||
last_crash *prometheus.GaugeVec
|
||||
last_hang *prometheus.GaugeVec
|
||||
execs_since_crash *prometheus.GaugeVec
|
||||
exec_timeout *prometheus.GaugeVec
|
||||
slowest_exec_ms *prometheus.GaugeVec
|
||||
peak_rss_mb *prometheus.GaugeVec
|
||||
// missing start_time, last_update, afl_version, target_mode and command_line, and to be honest, I don't see a reason to export these values via prometheus
|
||||
)
|
||||
|
||||
// Initializes all gauges and register them
|
||||
func InitializeGauges() {
|
||||
// Set up gauges
|
||||
fuzzer_pid = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "fuzzer_pid"}, []string{"name"})
|
||||
cycles_done = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "cycles_done"}, []string{"name"})
|
||||
execs_done = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "execs_done"}, []string{"name"})
|
||||
execs_per_sec = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "execs_per_sec"}, []string{"name"})
|
||||
paths_total = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "paths_total"}, []string{"name"})
|
||||
paths_favored = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "paths_favored"}, []string{"name"})
|
||||
paths_found = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "paths_found"}, []string{"name"})
|
||||
paths_imported = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "paths_imported"}, []string{"name"})
|
||||
max_depth = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "max_depth"}, []string{"name"})
|
||||
cur_path = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "cur_path"}, []string{"name"})
|
||||
pending_favs = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "pending_favs"}, []string{"name"})
|
||||
pending_total = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "pending_total"}, []string{"name"})
|
||||
variable_paths = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "variable_paths"}, []string{"name"})
|
||||
stability = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "stability"}, []string{"name"})
|
||||
bitmap_cvg = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "bitmap_cvg"}, []string{"name"})
|
||||
unique_crashes = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "unique_crashes"}, []string{"name"})
|
||||
unique_hangs = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "unique_hangs"}, []string{"name"})
|
||||
last_path = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "last_path"}, []string{"name"})
|
||||
last_crash = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "last_crash"}, []string{"name"})
|
||||
last_hang = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "last_hang"}, []string{"name"})
|
||||
execs_since_crash = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "execs_since_crash"}, []string{"name"})
|
||||
exec_timeout = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "exec_timeout"}, []string{"name"})
|
||||
slowest_exec_ms = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "slowest_exec_ms"}, []string{"name"})
|
||||
peak_rss_mb = prometheus.NewGaugeVec(prometheus.GaugeOpts{Name: "peak_rss_mb"}, []string{"name"})
|
||||
|
||||
// Register gauges
|
||||
prometheus.MustRegister(fuzzer_pid)
|
||||
prometheus.MustRegister(cycles_done)
|
||||
prometheus.MustRegister(execs_done)
|
||||
prometheus.MustRegister(execs_per_sec)
|
||||
prometheus.MustRegister(paths_total)
|
||||
prometheus.MustRegister(paths_favored)
|
||||
prometheus.MustRegister(paths_found)
|
||||
prometheus.MustRegister(paths_imported)
|
||||
prometheus.MustRegister(max_depth)
|
||||
prometheus.MustRegister(cur_path)
|
||||
prometheus.MustRegister(pending_favs)
|
||||
prometheus.MustRegister(pending_total)
|
||||
prometheus.MustRegister(variable_paths)
|
||||
prometheus.MustRegister(stability)
|
||||
prometheus.MustRegister(bitmap_cvg)
|
||||
prometheus.MustRegister(unique_crashes)
|
||||
prometheus.MustRegister(unique_hangs)
|
||||
prometheus.MustRegister(last_path)
|
||||
prometheus.MustRegister(last_crash)
|
||||
prometheus.MustRegister(last_hang)
|
||||
prometheus.MustRegister(execs_since_crash)
|
||||
prometheus.MustRegister(exec_timeout)
|
||||
prometheus.MustRegister(slowest_exec_ms)
|
||||
prometheus.MustRegister(peak_rss_mb)
|
||||
}
|
||||
15
watch.go
15
watch.go
@ -26,19 +26,14 @@ func getFuzzersToWatch() ([]string, error) {
|
||||
|
||||
// Registers the given paths as fuzzer directories which should be monitored
|
||||
func registerFuzzers(targets []string) {
|
||||
// First, create fuzzer instances based on the directory
|
||||
for _, f := range targets {
|
||||
var tmpFuzzer Fuzzer
|
||||
tmpFuzzer.afl_directory = f
|
||||
parseErr := tmpFuzzer.ParseStatsFile()
|
||||
|
||||
if parseErr != nil {
|
||||
log.Printf("Encountered error while parsing %s: %s", f, parseErr.Error())
|
||||
break
|
||||
}
|
||||
|
||||
// Append fuzzer to our list of registered fuzzers
|
||||
tmpFuzzer := CreateFuzzer(f)
|
||||
registeredFuzzers = append(registeredFuzzers, tmpFuzzer)
|
||||
}
|
||||
|
||||
// Create gauges
|
||||
InitializeGauges()
|
||||
}
|
||||
|
||||
// Watch over the fuzzer(s)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user