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