8.6周末作业（进程间通信：消息队列、共享内存）

作业内容

1. 要求AB进程通过消息队列做通信
   1)A进程发送一句话，B进程接收打印
   2)然后B进程发送给A进程一句话，A进程接收打印
   3)重复1,2步骤，直到A进程或者B进程收到quit，退出AB进程；
   4)AB进程能够随时收发数据
2. A进程写入一个整型，在该整型后，写入一个字符串
   B进程将共享内存中的整型以及字符串读取出来；

1. AB进程通过消息队列进行通信

进程A代码

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

typedef void (*sighandler_t)(int);

//17号信号处理函数
void handler(int sig)
{
	wait(NULL);
	exit(0);
}

struct msgbuf
{
	long mtype; 		//消息类型
	char mtext[128]; 	//消息内容
};

int main(int argc, const char *argv[])
{
	//计算键值
	key_t key = ftok("./",1);
	if(key < 0)
	{
		perror("ftok");
		return -1;
	}

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

	//创建子进程
	pid_t pid = fork();
	if(pid > 0)
	{
		//父进程读取消息
		struct msgbuf ra;
		while(1)
		{
			//捕获子进程发送的信号
			sighandler_t s = signal(SIGCHLD,handler);
			if(SIG_ERR == s)
			{
				perror("signal");
				return -1;
			}

			//接收2类消息
			if(msgrcv(msqid,&ra,sizeof(ra.mtext),2,0) < 0)
			{
				//可能被信号捕获中断系统调用，因此除去此类错误
				if(errno != 4)
				{
					perror("msgrcv");
					return -1;
				}
			}

			if(strcasecmp(ra.mtext,"quit") == 0)
			{
				printf("对方已退出\n");
				//杀死子进程
				kill(pid,SIGKILL);
				wait(NULL);
				exit(0);
			}

			printf("A接收到B的消息：%s\n",ra.mtext);
			printf("A:请输入消息内容：\n");
		}
	}
	else if(0 == pid)
	{
		//子进程写入消息
		struct msgbuf wa;
		//规定A进程发送1类消息，接收2类消息
		wa.mtype = 1;
		while(1)
		{
			printf("A:请输入消息内容：\n");
			fgets(wa.mtext,sizeof(wa.mtext),stdin);
			wa.mtext[strlen(wa.mtext)-1] = 0;
			msgsnd(msqid,&wa,sizeof(wa.mtext),0);
			if(strcasecmp(wa.mtext,"quit") == 0)
			{
				printf("退出成功\n");
				exit(0); 				//子进程退出进程
			}
		}
	}
	else
	{
		perror("fork");
		return -1;
	}

	return 0;
}

进程B代码

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

typedef void (*sighandler_t)(int);

//17号信号处理函数
void handler(int sig)
{
	wait(NULL);
	exit(0);
}

struct msgbuf
{
	long mtype; 		//消息类型
	char mtext[128]; 	//消息内容
};

int main(int argc, const char *argv[])
{
	//计算键值
	key_t key = ftok("./",1);
	if(key < 0)
	{
		perror("ftok");
		return -1;
	}

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

	//创建子进程
	pid_t pid = fork();
	if(pid > 0)
	{
		//父进程读取消息
		struct msgbuf rb;
		while(1)
		{
			//捕获子进程发送的信号
			sighandler_t s = signal(SIGCHLD,handler);
			if(SIG_ERR == s)
			{
				perror("signal");
				return -1;
			}

			//接收1类消息
			if(msgrcv(msqid,&rb,sizeof(rb.mtext),1,0) < 0)
			{
				//可能被信号捕获中断系统调用，因此除去此类错误
				if(errno != 4)
				{
					perror("msgrcv");
					return -1;
				}
			}

			if(strcasecmp(rb.mtext,"quit") == 0)
			{
				printf("对方已退出\n");
				//杀死子进程
				kill(pid,SIGKILL);
				wait(NULL);
				exit(0);
			}
			
			printf("B接收到A的消息：%s\n",rb.mtext);
			printf("B:请输入消息内容：\n");
		}
	}
	else if(0 == pid)
	{
		//子进程写入消息
		struct msgbuf wb;
		//规定B进程发送2类消息，接收1类消息
		wb.mtype = 2;
		while(1)
		{
			printf("B:请输入消息内容：\n");
			fgets(wb.mtext,sizeof(wb.mtext),stdin);
			wb.mtext[strlen(wb.mtext)-1] = 0;
			msgsnd(msqid,&wb,sizeof(wb.mtext),0);
			if(strcasecmp(wb.mtext,"quit") == 0)
			{
				printf("退出成功\n");
				exit(0); 				//子进程退出进程
			}
		}
	}
	else
	{
		perror("fork");
		return -1;
	}

	return 0;
}

测试结果

2. AB进程通过共享内存进行通信

进程A代码

#include <stdio.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>

int main(int argc, const char *argv[])
{
	//计算键值
	key_t key = ftok("./",3);
	if(key == -1)
	{
		perror("ftok");
		return -1;
	}

	//创建共享内存
	int shmid = shmget(key,128,IPC_CREAT|0664);
	if(shmid < 0)
	{
		perror("shmget");
		return -1;
	}

	//读写方式映射用户空间
	void* shmaddr = shmat(shmid,NULL,0);
	
	int* pa1 = (int*)shmaddr;
	printf("请输入一个整型数：\n");
	scanf("%d",pa1);
	getchar();
	
	//字符串首地址紧挨着整型后面
	char* pa2 = (char*)(pa1+1);
	printf("请输入一个字符串：\n");
	fgets(pa2,128,stdin);

	return 0;
}

进程B代码

#include <stdio.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>

int main(int argc, const char *argv[])
{
	key_t key = ftok("./",3);
	if(key == -1)
	{
		perror("ftok");
		return -1;
	}

	int shmid = shmget(key,128,IPC_CREAT|0664);
	if(shmid < 0)
	{
		perror("shmget");
		return -1;
	}

	//只读方式映射用户空间
	void* shmaddr = shmat(shmid,NULL,SHM_RDONLY);
	
	//读取共享内存数据
	int* pb1 = (int*)shmaddr;
	char* pb2 = (char*)(pb1+1);
	
	printf("整型：%d\n",*pb1);
	printf("字符串：%s",pb2);

	return 0;
}

测试结果