2024.2.26

sem.h

#ifndef _SEM_H_
#define _SEM_H_

//创建或打开共享内存：参数为要申请的信号灯集中包含的灯的个数，返回值信号灯集的id
int open_sem(int semcount);


//进行申请资源操作:参数为要申请的信号灯id，以及灯编号
int p(int semid,int semno);


//进行释放资源操作:参数为要申请的信号灯id，以及灯编号
int v(int semid,int semno);


//删除信号灯集的操作:参数为要删除的信号灯集id
int del_sem(int semid);

#endif

sem.c

#include<myhead.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 init_semno(int semid,int semno)
{
	union semun buf;
	printf("请输入要给编号为%d的灯设置的值",semno);
	scanf("%d",&buf.val);


	//调用控制函数
	if(semctl(semid,semno,SETVAL,buf)==-1)
	{
		perror("semctl error");
		return -1;
	}

	return 0;

}



//创建或打开共享内存：参数为要申请的信号灯集中包含的灯的个数，返回值
int open_sem(int semcount)
{
	//创建key值
	key_t key=-1;
	if((key=ftok("/",'s'))==-1)
	{
		perror("ftok error");
		return -1;
	}

	//通过key值创建一个信号灯集
	int semid=-1;
	if((semid=semget(key,semcount,IPC_CREAT|IPC_EXCL|0664))==-1)
	{
		//对错误码进行判断，如果错误码为EEXIST，说明信号灯集已经存在
		if(errno==EEXIST)
		{
			//直接打开信号灯集即可
			semid=semget(key,semcount,IPC_CREAT|0664);
			return semid;
		}


		perror("semget error");
		return -1;
	}


	//给信号灯集中的每个灯进行初始化操作
	for(int i=0;i<semcount;i++)
	{
		init_semno(semid,i);
	}

	return semid;
}

   
//进行申请资源操作:参数为要申请的信号灯id，以及灯编号
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;
}
  
 
//进行释放资源操作:参数为要申请的信号灯id，以及灯编号
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;
} 
//删除信号灯集的操作:参数为要删除的信号灯集id
int del_sem(int semid)
{
	//删除信号灯集
	if(semctl(semid,0,IPC_RMID)==-1)
	{
		perror("deletc error");
		return -1;
	}

	return 0;

}

shmsnd.c

#include<myhead.h>
#include"sem.h"
#define PAGE_SIZE 4096

int main(int argc, const char *argv[])
{
	int semid=open_sem(2);

	key_t key=-1;
	if((key=ftok("/",'k'))==-1){
		perror("ftok error");
		return -1;
	}

	int shmid=-1;
	if((shmid=shmget(key,PAGE_SIZE,IPC_CREAT|0664))==-1){
			perror("shmget error");
			return -1;			
			}

	char *addr=shmat(shmid,NULL,0);

	if(addr==(void *)-1){
		perror("shmat error");
		return -1;
	}

	while(1){
		
		p(semid,0);

		printf("请输入:");

		fgets(addr,PAGE_SIZE,stdin);
		addr[strlen(addr)-1]=0;

		printf("发送成功\n");
		
		v(semid,1);

		if(strcmp(addr,"quit")==0){
			break;
		}
		if(strcmp(addr,"q")==0)
		{
			break;
		}

				
	}

	
	if(shmdt(addr)==-1){
		perror("shmdt error");
		return -1;
	}

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


	return 0;
}

shmrcv.c

#include<myhead.h>
#include"sem.h"

#define PAGE_SIZE 4096

int main(int argc, const char *argv[])
{

	int semid=open_sem(2);

	key_t key=-1;
	if((key=ftok("/",'k'))==-1){
		perror("ftok error");
		return -1;
	}
printf("%d\n",key);
	int shmid=-1;
	if((shmid=shmget(key,PAGE_SIZE,IPC_CREAT|0664))==-1){
			perror("shmget error");
			return -1;			
			}

printf("%d\n",shmid);

	char *addr=shmat(shmid,NULL,0);

	if(addr==(void *)-1){
		perror("shmat error");
		return -1;
	}

	while(1){
		
		p(semid,1);


		printf("要读取的数据为:%s\n",addr);
		
		if(strcmp(addr,"quit")==0){
			break;
		}

		v(semid,0);
	}

	if(shmdt(addr)==-1){
		perror("shmdt error");
		return -1;
	}

del_sem(semid);


	return 0;
}