Linux下进程间通信之消息队列

    消息队列是进程间通信之一，当然还有其他通信方式：管道，有名管道，内存共享，套接字等，下面详细介绍消息队列的用法。

首先介绍几个函数

int msgctl(int msgid , ing cmd , struct msgid_ds *buf);

int msgget(key_t key , ingt msgflg);

int msgrcv(ing msgid , void* msg_ptr , size_t msg_sz , long int msgtype , int msgflg);

int msgsnd(int msgid , const void* msg_ptr , size_t msg_sz , int msgflg);

这些函数的作用跟共享内存和信号量的作用是相同的。

int msgctl(int msgid , ing cmd , struct msgid_ds *buf);： 控制函数，用来控制消息列队的关闭等。

int msgget(key_t key , ingt msgflg);用来创建一个消息列队

int msgrcv(ing msgid , void* msg_ptr , size_t msg_sz , long int msgtype , int msgflg);用来接收一个消息

int msgsnd(int msgid , const void* msg_ptr , size_t msg_sz , int msgflg);用来发送一个消息

下面来看一个具体的例子。在这个例子中，我们将在客户端中对消息进行接收，在接收完所有的消息后，删除消息列队。

在服务器中，我们将发送消息到消息列队中，直到遇到end结束符为止。

发送消息源程序，命名为msgsend.c

#include<stdlib.h>  
#include<stdio.h>  
#include<string.h>  
#include<errno.h>  
#include<unistd.h>  
#include<sys/msg.h>  

#define BUFSIZE 1024

struct msg_st
{
	int msg_type;
	char msg_buf[BUFSIZE];
};

int main(int argc, char *argv[])
{
	int msgid;//定义消息队列id
	key_t key;//唯一识别两通信的消息队列，要与接收方一致
	struct msg_st msg_data;
	char buffer[BUFSIZE];
	int runnig = 1;

	//消息队列是否创建成功
	if((msgid = msgget(key = 1234, 0666 | IPC_CREAT)) == -1)
	{
		fprintf(stderr, "msgget failed with error:%d\n",errno);
		exit(-1);	
	}


	while(runnig)
	{
		//get some data
		printf("enter some data:");
		fgets(buffer,BUFSIZE,stdin);
		msg_data.msg_type = 1;
		strcpy(msg_data.msg_buf,buffer);
		
		//send data
		if(msgsnd(msgid, (void *)&msg_data, BUFSIZE, 0) == -1)
		{
			fprintf(stderr, "msgsend failed with error:%d\n",errno);
			exit(-1);
		}

		//exit with end
		if(strncmp(msg_data.msg_buf,"end",3) == 0)
		{
			runnig = 0;
		}
		
	}

	if(msgctl(msgid,IPC_RMID,0) == -1)
	{
		fprintf(stderr, "msgctl failed with error:%d\n",errno);
		exit(-1);
	}
	
	exit(0);
	
}

接收消息源程序，命名为 msgrcv.c

#include<stdlib.h>  
#include<stdio.h>  
#include<string.h>  
#include<errno.h>  
#include<unistd.h>  
#include<sys/msg.h>  

#define BUFSIZE 1024

struct msg_st
{
	int msg_type;
	char msg_buf[BUFSIZE];
};

int main(int argc, char *argv[])
{
	int msgid;
	key_t key;
	struct msg_st msg_data;
	int msg_recv = 0;
	int runnig = 1;

	if((msgid = msgget(key = 1234, 0666 | IPC_CREAT)) == -1)
	{
		fprintf(stderr, "msgget failed with error:%d\n",errno);
		exit(-1);	
	}


	while(runnig)
	{
		if(msgrcv(msgid, (void *)&msg_data, BUFSIZE, msg_recv, 0) == -1)
		{
			fprintf(stderr, "msgrcv failed with error:%d\n",errno);
			exit(-1);
		}

		//以end结束
		printf("receive data: %s\n",msg_data.msg_buf);
		if(strncmp(msg_data.msg_buf,"end",3) == 0)
		{
			runnig = 0;
		}
		
	}
	

	//关闭消息队列
	if(msgctl(msgid,IPC_RMID,0) == -1)
	{
		fprintf(stderr, "msgctl failed with error:%d\n",errno);
		exit(-1);
	}
	exit(0);
}

消息队列的用法比较简单，大概分为以下几步：

1.定义一个结构体用来存放数据，msg_data和msg_type

2.定义消息队列id，创建消息队列 msgget(){}

3.消息队列接收和发送函数 msgrcv（）{}和 msgsnd(){}

4.关闭消息队列 msgctl(){}

注意：在定义key_t key 的时候，发送和接收的key值要相同