消息队列实现进程间通信

发送方代码:

#include 

//消息类型结构体
typedef struct{

	long msgtype;//消息类型
	char data[1024];//消息正文
}Msg_ds;

#define SIZE sizeof(Msg_ds)-sizeof(long)//正文大小

int main(int argc, const char *argv[])
{	
	//创建key值
	key_t key;
	if((key=ftok("/",'k'))==-1){
		perror("ftok error");
		return -1;
	}
	//创建消息队列
	int msgid;
	if((msgid=msgget(key,IPC_CREAT|0664))==-1){
		perror("msgget error");
		return -1;
	}


	//创建进程
	pid_t pid=fork();
	if(pid<0){
		perror("fork error");
		return -1;
	}
	//子进程
	else if(pid==0){


		//向消息队列中发送数据
		Msg_ds msg1={.msgtype=100};

		while(1){
			printf("请输入消息内容:");
			//从终端输入字符串
			fgets(msg1.data,sizeof(msg1.data),stdin);
			//将字符串'\n'换成'\0';
			msg1.data[strlen(msg1.data)-1]='\0';
			//将消息发送到消息队列中
			if(msgsnd(msgid,&msg1,SIZE,0)==-1){
				perror("msgsnd error");
				return -1;
			}
			if(strcmp(msg1.data,"quit")==0){
				break;
			}
		}
		//回收子进程
		exit(EXIT_SUCCESS);


	}
	else{
		//从消息队列中取数据
		Msg_ds msg;

		while(1)
		{


			//从消息队列中取数据
			//第一个0表示取消息的类型,每次都是取第一个
			//第二个0表示阻塞方式从消息队列中取数据
			if(msgrcv(msgid, &msg, SIZE, 200 ,0) == -1)
			{
				perror("msgsnd error");
				return -1;
			}

			if(strcmp(msg.data, "quit") == 0)
			{
				break;
			}

			printf("rcv: %s\n", msg.data);
		}
		wait(NULL);	
	}
	msgctl(msgid,IPC_RMID,NULL);
	return 0;
}

接收方代码:

#include 

//消息结构体
typedef struct
{
	long msgtype;        //消息类型
	char data[1024];        //消息正文
}Msg_ds;

#define SIZE sizeof(Msg_ds)-sizeof(long)      //正文大小
int main(int argc, const char *argv[])

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

	//2、创建消息队列
	int msgid; 
	if((msgid = msgget(key, IPC_CREAT|0664)) == -1)
	{
		perror("msgget error");
		return -1;
	}

	pid_t pid=fork();
	if(pid<0){
		perror("fork error");
		return -1;
	}
	//子进程
	else if(pid==0){

		//向消息队列中发送数据
		Msg_ds msg = {.msgtype = 200};

		while(1)
		{
			printf("请输入消息内容:");
			fgets(msg.data, sizeof(msg.data), stdin);    //从终端输入字符串
			//将字符串的'\n'换成'\0'
			msg.data[strlen(msg.data)-1] = '\0';

			//将消息发送到消息队列中
			if(msgsnd(msgid, &msg, SIZE, 0) == -1)
			{
				perror("msgsnd error");
				return -1;
			}

			if(strcmp(msg.data, "quit") == 0)
			{
				break;
			}

		}
		exit(EXIT_SUCCESS);
	}
	else{
		//从消息队列中取数据
		Msg_ds msg1;

		while(1)
		{


			//从消息队列中取数据
			//第一个0表示取消息的类型,每次都是取第一个
			//第二个0表示阻塞方式从消息队列中取数据
			if(msgrcv(msgid, &msg1, SIZE, 100 ,0) == -1)
			{
				perror("msgsnd error");
				return -1;
			}

			if(strcmp(msg1.data, "quit") == 0)
			{
				break;
			}

			printf("rcv: %s\n", msg1.data);

		}

		wait(NULL);
	}

	msgctl(msgid, IPC_RMID, NULL);

	return 0;
}

你可能感兴趣的:(开发语言)