Linux 下的消息队列

最新推荐文章于 2025-03-15 14:48:15 发布

LordWingF

最新推荐文章于 2025-03-15 14:48:15 发布

阅读量876

点赞数 1

分类专栏：学习文章标签： linux

本文链接：https://blog.youkuaiyun.com/LordWingF/article/details/72974488

版权

学习专栏收录该内容

30 篇文章

订阅专栏

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

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

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

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

[cpp] view plain copy print?
int msgget(key_t, key, int msgflg);

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

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

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

[cpp] view plain copy print?
int msgsend(int msgid, const void *msg_ptr, size_t msg_sz, int msgflg);

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

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

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

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

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

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

[cpp] view plain copy print?
int msgrcv(int msgid, void *msg_ptr, size_t msg_st, long int msgtype, int msgflg);

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

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

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

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

[cpp] view plain copy print?
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结构的指针，它指向消息队列模式和访问权限的结构。

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

三、实现消息队列
下面我们来看看如何实现消息队列的基本操作

/*Makefile*/

.PHONY:all
all:client server
client:client.c comm.c
    gcc -o $@ $^
server:server.c comm.c
    gcc -o $@ $^

.PHNOY:clean
clean:
    rm -f server client


/*comm.h*/

/*************************************************************************
    > File Name: comm.h
    > Author:Lord'WingF'
    > Mail:暂无
    > Created Time: Sat 03 Jun 2017 01:19:23 PM PDT
 ************************************************************************/

#ifndef _COMM_H
#define _COMM_H

#include<stdio.h>
#include<sys/types.h>
#include<sys/wait.h>
#include<sys/stat.h>
#include<sys/msg.h>
#include<string.h>
#include<stdlib.h>
#include<sys/ipc.h>

#define PROJ_ID 0x6666
#define SERVER_TYPE 1
#define CLIENT_TYPE 2
#define PATHNAME "."

struct msgbuf{
    long mytype;
    char metext[1024];
};

int CreatMsgQueue();
int GetMsgQueue();
int DestoryMsgQueue(int msgid);
int RecvMsgQueue(int msdid, int type, char* outmsg);
int SendMsgQueue(int msgid, int who, const char* msg);

#endif


/*comm.c*/

#include"comm.h"

/********************************************************/
int CommMsgQueue(int flags)
{
    key_t _key = ftok(PATHNAME, PROJ_ID);  
    if(_key < 0){
        perror("ftok");
        return 1;
    }

    int msgid = msgget(_key, flags);
    if(msgid < 0){
        perror("msgget");
        return 2;
    }

    return msgid;
}
/********************************************************/

/********************************************************/
int CreateMsgQueue()
{
    return CommMsgQueue(IPC_CREAT | IPC_EXCL | 0666);
}
/********************************************************/

/********************************************************/
int GetMsgQueue()
{
    return CommMsgQueue(IPC_CREAT);
}
/********************************************************/

/********************************************************/
int DestoryMsgQueue(int msgid)
{
    if(msgctl(msgid, IPC_RMID, NULL) < 0){
        perror("msgctl");
        return 1;
    }
    return 0;
}
/********************************************************/

/********************************************************/
int SendMsgQueue(int msgid, int who, const char* msg)
{
    struct msgbuf _mb;
    _mb.mytype = who;
    strcpy(_mb.metext, msg);
    if(msgsnd(msgid, (void*)&_mb, sizeof(_mb.metext), 0) < 0)
    {
        perror("msgsnd");
        return -1;
    }
    return 0;
}
/********************************************************/

/********************************************************/
int RecvMsgQueue(int msgid, int type, char *outMsg)
{
    struct msgbuf _mb;
    if(msgrcv(msgid, (void*)&_mb, sizeof(_mb.metext),type, 0) < 0)
    {
        perror("msgrcv");
        return -1;
    }
    strcpy(outMsg, _mb.metext);
    return 0;
}
/********************************************************/


/*sever.c*/

#include"comm.h"

int main()
{
    char buf[1024];
    int msgid = CreateMsgQueue();

    while(1){
        if(RecvMsgQueue(msgid, CLIENT_TYPE, buf) < 0){
            break;
        }
        printf("client say# %s\n", buf);

        printf("please enter# ");
        fflush(stdout);
        ssize_t s = read(0, buf, sizeof(buf)-1);
        if(s > 0){
            buf[s-1] = 0;
            if(SendMsgQueue(msgid, SERVER_TYPE, buf) < 0){
                break;
            }
        }
    }

    DestoryMsgQueue(msgid);
    return 0;
}


/*client.c*/

#include"comm.h"

int main()
{
    char buf[1024];
    int msgid = GetMsgQueue();
    while(1){
        printf("please enter# ");
        fflush(stdout);

        ssize_t s = read(0, buf, sizeof(buf)-1);
        if(s > 0)
        {
            buf[s-1] = 0;
            if(SendMsgQueue(msgid, CLIENT_TYPE, buf) < 0){
                break;
            }
        }

        if(RecvMsgQueue(msgid, SERVER_TYPE, buf) < 0){
            break;
        }
        printf("server say : %s\n",buf);
    }
    return 0;
}