Fixup t/ test apps for lfsr_next change()

[fio.git] / t / axmap.c

diff --git a/t/axmap.c b/t/axmap.c
index 1f8c3e9a90f6855878f37c0fb574217f6363094e..e32ff98d09f1323a12b9cb1c708cb49370ca0a38 100644 (file)
--- a/t/axmap.c
+++ b/t/axmap.c
@@ -6,10 +6,7 @@
 #include <inttypes.h>
 
 #include "../lib/lfsr.h"
-
-struct axmap;
-void axmap_set(struct axmap *, uint64_t);
-struct axmap *axmap_new(uint64_t size);
+#include "../lib/axmap.h"
 
 void *smalloc(size_t size)
 {
@@ -21,26 +18,125 @@ void sfree(void *ptr)
        free(ptr);
 }
 
-int main(int argc, char *argv[])
+static int test_regular(size_t size, int seed)
 {
        struct fio_lfsr lfsr;
-       size_t size = (1UL << 28) - 200;
        struct axmap *map;
+       size_t osize;
+       uint64_t ff;
+       int err;
 
-       if (argc > 1)
-               size = strtoul(argv[1], NULL, 10);
-
-       printf("Using %llu entries\n", (unsigned long long) size);
+       printf("Using %llu entries...", (unsigned long long) size);
+       fflush(stdout);
 
-       lfsr_init(&lfsr, size);
+       lfsr_init(&lfsr, size, seed, seed & 0xF);
        map = axmap_new(size);
+       osize = size;
+       err = 0;
 
        while (size--) {
                uint64_t val;
 
-               lfsr_next(&lfsr, &val);
+               if (lfsr_next(&lfsr, &val)) {
+                       printf("lfsr: short loop\n");
+                       err = 1;
+                       break;
+               }
+               if (axmap_isset(map, val)) {
+                       printf("bit already set\n");
+                       err = 1;
+                       break;
+               }
                axmap_set(map, val);
+               if (!axmap_isset(map, val)) {
+                       printf("bit not set\n");
+                       err = 1;
+                       break;
+               }
+       }
+
+       if (err)
+               return err;
+
+       ff = axmap_next_free(map, osize);
+       if (ff != (uint64_t) -1ULL) {
+               printf("axmap_next_free broken: got %llu\n", (unsigned long long) ff);
+               return 1;
+       }
+
+       printf("pass!\n");
+       axmap_free(map);
+       return 0;
+}
+
+static int test_multi(size_t size, unsigned int bit_off)
+{
+       unsigned int map_size = size;
+       struct axmap *map;
+       uint64_t val = bit_off;
+       int i, err;
+
+       printf("Test multi %llu entries %u offset...", (unsigned long long) size, bit_off);
+       fflush(stdout);
+
+       map = axmap_new(map_size);
+       while (val + 128 <= map_size) {
+               err = 0;
+               for (i = val; i < val + 128; i++) {
+                       if (axmap_isset(map, val + i)) {
+                               printf("bit already set\n");
+                               err = 1;
+                               break;
+                       }
+               }
+
+               if (err)
+                       break;
+
+               err = axmap_set_nr(map, val, 128);
+               if (err != 128) {
+                       printf("only set %u bits\n", err);
+                       break;
+               }
+
+               err = 0;
+               for (i = 0; i < 128; i++) {
+                       if (!axmap_isset(map, val + i)) {
+                               printf("bit not set: %llu\n", (unsigned long long) val + i);
+                               err = 1;
+                               break;
+                       }
+               }
+
+               val += 128;
+               if (err)
+                       break;
        }
 
+       if (!err)
+               printf("pass!\n");
+
+       axmap_free(map);
+       return err;
+}
+
+int main(int argc, char *argv[])
+{
+       size_t size = (1UL << 23) - 200;
+       int seed = 1;
+
+       if (argc > 1) {
+               size = strtoul(argv[1], NULL, 10);
+               if (argc > 2)
+                       seed = strtoul(argv[2], NULL, 10);
+       }
+
+       if (test_regular(size, seed))
+               return 1;
+       if (test_multi(size, 0))
+               return 2;
+       if (test_multi(size, 17))
+               return 3;
+
        return 0;
 }