System V共享内存 信号量： 生产者消费者实例（完善版）

具体思路参考此blog，但运行此blog代码时发现各种报错，踩了无数坑，因此把最终解决方案分享给大家~
源码结构分布图：

consumer.c

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include "sem_com.h"
#include "shm_com.h"
 

#define IPC_KEY_PATH    "/"
#define SHM_KEY_PROJ    1
#define SEM_KEY_PROJ    2

int main()
{
    
    int shm_id,sem_id;
    shm_buf *shmbuf;
    /*获得共享内存*/    
    if((shm_id = shmget(ftok(IPC_KEY_PATH,SHM_KEY_PROJ),sizeof(shm_buf),IPC_CREAT|0666)) <0 )
    {
        perror("shmget");
        exit(-1);
    }
    printf("consumer\'s shm_id:%d\n",shm_id);
    /*获得信号量集合*/
    if((sem_id = semget(ftok(IPC_KEY_PATH,SEM_KEY_PROJ),1,IPC_CREAT|0666)) < 0 )/*只有一个信号量*/
    {
        perror("semget");
        exit(-1);
    }
    printf("consumer\'s sem_id:%d\n",sem_id);
    /*映射共享内存到当前进程*/  
    if((shmbuf = (shm_buf *)shmat(shm_id,NULL,0)) <0 )
    {
        perror("shmat");
        exit(-1);
    }
    printf("shared memory init  sucessful\n");
    /*读取共享内存中数据*/
    do
    {
        sem_p(sem_id);
        printf("Data from shared memory : process[%d],%s \n",shmbuf->pid,shmbuf->buffer);
        /*读取数据后将共享内存清空*/
        if (strncmp(shmbuf->buffer,"quit",4) == 0 )
        {
            break;
        }
        shmbuf->pid = 0 ;
        memset(shmbuf->buffer,0,SHM_BUFSIZE);
        sem_v(sem_id);

    } while (1); 

    /*解除共享内存映射*/
    if (shmdt(shmbuf) < 0 ) 
    {
        perror("shmdt");
        exit(-1);
    }
    /*删除共享内存*/
    if(shmctl(shm_id,IPC_RMID,NULL) < 0 )
    {
        perror("shmctl_remove");
        exit(-1);
    }
    exit(0);
}

producer.c

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include "sem_com.h"
#include "shm_com.h"
 

#define IPC_KEY_PATH    "/"
#define SHM_KEY_PROJ    1
#define SEM_KEY_PROJ    2

int main()
{
    
    int shm_id,sem_id;
    shm_buf *shmbuf;
    /*创建共享内存*/    
    if((shm_id = shmget((ftok(IPC_KEY_PATH,SHM_KEY_PROJ)),sizeof(shm_buf),IPC_CREAT|0666)) <0 )
    {
        perror("shmget");
        exit(-1);
    }
    printf("producer\'s shm_id:%d\n",shm_id);
    /*创建信号量集合*/
    if((sem_id = semget(ftok(IPC_KEY_PATH,SEM_KEY_PROJ),1,IPC_CREAT|0666)) < 0 )/*只有一个信号量*/
    {
        perror("semget");
        exit(-1);
    }
    printf("producer\'s sem_id:%d\n",sem_id);
    /*信号量初始值为1*/
    sem_init(sem_id,1);
    
    /*映射共享内存到当前进程*/  
    if((shmbuf = (shm_buf *)shmat(shm_id,NULL,0)) <0 )
    {
        perror("shmat");
        exit(-1);
    }
    printf("shared memory init  sucessful\n");
    /*向共享内存中写数据*/
    do
    {
        printf("Please Enter some text to shared memory\n");
        sem_p(sem_id);
        if (fgets(shmbuf->buffer,SHM_BUFSIZE,stdin) ==NULL)
        {
            perror("fgets");
            sem_v(sem_id);
            break;
        }
        shmbuf->pid = getpid();
        sem_v(sem_id);
    } while (strncmp(shmbuf->buffer,"quit",4) != 0 ); /*将quit写进共享内存中退出*/

    /*删除信号量集*/
    sem_delete(sem_id);
    /*解除共享内存映射*/
    if (shmdt(shmbuf) < 0 ) 
    {
        perror("shmdt");
        exit(-1);
    }
    exit(0);
}

sem_com.h（这里踩了坑，原作者的代码中sem_com.h的具体实现函数是放在了sem_com.c的文件中，应该将sem_com.c文件中的具体实现函数复制到sem_com.h才可以）

#ifndef _SEM_COM_H
#define _SEM_COM_H
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>

union semun
{
	int val;
	struct semid_ds *buf;
	unsigned short *array;
	struct seminfo * _buf;
};

/*  信号量初始化函数
 *  参数        sem_id  由KEY值得到的消耗量集ID
 *             init_value 信号量初始化值
 *  返回值      错误返回-1 成功返回0
 */
int sem_init(int sem_id,int init_value)
{
    union semun sem_union;
    sem_union.val = init_value;
    if (semctl(sem_id,0,SETVAL,sem_union) ==-1 )
    {
       
        return -1;
    }
    else 
        return 0;
}
/* 信号量集删除函数
 * 参数        sem_id 由KEY值得到的信号量集ID
 * 返回值       成功 0  失败-1
 */
int sem_delete(int sem_id)
{
  if(semctl(sem_id,0,IPC_RMID,NULL) < 0)
  {
      perror("semctl_delete");
      return(-1);
  }
  else 
    return 0;

}
/************************************
 * p操作函数
 * 参数         sem_id 由KEY值得到的信号量集ID
 * ***********************************/
int sem_p(int sem_id)
{
    struct sembuf buf;
    buf.sem_num = 0;
    buf.sem_op = -1; /*-1代表P操作*/
    buf.sem_flg = SEM_UNDO; /*进程终止时，系统自动释放残余信号量*/
    if(semop(sem_id,&buf,1) < 0 )
    {
        perror("semop1");
        return -1;
    }
    return 0;
}

/*************************
 * V操作函数
 * 参数      sem_id 由KEY值得到的信号量集ID
 ************************/
int sem_v(int sem_id)
{
    struct sembuf buf;
    buf.sem_num = 0;
    buf.sem_op = 1; /*-1代表P操作*/
    buf.sem_flg = SEM_UNDO; /*进程终止时，系统自动释放残余信号量*/
    if(semop(sem_id,&buf,1) < 0 )
    {
        perror("semop2");
        return -1;
    }
    return 0;
}

#endif

shm_com.h(这里踩了坑，原作者这里是.c文件，应该改为.h文件)

#ifndef  _SHM_COM_H
#define _SHM_COM_H
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#define SHM_BUFSIZE 1024
/*定义共享内存数据结构*/
typedef struct 
{
    pid_t pid;
    char buffer[SHM_BUFSIZE];
}shm_buf;



#endif

在全部代码成功编译时，博主犯了一个低级错误o(╥﹏╥)o，博主运行代码的做法是先运行producer，后运行consumer，发现producer可以向共享内存写入数据，但是consumer永远接收不到数据，然后博主尝试打印出二者的信号量id和共享内存id是否一致。

如上图，博主发现二者sem_id的数据并不一致，于是开始阅读源代码，发现每次producer在结束之后都会删除信号量集，于是博主灵感一现，既然producer结束后会删除信号量集，那同时开两个终端不就可以实现producer和consumer共存，然后再想想自己学过的操作系统，的确是生产者和消费者线程同时开启的。因此，大家运行最终代码时，记得开启两个终端。最后附上程序成功运行图：