进程间使用信号量协调对共享资源访问c++代码实现

//代码不是本人原创，只是看到后自己写一遍<a target=_blank href="http://blog.csdn.net/ljianhui/article/details/10243617#comments">http://blog.csdn.net/ljianhui/article/details/10243617#comments</a>

#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
static int semid;

//union semun要自己定义一下，否则gcc编译会报错
union semun
{
	int val;
	struct semid_ds *buf;
	unsigned short *arry;
};

bool initSem()
{
	semid = semget((key_t)1235,1,0666|IPC_CREAT);//创建或者取得信号量集
	if(semid < 0)
	{
		perror("semget error\n");
		return false;
	}
	return true;
}

bool initSemTo1()
{
	semun buf;//semctl使用的是semun
	buf.val = 1;//SETVAL命令使用
	
	//semctl使用4个参数，第二个参数是要操作的信号量在信号量集中的编号
	if(semctl(semid,0,SETVAL,buf) < 0)//buf是实例，不是指针。
	{
		perror("initSemTo1 error");
		return false;
	}
	return true;
}

bool pOperation()
{
	sembuf buf;//semop使用的是sembuf
	buf.sem_num = 0;//信号量在信号量集中的编号，第一个信号的编号是0
	buf.sem_op = -1;
	buf.sem_flg = SEM_UNDO;
	int ret = semop(semid,&buf,1);//第3个参数是要操作几个信号量
	if(ret <0)
	{
		perror("pOperation error\n");
		return false;
	}
	return true;
	
}
bool vOperation()
{
	sembuf buf;
	buf.sem_num = 0;//信号量在信号量集中的编号，第一个信号的编号是0
	buf.sem_op = 1;
	buf.sem_flg = SEM_UNDO;
	int ret = semop(semid,&buf,1);
	if(ret <0)
	{
		perror("vOperation error\n");
		return false;
	}
	return true;
}
int deleteSem()
{
	semun buf;
	if(semctl(semid,0,IPC_RMID,buf) < 0)
	{
		perror("deleteSem error\n");
		return -1;
	}
	return 0;
}
int main(int argc,char*argv[])
{
	char c ='x';
	int semid = 0;
	if(!initSem())//创建或者取得信号量
	{
		return -1;
	}
	if(argc == 2)//加了参数则负责初始化信号量
	{
		c=argv[1][0];	

		if(!initSemTo1())//信号量集在使用前一定要初始化信号量
		{
			deleteSem();
			return -1;
		}		
	}
	int i = 0;
	for(;i<10;i++)
	{
		if(!pOperation())
		{
			exit(-1);
		}	
		printf("%c",c);//不会立刻打印，要调用fflush(stdout)才可以，或者用cout << endl;
		fflush( stdout );
		sleep(rand()%3);
		printf("%c",c);
		fflush( stdout );
		if(!vOperation())
		{
			exit(-1);
		}	
		sleep(rand()%3);
	}
	sleep(5);//为了让第二个进程能够先执行完
	if(argc == 2)//第一个进程要负责删除信号量
	{
		return deleteSem();
	}
	return 0;	
}

g++ test1.cpp -g -o test1.exe

 

<77 mdsp209 [mct] :/home/mdcc/mct/cpp/shm>./test1.exe 0 & test1.exe
[1] 28440
00xx00xx00xx00xx00xxxx00xx00xx00xxxx0000<78 mdsp209 [mct] :/home/mdcc/mct/cpp/shm>
[1]    Done                          ./test1.exe 0

两个进程间交替成对打印