用信号量实现进程互斥.c

/*用信号量实现进程“互斥”

 * 进程共享一套代码（fork之前的代码），fork之后相互独立

 * sudo kill -9 $(pidof ./a.out) //杀死所有a.out进程

 */

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

#include <sys/types.h>

#include <sys/ipc.h>

#include <sys/sem.h>

#define DEBUG 1//0 不采用信号量互斥 1 采用信号量互斥

#if defined(__GNU_LIBRARY__) && !defined(_SEM_SEMUN_UNDEFINED)   

#else

union semun {

int val;                             

struct semid_ds *buf;    

unsigned short int *array; 

struct seminfo *__buf;  

};

#endif   

//初始化信号量

void sem_init(int semid, int val) {

int ret;

union semun semun;

semun.val = val;

ret = semctl(semid, 0, SETVAL,  semun);

if (ret < 0) {

perror("semctl error.");

exit (1);

}

}

//信号量P操作，即信号量大于0则减1等于0则挂起

void sem_p(int semid) {

int ret;

struct sembuf sembuf;

sembuf.sem_num = 0;

sembuf.sem_op = -1;

sembuf.sem_flg = 0;

ret = semop(semid, &sembuf, 1);

if (ret < 0) {

perror("semop error");

exit(1);

}

}

//信号量V操作，如有该信号量挂起的进程则恢复，没有则信号量加1

void sem_v(int semid) {

int ret;

struct sembuf sembuf;

sembuf.sem_num = 0;

sembuf.sem_op = 1;

sembuf.sem_flg = 0;

ret = semop(semid, &sembuf, 1);

if (ret < 0) {

perror("semop error");

exit(1);

}

}

int main() {

int semid;

int i;

//获取信号量，几个进程将共用

semid = semget(1234, 1, IPC_CREAT);

if (semid < 0) {

perror("semget error.");

exit(1);

}

//初始化信号量

sem_init(semid,1);

printf ("semid = : %d\n ",semid);

//创建进程

fork();

fork();

fork();

//P操作，该代码段内容被锁起

#if DEBUG

sem_p(semid);

# endif

// #endif

/******** 临界区 begin ********/

printf("\033[31m进程(%d)开始使用数据...\033[0m\n", getpid());

for (i=0; i<8; i++) {

printf(" 进程(%d)正在使用...\n", getpid());

sleep(1);

}

printf(" \033[32m进程(%d)释放数据...\033[0m\n", getpid());

/******** 临界区 end ********/

//V操作，执行之后其他进程可使用该代码段

#if DEBUG

sem_v(semid);

#endif

getchar();//使进程不结束,一个进程结束，其他进程依然可以执行，不过会跳出会话窗口,用getchar每一个进程需输入一个回车，故结束时应键入八次回车

return 0;

}