父子进行通过信号灯同步对共享内存进行读写
1、父进程从键盘输入字符串到共享内存
2、子进程删除字符串中的空格并打印
3、父进程输入'quit'后删除共享内存和信号灯集,程序结束。
#include<stdio.h>
#include <unistd.h>
#include<string.h>
#include<stdlib.h>
#include<fcntl.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<signal.h>
#include<sys/ipc.h>
#include<sys/shm.h>
#include<sys/msg.h>
#include<sys/sem.h>
#define WRITE 0 //信号灯操作中的op
#define READ 1 //信号灯操作中的op
#define N 64 缓冲区大小
union semun
{
int val; /* Value for SETVAL */
struct semid_ds *buf; /* Buffer for IPC_STAT, IPC_SET */
unsigned short *array; /* Array for GETALL, SETALL */
struct seminfo *__buf; /* Buffer for IPC_INFO */
};
void init_sem(int semid,int s[],int n);
void pv(int semid,int num,int op);
int main()
{
pid_t pid; //用于创建进程
key_t key; //创建PCI对象指定ID
int shmid; //共享内存ID
int semid; //信号灯集ID
char *shmaddr; //共享内存的首地址
int s[]={0,1}; //信号灯集里信号灯的编号,分别为0号和1号
if( (key = ftok(".",'m')) == EOF) //创建PCI返回key
{
perror("ftok");
exit(-1);
}
if( (shmid = shmget(key,N,IPC_CREAT|0666)) == EOF ) //创建共享内存
{
perror("shmget");
exit(-1);
}
if((semid = semget(key,2,IPC_CREAT|0666)) == EOF) //创建信号灯集
{
perror("semget");
goto _error1;
}
init_sem(semid,s,2); //初始信号灯集
if((shmaddr = (char*)shmat(shmid,NULL,0)) == (char*)-1) //将共享内存映射的地址返回
{
perror("shmat");
goto _error2; //创建失败销毁已经申请的共享内存
}
if((pid = fork())<0) //通过fork()创建进程,返回值小于0则创建失败
{
perror("fork");
goto _error2; //创建失败销毁信号灯集
}else if(pid == 0) //pid==0则为子进程
{
char *p,*q;
while(1)
{
p = q = shmaddr;
while(*q)
{
if(*q != ' ')
{
*p++ = *q;
}
q++;
}
*p = '\0';
printf("%s",shmaddr);
sleep(10);
pv(semid,WRITE,1);
}
}else{ //父进程
while(1)
{
pv(semid,WRITE,-1); //-1代表P操作
printf("intput : ");
fgets(shmaddr,N,stdin);
//从输入流中获取一个字符串到共享内存当中,每次获取N个,shmaddr为共享内存首地址
if(strcmp(shmaddr,"quit\n") == 0) break; //如果输入quit则结束
pv(semid,READ,1); //1代表V操作
}
kill(pid,SIGUSR1);
}
_error1:
shmctl(shmid,IPC_RMID,NULL); //销毁共享内存
_error2:
semctl(semid,0,IPC_RMID); //销毁信号灯记
}
//传入信号灯集ID,s[]={0,1} s数组代表信号灯的编号,n为信号灯个数
void init_sem(int semid,int s[],int n)
{
union semun myun;
for(int i=0;i<n;i++) //一共有i个信号灯
{
myun.val=s[i]; //设置信号灯的编号
semctl(semid,i,SETVAL,myun);
//设置每个信号灯的属性,传入信号灯集ID,信号灯编号i,操作SETVAL(表示设置信号灯的值),传入共用
//体,改共用体表示了该信号灯集中每个信号灯的各属性 。
}
}
void pv(int semid,int num,int op) //对信号灯集的两种信号进行pv操作,传入op WRITE:0 READ 1
{
//定义信号灯集中每个信号灯的操作,这里每个信号灯的操作封装在一个结构体buf当中
struct sembuf buf;
buf.sem_num = num; //信号灯集中的信号灯编号
buf.sem_op = op; //-1带边p操作 1代表V操作
buf.sem_flg = 0; //阻塞方式 0:等待 IPC_NOWITE :直接返回
semop(semid,&buf,1); //传入操作对象,1表示每次操作的信号灯个数
}