IO线程day8

作业

题目1

1. A进程写入一个整型，在该整型后，写入一个字符串
2. B进程将共享内存中的整型以及字符串读取出来;

代码

A进程

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

#define PER(per) { fprintf(stderr,"line: %d ",__LINE__);\
					perror(per);\
					return -1;	}
int main(int argc, const char *argv[])
{
	key_t key=ftok("./",3);
	if(key==-1)
	{
		PER("ftok")
	}

	//创建共享内存
	int shmid=shmget(key,64,IPC_CREAT|0664);
	if(shmid==-1)
	{
		PER("shmget")
	}

	//映射内存
	int *a=(int *)shmat(shmid,NULL,0);
	if((void*)a==(void*)-1)
	{
		PER("shmat")
	}
	//写入数据
	*a=16;
	char *b= (char *)(a+1);
	strcpy(b,"hello");

	return 0;
}

B进程

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

#define PER(per) { fprintf(stderr,"line: %d ",__LINE__);\
					perror(per);\
					return -1;	}
int main(int argc, const char *argv[])
{
	key_t key=ftok("./",3);
	if(key==-1)
	{
		PER("ftok")
	}

	//创建共享内存
	int shmid=shmget(key,64,IPC_CREAT|0664);
	if(shmid==-1)
	{
		PER("shmget")
	}

	//映射内存
	int *a=(int *)shmat(shmid,NULL,0);
	if((void*)a==(void*)-1)
	{
		PER("shmat")
	}
	//读取数据
	printf("%d ",*a);
	char *b= (char *)(a+1);
	printf("%s\n",b);

	return 0;
}

题目2

1）要求AB进程做通信（消息队列方式）

1.A进程发送一句话，B进程接收打印
2. 然后B进程发送给A进程一句话，A进程接收打印
3.重复1,2步骤，直到A进程或者B进程收到quit，退出AB进程；

2）在第一题的基础上实现，AB进程能够随时收发数据（附加题）

进程A

#include<stdio.h>
#include<unistd.h>
#include<sys/types.h>
#include<sys/ipc.h>
#include<sys/msg.h>
#include<stdlib.h>
#include<string.h>
#include<signal.h>
#include<wait.h>
//消息包结构体
struct msgbuf
{
	long mtype;              //消息类型
	char mtext[128];         //消息内容
};

typedef void (*sighandler_t)(int);

//捕获信号SIGUSR1后的处理函数
void handler2(int sig)
{
	printf("\n子进程接收到退出请求,子进程退出\n");
	exit(0);
}


//捕获信号SIGCHLD后的处理函数
void handler1(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0);
	printf("程序退出成功\n");
	exit(0);
}


int main(int argc, const char *argv[])
{

	//建立父子进程
	pid_t pid=fork();
	if(pid==-1)
	{
		perror("fork");
		return -1;
	}


	//父进程发数据
	if(pid>0)
	{
		//捕获信号SIGCHLD
		if(signal(SIGCHLD,handler1)==SIG_ERR)
		{
			perror("signal");
			return -1;
		}

		//计算key值1
		key_t key=ftok("./",1);
		if(key==-1)
		{
			perror("ftok");
			return -1;
		}
		//创建消息队列1
		int msqid=msgget(key,IPC_CREAT|0664);
		if(msqid==-1)
		{
			perror("msgget");
			return -1;
		}

		//发消息
		struct msgbuf snd;
		snd.mtype=1;
		while(1)
		{
			bzero(snd.mtext,sizeof(snd.mtext));	
			printf("请输入消息-->>");
			fgets(snd.mtext,sizeof(snd.mtext),stdin);
			snd.mtext[strlen(snd.mtext)-1]=0;


			//阻塞方式发送
			if(msgsnd(msqid,&snd,sizeof(snd.mtext),0)==-1)
			{
				perror("msgsnd");
				return -1;
			}

			//发送quit时
			if(strcasecmp(snd.mtext,"quit")==0)
			{
				printf("程序退出中\n");
				if(kill(pid,SIGUSR1)==-1)
				{
					perror("kill_SIGUSR1");
					return -1;
				}
				wait(NULL);

			}

		}
		exit(0);
	}
	//子进程接收
	if(pid==0)
	{
		//捕获信号SIGUSR1
		if(signal(SIGUSR1,handler2)==SIG_ERR)
		{
			perror("signal");
			return -1;
		}
		//计算key值2
		key_t key=ftok("./",2);
		if(key==-1)
		{
			perror("ftok");
			return -1;
		}

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

		//接收消息
		ssize_t res=0;
		struct msgbuf rcv;
		while(1)
		{
			bzero(rcv.mtext,sizeof(rcv.mtext));
			res=msgrcv(msqid,&rcv,sizeof(rcv.mtext),0,0);
			if(res==-1)
			{
				perror("msgrcv");
				return -1;
			}
			//接收到quit时
			if (strcasecmp(rcv.mtext,"quit")==0)
			{
				printf("\n\n子程序接收到退出请求,子程序退出\n");
				break;
			}
			printf("\nB的消息为: %s\n",rcv.mtext);

			printf("请继续发送消息-->>");


			fflush(stdout);

		}
		exit(0);
	}
	return 0;
}

进程B

#include<stdio.h>
#include<unistd.h>
#include<sys/types.h>
#include<sys/ipc.h>
#include<sys/msg.h>
#include<stdlib.h>
#include<string.h>
#include<signal.h>
#include<wait.h>
//消息包结构体
struct msgbuf
{
	long mtype;              //消息类型
	char mtext[128];         //消息内容
};

typedef void (*sighandler_t)(int);

void handler2(int sig)
{
	printf("\n子进程接收到退出请求,子进程退出\n");
	exit(0);
}



void handler1(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0);
	printf("程序退出成功\n");
	exit(0);
}


int main(int argc, const char *argv[])
{

	//建立父子进程
	pid_t pid=fork();
	if(pid==-1)
	{
		perror("fork");
		return -1;
	}


	//父进程发数据
	if(pid>0)
	{

		if(signal(SIGCHLD,handler1)==SIG_ERR)
		{
			perror("signal");
			return -1;
		}

		//计算key值1
		key_t key=ftok("./",2);
		if(key==-1)
		{
			perror("ftok");
			return -1;
		}
		//创建消息队列1
		int msqid=msgget(key,IPC_CREAT|0664);
		if(msqid==-1)
		{
			perror("msgget");
			return -1;
		}

		//发消息
		struct msgbuf snd;
		snd.mtype=1;
		while(1)
		{
			bzero(snd.mtext,sizeof(snd.mtext));	
			printf("请输入消息-->>");
			fgets(snd.mtext,sizeof(snd.mtext),stdin);
			snd.mtext[strlen(snd.mtext)-1]=0;


			//阻塞方式发送
			if(msgsnd(msqid,&snd,sizeof(snd.mtext),0)==-1)
			{
				perror("msgsnd");
				return -1;
			}

			//发送quit时
			if(strcasecmp(snd.mtext,"quit")==0)
			{
				printf("程序退出中\n");
				if(kill(pid,SIGUSR1)==-1)
				{
					perror("kill_SIGUSR1");
					return -1;
				}
				wait(NULL);            //

			}

		}
		exit(0);
	}
	//子进程接收
	if(pid==0)
	{

		if(signal(SIGUSR1,handler2)==SIG_ERR)
		{
			perror("signal");
			return -1;
		}
		//计算key值2
		key_t key=ftok("./",1);
		if(key==-1)
		{
			perror("ftok");
			return -1;
		}

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

		//接收消息
		ssize_t res=0;
		struct msgbuf rcv;
		while(1)
		{
			bzero(rcv.mtext,sizeof(rcv.mtext));
			res=msgrcv(msqid,&rcv,sizeof(rcv.mtext),0,0);
			if(res==-1)
			{
				perror("msgrcv");
				return -1;
			}
			if (strcasecmp(rcv.mtext,"quit")==0)
			{
				printf("\n\n子程序接收到退出请求,子程序退出\n");
				break;
			}
			printf("\nB的消息为: %s\n",rcv.mtext);

			printf("请继续发送消息-->>");


			fflush(stdout);

		}
		exit(0);
	}
	return 0;
}