1 /*
2  * Copyright 2024 The Hafnium Authors.
3  *
4  * Use of this source code is governed by a BSD-style
5  * license that can be found in the LICENSE file or at
6  * https://opensource.org/licenses/BSD-3-Clause.
7  */
8 
9 #include "gtest/gtest.h"
10 
11 extern "C" {
12 #include "hf/bits.h"
13 }
14 
TEST(bits,is_bit_set)15 TEST(bits, is_bit_set)
16 {
17 	ASSERT_EQ(IS_BIT_SET(UINT64_C(1) << 63, 63), true);
18 	ASSERT_EQ(IS_BIT_SET(0, 63), false);
19 }
20 
TEST(bits,is_bit_unset)21 TEST(bits, is_bit_unset)
22 {
23 	ASSERT_EQ(IS_BIT_UNSET(UINT64_C(1) << 63, 63), false);
24 	ASSERT_EQ(IS_BIT_UNSET(0, 63), true);
25 }
26 
TEST(bits,get_bits_mask)27 TEST(bits, get_bits_mask)
28 {
29 	ASSERT_EQ(GET_BITS_MASK(0, 0), 0b0001);
30 	ASSERT_EQ(GET_BITS_MASK(1, 0), 0b0011);
31 	ASSERT_EQ(GET_BITS_MASK(2, 0), 0b0111);
32 	ASSERT_EQ(GET_BITS_MASK(7, 0), 0b1111'1111);
33 	ASSERT_EQ(GET_BITS_MASK(7, 1), 0b1111'1110);
34 
35 	ASSERT_EQ(GET_BITS_MASK(63, 0), UINT64_MAX);
36 	ASSERT_EQ(GET_BITS_MASK(63, 1), UINT64_MAX - 1);
37 }
38 
TEST(bits,get_bits)39 TEST(bits, get_bits)
40 {
41 	ASSERT_EQ(GET_BITS(0xAF, 3, 0), 0x0F);
42 	ASSERT_EQ(GET_BITS(0xAF, 7, 4), 0xA0);
43 	ASSERT_EQ(GET_BITS(0x10, 7, 4), 0x10);
44 	ASSERT_EQ(GET_BITS(0x0F, 7, 4), 0x00);
45 }
46 
TEST(bits,any_bits_set)47 TEST(bits, any_bits_set)
48 {
49 	ASSERT_EQ(ANY_BITS_SET(0x01, 3, 0), true);
50 	ASSERT_EQ(ANY_BITS_SET(0x00, 3, 0), false);
51 	ASSERT_EQ(ANY_BITS_SET(0x10, 7, 4), true);
52 	ASSERT_EQ(ANY_BITS_SET(0x0F, 7, 4), false);
53 }
54 
TEST(bits,all_bits_set)55 TEST(bits, all_bits_set)
56 {
57 	ASSERT_EQ(ALL_BITS_SET(0b1111, 3, 0), true);
58 	ASSERT_EQ(ALL_BITS_SET(0b0111, 3, 0), false);
59 	ASSERT_EQ(ALL_BITS_SET(0xF0, 7, 4), true);
60 	ASSERT_EQ(ALL_BITS_SET(0xA0, 7, 4), false);
61 }
62 
TEST(bits,any_bits_unset)63 TEST(bits, any_bits_unset)
64 {
65 	ASSERT_EQ(ANY_BITS_UNSET(0x00, 3, 0), true);
66 	ASSERT_EQ(ANY_BITS_UNSET(0x01, 3, 0), true);
67 	ASSERT_EQ(ANY_BITS_UNSET(0x0F, 3, 0), false);
68 	ASSERT_EQ(ANY_BITS_UNSET(0xF0, 7, 4), false);
69 	ASSERT_EQ(ANY_BITS_UNSET(0xA0, 7, 4), true);
70 }
71 
TEST(bits,all_bits_unset)72 TEST(bits, all_bits_unset)
73 {
74 	ASSERT_EQ(ALL_BITS_UNSET(0x01, 3, 0), false);
75 	ASSERT_EQ(ALL_BITS_UNSET(0x00, 3, 0), true);
76 	ASSERT_EQ(ALL_BITS_UNSET(0x10, 7, 4), false);
77 	ASSERT_EQ(ALL_BITS_UNSET(0x0F, 7, 4), true);
78 }
79