进程间通信之分别用共享内存和信号量实现卖票

【进程间通信之分别用共享内存和信号量实现卖票】利用共享内存实现的卖票系统:
利用flag来保证同一时间只有一个程序使用内存,使用结束还原。

#include #include #include #include #include #include typedef struct _shm { int flag; int ticket; }SHM; void sellTicket(SHM* pshm) {while (1) { int time = rand() % 10 + 1; usleep(time*100000); if (pshm->flag == 1) { pshm->flag = 0; if (pshm->ticket == 0)// 票卖完 { pshm->flag = 1; break; }printf ("卖掉一张票,座位号是 : %d\n", pshm->ticket); pshm->ticket--; pshm->flag = 1; } } }int main(int argc, char **argv) { srand ((unsigned int)time(NULL)); // 1、创建或者获取一个共享内存 int shmid = shmget((key_t)1234, sizeof(SHM), 0666 | IPC_CREAT); if (shmid == -1) { perror ("shmget"); return -1; }// 2、将共享内存映射到当前的进程空间 SHM* pshm = (SHM*)shmat(shmid, NULL, 0); if(pshm == (SHM*)-1) { perror ("shmat"); return -1; }// 如果命令行参数等于2 负责对共享内存进行初始化 if (argc == 2) { pshm->flag= 1; pshm->ticket = 100; }// 开始卖票 sellTicket(pshm); // 如果命令行参数等于2 负责对共享内存进行删除 if (argc == 2) { shmctl(shmid, IPC_RMID, NULL); }return 0; }

利用信号量实现:
头文件:
#ifndef __SEMAPHORE_H__ #define __SEMAPHORE_H__#include #include #include union semun { intval; /* Value for SETVAL */ struct semid_ds *buf; /* Buffer for IPC_STAT, IPC_SET */ unsigned short*array; /* Array for GETALL, SETALL */ struct seminfo*__buf; /* Buffer for IPC_INFO (Linux specific) */ }; // 信号量的初始化函数 int sem_init(int sem_id) { union semun sem; sem.val = 1; int ret = semctl(sem_id, 0, SETVAL, sem); return ret; }// 信号量的 P 操作 int sem_p(int sem_id) { struct sembuf sem; sem.sem_num = 0; sem.sem_op= -1; sem.sem_flg = SEM_UNDO; int ret = semop(sem_id, &sem,1); return ret; }// 信号量的 v 操作 int sem_v(int sem_id) { struct sembuf sem; sem.sem_num = 0; sem.sem_op= 1; sem.sem_flg = SEM_UNDO; int ret = semop(sem_id, &sem,1); return ret; }// 销毁信号量 int sem_del(int sem_id) { int ret = semctl(sem_id, 0, IPC_RMID); return ret; }#endif // __SEMAPHORE_H__

主函数:
#include #include #include #include #include #include #include "semaphore.h"typedef struct _shm { int flag; int ticket; }SHM; void sellTicket(SHM* pshm, int sem_id) {while (1) { int time = rand() % 10 + 1; usleep(time*100000); // 信号量的P操作 sem_p (sem_id); if (pshm->ticket == 0)// 票卖完 { sem_v (sem_id); break; }printf ("卖掉一张票,座位号是 : %d\n", pshm->ticket); pshm->ticket--; sem_v (sem_id); } }int main(int argc, char **argv) { srand ((unsigned int)time(NULL)); // 1、创建或者获取一个共享内存 int shmid = shmget((key_t)1234, sizeof(SHM), 0666 | IPC_CREAT); if (shmid == -1) { perror ("shmget"); return -1; }// 创建一个信号量 int sem_id = semget((key_t)5678, 1, 0666 | IPC_CREAT); if (sem_id== -1) { perror ("semget"); return -1; }// 2、将共享内存映射到当前的进程空间 SHM* pshm = (SHM*)shmat(shmid, NULL, 0); if(pshm == (SHM*)-1) { perror ("shmat"); return -1; }// 如果命令行参数等于2 负责对共享内存和信号量进行初始化 if (argc == 2) { pshm->ticket = 100; sem_init(sem_id); }// 开始卖票 sellTicket(pshm, sem_id); // 如果命令行参数等于2 负责对共享内存和信号量进行删除 if (argc == 2) { shmctl(shmid, IPC_RMID, NULL); sem_del(sem_id); }return 0; }

    推荐阅读