Linux应用程序：使用互斥锁保护临界区

本文介绍Linux应用编程中怎样使用互斥锁保护临界区。示例中两个线程操作同一个全局变量来模拟临界区。

#include <stdio.h>
#include <errno.h>
#include <pthread.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/prctl.h>
#include <semaphore.h>

#define errExit(err, msg) \
	do { errno = err; perror(msg); \
		exit(EXIT_FAILURE); } while(0)

#define TEST_COUNT 100000

int count;
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;

void *__thread1_func(void *arg)
{
	char *name = "thread1";
	int ret;
	ret = prctl(PR_SET_NAME, (unsigned long)name, NULL, NULL, NULL);
	if(ret < 0) {
		perror("prctl PR_SET_NAME");
		exit(EXIT_FAILURE);
	}
	for(int i = 0; i < TEST_COUNT; i++) {
		pthread_mutex_lock(&mutex);
		count++;
		pthread_mutex_unlock(&mutex);
	}
}

void *__thread2_func(void *arg)
{
	char *name = "thread2";
	int ret;
	ret = prctl(PR_SET_NAME, (unsigned long)name, NULL, NULL, NULL);
	if(ret < 0) {
		perror("prctl PR_SET_NAME");
		exit(EXIT_FAILURE);
	}
	for(int i = 0; i < TEST_COUNT; i++) {
		pthread_mutex_lock(&mutex);
                count++;
                pthread_mutex_unlock(&mutex);
	}
}
int main()
{
	pthread_t thread1;
	pthread_t thread2;
	int ret;
	
	count = 0;
	ret = pthread_create(&thread1, NULL, __thread1_func, NULL);
	if(ret) {
		errExit(ret, "create thread1");
	}

	ret = pthread_create(&thread2, NULL, __thread2_func, NULL);
	if(ret) {
		errExit(ret, "thread2");
	}


	pthread_join(thread1, NULL);
	pthread_join(thread2, NULL);
	pthread_mutex_destroy(&mutex);
	printf("count = %d\n", count);
	printf("test end\n");
	return 0;
}