5.12 消息队列应用：多人聊天室

最新推荐文章于 2023-08-17 20:43:00 发布

大狮熊

最新推荐文章于 2023-08-17 20:43:00 发布

阅读量480

点赞数 1

分类专栏： Linux系统编程文章标签： linux

本文链接：https://blog.csdn.net/qq_38105227/article/details/124016073

版权

Linux系统编程专栏收录该内容

35 篇文章 0 订阅

订阅专栏

1、实验代码
client.c

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

#define handle_error(msg) \
    {perror(msg);exit(EXIT_FAILURE);}

#define TO_SERVER_MSGTYPE 1000

struct msgbuf
{
	long mtype;
	int client_id;
	char mtext[80];
};

int main (void)
{
//	key_t key = ftok ("./", 512);
	key_t key = 512;
	int msg_id = msgget (key, IPC_CREAT | 0666);

	struct msgbuf msg_snd, msg_rcv;

	printf ("input guest ID:");
	int client_id;
	scanf ("%d", &client_id);
	printf ("client_id: %d\n", client_id);

	int ret_from_fork;
	ret_from_fork = fork ();
	if (ret_from_fork == -1)
        handle_error("fork")	
	else if (ret_from_fork == 0)
	{
		while (1)
		{
			gets (msg_snd.mtext);
			msg_snd.mtype = TO_SERVER_MSGTYPE;
			msg_snd.client_id = client_id;
			msgsnd (msg_id, &msg_snd, 80, 0);
		}
	}
	else
	{
		while (1)
		{
			memset (&msg_rcv, 0, sizeof(msg_rcv));
			if (msgrcv (msg_id, (void *)&msg_rcv, \
                        sizeof(msg_rcv.mtext), client_id, 0) == -1)
			    handle_error("msg_rcv")
			else
				printf ("client%d: %s\n", msg_rcv.client_id, msg_rcv.mtext);
		}
	}
	return 0;
}

server.c

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


#define TO_SERVER_MSGTYPE 1000
#define CLIENT_NUMBER 3

struct msgbuf
{
	long mtype;
	int client_id;
	char mtext[80];
};

int main (void)
{
//	key_t key = ftok ("./", 512);
	key_t key = 512;
	int msg_id = msgget (key, IPC_CREAT | 0666);
	struct msgbuf msg;

    while (1)
	{
	    memset (&msg, 0, sizeof(msg));
		if (msgrcv (msg_id, (void *)&msg, sizeof(msg.mtext), \
					TO_SERVER_MSGTYPE, 0) == -1)
		{
			perror ("msgrcv");
			exit (EXIT_FAILURE);
		}
		else
		{
			for (int i = 1; i <= CLIENT_NUMBER; i++)
			{
				if (i == msg.client_id)
					continue;
				else
				{
					msg.mtype = i;
					msgsnd (msg_id, &msg, 80, 0);
				}
			}
		}
		printf ("server: %s\n", msg.mtext);
	}
	msgctl (msg_id, IPC_RMID, NULL);
	return 0;
}