Code

src/oping.c: Add median and 'n'th percentile, remove average and stddev.
[liboping.git] / src / oping.c
index b7ef0454395a60c8da3d4666ee16e861b01651c1..ff0667063a838680141654cc9df948aa3c68bef1 100644 (file)
@@ -95,6 +95,9 @@
 # define OPING_YELLOW_HIST 5
 # define OPING_RED_HIST 6
 
+double const threshold_green = 0.8;
+double const threshold_yellow = 0.95;
+
 static char const * const hist_symbols_utf8[] = {
        "▁", "▂", "▃", "▄", "▅", "▆", "▇", "█" };
 static size_t const hist_symbols_utf8_num = sizeof (hist_symbols_utf8)
@@ -123,14 +126,15 @@ static size_t const hist_colors_num = sizeof (hist_colors_utf8)
 #endif
 
 /* "─" */
-#define BOXPLOT_WHISKER_BAR (113 | A_ALTCHARSET)
+#define BOXPLOT_WHISKER_BAR       (113 | A_ALTCHARSET)
 /* "├" */
-#define BOXPLOT_WHISKER_LEFT_END (116 | A_ALTCHARSET)
+#define BOXPLOT_WHISKER_LEFT_END  (116 | A_ALTCHARSET)
 /* "┤" */
 #define BOXPLOT_WHISKER_RIGHT_END (117 | A_ALTCHARSET)
-#define BOXPLOT_BOX ' '
-/* "│" */
-#define BOXPLOT_MEDIAN (120 | A_ALTCHARSET)
+/* Inverted */
+#define BOXPLOT_BOX               ' '
+/* "│", inverted */
+#define BOXPLOT_MEDIAN            (120 | A_ALTCHARSET)
 
 #include "oping.h"
 
@@ -159,11 +163,12 @@ typedef struct ping_context
        double latency_min;
        double latency_max;
        double latency_total;
-       double latency_total_square;
 
 /* 1000 + one "infinity" bucket. */
 #define OPING_HISTOGRAM_BUCKETS 1001
-       uint32_t *latency_histogram;
+       uint32_t *histogram_counters;
+       uint32_t *histogram_accumulated;
+       double *histogram_ratio;
        size_t latency_histogram_size;
 
 #if USE_NCURSES
@@ -183,6 +188,7 @@ static uint8_t opt_send_qos   = 0;
 static double  opt_percentile = -1.0;
 static double  opt_exit_status_threshold = 1.0;
 #if USE_NCURSES
+static int     opt_show_graph = 1;
 static int     opt_utf8       = 0;
 #endif
 
@@ -212,11 +218,14 @@ static ping_context_t *context_create (void) /* {{{ */
        ret->latency_min   = -1.0;
        ret->latency_max   = -1.0;
        ret->latency_total = 0.0;
-       ret->latency_total_square = 0.0;
 
        ret->latency_histogram_size = (size_t) OPING_HISTOGRAM_BUCKETS;
-       ret->latency_histogram = calloc (ret->latency_histogram_size,
-                       sizeof (*ret->latency_histogram));
+       ret->histogram_counters = calloc (ret->latency_histogram_size,
+                       sizeof (*ret->histogram_counters));
+       ret->histogram_accumulated = calloc (ret->latency_histogram_size,
+                       sizeof (*ret->histogram_accumulated));
+       ret->histogram_ratio = calloc (ret->latency_histogram_size,
+                       sizeof (*ret->histogram_ratio));
 
 #if USE_NCURSES
        ret->window = NULL;
@@ -238,51 +247,32 @@ static void context_destroy (ping_context_t *context) /* {{{ */
        }
 #endif
 
-       free (context->latency_histogram);
-       context->latency_histogram = NULL;
-
-       free (context);
-} /* }}} void context_destroy */
-
-static double context_get_average (ping_context_t *ctx) /* {{{ */
-{
-       double num_total;
+       free (context->histogram_counters);
+       context->histogram_counters = NULL;
 
-       if (ctx == NULL)
-               return (-1.0);
+       free (context->histogram_accumulated);
+       context->histogram_accumulated = NULL;
 
-       if (ctx->req_rcvd < 1)
-               return (-0.0);
+       free (context->histogram_ratio);
+       context->histogram_ratio = NULL;
 
-       num_total = (double) ctx->req_rcvd;
-       return (ctx->latency_total / num_total);
-} /* }}} double context_get_average */
+       free (context);
+} /* }}} void context_destroy */
 
 static double context_get_percentile (ping_context_t *ctx, /* {{{ */
                double percentile)
 {
        double threshold = percentile / 100.0;
-       uint32_t accumulated[ctx->latency_histogram_size];
-       double ratios[ctx->latency_histogram_size];
        double index_to_ms_factor;
-       uint32_t num;
        size_t i;
+       double ret;
 
-       if (ctx->latency_histogram == NULL)
+       if (ctx->histogram_ratio == NULL)
                return (NAN);
 
-       accumulated[0] = ctx->latency_histogram[0];
-       for (i = 1; i < ctx->latency_histogram_size; i++)
-               accumulated[i] = accumulated[i - 1]
-                       + ctx->latency_histogram[i];
-       num = accumulated[ctx->latency_histogram_size - 1];
-
        for (i = 0; i < ctx->latency_histogram_size; i++)
-       {
-               ratios[i] = ((double) accumulated[i]) / ((double) num);
-               if (ratios[i] >= threshold)
+               if (ctx->histogram_ratio[i] >= threshold)
                        break;
-       }
 
        if (i >= ctx->latency_histogram_size)
                return (NAN);
@@ -293,26 +283,14 @@ static double context_get_percentile (ping_context_t *ctx, /* {{{ */
 
        /* Multiply with i+1, because we're interested in the _upper_ bound of
         * each bucket. */
-       return (index_to_ms_factor * ((double) (i + 1)));
-} /* }}} double context_get_percentile */
+       ret = (index_to_ms_factor * ((double) (i + 1)));
 
-static double context_get_stddev (ping_context_t *ctx) /* {{{ */
-{
-       double num_total;
-
-       if (ctx == NULL)
-               return (-1.0);
+       /* Avoid reporting a higher latency than latency_max. */
+       if (ret > ctx->latency_max)
+               ret = ctx->latency_max;
 
-       if (ctx->req_rcvd < 1)
-               return (-0.0);
-       else if (ctx->req_rcvd < 2)
-               return (0.0);
-
-       num_total = (double) ctx->req_rcvd;
-       return (sqrt (((num_total * ctx->latency_total_square)
-                                       - (ctx->latency_total * ctx->latency_total))
-                               / (num_total * (num_total - 1.0))));
-} /* }}} double context_get_stddev */
+       return (ret);
+} /* }}} double context_get_percentile */
 
 static double context_get_packet_loss (const ping_context_t *ctx) /* {{{ */
 {
@@ -376,6 +354,7 @@ static void usage_exit (const char *name, int status) /* {{{ */
                        "  -f filename  filename to read hosts from\n"
 #if USE_NCURSES
                        "  -u / -U      force / disable UTF-8 output\n"
+                       "  -g graph     graph type to draw\n"
 #endif
                        "  -P percent   Report the n'th percentile of latency\n"
                        "  -Z percent   Exit with non-zero exit status if more than this percentage of\n"
@@ -584,7 +563,7 @@ static int read_options (int argc, char **argv) /* {{{ */
        {
                optchar = getopt (argc, argv, "46c:hi:I:t:Q:f:D:Z:P:"
 #if USE_NCURSES
-                               "uU"
+                               "uUg:"
 #endif
                                );
 
@@ -678,6 +657,19 @@ static int read_options (int argc, char **argv) /* {{{ */
                                break;
 
 #if USE_NCURSES
+                       case 'g':
+                               if (strcasecmp ("none", optarg) == 0)
+                                       opt_show_graph = 0;
+                               else if (strcasecmp ("prettyping", optarg) == 0)
+                                       opt_show_graph = 1;
+                               else if (strcasecmp ("boxplot", optarg) == 0)
+                                       opt_show_graph = 2;
+                               else if (strcasecmp ("histogram", optarg) == 0)
+                                       opt_show_graph = 3;
+                               else
+                                       fprintf (stderr, "Unknown graph option: %s\n", optarg);
+                               break;
+
                        case 'u':
                                opt_utf8 = 2;
                                break;
@@ -707,6 +699,7 @@ static int read_options (int argc, char **argv) /* {{{ */
                        case 'h':
                                usage_exit (argv[0], 0);
                                break;
+
                        default:
                                usage_exit (argv[0], 1);
                }
@@ -783,40 +776,27 @@ static _Bool has_utf8() /* {{{ */
 # endif
 } /* }}} _Bool has_utf8 */
 
-static int update_boxplot (ping_context_t *ctx) /* {{{ */
+static int update_graph_boxplot (ping_context_t *ctx) /* {{{ */
 {
-       uint32_t *accumulated;
        double *ratios;
-       uint32_t num;
        size_t i;
        size_t x_max;
        size_t x;
 
        x_max = (size_t) getmaxx (ctx->window);
-       if (x_max <= 4)
+       if (x_max <= 8)
                return (EINVAL);
        x_max -= 4;
 
-       accumulated = calloc (x_max, sizeof (*accumulated));
        ratios = calloc (x_max, sizeof (*ratios));
 
        /* Downsample */
        for (i = 0; i < ctx->latency_histogram_size; i++)
        {
                x = i * x_max / ctx->latency_histogram_size;
-               accumulated[x] += ctx->latency_histogram[i];
+               ratios[x] = ctx->histogram_ratio[i];
        }
 
-       /* Sum */
-       for (x = 1; x < x_max; x++)
-               accumulated[x] += accumulated[x - 1];
-
-       num = accumulated[x_max - 1];
-
-       /* Calculate ratios */
-       for (x = 0; x < x_max; x++)
-               ratios[x] = ((double) accumulated[x]) / ((double) num);
-
        for (x = 0; x < x_max; x++)
        {
                int symbol = ' ';
@@ -876,11 +856,10 @@ static int update_boxplot (ping_context_t *ctx) /* {{{ */
        }
 
        free (ratios);
-       free (accumulated);
        return (0);
-} /* }}} int update_boxplot */
+} /* }}} int update_graph_boxplot */
 
-static int update_prettyping_graph (ping_context_t *ctx, /* {{{ */
+static int update_graph_prettyping (ping_context_t *ctx, /* {{{ */
                double latency, unsigned int sequence)
 {
        int color = OPING_RED;
@@ -957,7 +936,97 @@ static int update_prettyping_graph (ping_context_t *ctx, /* {{{ */
 
        wprintw (ctx->window, " ");
        return (0);
-} /* }}} int update_prettyping_graph */
+} /* }}} int update_graph_prettyping */
+
+static int update_graph_histogram (ping_context_t *ctx) /* {{{ */
+{
+       uint32_t *counters;
+       uint32_t *accumulated;
+       uint32_t num;
+       uint32_t max;
+       size_t i;
+       size_t x_max;
+       size_t x;
+
+       size_t symbols_num = hist_symbols_acs_num;
+
+       if (has_utf8 ())
+               symbols_num = hist_symbols_utf8_num;
+
+       x_max = (size_t) getmaxx (ctx->window);
+       if (x_max <= 4)
+               return (EINVAL);
+       x_max -= 4;
+
+       counters = calloc (x_max, sizeof (*counters));
+       accumulated = calloc (x_max, sizeof (*accumulated));
+
+       /* Downsample */
+       max = 0;
+       for (i = 0; i < ctx->latency_histogram_size; i++)
+       {
+               x = i * x_max / ctx->latency_histogram_size;
+               counters[x] += ctx->histogram_counters[i];
+               accumulated[x] = counters[x];
+
+               if (max < counters[x])
+                       max = counters[x];
+       }
+
+       /* Sum */
+       for (x = 1; x < x_max; x++)
+               accumulated[x] += accumulated[x - 1];
+       num = accumulated[x_max - 1];
+
+       /* Calculate ratios */
+       for (x = 0; x < x_max; x++)
+       {
+               double height = ((double) counters[x]) / ((double) max);
+               double ratio_this = ((double) accumulated[x]) / ((double) num);
+               double ratio_prev = 0.0;
+               size_t index;
+               int color = 0;
+
+               index = (size_t) (height * ((double) symbols_num));
+               if (index >= symbols_num)
+                       index = symbols_num - 1;
+
+               if (x > 0)
+                       ratio_prev = ((double) accumulated[x - 1]) / ((double) num);
+
+               if (has_colors () == TRUE)
+               {
+                       if ((ratio_this <= threshold_green)
+                                       || ((ratio_prev < threshold_green)
+                                               && (ratio_this > threshold_green)))
+                               color = OPING_GREEN;
+                       else if ((ratio_this <= threshold_yellow)
+                                       || ((ratio_prev < threshold_yellow)
+                                               && (ratio_this > threshold_yellow)))
+                               color = OPING_YELLOW;
+                       else
+                               color = OPING_RED;
+
+                       wattron (ctx->window, COLOR_PAIR(color));
+               }
+
+               if (counters[x] == 0)
+                       mvwaddch (ctx->window, /* y = */ 3, /* x = */ x + 2, ' ');
+               else if (has_utf8 ())
+                       mvwprintw (ctx->window, /* y = */ 3, /* x = */ x + 2,
+                                       hist_symbols_utf8[index]);
+               else
+                       mvwaddch (ctx->window, /* y = */ 3, /* x = */ x + 2,
+                                       hist_symbols_acs[index] | A_ALTCHARSET);
+
+               if (has_colors () == TRUE)
+                       wattroff (ctx->window, COLOR_PAIR(color));
+
+       }
+
+       free (accumulated);
+       return (0);
+} /* }}} int update_graph_histogram */
 
 static int update_stats_from_context (ping_context_t *ctx, pingobj_iter_t *iter) /* {{{ */
 {
@@ -992,26 +1061,25 @@ static int update_stats_from_context (ping_context_t *ctx, pingobj_iter_t *iter)
                        ctx->latency_total);
        if (ctx->req_rcvd != 0)
        {
-               double average;
-               double deviation;
+               double median;
                double percentile;
 
-               average = context_get_average (ctx);
-               deviation = context_get_stddev (ctx);
+               median = context_get_percentile (ctx, 50.0);
                percentile = context_get_percentile (ctx, opt_percentile);
 
                mvwprintw (ctx->window, /* y = */ 2, /* x = */ 2,
-                               "rtt min/avg/%.0f%%/max/sdev = "
-                               "%.3f/%.3f/%.0f/%.3f/%.3f ms\n",
-                               opt_percentile,
-                               ctx->latency_min,
-                               average,
-                               percentile,
-                               ctx->latency_max,
-                               deviation);
+                               "RTT[ms]: min = %.0f, median = %.0f, p(%.0f) = %.0f, max = %.0f",
+                               ctx->latency_min, median,
+                               opt_percentile, percentile,
+                               ctx->latency_max);
        }
 
-       update_prettyping_graph (ctx, latency, sequence);
+       if (opt_show_graph == 1)
+               update_graph_prettyping (ctx, latency, sequence);
+       else if (opt_show_graph == 2)
+               update_graph_boxplot (ctx);
+       else if (opt_show_graph == 3)
+               update_graph_histogram (ctx);
 
        wrefresh (ctx->window);
 
@@ -1024,12 +1092,13 @@ static int on_resize (pingobj_t *ping) /* {{{ */
        int width = 0;
        int height = 0;
        int main_win_height;
+       int box_height = (opt_show_graph == 0) ? 4 : 5;
 
        getmaxyx (stdscr, height, width);
        if ((height < 1) || (width < 1))
                return (EINVAL);
 
-       main_win_height = height - (5 * host_num);
+       main_win_height = height - (box_height * host_num);
        wresize (main_win, main_win_height, /* width = */ width);
        /* Allow scrolling */
        scrollok (main_win, TRUE);
@@ -1053,9 +1122,9 @@ static int on_resize (pingobj_t *ping) /* {{{ */
                        delwin (context->window);
                        context->window = NULL;
                }
-               context->window = newwin (/* height = */ 5,
+               context->window = newwin (/* height = */ box_height,
                                /* width = */ width,
-                               /* y = */ main_win_height + (5 * context->index),
+                               /* y = */ main_win_height + (box_height * context->index),
                                /* x = */ 0);
        }
 
@@ -1087,6 +1156,7 @@ static int pre_loop_hook (pingobj_t *ping) /* {{{ */
        int width = 0;
        int height = 0;
        int main_win_height;
+       int box_height = (opt_show_graph == 0) ? 4 : 5;
 
        initscr ();
        cbreak ();
@@ -1108,7 +1178,7 @@ static int pre_loop_hook (pingobj_t *ping) /* {{{ */
                init_pair (OPING_RED_HIST,    COLOR_RED,    COLOR_YELLOW);
        }
 
-       main_win_height = height - (5 * host_num);
+       main_win_height = height - (box_height * host_num);
        main_win = newwin (/* height = */ main_win_height,
                        /* width = */ width,
                        /* y = */ 0, /* x = */ 0);
@@ -1135,9 +1205,9 @@ static int pre_loop_hook (pingobj_t *ping) /* {{{ */
                        delwin (context->window);
                        context->window = NULL;
                }
-               context->window = newwin (/* height = */ 5,
+               context->window = newwin (/* height = */ box_height,
                                /* width = */ width,
-                               /* y = */ main_win_height + (5 * context->index),
+                               /* y = */ main_win_height + (box_height * context->index),
                                /* x = */ 0);
        }
 
@@ -1203,28 +1273,46 @@ static int post_sleep_hook (__attribute__((unused)) pingobj_t *ping) /* {{{ */
 } /* }}} int post_sleep_hook */
 #endif
 
+static size_t latency_to_bucket (ping_context_t *ctx, double latency) /* {{{ */
+{
+       size_t bucket;
+
+       /* latency is in ms, opt_interval is in s. */
+       bucket = (size_t) ((latency * (ctx->latency_histogram_size - 1))
+                       / (1000.0 * opt_interval));
+       if (bucket >= ctx->latency_histogram_size)
+               bucket = ctx->latency_histogram_size - 1;
+
+       return (bucket);
+} /* }}} size_t latency_to_bucket */
+
 static void update_context (ping_context_t *context, double latency) /* {{{ */
 {
        size_t bucket;
+       size_t i;
+       double num;
 
        context->req_rcvd++;
        context->latency_total += latency;
-       context->latency_total_square += (latency * latency);
 
        if ((context->latency_max < 0.0) || (context->latency_max < latency))
                context->latency_max = latency;
        if ((context->latency_min < 0.0) || (context->latency_min > latency))
                context->latency_min = latency;
 
-       if (context->latency_histogram == NULL)
-               return;
+       bucket = latency_to_bucket (context, latency);
+       num = (double) context->req_rcvd;
 
-       /* latency is in ms, opt_interval is in s. */
-       bucket = (size_t) ((latency * (context->latency_histogram_size - 1))
-                       / (1000.0 * opt_interval));
-       if (bucket >= context->latency_histogram_size)
-               bucket = context->latency_histogram_size - 1;
-       context->latency_histogram[bucket]++;
+       context->histogram_counters[bucket]++;
+
+       context->histogram_accumulated[0] = context->histogram_counters[0];
+       context->histogram_ratio[0] = ((double) context->histogram_accumulated[0]) / num;
+       for (i = 1; i < context->latency_histogram_size; i++)
+       {
+               context->histogram_accumulated[i] = context->histogram_accumulated[i - 1]
+                       + context->histogram_counters[i];
+               context->histogram_ratio[i] = ((double) context->histogram_accumulated[i]) / num;
+       }
 } /* }}} void update_context */
 
 static void update_host_hook (pingobj_iter_t *iter, /* {{{ */
@@ -1279,16 +1367,28 @@ static void update_host_hook (pingobj_iter_t *iter, /* {{{ */
 #if USE_NCURSES
                if (has_colors () == TRUE)
                {
+                       size_t bucket;
+                       double ratio_this;
+                       double ratio_prev;
                        int color = OPING_GREEN;
-                       double average = context_get_average (context);
-                       double stddev = context_get_stddev (context);
 
-                       if ((latency < (average - (2 * stddev)))
-                                       || (latency > (average + (2 * stddev))))
-                               color = OPING_RED;
-                       else if ((latency < (average - stddev))
-                                       || (latency > (average + stddev)))
+                       bucket = latency_to_bucket (context, latency);
+                       ratio_this = context->histogram_ratio[bucket];
+                       if (bucket > 0)
+                               ratio_prev = context->histogram_ratio[bucket - 1];
+                       else
+                               ratio_prev = 0.0;
+
+                       if ((ratio_this <= threshold_green)
+                                       || ((ratio_prev < threshold_green)
+                                               && (ratio_this > threshold_green)))
+                               color = OPING_GREEN;
+                       else if ((ratio_this <= threshold_yellow)
+                                       || ((ratio_prev < threshold_yellow)
+                                               && (ratio_this > threshold_yellow)))
                                color = OPING_YELLOW;
+                       else
+                               color = OPING_RED;
 
                        HOST_PRINTF ("%zu bytes from %s (%s): icmp_seq=%u ttl=%i ",
                                        data_len, context->host, context->addr,
@@ -1322,7 +1422,7 @@ static void update_host_hook (pingobj_iter_t *iter, /* {{{ */
                }
 #endif
        }
-       else
+       else /* if (!(latency > 0.0)) */
        {
 #if USE_NCURSES
                if (has_colors () == TRUE)
@@ -1387,22 +1487,16 @@ static int post_loop_hook (pingobj_t *ping) /* {{{ */
 
                if (context->req_rcvd != 0)
                {
-                       double average;
-                       double deviation;
+                       double median;
                        double percentile;
 
-                       average = context_get_average (context);
-                       deviation = context_get_stddev (context);
+                       median = context_get_percentile (context, 50.0);
                        percentile = context_get_percentile (context, opt_percentile);
 
-                       printf ("rtt min/avg/%.0f%%/max/sdev = "
-                                       "%.3f/%.3f/%.0f/%.3f/%.3f ms\n",
-                                       opt_percentile,
-                                       context->latency_min,
-                                       average,
-                                       percentile,
-                                       context->latency_max,
-                                       deviation);
+                       printf ("RTT[ms]: min = %.0f, median = %.0f, p(%.0f) = %.0f, max = %.0f\n",
+                                       context->latency_min, median,
+                                       opt_percentile, percentile,
+                                       context->latency_max);
                }
 
                ping_iterator_set_context (iter, NULL);
@@ -1618,6 +1712,9 @@ int main (int argc, char **argv) /* {{{ */
                exit (EXIT_FAILURE);
        }
 
+       if (host_num == 0)
+               exit (EXIT_FAILURE);
+
 #if _POSIX_SAVED_IDS
        saved_set_uid = (uid_t) -1;
 #endif