在进一步分析之前，本文我们先来熟悉下OpenHarmony鸿蒙轻内核提供的位操作模块，在互斥锁等模块对位操作有使用。位操作是指对二进制数的bit位进行操作。程序可以设置某一变量为状态字，状态字中的每一bit位(标志位)可以具有自定义的含义。
1 位操作的宏定义
位操作模块提供对32位无符号整数数值的bit位进行操作，bit位取值为0-31，以0开始计算，从左向右，第0位，第1位。。。第31位等。⑴处定义的宏OS_BITMAP_MASK如下，也就是十进制31。如果传入的比特位pos大于31，会通过逻辑与运算截断(pos & OS_BITMAP_MASK)，只取低5位，确保不会大于31，避免溢出。⑵处定义的位图掩码全是1。
- ⑴ #define OS_BITMAP_MASK 0x1FU
- ⑵ #define OS_BITMAP_WORD_MASK ~0UL
在文件kernel\include\los_bitmap.h中定义了常用的位操作相关的宏。宏BITMAP_WORD根据参数x计算出需要操作第几个状态字，由于计算状态字的使用的是UINTPTR，状态字可以是32位、也可以是64位。后文，我们默认以32位进行讲解。宏BITMAP_FIRST_WORD_MASK传入的参数是位操作的开始bit位数，用于计算需要进行位操作的掩码，从开始位全部是1，宏BITMAP_LAST_WORD_MASK传入的参数是位操作的结束bit位数，用于计算需要进行位操作的掩码，结束位之前全部是1。宏BITMAP_NUM_WORDS传入位数，计算状态字的数量。
- #define _ONE(x) (1 + ((x) - (x)))
- #define BIT(n) (1U << (n))
- #define BIT_GET(x, bit) ((x) & (_ONE(x) << (bit)))
- #define BIT_SHIFT(x, bit) (((x) >> (bit)) & 1)
- #define BITS_GET(x, high, low) ((x) & (((_ONE(x) << ((high) + 1)) - 1) & ~((_ONE(x) << (low)) - 1)))
- #define BITS_SHIFT(x, high, low) (((x) >> (low)) & ((_ONE(x) << ((high) - (low) + 1)) - 1))
- #define BIT_SET(x, bit) (((x) & (_ONE(x) << (bit))) ? 1 : 0)
- #define BITMAP_BITS_PER_WORD (sizeof(UINTPTR) * 8)
- #define BITMAP_NUM_WORDS(x) (((x) + BITMAP_BITS_PER_WORD - 1) / BITMAP_BITS_PER_WORD)
- #define BITMAP_WORD(x) ((x) / BITMAP_BITS_PER_WORD)
- #define BITMAP_BIT_IN_WORD(x) ((x) & (BITMAP_BITS_PER_WORD - 1))
- #define BITMAP_FIRST_WORD_MASK(start) (~0UL << ((start) % BITMAP_BITS_PER_WORD))
- #define BITMAP_LAST_WORD_MASK(nbits) \
- (((nbits) % BITMAP_BITS_PER_WORD) ? (1UL << ((nbits) % BITMAP_BITS_PER_WORD)) - 1 : ~0UL)
- #define BITMAP_BITS_PER_INT (sizeof(INTPTR) * 8)
- #define BITMAP_BIT_IN_INT(x) ((x) & (BITMAP_BITS_PER_INT - 1))
- #define BITMAP_INT(x) ((x) / BITMAP_BITS_PER_INT)
- #define BIT_MASK(x) (((x) >= sizeof(UINTPTR) * 8) ? (0UL - 1) : ((1UL << (x)) - 1))
2 位操作常用功能
OpenHarmony鸿蒙轻内核的位操作模块提供标志位的置1和清0操作，可以改变标志位的内容，同时还提供获取状态字中标志位为1的最高位和最低位的功能。用户也可以对系统的寄存器进行位操作。位操作提供了7个API，进行置1、清0、获取为1的最高、最低位等操作，如下：
下面，我们剖析下位操作的源代码。
2.1 LOS_BitmapSet()对状态字的某一标志位进行置1操作
对状态字的某一标志位进行置1操作。我们先看看传入的参数，需要的2个参数分别是：需要改变bit位内容的状态字UINT32 *bitmap，需要改变的bit位位数UINT16 pos。
代码很简单，首先进行基础的校验，如果状态字为空，则返回。然后计算pos & OS_BITMAP_MASK，只取二进制的低5位，最大位值为31，避免左移的时候发生溢出。1U << (pos & OS_BITMAP_MASK)就是需要改变内容的状态字的bit位，通过按位或运算设置状态字UINT32 *bitmap的指定bit位的内容为1。
- VOID LOS_BitmapSet(UINT32 *bitmap, UINT16 pos)
- {
- if (bitmap == NULL) {
- return;
- }
- *bitmap |= 1U << (pos & OS_BITMAP_MASK);
- }
2.2 LOS_BitmapClr()对状态字的某一标志位进行清0操作
对状态字的某一标志位进行清0操作，代码和置1操作对应，比较简单，~(1U << (pos & OS_BITMAP_MASK))表示需要改变内容的状态字的bit位为0，其余位为1，然后通过按位与运算设置状态字UINT32 *bitmap的指定bit位的内容为0。
- VOID LOS_BitmapClr(UINT32 *bitmap, UINT16 pos)
- {
- if (bitmap == NULL) {
- return;
- }
- *bitmap &= ~(1U << (pos & OS_BITMAP_MASK));
- }
2.3 LOS_HighBitGet()获取状态字中为1的最高位
代码中CLZ(bitmap)是宏，展开为(__builtin_clz(bitmap))，这是编译器内置的高效位运算的库函数，clz是count leading zeros的缩写，就是统计二进制数值中高位区开头的全是0的数目。使用OS_BITMAP_MASK减去该值，结果就是状态字中的1的最高位。
- UINT16 LOS_HighBitGet(UINT32 bitmap)
- {
- if (bitmap == 0) {
- return LOS_INVALID_BIT_INDEX;
- }
- return (OS_BITMAP_MASK - CLZ(bitmap));
- }
2.4 LOS_LowBitGet()获取状态字中为1的最低位
代码其中CTZ(bitmap)是宏，展开为(__builtin_ctz(value))，这是编译器内置的高效位运算的库函数，ctz是count trailing zeros的缩写，就是统计二进制数值中低位区结尾的全是0的数目，该结果就是状态字中的1的最低位。
- UINT16 LOS_LowBitGet(UINT32 bitmap)
- {
- if (bitmap == 0) {
- return LOS_INVALID_BIT_INDEX;
- }
- return CTZ(bitmap);
- }
2.5 LOS_BitmapSetNBits()对状态字的连续标志位进行置1操作
可以使用LOS_BitmapSetNBits()函数对状态字的连续比特位进行置1操作，第一个参数是需要改变bit位内容的状态字UINT32 *bitmap，第二个参数是需要置1的bit位开始数start，第三个参数是需要置1的数量numsSet。由于bit位开始数start并没有限制在[0,31]，所以实际上设置的可能是UINT32 *bitmap状态字后面的状态字，需要根据业务实际情况进行设置，避免覆写其他内存。同样，需要置1的数量numsSet也可能跨多个状态字。如图所示：
我们看下代码，
⑴处计算出需要操作的状态字，其中BITMAP_WORD(start)计算相对状态字bitmap需要偏移的数量，如果start处于区间[0,31]，BITMAP_WORD(start)等于0，操作的就是状态字bitmap。如果start处于区间[32,63]，BITMAP_WORD(start)等于1，操作的就是状态字bitmap后面的第一个状态字，以此类推。
⑵处size可以和bit位开始数start结合来理解，size就是需要置1的bit位结束位数。
⑶处需要置1操作的bit位的位数。
⑷是对应需要置1操作的bit位的掩码。
⑸处如果条件成立，说明需要置1操作需要跨多个状态字进行操作，代码会一个状态字处理完毕，再去处理下一个状态字。
⑹处把当前状态字的相应的bit位进行置1操作，然后执行⑺把剩余需要置1的位数减去已经置1的位数。
⑻处更新bitsToSet和maskToSet，然后指针p指向下一个状态字。
⑼处如果需要置1的位数大于0，并且此时已经可以在一个状态字内完成操作，执行⑽处计算需要置1操作的掩码，从bit开始位到结束位需要进行置1。
⑾处代码执行置1操作，完成对状态字的连续标志位进行置1操作。
- VOID LOS_BitmapSetNBits(UINTPTR *bitmap, UINT32 start, UINT32 numsSet)
- {
- ⑴ UINTPTR *p = bitmap + BITMAP_WORD(start);
- ⑵ const UINT32 size = start + numsSet;
- ⑶ UINT16 bitsToSet = BITMAP_BITS_PER_WORD - (start % BITMAP_BITS_PER_WORD);
- ⑷ UINTPTR maskToSet = BITMAP_FIRST_WORD_MASK(start);
- ⑸ while (numsSet > bitsToSet) {
- ⑹ *p |= maskToSet;
- ⑺ numsSet -= bitsToSet;
- ⑻ bitsToSet = BITMAP_BITS_PER_WORD;
- maskToSet = OS_BITMAP_WORD_MASK;
- p++;
- }
- ⑼ if (numsSet) {
- ⑽ maskToSet &= BITMAP_LAST_WORD_MASK(size);
- *p |= maskToSet;
- }
- }
2.6 LOS_BitmapClrNBits()对状态字的连续标志位进行清0操作
可以使用LOS_BitmapClrNBits()函数对状态字的连续比特位进行清0操作，第一个参数是需要改变bit位内容的状态字UINT32 *bitmap，第二个参数是需要清0的bit位开始数start，第三个参数是需要清0的数量numsClear。该函数是函数LOS_BitmapSetNBits()的反向操作，代码解释可以参考函数LOS_BitmapSetNBits()。
- VOID LOS_BitmapClrNBits(UINTPTR *bitmap, UINT32 start, UINT32 numsClear)
- {
- UINTPTR *p = bitmap + BITMAP_WORD(start);
- const UINT32 size = start + numsClear;
- UINT16 bitsToClear = BITMAP_BITS_PER_WORD - (start % BITMAP_BITS_PER_WORD);
- UINTPTR maskToClear = BITMAP_FIRST_WORD_MASK(start);
- while (numsClear >= bitsToClear) {
- *p &= ~maskToClear;
- numsClear -= bitsToClear;
- bitsToClear = BITMAP_BITS_PER_WORD;
- maskToClear = OS_BITMAP_WORD_MASK;
- p++;
- }
- if (numsClear) {
- maskToClear &= BITMAP_LAST_WORD_MASK(size);
- *p &= ~maskToClear;
- }
- }
2.7 LOS_BitmapFfz()获取从最低有效位开始的第一个0的bit位
可以使用LOS_BitmapFfz()函数获取从最低有效位开始的第一个0的bit位位数，第一个参数是需要改变bit位内容的状态字UINT32 *bitmap，第二个参数numBits表示最大的位数，对返回值进行限制，需要在指定的位数内找到符合条件的位数，否则返回-1。
在看函数代码之前，先了解下Ffz()函数，如下：调用内嵌函数__builtin_ffsl()可以获取一个unsigned long类型数字的二进制形式的从左开始的第一个1的位数，这个位数从1开始计数。比如对于二进制数字0110，该函数会返回2。在下面的函数中，给函数__builtin_ffsl()传入的参数进行了取反，并减去了1，所以Ffz()函数返回一个数字从左开始的第一个0的位数，这个位数从0开始计数。
- /* find first zero bit starting from LSB */
- STATIC INLINE UINT16 Ffz(UINTPTR x)
- {
- return __builtin_ffsl(~x) - 1;
- }
我们接着看下函数LOS_BitmapFfz()的代码。⑴处根据位数numBits计算出对应的状态字的数量，然后依次循环每一个状态字，⑵处如果状态字全为1，则继续循环，否则执行⑶。执行到⑶说明，，前面有i个状态字的各个位全为1。i * BITMAP_BITS_PER_WORD + Ffz(bitmap[i])就表示各个状态字的二进制位中，从左到右第一个0的位置。⑷处如果获取的位数小于第二个参数，则返回获取的位数，否则返回-1。如下图所示：
源代码如下：
- INT32 LOS_BitmapFfz(UINTPTR *bitmap, UINT32 numBits)
- {
- INT32 bit, i;
- ⑴ for (i = 0; i < BITMAP_NUM_WORDS(numBits); i++) {
- ⑵ if (bitmap[i] == OS_BITMAP_WORD_MASK) {
- continue;
- }
- ⑶ bit = i * BITMAP_BITS_PER_WORD + Ffz(bitmap[i]);
- ⑷ if (bit < numBits) {
- return bit;
- }
- return -1;
- }
- return -1;
- }
小结
本文带领大家一起剖析了鸿蒙轻内核的位操作模块的源代码。
