用共享内存实现生产者和消费者的机制

shm_com.h

#include<unistd.h>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<sys/types.h>
#include<sys/ipc.h>
#include<sys/shm.h>

#define SHM_BUFF_SZ  2048
struct shm_buff
{
    int pid;
    char buffer[SHM_BUFF_SZ];
};

sem_com.h

#include<sys/types.h>
#include<sys/ipc.h>
#include<sys/sem.h>
#include<stdio.h>
#include<stdlib.h>
union semun
{
    int val;
    struct semid_ds *buf;
    unsigned short *array;
};

int init_sem(int sem_id,int init_value);
int del_sem(int sem_id);
int sem_p(int sem_id);
int sem_v(int sem_id);

producer.c

#include"shm_com.h"
#include"sem_com.h"

#include<signal.h>

int init_sem(int sem_id,int init_value)
{
    union semun sem_union;
    sem_union.val=init_value;  
        if(semctl(sem_id,0,SETVAL,sem_union)==-1)
    {
        perror("Initialize semaphore");
        return -1;
    }
    return 0;
}

int del_sem(int sem_id)
{
    union semun sem_union;
    if(semctl(sem_id,0,IPC_RMID,sem_union)==-1)
    {
        perror("Delete semaphore");
        return -1;
    }
}

int sem_p(int sem_id)
{
    struct sembuf sem_b;
    sem_b.sem_op=-1;  
    sem_b.sem_flg=SEM_UNDO; 
    if(semop(sem_id,&sem_b,1)==-1)
    {
        perror("P operationn");
        return -1;
    }
    return 0;
}

int sem_v(int sem_id)
{
    struct sembuf sem_b;
    sem_b.sem_num=0;    sem_b.sem_op=1;  
    sem_b.sem_flg=SEM_UNDO; 
    if(semop(sem_id,&sem_b,1)==-1)
    {
        perror("V operationn");
        return -1;
    }
    return 0;
}


int ignore_signal(void)
{ 
    signal(SIGINT,SIG_IGN);
    signal(SIGSTOP,SIG_IGN);
    signal(SIGQUIT,SIG_IGN);
    return 0;
}

int main()
{
    char *shared_memory;
    struct shm_buff *shm_buff_inst;
    char buffer[BUFSIZ];
    int shmid,semid;
    ignore_signal();    semid=semget(ftok(".",'a'),1,0666|IPC_CREAT);   init_sem(semid,1); 

    shmid=shmget(ftok(".",'c'),sizeof(struct shm_buff),0666|IPC_CREAT);
    if(shmid==-1)
    {
        perror("shmget failed");
        del_sem(semid);
        exit(1);
    }

        shared_memory=shmat(shmid,0,0);
    if(shared_memory==(void*)-1)
    {
        perror("shmat");
        del_sem(semid);
        exit(1);
    }

    printf("Memory attached at %p\n",shared_memory);
    shm_buff_inst=(struct shm_buff *)shared_memory;
    do
    {
        sem_p(semid);
        printf("Emter some text to the shared memory(enter 'quit' to exit):");
        if(fgets(shm_buff_inst->buffer,SHM_BUFF_SZ,stdin)==NULL)
        {
            perror("fgets");
            sem_v(semid);
            break;
        }
        shm_buff_inst->pid=getpid();
        sem_v(semid);
    }while(strncmp(shm_buff_inst->buffer,"quit",4)!=0);

    del_sem(semid);

    if(shmdt(shared_memory)==-1)
    {
        perror("shmdt");
        exit(1);
    }

    exit(0);
}

customer.c

#include"shm_com.h"
#include"sem_com.h"

int init_sem(int sem_id,int init_value)
{
    union semun sem_union;
    sem_union.val=init_value;  
    if(semctl(sem_id,0,SETVAL,sem_union)==-1)
    {
        perror("Initialize semaphore");
        return -1;
    }
    return 0;
}

int del_sem(int sem_id)
{
    union semun sem_union;
    if(semctl(sem_id,0,IPC_RMID,sem_union)==-1)
    {
        perror("Delete semaphore");
        return -1;
    }
}

int sem_p(int sem_id)
{
    struct sembuf sem_b;
    sem_b.sem_num=0;  
    sem_b.sem_op=-1;  
    sem_b.sem_flg=SEM_UNDO; 
    if(semop(sem_id,&sem_b,1)==-1)
    {
        perror("P operationn");
        return -1;
    }
    return 0;
}

int sem_v(int sem_id)
{
    struct sembuf sem_b;
    sem_b.sem_num=0; 
    sem_b.sem_op=1; 
    sem_b.sem_flg=SEM_UNDO; 
    if(semop(sem_id,&sem_b,1)==-1)
    {
        perror("V operationn");
        return -1;
    }
    return 0;
}


main()
{
    char *shared_memory=NULL;
    struct shm_buff *shm_buff_inst;
    int shmid,semid;

    semid=semget(ftok(".",'a'),1,0666);
    if(semid==-1)
    {
        perror("Producer isn't exist");
        exit(1);
    }
    shmid=shmget(ftok(".",'c'),sizeof(struct shm_buff),0666|IPC_CREAT);
    if(shmid==-1)
    {
        perror("shmget");
        exit(1);
    }

    shared_memory=shmat(shmid,0,0);
    if(shared_memory==(void *)-1)
    {
        perror("shmat");
        exit(1);
    }
    printf("Memory attached at %p\n",shared_memory);

    shm_buff_inst=(struct shm_buff *)shared_memory;
    do
    {
        sem_p(semid);
        printf("Shared memory was written by process %d : %s",shm_buff_inst->pid,shm_buff_inst->buffer);
        if(strncmp(shm_buff_inst->buffer,"quit",4)==0)
        {
            perror("strncmp");
            exit(1);
        }
        shm_buff_inst->pid=0;
        memset(shm_buff_inst->buffer,0,SHM_BUFF_SZ);
        sem_v(semid);
    }while(1);

    if(shmdt(shared_memory)==-1)
    {
        perror("shmdt");
        exit(1);
    }

    if(shmctl(shmid,IPC_RMID,NULL)==-1)
    {
        perror("shmctl");
        exit(1);
    }
    exit(0);
}