#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/sem.h>
#include <sys/shm.h>
#include <sys/types.h>
int set_sem(key_t sem_key,int sem_val,int sem_flg) //set_sem函数建立一个具初值为semval信号量
{
int sem_id;
int sem_arg;
if((sem_id = semget(sem_key,1,sem_flg)) < 0)
{ //创建一个新的信号量或取得一个原有的信号量
perror("semaphore create error");
exit(1);
}
if(semctl(sem_id,0,SETVAL,sem_val) <0)
{ //初始化信号量
perror("semaphore set error");
exit(1);
}
return sem_id; //返回ipc的id
}
char * set_shm(key_t shm_key,int shm_num,int shm_flg) //set_shm函数建立一个具有n个字节的共享内存区
{
int i,shm_id;
char * shm_buf;
//shmget 新建一个长度为shm_num字节的共享内存,其标号返回到shm_id,权限为shm_flg
if((shm_id = shmget(shm_key,shm_num,shm_flg)) <0)
{
perror("shareMemory set error");
exit(1);
}
//shmat 将由shm_id标识的共享内存附加给指针shm_buf,该函数用来启用对该共享内存的访问
if((shm_buf = (char *)shmat(shm_id,0,0)) < (char *)0)
{
perror("get shareMemory error");
exit(1);
}
for(i=0; i<shm_num; i++) //共享内存区初始化
shm_buf[i] = 0; //初始为0
return shm_buf;
}
//请求P操作
int P_operation(int sem_id) //信号灯数组标识符
{
struct sembuf buf={0,-1,SEM_UNDO}; //信号灯结构,0信号量编号,-1 p操作
if((semop(sem_id,&buf,1)) <0) //调用semop,成功就使之等待,不成功显示错误
{
perror("请求失败");
exit(1);
}
return 0;
}
//释放 V操作
int V_operation(int sem_id)
{
struct sembuf buf={0,1,SEM_UNDO}; //信号灯结构
if((semop(sem_id,&buf,1)) < 0) //调用semop,成功释放占用,使之运行
{
perror("释放失败");
exit(1);
}
return 0;
}
int main()
{
int rate = 3;
pid_t fpid = fork();
fpid = fork(); //两个fork,模拟多生产者消费者
//共享内存使用的变量,其中键值任给
key_t buff_key = 104; //缓冲区任给的键值
int buff_num = 8; //缓冲区任给的长度
key_t pput_key = 105; //生产者放产品指针的键值
int pput_num = 1; //指针数
key_t cget_key = 106; //消费者取产品指针的键值
int cget_num = 1; //指针数
int shm_flg = IPC_CREAT | 0644; //共享内存读写权限,用户具有读写权限,组用户和其它用户具有只读权限
char* buff_ptr = (char *)set_shm(buff_key,buff_num,shm_flg); //获取缓冲区使用的共享内存,buff_ptr 指向缓冲区首地址 char*
int* pput_ptr = (int *)set_shm(pput_key,pput_num,shm_flg); //获取生产者放产品位置指针
int* cget_ptr = (int *)set_shm(cget_key,cget_num,shm_flg); //获取消费者取产品指针
key_t prod_key = 205; //生产者同步信号灯键值
key_t mtx_key = 206; //互斥信号灯键值
key_t cons_key = 207; //消费者同步信号灯键值
int sem_flg = IPC_CREAT | 0644; //权限
int prod_sem = set_sem(prod_key,buff_num,sem_flg); //得到信号量id
int cons_sem = set_sem(cons_key,0,sem_flg); //得到信号量id
int mtx_sem = set_sem(mtx_key,1,sem_flg); //得到信号量id
if (fpid != 0) //父进程生产
{
for (int i = 0; i < 20; i++) //循环执行模拟生产者不断放产品
{
int s = rand()%2+1; //随机睡眠一段时间
P_operation(prod_sem); //如果缓冲区满则生产者阻塞
P_operation(mtx_sem); //如果有进程进入,本生产者阻塞
sleep(s); //挂起s秒
buff_ptr[*pput_ptr] = 65 + (*pput_ptr); //用写一字符的形式模拟生产者放产品,报告本进程号和放入的字符及存放的位置,原本代码这个操作执行完后就是在白做,放了又放一遍
printf("%d producer put: %c to Buffer[%d]|\n",getpid(),buff_ptr[*pput_ptr],*pput_ptr);
*pput_ptr = (*pput_ptr+1) % buff_num; //存放位置循环下移
V_operation(mtx_sem); //唤醒阻塞的进程
V_operation(cons_sem); //唤醒阻塞的消费者
}
}
else //子进程消费
{
for (int i = 0; i < 20; i++) //循环执行模拟消费者不断取产品
{
int s = rand()%2+1; //随机睡眠一段时间
P_operation(cons_sem); //如果无产品消费者阻塞
P_operation(mtx_sem); //如果有进程进入,本消费者阻塞s
sleep(s); //挂起s秒,用读一字符的形式模拟消费者取产品,报告本进程号和获取的字符及读取的位置
printf(" |");
printf("%d consumer get: %c fromBuffer[%d]\n", getpid(),buff_ptr[*cget_ptr], *cget_ptr);
buff_ptr[*cget_ptr] = 0; //该物品被取走,原本代码没有取走操作!!
*cget_ptr = (*cget_ptr+1) % buff_num; //读取位置循环下移
V_operation(mtx_sem); //唤醒阻塞的进程
V_operation(prod_sem); //唤醒阻塞的生产者
}
}
semctl(cons_sem, 0, IPC_RMID);
semctl(prod_sem, 0, IPC_RMID);
semctl(mtx_sem, 0, IPC_RMID);
return 0;
}