Interlocked原子访问系列函数

灰太狼 2022-08-20 14:13 117阅读 0赞

所谓原子访问，指的是一个线程在访问某个资源的同时能够保证没有其他线程会在同一时刻访问同一资源。Interlocked系列函数提供了这样的操作。所有这些函数会以原子方式来操控一个值。

Interlocked函数的工作原理取决于代码运行的CPU平台，如果是x86系列CPU，那么Interlocked函数会在总线上维持一个硬件信号，这个信号会阻止其他CPU访问同一个内存地址。我们必须确保传给这些函数的变量地址是经过对齐的，否则这些函数可能会失败。C运行库提供了一个\_aligned\_malloc函数，我们可以使用这个函数来分配一块对齐过的内存：

void \* \_aligned\_malloc(

size\_t size, //要分配的字节数

size\_t alignment //要对齐到的字节边界，传给alignment的值必须是2的整数幂次方

);

Interlocked函数的另一个需要注意的点是它们执行得很快。调用一次Interlocked函数通常只占用几个CPU周期（通常小于50），而且不需要在用户模式和内核模式之间进行切换（这个切换通常需要占用1000个CPU周期以上）。

1）原子加减操作InterlockedExchangeAdd函数原型如下：

LONG \_\_cdecl InterlockedExchangeAdd( //对32位值进行操作

\_\_inout LONG volatile \*Addend, //需要递增的变量地址

\_\_in LONG Value //增量值，可为负值表示减法

);

LONGLONG \_\_cdecl InterlockedExchangeAdd64( //对64位值进行操作

\_\_inout LONGLONG volatile \*Addend,

\_\_in LONGLONG Value

);

2）InterlockedExchange函数用于原子地将32位整数设为指定的值：

LONG \_\_cdecl InterlockedExchange(

\_\_inout LONG volatile \*Target, //指向要替换的32位值的指针

\_\_in LONG Value //替换的值

);

返回值是指向原先的32位整数值。

InterlockedExchangePointer函数原子地用于替换地址值：

PVOID \_\_cdecl InterlockedExchangePointer(

\_\_inout PVOID volatile \*Target, //指向要替换的地址值的指针

\_\_in PVOID Value //替换的地址值

);

返回值是原来的地址值。

对32位应用程序来说，以上两个函数都用一个32位值替换另一个32位值，但对64位应用程序来说，InterlockedExchange替换的是32位值，而InterlockedExchangePointer替换的是64位值。

当然，还有一个函数InterlockedExchange64专门用来原子地操作64位值的：

LONGLONG \_\_cdecl InterlockedExchange64(

\_\_inout LONGLONG volatile \*Target,

\_\_in LONGLONG Value

);

在实现旋转锁时，InterlockedExchange函数极其有用：

//标识一个共享资源是否正在被使用的全局变量

BOOL g\_fResourceInUse = FALSE;

...

void ASCEFunc()

//等待访问共享资源

while(InterlockedExchange(&g\_fResourceInUse, TRUE) == TRUE)

sleep(0);

//访问共享资源

...

//结束访问

InterlockedExchange(&g\_fResourceInUse, FALSE);

注意，在使用这项技术时要小心，因为旋转锁会耗费CPU时间。特别是在单CPU机器上应该避免使用旋转锁，如果一个线程不停地循环，那么这会浪费宝贵的CPU时间，而且会阻止其他线程改变该锁的值。

3）函数InterlockedCompareExchange函数和InterlockedCompareExchangePointer函数原型如下：

LONG \_\_cdecl InterlockedCompareExchange(

\_\_inout LONG volatile \*Destination, //当前值

\_\_in LONG Exchange, //

\_\_in LONG Comparand //比较值

);

PVOID \_\_cdecl InterlockedCompareExchangePointer(

\_\_inout PVOID volatile \*Destination,

\_\_in PVOID Exchange,

\_\_in PVOID Comparand

);

这两个函数以原子方式执行一个测试和设置操作。对32位应用程序来说，这两个函数都对32位值进行操作；在64位应用程序中，InterlockedCompareExchange对32位值进行操作而InterlockedCompareExchangePointer对64位值进行操作。函数会将当前值（Destination指向的）与参数Comparand进行比较，如果两个值相同，那么函数会将\*Destination修改为Exchange参数指定的值。若不等，则\*Destination保持不变。函数会返回\*Destination原来的值。所有这些操作都是一个原子执行单元来完成的。

当然，这两个函数的64位版本是：

LONGLONG \_\_cdecl InterlockedCompareExchange64(

\_\_inout LONGLONG volatile \*Destination,

\_\_in LONGLONG Exchange,

\_\_in LONGLONG Comparand

);