tests/test-bitmap.c

   1 /*
   2  * SPDX-License-Identifier: GPL-2.0-or-later
   3  *
   4  * Bitmap.c unit-tests.
   5  *
   6  * Copyright (C) 2019, Red Hat, Inc.
   7  *
   8  * Author: Peter Xu <peterx@redhat.com>
   9  */
  10
  11 #include <stdlib.h>
  12 #include "qemu/osdep.h"
  13 #include "qemu/bitmap.h"
  14
  15 #define BMAP_SIZE  1024
  16
  17 static void check_bitmap_copy_with_offset(void)
  18 {
  19     unsigned long *bmap1, *bmap2, *bmap3, total;
  20
  21     bmap1 = bitmap_new(BMAP_SIZE);
  22     bmap2 = bitmap_new(BMAP_SIZE);
  23     bmap3 = bitmap_new(BMAP_SIZE);
  24
  25     bmap1[0] = random();
  26     bmap1[1] = random();
  27     bmap1[2] = random();
  28     bmap1[3] = random();
  29     total = BITS_PER_LONG * 4;
  30
  31     /* Shift 115 bits into bmap2 */
  32     bitmap_copy_with_dst_offset(bmap2, bmap1, 115, total);
  33     /* Shift another 85 bits into bmap3 */
  34     bitmap_copy_with_dst_offset(bmap3, bmap2, 85, total + 115);
  35     /* Shift back 200 bits back */
  36     bitmap_copy_with_src_offset(bmap2, bmap3, 200, total);
  37
  38     g_assert_cmpmem(bmap1, total / BITS_PER_LONG,
  39                     bmap2, total / BITS_PER_LONG);
  40
  41     bitmap_clear(bmap1, 0, BMAP_SIZE);
  42     /* Set bits in bmap1 are 100-245 */
  43     bitmap_set(bmap1, 100, 145);
  44
  45     /* Set bits in bmap2 are 60-205 */
  46     bitmap_copy_with_src_offset(bmap2, bmap1, 40, 250);
  47     g_assert_cmpint(find_first_bit(bmap2, 60), ==, 60);
  48     g_assert_cmpint(find_next_zero_bit(bmap2, 205, 60), ==, 205);
  49     g_assert(test_bit(205, bmap2) == 0);
  50
  51     /* Set bits in bmap3 are 135-280 */
  52     bitmap_copy_with_dst_offset(bmap3, bmap1, 35, 250);
  53     g_assert_cmpint(find_first_bit(bmap3, 135), ==, 135);
  54     g_assert_cmpint(find_next_zero_bit(bmap3, 280, 135), ==, 280);
  55     g_assert(test_bit(280, bmap3) == 0);
  56
  57     g_free(bmap1);
  58     g_free(bmap2);
  59     g_free(bmap3);
  60 }
  61
  62 typedef void (*bmap_set_func)(unsigned long *map, long i, long len);
  63 static void bitmap_set_case(bmap_set_func set_func)
  64 {
  65     unsigned long *bmap;
  66     int offset;
  67
  68     bmap = bitmap_new(BMAP_SIZE);
  69
  70     /* Both Aligned, set bits [BITS_PER_LONG, 3*BITS_PER_LONG] */
  71     set_func(bmap, BITS_PER_LONG, 2 * BITS_PER_LONG);
  72     g_assert_cmpuint(bmap[1], ==, -1ul);
  73     g_assert_cmpuint(bmap[2], ==, -1ul);
  74     g_assert_cmpint(find_first_bit(bmap, BITS_PER_LONG), ==, BITS_PER_LONG);
  75     g_assert_cmpint(find_next_zero_bit(bmap, 3 * BITS_PER_LONG, BITS_PER_LONG),
  76                     ==, 3 * BITS_PER_LONG);
  77
  78     for (offset = 0; offset <= BITS_PER_LONG; offset++) {
  79         bitmap_clear(bmap, 0, BMAP_SIZE);
  80         /* End Aligned, set bits [BITS_PER_LONG - offset, 3*BITS_PER_LONG] */
  81         set_func(bmap, BITS_PER_LONG - offset, 2 * BITS_PER_LONG + offset);
  82         g_assert_cmpuint(bmap[1], ==, -1ul);
  83         g_assert_cmpuint(bmap[2], ==, -1ul);
  84         g_assert_cmpint(find_first_bit(bmap, BITS_PER_LONG),
  85                         ==, BITS_PER_LONG - offset);
  86         g_assert_cmpint(find_next_zero_bit(bmap,
  87                                            3 * BITS_PER_LONG,
  88                                            BITS_PER_LONG - offset),
  89                         ==, 3 * BITS_PER_LONG);
  90     }
  91
  92     for (offset = 0; offset <= BITS_PER_LONG; offset++) {
  93         bitmap_clear(bmap, 0, BMAP_SIZE);
  94         /* Start Aligned, set bits [BITS_PER_LONG, 3*BITS_PER_LONG + offset] */
  95         set_func(bmap, BITS_PER_LONG, 2 * BITS_PER_LONG + offset);
  96         g_assert_cmpuint(bmap[1], ==, -1ul);
  97         g_assert_cmpuint(bmap[2], ==, -1ul);
  98         g_assert_cmpint(find_first_bit(bmap, BITS_PER_LONG),
  99                         ==, BITS_PER_LONG);
 100         g_assert_cmpint(find_next_zero_bit(bmap,
 101                                            3 * BITS_PER_LONG + offset,
 102                                            BITS_PER_LONG),
 103                         ==, 3 * BITS_PER_LONG + offset);
 104     }
 105
 106     g_free(bmap);
 107 }
 108
 109 static void check_bitmap_set(void)
 110 {
 111     bitmap_set_case(bitmap_set);
 112     bitmap_set_case(bitmap_set_atomic);
 113 }
 114
 115 int main(int argc, char **argv)
 116 {
 117     g_test_init(&argc, &argv, NULL);
 118
 119     g_test_add_func("/bitmap/bitmap_copy_with_offset",
 120                     check_bitmap_copy_with_offset);
 121     g_test_add_func("/bitmap/bitmap_set",
 122                     check_bitmap_set);
 123
 124     g_test_run();
 125
 126     return 0;
 127 }