多进程通信之消息队列

最近项目里要用到消息队列，服务器从串口获取数据，再通过TCP发送给APP。

此文借鉴于：http://blog.csdn.net/ljianhui/article/details/10287879

一、什么是消息队列

消息队列提供了一种从一个进程向另一个进程发送一个数据块的方法。  每个数据块都被认为含有一个类型，接收进程可以独立地接收含有不同类型的数据结构。我们可以通过发送消息来避免命名管道的同步和阻塞问题。但是消息队列与命名管道一样，每个数据块都有一个最大长度的限制。

Linux用宏MSGMAX和MSGMNB来限制一条消息的最大长度和一个队列的最大长度。

二、在Linux中使用消息队列

Linux提供了一系列消息队列的函数接口来让我们方便地使用它来实现进程间的通信。它的用法与其他两个System V PIC机制，即信号量和共享内存相似。

1、msgget函数

该函数用来创建和访问一个消息队列。它的原型为：

[cpp] view plain copy

1. int msgget(key_t key, int msgflg);  

与其他的IPC机制一样，程序必须提供一个键来命名某个特定的消息队列。msgflg是一个权限标志，表示消息队列的访问权限，它与文件的访问权限一样。msgflg可以与IPC_CREAT做或操作，表示当key所命名的消息队列不存在时创建一个消息队列，如果key所命名的消息队列存在时，IPC_CREAT标志会被忽略，而只返回一个标识符。

它返回一个以key命名的消息队列的标识符（非零整数），失败时返回-1.

2、msgsnd函数

该函数用来把消息添加到消息队列中。它的原型为：

[cpp] view plain copy

1. int msgsend(int msgid, const void *msg_ptr, size_t msg_sz, int msgflg);  

msgid是由msgget函数返回的消息队列标识符。

msg_ptr是一个指向准备发送消息的指针，但是消息的数据结构却有一定的要求，指针msg_ptr所指向的消息结构一定要是以一个长整型成员变量开始的结构体，接收函数将用这个成员来确定消息的类型。所以消息结构要定义成这样：

[cpp] view plain copy

1. struct my_message{  
2.     long int message_type;  
3.     /* The data you wish to transfer*/  
4. };  

msg_sz是msg_ptr指向的消息的长度，注意是消息的长度，而不是整个结构体的长度，也就是说msg_sz是不包括长整型消息类型成员变量的长度。

msgflg用于控制当前消息队列满或队列消息到达系统范围的限制时将要发生的事情。

如果调用成功，消息数据的一分副本将被放到消息队列中，并返回0，失败时返回-1.

3、msgrcv函数

该函数用来从一个消息队列获取消息，它的原型为

[cpp] view plain copy

1. int msgrcv(int msgid, void *msg_ptr, size_t msg_st, long int msgtype, int msgflg);  

msgid, msg_ptr, msg_st的作用也函数msgsnd函数的一样。

msgtype可以实现一种简单的接收优先级。如果msgtype为0，就获取队列中的第一个消息。如果它的值大于零，将获取具有相同消息类型的第一个信息。如果它小于零，就获取类型等于或小于msgtype的绝对值的第一个消息。

msgflg用于控制当队列中没有相应类型的消息可以接收时将发生的事情。

调用成功时，该函数返回放到接收缓存区中的字节数，消息被复制到由msg_ptr指向的用户分配的缓存区中，然后删除消息队列中的对应消息。失败时返回-1.

4、msgctl函数

该函数用来控制消息队列，它与共享内存的shmctl函数相似，它的原型为：

[cpp] view plain copy

1. int msgctl(int msgid, int command, struct msgid_ds *buf);  

command是将要采取的动作，它可以取3个值，

    IPC_STAT：把msgid_ds结构中的数据设置为消息队列的当前关联值，即用消息队列的当前关联值覆盖msgid_ds的值。

    IPC_SET：如果进程有足够的权限，就把消息列队的当前关联值设置为msgid_ds结构中给出的值

    IPC_RMID：删除消息队列

buf是指向msgid_ds结构的指针，它指向消息队列模式和访问权限的结构。msgid_ds结构至少包括以下成员：

[cpp] view plain copy

1. struct msgid_ds  
2. {  
3.     uid_t shm_perm.uid;  
4.     uid_t shm_perm.gid;  
5.     mode_t shm_perm.mode;  
6. };  

成功时返回0，失败时返回-1.

下面是源码：

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

#define MAX_DATA_LENGTH  2048

struct Message_Queue  
{  
    int msg_type;  
    char Msg_Data[MAX_DATA_LENGTH];  
};  
int running = 1;  
int msgid = -1;  
long int msgtype = 0;   

char buffer[BUFSIZ+1] = {0};  

int main(void)
{
    pid_t pid;
    struct Message_Queue message;  //要是声明为 *message,则必须手动分配空间，不然会段错误
    //建立消息队列  
    msgid = msgget((key_t)1106, 0666 | IPC_CREAT);   //key值可通过ftok函数生成，也可以自己写一个整数
    if(msgid == -1)  
    {  
        fprintf(stderr, "msgget failed with error: %d\n", errno);  
        exit(EXIT_FAILURE);  
    }  

    if((pid = fork())==-1)
    {
        perror("Fail to fork");
        exit(EXIT_SUCCESS);
    }

    if(pid!=0)  //父进程负责发送消息
    {
        //向消息队列中写消息，直到写入end  
        while(running)  
        {  
            //输入数据 
            bzero(buffer,sizeof(buffer));	
            bzero(message.Msg_Data,sizeof(message.Msg_Data)); //清空数组，不然读取的时候会发生错误			
            printf("Enter some Msg_Data: ");  
            fgets(buffer, BUFSIZ, stdin);  
            message.msg_type = 1;    //注意2  

            strncpy(message.Msg_Data,buffer,strlen(buffer));
            //strncpy时是没有拷贝'\0'的，所以要加上，不然读的时候打印会有乱码
            message.Msg_Data[strlen(message.Msg_Data)-1] = '\0';
            //message.msg_type = 1;    //注意2 
            //向队列发送数据  
            if(msgsnd(msgid, (void*)&message, MAX_DATA_LENGTH, 0) == -1)  
            {  
                fprintf(stderr, "msgsnd failed\n");  
                exit(EXIT_FAILURE);  
            }  
            //输入end结束输入  
            if(strncmp(buffer, "end", 3) == 0)  
                running = 0;  
            sleep(1);  
        }  
        exit(EXIT_SUCCESS);  
    }

    if(pid==0)   //子进程负责读消息
    {
        //从队列中获取消息，直到遇到end消息为止  
        while(running)  
        {  
            if(msgrcv(msgid, (void*)&message, BUFSIZ, msgtype, 0) == -1)  
            {  
                fprintf(stderr, "msgrcv failed with errno: %d\n", errno);  
                exit(EXIT_FAILURE);  
            }  
            printf("You wrote: %s\n",message.Msg_Data);  
            //遇到end结束  
            if(strncmp(message.Msg_Data, "end", 3) == 0)  
                running = 0;  
        }  
        //删除消息队列  
        if(msgctl(msgid, IPC_RMID, 0) == -1)  
        {  
            fprintf(stderr, "msgctl(IPC_RMID) failed\n");  
            exit(EXIT_FAILURE);  
        }  
        exit(EXIT_SUCCESS);  
    }

    exit(EXIT_SUCCESS);
}

运行情况：