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 int main(int argc, char **argv) 63 { 64 g_test_init(&argc, &argv, NULL); 65 66 g_test_add_func("/bitmap/bitmap_copy_with_offset", 67 check_bitmap_copy_with_offset); 68 69 g_test_run(); 70 71 return 0; 72 } 73