Linux：使用多线程编程和消息队列，实现两个进程之间的聊天

思路：

一个文件：创建一个线程和主函数，或者创建两个线程主函数调用（我用这种）。

创建两个消息队列，

一共两个文件，两个队列，四个进程

a.c    一个进程写（消息类型为1）   ---->>队列     一个进程读（消息类型为2）

b.c   一个进程写（消息类型为2）   ---->>队列     一个进程读（消息类型为1）

 

a.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pthread.h>

#include <unistd.h>
#include <sys/ipc.h>
#include <sys/msg.h>

struct MSG{
    long mtype;//消息类型
    char buf[100];//消息数据
};

struct MSG_read{
    long mtype_read;//消息类型
    char buf_read[100];//消息数据
};


void *write1(void *arg)
{

	   //1.获取key
    key_t key = ftok("/",'a');
    if(key==-1)
    {
        perror("ftok");
        exit(-1);
    }


     int msqid = msgget(key,IPC_CREAT|0666);       //2.通过key创建消息队列
    if(msqid==-1)
    {
        perror("msgget");
        exit(-2);
    }

        //3.发送消息
    struct MSG msg1;
    
    while(1)
    {
    	msg1.mtype = 1;


    	fgets(msg1.buf,100,stdin);

	    int res = msgsnd(msqid,&msg1,sizeof(msg1.buf),0);

    }

}

void *read1(void *arg)
{


	key_t key = ftok("/",'b');
    if(key==-1)
    {
        perror("ftok");
        exit(-1);
    }


     int msqid = msgget(key,IPC_CREAT|0666);       //2.通过key创建消息队列
    if(msqid==-1)
    {
        perror("msgget");
        exit(-2);
    }

     struct MSG_read msg;
     while(1)
     {
     	    int res = msgrcv(msqid,&msg,sizeof(msg.buf_read),2,0);   //收类型为2 的信息
	    if(res==-1){
	        perror("msgrcv");
	        exit(-3);
	    }
	     printf("%s",msg.buf_read);
     }

}




int main()
{
	pthread_t id1;
	pthread_t id2;

    //创建线程
    int res = pthread_create(&id1,0,write1,0);
    if(res){
        printf("%s\n",strerror(res));
        //exit(-1);
    }

    int res2 = pthread_create(&id2,0,read1,0);
    if(res2){
        printf("%s\n",strerror(res2));
        //exit(-1);
    }

     pthread_join(id1,0);
     pthread_join(id2,0);

	return 0;
}

 

b.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pthread.h>

#include <unistd.h>
#include <sys/ipc.h>
#include <sys/msg.h>


struct MSG{
    long mtype;//消息类型
    char buf[100];//消息数据
};

struct MSG_read{
    long mtype_read;//消息类型
    char buf_read[100];//消息数据
};



void *read2(void *arg)
{

       //1.获取key
    key_t key = ftok("/",'a');
    if(key==-1)
    {
        perror("ftok");
        exit(-1);
    }


     int msqid = msgget(key,IPC_CREAT|0666);       //2.通过key创建消息队列
    if(msqid==-1)
    {
        perror("msgget");
        exit(-2);
    }

     struct MSG msg1;
     while(1)
     {
        int res = msgrcv(msqid,&msg1,sizeof(msg1.buf),1,0);   //收类型为2 的信息
        if(res==-1){
            perror("msgrcv");
            exit(-3);
        }
         printf("%s",msg1.buf);
     }
  
}

void *write2(void *arg)
{

    key_t key = ftok("/",'b');
    if(key==-1)
    {
        perror("ftok");
        exit(-1);
    }


     int msqid = msgget(key,IPC_CREAT|0666);       //2.通过key创建消息队列
    if(msqid==-1)
    {
        perror("msgget");
        exit(-2);
    }

    struct MSG_read msg1;
    msg1.mtype_read = 2;
    while(1)
    {
        
        
        fgets(msg1.buf_read,100,stdin);


        int res = msgsnd(msqid,&msg1,sizeof(msg1.buf_read),0);

    }
}



int main()
{
	pthread_t id1;
	pthread_t id2;

    //创建线程
    int res = pthread_create(&id1,0,write2,0);
    if(res){
        printf("%s\n",strerror(res));
        //exit(-1);
    }

    int res2 = pthread_create(&id2,0,read2,0);
    if(res2){
        printf("%s\n",strerror(res2));
        //exit(-1);
    }

     pthread_join(id1,0);
     pthread_join(id2,0);

	return 0;
}

 gcc a.c -pthread -o b.out

gcc b.c  -pthread