11月15日作业

共享进程实现双向通信

sem.h

#ifndef _SEM_H_
#define _SEM_H_
//声明一个创建信号灯集并初始化函数
int init_sem(int semnum);

//声明一个P操作函数
int P(int semid,int semno);

//声明一个V操作函数
int V(int semid,int semno);

//声明一个删除信号灯集的函数
int del_sem(int semid);
#endif

sem.c

#include <myhd.h>
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
                                (Linux-specific) */
};
//自定义初始化信号灯函数
int set_semno_value(int semid,int semno,int semval)
{
	//定义共用体
	union semun us;
	us.val=semval;
	//调用semctl对灯进行初始化
	if(semctl(semid,semno,SETVAL,us)==-1)
	{
		perror("semctl error");
		return -1;
	}
	return 0;
}
int init_sem(int semnum)
{
	//1、创建key值
	key_t key;
	if((key=ftok("/",'t'))==-1)
	{
		perror("ftok error");
		return -1;
	}
	//2、创建信号灯集
	int semid;
	if((semid=semget(key,semnum,IPC_CREAT|IPC_EXCL|0664))==-1)
	{
		if(errno==EEXIST)
		{
			//表明信号灯集已经存在，直接打开即可
			semid=semget(key,semnum,IPC_CREAT|0664);
			return semid; //将信号灯集打开后，直接将信号灯集id返回
		}
		else
		{
			perror("semget error");
		}
		return -1;
	}
	//3、给信号灯集进行初始化
	for(int i=0;i<semnum;i++)
	{
		if(i==0||i==2)
		{
			//设置信号灯的初始值
			set_semno_value(semid,i,1);
			//参数1：信号灯集
			//参数2：要操作的信号灯编号
			//参数3：要设置的值
		}
		else
		{
			set_semno_value(semid,i,0);
		}
	}
	return semid;
}
int P(int semid,int semno)
{
	//定义一个操作结构体变量
	struct sembuf buf;
	buf.sem_num=semno;  //要操作的信号灯编号
	buf.sem_op=-1;  //表示要进行申请资源
	buf.sem_flg=0;  //表示阻塞申请资源
	if(semop(semid,&buf,1)==-1)
	{
		perror("P error");
		return -1;
	}
	return 0;
}
int V(int semid,int semno)
{
	//定义一个操作结构体变量
	struct sembuf buf;
	buf.sem_num=semno;  //要操作的信号灯编号
	buf.sem_op=1;  //表示要进行资源释放
	buf.sem_flg=0;  //表示阻塞申请资源
	if(semop(semid,&buf,1)==-1)
	{
		perror("V error");
		return -1;
	}
	return 0;
}
int del_sem(int semid)
{
	//删除信号灯集
	if(semctl(semid,0,IPC_RMID,0)==-1)
	{
		perror("delete error");
		return -1;
	}
	return 0;
}

snd.c

#include <myhd.h>
#include "sem.h"
int main(int argc, const char *argv[])
{
	//11、打开信号灯集
	int semid=init_sem(4);
	//1、创建key值
	key_t key;
	if((key=ftok("/",'k'))==-1)
	{
		perror("ftok error");
		return -1;
	}
	//2、创建共享内存
	int shmid;
	if((shmid=shmget(key,4096,IPC_CREAT|0664))==-1)
	{
		perror("shmget error");
		return -1;
	}
	//3、映射内容
	char *addr=NULL;  //指向共享内存段
	if((addr=shmat(shmid,NULL,0))==(void *)-1)
	{
		perror("shmat error");
		return -1;
	}
	pid_t pid;
	pid=fork();
	if(pid<0)
	{
		perror("fork error");
		return -1;
	}
	else if(pid>0)
	{
		while(1)
		{
			//22、等待0号灯的资源
			P(semid,0);
			//4、向共享内存中写入数据
			fgets(addr,4096,stdin);  //从终端获取数据放入共享内存中
			addr[strlen(addr)-1]='\0';
			//3、释放1号灯的资源
			V(semid,1);
			if(strcmp(addr,"quit")==0)
			{
				break;
			}
		}
		wait(NULL);
	}
	else
	{
		while(1)
		{
			P(semid,3);
			printf("usre2：%s\n",addr);
			V(semid,2);
			if(strcmp(addr,"quit")==0)
			{
				break;
			}
		}
		exit(EXIT_SUCCESS);
	}
	//5、取消映射
	if(shmdt(addr)==-1)
	{
		perror("shmdt error");
		return -1;
	}
	//6、删除共享内存
	return 0;
}

rev.c

#include <myhd.h>
#include "sem.h"
int main(int argc, const char *argv[])
{
	//11、打开信号灯集
	int semid=init_sem(4);
	//1、创建key值
	key_t key;
	if((key=ftok("/",'k'))==-1)
	{
		perror("ftok error");
		return -1;
	}
	//2、创建共享内存
	int shmid;
	if((shmid=shmget(key,4096,IPC_CREAT|0664))==-1)
	{
		perror("shmget error");
		return -1;
	}
	//3、映射内容
	char *addr=NULL;  //指向共享内存段
	if((addr=shmat(shmid,NULL,0))==(void *)-1)
	{
		perror("shmat error");
		return -1;
	}
	pid_t pid;
	pid=fork();
	if(pid<0)
	{
		perror("fork error");
		return -1;
	}
	else if(pid>0)
	{
		while(1)
		{
			//22、等待1号灯的资源
			P(semid,1);
			//4、从共享内存中读取数据
			printf("user2：%s\n",addr);
			//33、释放0号灯的资源
			V(semid,0);
			if(strcmp(addr,"quit")==0)
			{
				break;
			}
		}
		wait(NULL);
	}
	else
	{
		while(1)
		{
			P(semid,2);
			fgets(addr,4096,stdin);
			addr[strlen(addr)-1]='\0';
			V(semid,3);
			if(strcmp(addr,"quit")==0)
			{
				break;
			}
		}
		exit(EXIT_SUCCESS);
	}
	//5、取消映射
	if(shmdt(addr)==-1)
	{
		perror("shmdt error");
		return -1;
	}
	//6、删除共享内存
	if((shmctl(shmid,IPC_RMID,NULL))==-1)
	{
		perror("shmctl error");
		return -1;
	}
	//44、删除信号灯集
	del_sem(semid);
	return 0;
}