2024.2.26

信号灯集和共享内存实现收发

自定义头文件

#ifndef SEM_H_
#define SEM_H_
//创建信号灯集,
int creat_t(int number);
//申请释放资源
int P(int semid,int semno);
//申请释放资源
int V(int semid,int semno);
//删除信号灯集
int del(int semid);
#endif

信号灯集函数集合

#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 creat_light(int semid,int i)
{
	union semun buf;
	printf("请输入要给编号为%d的灯设置的值：", i);
	scanf("%d", &buf.val);
	getchar();
	if(semctl(semid,i,SETVAL,buf) == -1)
	{
		perror("semctl error");
		return -1;
	}
 
	return 0;
}
int creat_t(int number)
{
	//创建信号灯集的key值
	key_t key=ftok("./",'k');
	if(key==-1)
	{
		perror("ftok error");
		return -1;
	}
	//创建信号灯集
	int semid;
	if((semid=semget(key,number,IPC_CREAT|IPC_EXCL|0664))==-1)
	{
		//对错误码进行判断，如果错误码为EEXIST，说明信号灯集已经存在
        if(errno == EEXIST)
        {
            //直接打开信号灯集即可
            semid=semget(key,number,IPC_CREAT|0664);
            return semid;
        }
		perror("semget error");
		return -1;
	}
	//定义信号灯集中灯的编号和每个灯中机构体内容
	for(int i=0;i<number;i++)
	{
		creat_light(semid,i);
	}
			
	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("P error");
        return -1;
    }
 
    return 0;
}
int del(int semid)
{
	if(semctl(semid,0,IPC_RMID)==-1)
	{
		perror("delete error");
		return -1;
	}
	return 0;
}

发送代码

#include <myhead.h>
#include "sem.h"
#define size 4096
int main(int argc, const char *argv[])
{
	//创建信号灯集
	int semid=creat_t(2);
	//创建共享内存的key值
	key_t key=ftok("./",'k');
		if(key==-1)
		{
			perror("ftok error");
			return -1;
		}
	//创建共享内存空间 
	int shmpid;
	if((shmpid=shmget(key,size,IPC_CREAT|0664))==-1)
	{
		perror("shmget error");
		return -1;
	}
	//映射内存到用户空间
	char *addr=shmat(shmpid,NULL,0);
	if(addr==(void*)-1)
		{
			perror("shmat error");
			return -1;
		}
	while(1)
	{
		//申请0号灯资源
		P(semid,0);
		printf("请输入:");
		fgets(addr,size,stdin);
		addr[strlen(addr)-1]='\0';
		printf("发送成功\n");
		//释放一号灯资源
		V(semid,1);
		if(strcmp(addr,"end")==0)
		{
			break;
		}
 
	}
	//分离共享内存
	if(shmdt(addr)==-1)
	{
		perror("shmdt error");
	}
	return 0;
}

接收代码

#include <myhead.h>
#include "sem.h"
#define size 4096
int main(int argc, const char *argv[])
{
	//创建信号灯集
	int semid=creat_t(2);
	//创建共享内存的key值
	key_t key=ftok("./",'k');
		if(key==-1)
		{
			perror("ftok error");
			return -1;
		}
	//创建共享内存空间 
	int shmpid;
	if((shmpid=shmget(key,size,IPC_CREAT|0664))==-1)
	{
		perror("shmget error");
		return -1;
	}
	//映射内存到用户空间
	char *addr=shmat(shmpid,NULL,0);
	if(addr==(void*)-1)
		{
			perror("shmat error");
			return -1;
		}
	while(1)
	{
		P(semid,1);//申请1号灯资源
		sleep(1);
		printf("收到的数据为:%s\n",addr);
		V(semid,0);//释放0号灯资源
		if(strcmp(addr,"end")==0)
		{
			break;
		}
 
	}
	//分离共享内存
	if(shmdt(addr)==-1)
	{
		perror("shmdt error");
	}
	//删出共享内存
	if(shmctl(shmpid,IPC_RMID,NULL)==-1)
	{
		perror("shmctl error");
		return -1;
	}
	//删除信号灯集
	del(semid);
	return 0;
}