linux 实验三

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

#define SOLVENUM 0
#define SOLVESTR 1

int sum = 0;
void *p = NULL;

void catch_signal(int signo,siginfo_t *resdata,void *unkonwp)
{
//处理信号函数

if(resdata->si_value.sival_int == SOLVENUM)
{//处理数字
	sum += *(int*)p;
	printf("数字总和：%d\n", sum);
	if(sum > 100)
	{
		printf("my work done!\n");
		kill(getppid(),SIGINT);
		exit(0);
	}
}
else if(resdata->si_value.sival_int == SOLVESTR)
{//处理字符串
	printf("获得字符串：%s\n", (char*)p);
}

}

//安装信号
int mysignal(int sign,void (func)(int, siginfo_t, void*))
{
struct sigaction act;
//初始化sa_mask
sigemptyset(&act.sa_mask);
act.sa_sigaction=func;
//一旦使用了sa_sigaction属性，那么必须设置sa_flags属性的值为SA_SIGINFO
act.sa_flags=SA_SIGINFO;
return sigaction(sign,&act,NULL);
}

int main()
{
char buffer[64]; //输入
char c;
int i;
int isstr; //判断是否为字符串
int n; //输入的数字
int shmid;

pid_t p1 = fork();
if(p1 < 0)
{
	printf("error in fork!");
	exit(0);
}
else if(p1 == 0)
{//子进程，处理数据
	
	shmid = shmget(0x2234, 0, 0); 
	if (shmid == -1)
	{
		perror("shmget err");
		return errno;
	}
	//将共享内存段连接到进程地址空间
	p = shmat(shmid, NULL, 0);
	if (p == (void *)-1 )
	{
		perror("shmget err");
		return errno;
	}
	
	//安装信号
	if(mysignal(SIGUSR1, catch_signal)!=0)
	{
		printf("sigaction() failed! \n");
		return -1;
	}
	while(1)
	{
		//等待父进程返回信号
		int status=0;
		wait(&status);
	}
}
else
{
	//创建共享内存 64字节 ，相当于打开文件，文件不存在则创建
	shmid = shmget(0x2234, 64, IPC_CREAT | 0666); 
	if (shmid == -1)
	{
		perror("shmget err");
		return errno;
	}
	
	//将共享内存段连接到进程地址空间
	p = shmat(shmid, NULL, 0);//第二个参数shmaddr为NULL，核心自动选择一个地址
	if (p == (void *)-1 )
	{
		perror("shmget err");
		return errno;
	}
	
	//父进程A，获取输入
	while(1)
	{
		//printf("取输入字符串或数字：\n");
		fgets(buffer, 63, stdin);
		i = 0;
		isstr = 0;
		buffer[strlen(buffer)-1] = '\0';
		
		if(strlen(buffer)==0)
		{
			printf("exit..\n");
			kill(p1,SIGKILL);
			exit(0);
		}
		//检查是字符串还是数字
		while((c = buffer[i++]) != '\0')
		{
			if(c < '0' || c > '9')
			{
				isstr = 1;		//若存在非数字字符则为字符串
				break;
			}
		}
		
		//向子进程发送带数据的信号
		union sigval sigvalue;
		
		if(isstr)
		{
			sigvalue.sival_int = SOLVESTR;
			stpcpy((char*)p,buffer);
			
			//发送信号
			if(sigqueue(p1,SIGUSR1,sigvalue)==-1)
			{
				printf("sigqueue() failed ! error message:%s\n",strerror(errno));
				exit(0);
			}
		}
		else
		{
			n = atoi(buffer);
			
			sigvalue.sival_int = SOLVENUM;
			*(int*)p = n;
			
			//发送信号
			if(sigqueue(p1,SIGUSR1,sigvalue)==-1)
			{
				printf("sigqueue() failed ! error message:%s\n",strerror(errno));
				exit(0);
			}
		}
		//sleep(2);
	}
}

return 0;

}