tally: don't add extra bucket unless buckets == 0

[ccan] / ccan / tally / tally.c

diff --git a/ccan/tally/tally.c b/ccan/tally/tally.c
index 9f2a4591f8c3ce5ddd05b8f29702ff2761784dad..b3ac863ba68119aedc59973b72c1d3716bb35867 100644 (file)
--- a/ccan/tally/tally.c
+++ b/ccan/tally/tally.c
@@ -16,25 +16,28 @@ struct tally {
        size_t total[2];
        /* This allows limited frequency analysis. */
        unsigned buckets, step_bits;
        size_t total[2];
        /* This allows limited frequency analysis. */
        unsigned buckets, step_bits;

-       size_t counts[1 /* [buckets] */ ];
+       size_t counts[1 /* Actually: [buckets] */ ];

 };
 
 struct tally *tally_new(unsigned buckets)
 {
        struct tally *tally;
 
 };
 
 struct tally *tally_new(unsigned buckets)
 {
        struct tally *tally;
 

+       /* There is always 1 bucket. */
+       if (buckets == 0)
+               buckets = 1;
+

        /* Check for overflow. */
        if (buckets && SIZE_MAX / buckets < sizeof(tally->counts[0]))
                return NULL;
        /* Check for overflow. */
        if (buckets && SIZE_MAX / buckets < sizeof(tally->counts[0]))
                return NULL;

-       tally = malloc(sizeof(*tally) + sizeof(tally->counts[0])*buckets);
+       tally = malloc(sizeof(*tally) + sizeof(tally->counts[0])*(buckets-1));

        if (tally) {
                tally->max = ((size_t)1 << (SIZET_BITS - 1));
                tally->min = ~tally->max;
                tally->total[0] = tally->total[1] = 0;
        if (tally) {
                tally->max = ((size_t)1 << (SIZET_BITS - 1));
                tally->min = ~tally->max;
                tally->total[0] = tally->total[1] = 0;

-               /* There is always 1 bucket. */
-               tally->buckets = buckets+1;
+               tally->buckets = buckets;

                tally->step_bits = 0;
                tally->step_bits = 0;

-               memset(tally->counts, 0, sizeof(tally->counts[0])*(buckets+1));
+               memset(tally->counts, 0, sizeof(tally->counts[0])*buckets);

        }
        return tally;
 }
        }
        return tally;
 }


@@ -422,7 +425,7 @@ char *tally_histogram(const struct tally *tally,
        } else {
                /* We create a temporary then renormalize so < height. */
                /* FIXME: Antialias properly! */
        } else {
                /* We create a temporary then renormalize so < height. */
                /* FIXME: Antialias properly! */

-               tmp = tally_new(tally->buckets-1);
+               tmp = tally_new(tally->buckets);

                if (!tmp)
                        return NULL;
                tmp->min = tally->min;
                if (!tmp)
                        return NULL;
                tmp->min = tally->min;