system-V 消息队列（实现多人聊天方法） 学习笔记

目录

一、system-V ipc特点

二、消息队列用法

三、源代码：

---

一、system-V ipc特点

- 独立于进程
- 没有文件名和文件描述符
- IPC对象具有key和ID

---

二、消息队列用法

• 定义一个唯一key(ftok)
  参数：
  - path：一个合法路径
  - proj_id：一个整数
• 构造消息对象(msgget)
  参数：
  - key：消息队列的键值
  - msgflg:
      - IPC_CREAT：如果消息队列不存在，则创建
      - mode：访问权限
• 发送特定类型消息(msgsnd)|
  参数：
  - msqid：消息队列ID
  - msgp：消息缓存区
  - struct msgbuf
  {
      long mtype;     //消息标识
      char mtext[1]; //消息内容
            ......
  }
• - msgsz：消息正文的字节数
  - msgflg：
      - IPC_NOWAIT：非阻塞发
      - 0：阻塞发送
• 接受特定类型消息(msgrcv)
  参数：
  - msqid：消息队列ID
  - msgp：消息缓存区
  - msgsz：消息正文的字节数
  - msgtyp：要接受消息的标识
  - msgflg：
    - IPC_NOWAIT：非阻塞读取
    - MSG_NOERROR：截断消息
    - 0：阻塞读取
• 删除消息队列(msgctl)
  参数：
  - msgqid：消息队列的ID
  - cmd
  - IPC_STAT：获取消息队列的属性信息
  - IPC_SET：设置消息队列的属性
  - IPC_RMID：删除消息队列
  - buf：相关结构体缓冲区

  ---

   

三、源代码：

实现n人聊天：将代码cp n份，将其中的（接收/发送）数据类型以及消息对象（用户）更改即可。

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


#define BUFF_SIZE 512
#define IPC "/tmp"

#define user_type 1


struct message
{
	long msg_type;/*类型*/
	char msg_text[BUFF_SIZE];/*内容*/
	char msg_name[10];/*用户*/
};

char *p_name[] = {"user_1","user_2","user_3"};  /*定义用户名*/

int user_n = sizeof(p_name)/sizeof(p_name[0]);

int p_type[] = {1,2,3};//每个用户名对应的数据类型


int main()
{
	pid_t fd;
	int msg_id;
	key_t key;

	key = ftok(IPC,p_type[2]);/*定义一个key*/
	if(key < 0)
	{
		printf("ftok error!\r\n");
		exit(1);
	}

	msg_id = msgget(key,IPC_CREAT | 0666);//如果没有消息队列则创建并给权限666
	if(msg_id < 0)
	{
		printf("msgget error!\r\n");
		exit(1);
	}

	fd = fork();//创建一个子进程
	if(fd < 0)
	{
		printf("fork error!\r\n");
		exit(1);
	}
	if(fd == 0)//子进程发送数据
	{
		while(1)
		{
			struct message msg;
			int user_i;
			printf("Please Input User:");
			fgets(msg.msg_name,BUFF_SIZE,stdin);
			msg.msg_name[strlen(msg.msg_name)-1] = '\0';//将fgets读取的回车换行符覆盖
			for(user_i = 0;user_i < user_n;user_i++)//判断输入的数据类型（用户）是否正确
			{
				if(strncmp(msg.msg_name,p_name[user_i],6) == 0)
					break;
			}
			if(user_i == user_n)//输入错误
			{
				printf("user_name error!\r\n");
				continue;
			}
			else
			{
				printf("Please Input Message:");
				fgets(msg.msg_text,BUFF_SIZE,stdin);
				msg.msg_text[strlen(msg.msg_text)-1] = '\0';//将fgets读取的回车换行符覆盖
				//给结构体成员赋值（发送缓存区）
				msg.msg_type = p_type[user_i];
				strcpy(msg.msg_name, "user_1");
				
				if((msgsnd(msg_id,&msg,sizeof(msg)-sizeof(long),0)) < 0)//发送将数据缓存区的数据
				{
					printf("msgsnd error!\r\n");
					continue;
				}
				sleep(1);
				if(strncmp(msg.msg_text,"quit",4) == 0)//如果发送的数据里有”quit“字符串则退出子进程
				{
					exit(0);
				}
			}
				
		}
		
	}
	else//父进程读取数据
	{
		while(1)
		{
			struct message msg;
			memset(&msg,0,BUFF_SIZE);//清空接收缓存区
			if(msgrcv(msg_id,&msg,sizeof(msg)-sizeof(long),user_type,0) < 0)//读取数据
			{
				printf("msgrcv error!\r\n");
				continue;
			}
			printf("\r\n%ld byte From %s Message : %s\r\n",strlen(msg.msg_text),msg.msg_name,msg.msg_text);
			printf("Please Input User:");
			/*下一个数据再上一个数据还没输出完毕，还在输出缓冲区中时，
			下一个printf就把另一个数据加入输出缓冲区，
			结果冲掉了原来的数据，出现输出错误，所以用fflush(stdout);强制输出*/
			fflush(stdout);
			if(strncmp(msg.msg_text,"quit",4) == 0)//如果接受的数据里有quit，则退出
			{
				if((msgctl(msg_id,IPC_RMID,NULL)) < 0)//删除消息队列
				{
					printf("msgclt error!\r\n");
					exit(1);
				}
				wait(NULL);//等待子进程退出
				exit(0);//父进程退出
			}
			
		}
		
	}
	
}