Linux C messge-queue

1要求实现AB进程对话 (消息队列加多进程实现)

A进程先发送一句话给B进程，B进程接收后打印

B进程再回复一句话给A进程，A进程接收后打印

重复1.2步骤，当收到quit后，要结束AB进程

在上述练习基础上实现AB进程对话，要求AB进程能够随时收发。

提示：多进程，多线程。

01_msg_snd.c

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

//定义消息报类型
struct msgbuf
{
    long mtype; //消息类型，必须 > 0
    char mtext[128]; //消息内容
};

/*
    规定死msgtype: 使用 100 和 101

*/

//回收僵尸进程
void handler(int sig)
{
    while (waitpid(-1,NULL,WNOHANG) > 0); //此处WNOHANG不能填0
}


int main(int argc, char const *argv[])
{
    //捕获17号信号
    __sighandler_t s = signal(SIGCHLD,handler);
    if(SIG_ERR == s)
    {
        perror("signal");
        return -1;
    }
    //
    //计算Key
    key_t key = ftok("/home/linux/",1);
    if (key < 0)
    {
        perror("ftok");
        return -1;
    }
    printf("key = %#x\n",key);

    //消息队列创建
    int msqid = msgget(key,IPC_CREAT|0664);
    if (msqid < 0)
    {
        perror("msgget");
        return -1;
    }

    printf("msqid = %d\n",msqid);

    system("ipcs -q");

    //创建两个进程，一个进程往消息队列中发送数据，一个进程从队列中写数据

    
    pid_t cpid = fork();

    if(cpid > 0)
    {
        

         /*******************************************************/
         //子进程从msgtype为101的进程中读取数据
        struct msgbuf rcv;
        
        while (1)
        {
            memset(&rcv,0,sizeof(rcv));
            rcv.mtype = 101; //消息类型
            //从消息队列中读取数据
            //阻塞的方式，读取消息队列中的第一条消息
            if(msgrcv(msqid,&rcv,sizeof(rcv.mtext),101,0) < 0)
            //if( msgrcv(msqid,&rcv,sizeof(rcv.mtext),101,IPC_NOWAIT) < 0)
            {
                    perror("msgrcv");
                    return -1;
            }

            printf("A==读取到的消息内容:%s\n",rcv.mtext);

            if (strcasecmp(rcv.mtext,"quit")==0)
            {
                //父进程推出前杀死子进程
                kill(cpid,9);
                break;
            }
        }
        
    } else if(0==cpid)
    {
        struct msgbuf snd;
        //父进程向msgtype为100的进程中发送数据
        while (1)
        {
            memset(&snd,0,sizeof(snd));
            snd.mtype = 100; //消息类型
            //从终端获取消息内容
            //printf("A==清输入消息内容>>>");
            fgets(snd.mtext,sizeof(snd.mtext),stdin);
            snd.mtext[strlen(snd.mtext)-1] = 0;
            //向消息队列中发送数据
            if( msgsnd(msqid,&snd,sizeof(snd.mtext),0) < 0)
            {
                perror("msgsnd");
                return -1;
            }
            if (strcasecmp(snd.mtext,"quit")==0)
            {
                
                break;
            }
            }
    } else {
        perror("fork");
        return -1;
    }







    
    

        return 0;
}

02_msg_rcv.c

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

//定义消息报类型
struct msgbuf
{
    long mtype; //消息类型，必须 > 0
    char mtext[128]; //消息内容
};

//回收僵尸进程
void handler(int sig)
{
    while (waitpid(-1,NULL,WNOHANG) > 0); //此处WNOHANG不能填0
}




int main(int argc, char const *argv[])
{
    //捕获17号信号
    __sighandler_t s = signal(SIGCHLD,handler);
    if(SIG_ERR == s)
    {
        perror("signal");
        return -1;
    }

    //计算Key
    key_t key = ftok("/home/linux/",1);
    if (key < 0)
    {
        perror("ftok");
        return -1;
    }
    
    printf("key = %#x\n",key);

    //消息队列创建
    int msqid = msgget(key,IPC_CREAT|0664);
    if (msqid < 0)
    {
        perror("msgget");
        return -1;
    }

    printf("msqid = %d\n",msqid);

    system("ipcs -q");

    //创建两个进程，一个进程往消息队列中发送数据，一个进程从队列中写数据
    pid_t cpid = fork();

    if(cpid > 0)
    {
        

        /***************************************************/

         //子进程向从msgtype为100的进程中读取数据
        struct msgbuf rcv;
        while (1)
        {
            memset(&rcv,0,sizeof(rcv));
            rcv.mtype = 100; //消息类型
            //从消息队列中读取数据
            //阻塞的方式，读取消息队列中的第一条消息
            if(msgrcv(msqid,&rcv,sizeof(rcv.mtext),100,0) < 0)
            //if( msgrcv(msqid,&rcv,sizeof(rcv.mtext),100,IPC_NOWAIT) < 0)
            {
                perror("msgrcv");
                return -1;
            }
            printf("B==读取到的消息内容:%s\n",rcv.mtext);
            if (strcasecmp(rcv.mtext,"quit")==0)
            {
                kill(cpid,9);
                break;
            }
        }

    } else if(0==cpid)
    {
        //父进程向msgtype为101的消息队列发送数据
        struct msgbuf snd;
        //父进程向msgtype为100的进程中发送数据
        while (1)
        {
            memset(&snd,0,sizeof(snd));
            snd.mtype = 101; //消息类型
            //从终端获取消息内容
            //printf("B==清输入消息内容>>>");
            fgets(snd.mtext,sizeof(snd.mtext),stdin);
            snd.mtext[strlen(snd.mtext)-1] = 0;
            //向消息队列中发送数据
            if( msgsnd(msqid,&snd,sizeof(snd.mtext),0) < 0)
            {
                perror("msgsnd");
                return -1;
            }

            if (strcasecmp(snd.mtext,"quit")==0)
            {
                
                break;
            }
        }
       
    } else {
        perror("fork");
        return -1;
    }








    
    return 0;
}

效果图