消息隊列程序

前面我分享了一篇有關於消息隊列的博客， 這裏附上我寫的一段消息隊列的程序：

\
send.c（發送端）

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

struct msgbuf
{
    long mtype;
    char mtext[1024];
};

int main()
{
    key_t key = ftok("./", 1);  //獲得與路徑pathname相對應的鍵值
    if (-1 == key)
    {
        perror("ftok");
        exit(1);
    }

    int msgid = msgget(key, IPC_CREAT);  //創建消息隊列，msgid爲消息隊列的描述字
    if (-1 == msgid)
    {
        perror("msgget");
        exit(2);
    }

    struct msgbuf buffer;
    memset(buffer.mtext, 0, 1024);

    int type = 1;
    char text[1024] = {0};

    while (1)
    {
        scanf("%d %s", &type, text);
        buffer.mtype = type;
        strcpy(buffer.mtext, text);

        if (-1 == msgsnd(msgid, &buffer, sizeof(buffer.mtext), 0))  //將數據寫入消息隊列
        {
            perror("msgsnd");
            exit(3);
        }
        memset(text, 0, 1024);

        if (0 == strcmp(buffer.mtext, "exit"))  //退出
        {
            break;
        }

        memset(text, 0, 1024);
        memset(buffer.mtext, 0, 1024);
    }

    return 0;
}

recv.c（接收端）

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

struct msgbuf
{
    long mtype;
    char mtext[1024];
};

int main()
{
    key_t key = ftok("./", 1);
    if (-1 == key)
    {
        perror("ftok");
        exit(1);
    }

    int msgid = msgget(key, IPC_CREAT);
    if (-1 == msgid)
    {
        perror("msgget");
        exit(2);
    }

    struct msgbuf buffer;
    memset(buffer.mtext, 0, 1024);

    int type = 0;
    scanf("%d", &type);
//    buffer.mtype = type;

    while (1)
    {
        int count = msgrcv(msgid, &buffer, sizeof(buffer.mtext), type, 0);  //從消息隊列讀出數據， type-消息類型
        if (-1 == count)
        {
            perror("msgrcv");
            exit(4);
        }

        if (0 == strcmp(buffer.mtext, "exit"))  //退出
        {
            msgctl(msgid, IPC_RMID, 0);  //刪除msgid標識的消息隊列
            break;
        }

        printf("recv:  mtype = %d  mtext = %s\n", buffer.mtype, buffer.mtext);
        memset(buffer.mtext, 0, 1024);
    }
    return 0;
}