c使用消息队列实现的2个终端之间的互相聊天

使用消息队列实现的2个终端之间的互相聊天

并使用信号控制消息队列的读取方式： 当键盘按ctrl+c的时候，切换消息读取方式，一般情况为读取指定编号的消息，按ctrl+c之后，指定的编号不读取，读取其他所有编号的消息

post.c 

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <wait.h>
#include <signal.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <semaphore.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/un.h>

struct msgbuf{
    long type;
    char buf[128];
};

  

// 接收消息的线程函数  
void *receive_messages(void *arg) {  
    int id = *((int *)arg);  
    struct msgbuf msg;  
  
    while (1) {  
        msgrcv(id, (char *)&msg, sizeof(struct msgbuf), 0, 0); // 读取所有类型的消息  
        printf(" 内容: %s\n", msg.buf);  
    }  
    return NULL;  
}  

int main(int argc, const char *argv[])
{
	if(argc != 2){
		printf("参数数量错误\n");
		return 1;
	}
    int msgtype = atoi(argv[1]);
	key_t key = ftok("./ipc",1);
	if(key == -1){
		perror("ftok");
		return 1;
	}
	int id = msgget(key,IPC_CREAT | 0666);
	if(id == -1){
		perror("msgget");
		return 1;
	}
    struct msgbuf msg;
    int size = 0;
	while(1){
		//写
		memset(&msg,0,sizeof(msg));
		msg.type = msgtype;//确认消息编号
		printf("请输入:");
		scanf("%128s",msg.buf);
		while(getchar()!=10);
		size = strlen(msg.buf);
		msgsnd(id,&msg,size,0);
		
		// 创建接收消息的线程  
		if (pthread_create(&recv_thread, 0, receive_messages, (void *)id) != 0) {  
			perror("pthread_create");  
			return 1;  
		}  
	}
	return 0;
}

get.c

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <wait.h>
#include <signal.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <semaphore.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/un.h>
struct msgbuf{
    long type;
    char buf[128];
};
int main(int argc, const char *argv[])
{
	if(argc != 2){
		printf("参数数量不正确\n");
		return 1;
	}

    int msgtype = atoi(argv[1]);
 //根据文件名和传入的编号获取秘钥

	key_t key = ftok("./ipc",1);
	if(key == -1){
		perror("ftok");
		return 1;
	}
	int id = msgget(key,IPC_CREAT | 0666);
	if(id == -1){
		perror("msgget");
		return 1;
	}
	struct msgbuf msg;
	struct msgbuf msg2;

	int size = 0;
	while(1){
		memset(&msg,0,sizeof(msg));
		memset(&msg2,0,sizeof(msg2));

		//读取消息队列，020000为MSG_EXCEPT的宏定义，参数msgflg
		msgrcv(id,&msg,128,msgtype,020000);
		printf("读取到的消息为:%s\n",msg.buf);

		printf("请输入:");
		scanf("%128s",msg2.buf);
		while(getchar()!=10);
		size = strlen(msg2.buf);
		msgsnd(id,&msg2,size,0);
	}
    return 0;
}