Question

我想以原子方式修改结构的某些元素。我当前的实现使用互斥锁来保护关键代码，可以在下面看到。

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <pthread.h>

pthread_mutex_t thread_mutex = PTHREAD_MUTEX_INITIALIZER;

#define ITER    100000

typedef struct global_status {
    int32_t context_delta;
    uint32_t global_access_count;
} global_status_t;

global_status_t g_status;

void *context0(void *ptr)
{
    unsigned int iter = ITER;
    while (iter--) {
        wait_event_from_device0();
        pthread_mutex_lock(&thread_mutex);
        g_status.context_delta++;
        g_status.global_access_count++;
        pthread_mutex_unlock(&thread_mutex);
    }

    return NULL;
}

void *context1(void *ptr)
{
    unsigned int iter = ITER;
    while (iter--) {
        wait_event_from_device1();
        pthread_mutex_lock(&thread_mutex);
        g_status.context_delta--;
        g_status.global_access_count++;
        pthread_mutex_unlock(&thread_mutex);
    }

    return NULL;
}

int main(int argc, char **argv)
{
    pthread_t tid0, tid1;
    int iret;

    if ((iret = pthread_create(&tid0, NULL, context0, NULL))) {
         fprintf(stderr, "context0 creation error!\n");
         return EXIT_FAILURE;
    }

    if ((iret = pthread_create(&tid1, NULL, context1, NULL))) {
         fprintf(stderr, "context1 creation error!\n");
         return EXIT_FAILURE;
    }

    pthread_join(tid0, NULL);
    pthread_join(tid1, NULL);

    printf("%d, %d\n", g_status.context_delta, g_status.global_access_count);
    return 0;
}

我打算将此代码移植到不支持posix的RTOS中，我想原子地执行此操作，而不使用互斥锁或禁用/启用中断。

我该怎么做这个操作？是否可以使用'原子比较和交换功能'（CAS）？

Answer 1

在您的示例中，您似乎有两个线程服务于不同的设备。您可以使用每个设备结构完全取消锁定。全局将是所有每个设备统计数据的总和。如果确实需要锁定，可以使用CAS，LL / SC或任何支持的基础原子构造。

Answer 2

我所做的是创建一个包含我想要同时更改的所有字段的联合。像这样：

union {
  struct {
    int            m_field1;
    unsigned short m_field2 : 2,
                   m_field3 : 1;
    BYTE           m_field4; 
  }
  unsigned long long m_n64;
  TData(const TData& r) { m_n64 = r.m_n64; }
} TData;

你在这样的大结构中嵌入这样的联盟：

struct {
  ...
  volatile TData m_Data;
  ...
} TBiggerStruct;

然后我做这样的事情：

while (1) {
  TData Old = BiggerSharedStruct.m_Data, New = Old;
  New.field1++;
  New.field4--;
  if (CAS(&SharedData.m_n64, Old.m_n64, New.m_n64))
    break; // success
}

我做了大量的字段打包，我想要同时更改为尽可能小的16位，32位或64位结构。我认为intel上的128位内容并不像64位内容那么快，所以我避免使用它。我暂时没有对它进行基准测试，所以我可能错了。

如何在不使用C的锁的情况下原子地修改结构元素？

2 个答案: