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);
}