ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-15 11:28:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adds other metrics

Browse Source
This commit is contained in:
Edznux
2020-10-04 03:22:28 +02:00
parent 26dcddab0c
commit ff8c6d2415
2 changed files with 57 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/afl-fuzz.h
View File

@ -956,7 +956,7 @@ void show_init_stats(afl_state_t *);
/* StatsD */ /* StatsD */
int statsd_init(char *host, int port); int statsd_socket_init(char *host, int port);
int send_statsd_metric(afl_state_t *afl); int send_statsd_metric(afl_state_t *afl);
int statsd_format_metric(afl_state_t *afl, char *buff, size_t bufflen); int statsd_format_metric(afl_state_t *afl, char *buff, size_t bufflen);

74
src/afl-fuzz-statsd.c
View File

@ -9,14 +9,15 @@
#include "afl-fuzz.h" #include "afl-fuzz.h"
#define MAX_STATSD_PACKET_SIZE 1400 #define MAX_STATSD_PACKET_SIZE 4096
#define MAX_TAG_LEN 200 #define MAX_TAG_LEN 200
#define METRIC_PREFIX "fuzzing"
int sock = 0; int sock = 0;
struct sockaddr_in server; struct sockaddr_in server;
int error = 0; int error = 0;
int statsd_init(char *host, int port){ int statsd_socket_init(char *host, int port){
if((sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1){ if((sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1){
perror("socket"); perror("socket");
exit(1); exit(1);
@ -45,11 +46,9 @@ int statsd_init(char *host, int port){
} }
int send_statsd_metric(afl_state_t *afl){ int send_statsd_metric(afl_state_t *afl){
u64 cur_ms = get_cur_time(); /* default port and host.
if (cur_ms - afl->stats_last_plot_ms < 1000) { Will be overwritten by AFL_STATSD_PORT and AFL_STATSD_HOST environment variable, if they exists.
return 0; */
}
u16 port = 8125; u16 port = 8125;
char* host = "127.0.0.1"; char* host = "127.0.0.1";
@ -64,7 +63,7 @@ int send_statsd_metric(afl_state_t *afl){
host = host_env; host = host_env;
} }
error = statsd_init(host, port); error = statsd_socket_init(host, port);
if (error){ if (error){
perror("Failed to init statsd client. Aborting"); perror("Failed to init statsd client. Aborting");
return -1; return -1;
@ -78,7 +77,7 @@ int send_statsd_metric(afl_state_t *afl){
char buff[MAX_STATSD_PACKET_SIZE] = {0}; char buff[MAX_STATSD_PACKET_SIZE] = {0};
statsd_format_metric(afl, buff, MAX_STATSD_PACKET_SIZE); statsd_format_metric(afl, buff, MAX_STATSD_PACKET_SIZE);
if (sendto(sock, buff, MAX_STATSD_PACKET_SIZE, 0, if (sendto(sock, buff, strlen(buff), 0,
(struct sockaddr *)&server, sizeof(server)) == -1) { (struct sockaddr *)&server, sizeof(server)) == -1) {
perror("sendto"); perror("sendto");
return -1; return -1;
@ -95,23 +94,62 @@ int statsd_format_metric(afl_state_t *afl, char *buff, size_t bufflen){
metric format: metric format:
<some.namespaced.name>:<value>|<type>|<tags> <some.namespaced.name>:<value>|<type>|<tags>
tags format: tags format:
#key:value,key:value,key */
#ifdef USE_STATSD_TAGS
/* #key:value,key:value,key
*/ */
char tags[MAX_TAG_LEN * 2] = {0}; char tags[MAX_TAG_LEN * 2] = {0};
snprintf(tags, MAX_TAG_LEN * 2, snprintf(tags, MAX_TAG_LEN * 2,
"banner:%s,afl_version:%s", "|#banner:%s,afl_version:%s",
afl->use_banner, afl->use_banner,
VERSION); VERSION);
#else
char *tags = "";
#endif
// Sends multiple metrics with one UDP Packet. // Sends multiple metrics with one UDP Packet.
// bufflen will limit to the max safe size. // bufflen will limit to the max safe size.
snprintf(buff, bufflen, snprintf(buff, bufflen,
"fuzzing.afl.cycle_done:%llu|g|#%s\n" METRIC_PREFIX".cycle_done:%llu|g%s\n"
"fuzzing.afl.total_path:%u|g|#%s\n" METRIC_PREFIX".cycles_wo_finds:%llu|g%s\n"
"fuzzing.afl.unique_crashes:%llu|g|#%s\n" METRIC_PREFIX".execs_done:%llu|g%s\n"
"fuzzing.afl.total_crashes:%llu|g|#%s\n" METRIC_PREFIX".execs_per_sec:%0.02f|g%s\n"
"fuzzing.afl.unique_hangs:%llu|g|#%s\n", METRIC_PREFIX".paths_total:%u|g%s\n"
afl->queue_cycle, tags, afl->queued_paths, tags, afl->unique_crashes, METRIC_PREFIX".paths_favored:%u|g%s\n"
tags, afl->total_crashes, tags, afl->unique_hangs, tags); METRIC_PREFIX".paths_found:%u|g%s\n"
METRIC_PREFIX".paths_imported:%u|g%s\n"
METRIC_PREFIX".max_depth:%u|g%s\n"
METRIC_PREFIX".cur_path:%u|g%s\n"
METRIC_PREFIX".pending_favs:%u|g%s\n"
METRIC_PREFIX".pending_total:%u|g%s\n"
METRIC_PREFIX".variable_paths:%u|g%s\n"
METRIC_PREFIX".unique_crashes:%llu|g%s\n"
METRIC_PREFIX".unique_hangs:%llu|g%s\n"
METRIC_PREFIX".total_crashes:%llu|g%s\n"
METRIC_PREFIX".slowest_exec_ms:%u|g%s\n"
METRIC_PREFIX".edges_found:%u|g%s\n"
METRIC_PREFIX".var_byte_count:%u|g%s\n"
METRIC_PREFIX".havoc_expansion:%u|g%s\n",
afl->queue_cycle ? (afl->queue_cycle - 1) : 0, tags,
afl->cycles_wo_finds, tags,
afl->fsrv.total_execs, tags,
afl->fsrv.total_execs / ((double)(get_cur_time() - afl->start_time) / 1000), tags,
afl->queued_paths, tags,
afl->queued_favored, tags,
afl->queued_discovered, tags,
afl->queued_imported, tags,
afl->max_depth, tags,
afl->current_entry, tags,
afl->pending_favored, tags,
afl->pending_not_fuzzed, tags,
afl->queued_variable, tags,
afl->unique_crashes, tags,
afl->unique_hangs, tags,
afl->total_crashes, tags,
afl->slowest_exec_ms, tags,
count_non_255_bytes(afl, afl->virgin_bits), tags,
afl->var_byte_count, tags,
afl->expand_havoc, tags
);
return 0; return 0;
} }