共享内存生产者消费者

ipc.h

#pragma once

#ifndef _ipc_h
#define _ipc_h

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include<unistd.h>
#define BUFSZ 256
//建立或获取ipc 的一组函数的原型说明

int P_operation(int sem_id);
int V_operation(int sem_id);
int get_ipc_id(char *proc_file,key_t key);
char *set_shm(key_t shm_key,int shm_num,int shm_flag);
int set_sem(key_t sem_key,int sem_val,int sem_flag);
/*信号灯控制用的共同体*/
typedef union semuns
{
    int val;
} Sem_uns;
//生产消费者共享缓冲区即其有关的变量
key_t buff_key;
int buff_num;
char *buff_ptr;
//生产者放产品位置的共享指针
key_t pput_key;
int pput_num;
int *pput_ptr;
//消费者取产品位置的共享指针
key_t cget_key;
int cget_num;
int *cget_ptr;
//生产者有关的信号量
key_t prod_key;     //IPC key标识
int prod_sem;       //生产者同步信号量

//消费者有关的信号量
key_t cons_key;     //IPC key标识
int cons_sem;       //消费者同步信号量
int sem_val;
int sem_flg;
int shm_flg;
/*互斥信号量*/
key_t mtx_key;      //IPC key标识
int mtx_sem;        //互斥信号量

#endif

ipc.c

#include "ipc.h"
/*
* get_ipc_id() 从/proc/sysvipc/文件系统中获取IPC 的id 号
* pfile: 对应/proc/sysvipc/目录中的IPC 文件分别为
* msg-消息队列,sem-信号量,shm-共享内存
* key: 对应要获取的IPC 的id 号的键值
*/
int get_ipc_id(char *proc_file,key_t key)
{
    FILE *pf;
    int i,j;
    char line[BUFSZ],colum[BUFSZ];
    if((pf = fopen(proc_file,"r")) == NULL)
    {
        perror("Proc file not open");
        exit(EXIT_FAILURE);
    }
    fgets(line, BUFSZ,pf);
    while(!feof(pf))
    {
        i = j = 0;
        fgets(line, BUFSZ,pf);
        while(line[i] == ‘ ‘)
            i++;
        while(line[i] !=‘ ‘)
            colum[j++] = line[i++];
        colum[j] = ‘/0‘;
        if(atoi(colum) != key)
            continue;
        j=0;
        while(line[i] == ‘ ‘)
            i++;
        while(line[i] !=‘ ‘)
            colum[j++] = line[i++];
        colum[j] = ‘/0‘;
        i = atoi(colum);
        fclose(pf);
        return i;
    } fclose(pf);
    return -1;
}
    {
        //shmget 新建一个长度为shm_num 字节的共享内存,其标号返回到shm_id
        if((shm_id = shmget(shm_key,shm_num,shm_flg)) <0)
        {
            perror("shareMemory set error");
            exit(EXIT_FAILURE);
        }
        //shmat 将由shm_id 标识的共享内存附加给指针shm_buf
        if((shm_buf = (char *)shmat(shm_id,0,0)) < (char *)0)
        {
            perror("get shareMemory error");
            exit(EXIT_FAILURE);
        }
        /*共享内存区初始化*/
        for(i=0; i<shm_num; i++)
            shm_buf[i] = 0; //初始为0
    }
    //shm_key 标识的共享内存区已经建立,将由shm_id 标识的共享内存附加给指针shm_buf
    if((shm_buf = (char *)shmat(shm_id,0,0)) < (char *)0)
    {
        perror("get shareMemory error");
        exit(EXIT_FAILURE);
    }
    return shm_buf;
}

producer.c

#include "ipc.h"
int main(int argc,char *argv[])
{
    int rate;
    //可在在命令行第一参数指定一个进程睡眠秒数，以调解进程执行速度
    if(argv[1] != NULL)
        rate = atoi(argv[1]);
    else
        rate = 3; //不指定为3 秒
    //共享内存使用的变量,其中键值任给,但是注意键值的
    //唯一性,在另外的文件中要用同一共享内存也要采用统一键值
    buff_key = 101;//缓冲区任给的键值
    buff_num = 8;//缓冲区任给的长度
    pput_key = 102;//生产者放产品指针的键值
    pput_num = 1; //指针数
    shm_flg = IPC_CREAT | 0644;//共享内存读写权限
    //获取缓冲区使用的共享内存，buff_ptr 指向缓冲区首地址
    buff_ptr = (char *)set_shm(buff_key,buff_num,shm_flg);
    //获取生产者放产品位置指针pput_ptr
    pput_ptr = (int *)set_shm(pput_key,pput_num,shm_flg);
    //信号量使用的变量
    prod_key = 201;//生产者同步信号灯键值
    mtx_key = 202;//互斥信号灯键值
    cons_key = 301;//消费者同步信号灯键值
    sem_flg = IPC_CREAT | 0644;
    //生产者同步信号灯初值设为缓冲区最大可用量
    sem_val = buff_num;
    //获取生产者同步信号灯，引用标识存prod_sem
    prod_sem = set_sem(prod_key,sem_val,sem_flg);
    //消费者初始无产品可取，同步信号灯初值设为0
    sem_val = 0;
    //获取消费者同步信号灯，引用标识存cons_sem
    cons_sem = set_sem(cons_key,sem_val,sem_flg);
    //生产者互斥信号灯初值为1
    sem_val = 1;
    //获取互斥信号灯，引用标识存mtx_sem
    mtx_sem = set_sem(mtx_key,sem_val,sem_flg);
    //循环执行模拟生产者不断放产品
    while(1)
    {
        //如果缓冲区满则生产者阻塞
        P_operation(prod_sem);
        //如果有进程进入，本生产者阻塞
        P_operation(mtx_sem);
        //用写一字符的形式模拟生产者放产品，报告本进程号和放入的字符及存放的位置
        buff_ptr[*pput_ptr] = ‘A‘+ *pput_ptr;
        //挂起rate秒
        //sleep(rate);
        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);
    }
    return EXIT_SUCCESS;
}

consumer

#include "ipc.h"
int main(int argc,char *argv[])
{
    int rate;
    //可在在命令行第一参数指定一个进程睡眠秒数，以调解进程执行速度
    if(argv[1] != NULL)
        rate = atoi(argv[1]);
    else
        rate = 3; //不指定为3 秒
    //共享内存使用的变量
    buff_key = 101; //缓冲区任给的键值
    buff_num = 8; //缓冲区任给的长度
    cget_key = 103; //消费者取产品指针的键值
    cget_num = 1; //指针数
    shm_flg = IPC_CREAT | 0644; //共享内存读写权限
    //获取缓冲区使用的共享内存，buff_ptr 指向缓冲区首地址
    buff_ptr = (char *)set_shm(buff_key,buff_num,shm_flg);
    //获取消费者取产品指针，cget_ptr 指向索引地址
    cget_ptr = (int *)set_shm(cget_key,cget_num,shm_flg);
    mtx_key = 202; //互斥信号灯键值
    //生产者同步信号灯初值设为缓冲区最大可用量
    sem_val = buff_num;
    //获取生产者同步信号灯，引用标识存prod_sem
    prod_sem = set_sem(prod_key,sem_val,sem_flg);
    //消费者初始无产品可取，同步信号灯初值设为0
    sem_val = 0;
    //获取消费者同步信号灯，引用标识存cons_sem
    cons_sem = set_sem(cons_key,sem_val,sem_flg);
    //消费者互斥信号灯初值为1
    sem_val = 1;
    //获取消费者同步信号灯，引用标识存cons_sem
    cons_sem = set_sem(cons_key,sem_val,sem_flg);
    //消费者互斥信号灯初值为1
    sem_val = 1;
    //获取互斥信号灯，引用标识存mtx_sem
    mtx_sem = set_sem(mtx_key,sem_val,sem_flg);
    //循环执行模拟消费者不断取产品
    while(1){
        //如果无产品消费者阻塞
        P_operation(cons_sem);
        //如果有进程进入,本消费者阻塞
        P_operation(mtx_sem);
        //用读一字符的形式模拟消费者取产品，报告本进程号和获取的字符及读取的位置
       // sleep(rate);
        printf("%d consumer get: %c fromBuffer[%d]\n",getpid(),buff_ptr[*cget_ptr],*cget_ptr);
        //读取位置循环下移
        *cget_ptr = (*cget_ptr+1) % buff_num;
        //唤醒阻塞的进程
        V_operation(mtx_sem);
        //唤醒阻塞的生产者
        V_operation(prod_sem);
    }
    return EXIT_SUCCESS;
}