用指针低三位存放额外信息的优化方法
2022-04-30 本文已影响0人
疾风2018
在8字节对齐的情况下指针的低三位都是0,在某些情况下我们需要维持跟指针对应的额外信息,例如标志位,但是又不想为此创建一个结构体,那就可以利用指针的低三位来存储额外信息。
比如说我们希望原子操作的类型是64位的数据,这样可以利用CMPXCHG机器指令实现CAS操作,即我们希望定义std::atomic<T *>
的变量实现原子操作,而不希望定义std::atomic<S>
,S为某个结构体。这时候就可以应用上述技巧。
具体实现方式如下所示:
std::uintptr_t
是定义一个可以表示地址值(指针值即地址值)的无符号整型。将指针转换成std::uintptr_t
然后进行位运算。以下代码作为示例:
#include <iostream>
#include <stdint.h>
struct Data
{
// 定义一些数据成员
long int a;
long int b;
long int c;
};
// 定义三个标志位,不用细究三个标志位的具体含义,这个在这里不重要。
static constexpr std::uintptr_t HAS_DATA = 1;
static constexpr std::uintptr_t NEED_DATA = 2;
static constexpr std::uintptr_t CLOSED = 4;
static constexpr std::uintptr_t FLAGS_MASK = HAS_DATA | NEED_DATA | CLOSED;
static constexpr std::uintptr_t PTR_MASK = ~FLAGS_MASK;
int main()
{
Data * dp = new Data();
dp->a = 88;
dp->b = 99;
dp->c = 77;
std::cout << "指针值为 " << std::hex << reinterpret_cast<int64_t>(dp) << std::endl;
// 在指针上附加上标志位
std::uintptr_t ptr_int = reinterpret_cast<std::uintptr_t>(dp) | HAS_DATA;
std::cout << "加过标记位后 " << std::hex << ptr_int << std::endl;
// 取标志位
std::uintptr_t flags = ptr_int & FLAGS_MASK;
std::cout << "标记位 " << flags << std::endl;
// 需要用指针的时候,清零低三位,恢复指针值原来的值
dp = reinterpret_cast<Data*>(ptr_int & PTR_MASK);
std::cout << "使用恢复后的指针" << std::dec << dp->a << ", " << dp->b << ", " << dp->c << std::endl;
return 0;
}