嵌入式学习

作业

        使用消息队列实现的2个终端之间的互相聊天。
        并使用信号控制消息队列的读取方式：当键盘按ctrl+c的时候，切换消息读取方式，一般情况为读取指定编号的消息，按ctrl+c之后，指定的编号不读取，读取其他所有编号的消息。

进程1 

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <wait.h>
#include <signal.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <semaphore.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/un.h>

struct msgbuf {
	long type;
	char buf[128];
};

int count=0;//用来记录切换的次数
void handler(int signum)
{
	if(signum == SIGINT)
	{
		count++;
		if(count/2 != 0)
		{
			printf("切换消息读取方式\n");
		}else if(count/2 == 0)
		{
			printf("切换为原来的消息读取方式\n");
		}
	}
	return;
}

void* run(void *arg)
{
	key_t key_02=ftok("./ipc",2);
	if(-1 == key_02)
	{
		perror("ftok_p");
		return NULL;
	}

	int id=msgget(key_02,IPC_CREAT | 0666);
	if(-1 == id)
	{
		perror("msgget_p");
		return NULL;
	}

	struct msgbuf msg;
	int size=0;
	signal(SIGINT,handler);
	if(count/2 != 0 )
	{
		while(1)
		{
			memset(&msg,0,sizeof(msg));
			msgrcv(id,&msg,128,2,020000);
			printf("读取到的消息为：%s\n",msg.buf);
		}
	}else if(count/2 == 0)
	{	
		while(1)
		{
			memset(&msg,0,sizeof(msg));
			msgrcv(id,&msg,128,2,0);
			printf("读取到的消息为：%s\n",msg.buf);
		}
	}
}

int main(int argc, const char *argv[])
{
	pthread_t pid;
	int retval=pthread_create(&pid,0,run,0);
	if(retval != 0)
	{
		perror("pthread_create");
		return 1;
	}
	pthread_detach(pid);

	//生成秘钥
	key_t key_01=ftok("./ipc",1);
	if(-1 == key_01)
	{
		perror("ftok");
		return 1;
	}

	//根据秘钥创建消息队列/访问消息队列
	int id=msgget(key_01,IPC_CREAT | 0666);
	if(-1 == id)
	{
		perror("msgget");
		return 1;
	}

	struct msgbuf msg;
	int size=0;
	signal(SIGALRM,SIG_IGN);
	while(1)
	{
		memset(&msg,0,sizeof(msg));
		msg.type=1;
		printf("请输入：");
		scanf("%128s",msg.buf);
		while(getchar()!=10);
		size=strlen(msg.buf);
		msgsnd(id,&msg,size,0);
	}

	return 0;
}

进程2

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <wait.h>
#include <signal.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <semaphore.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/un.h>

struct msgbuf {
	long type;
	char buf[128];
};

int count=0;//用来记录切换的次数
void handler(int signum)
{
    if(signum == SIGINT)
    {
        count++;
        if(count/2 != 0)
        {
            printf("切换消息读取方式\n");
        }else if(count/2 == 0)
        {
            printf("切换为原来的消息读取方式\n");
        }
    }
    return;
}

void* run(void *arg)
{
	key_t key_02=ftok("./ipc",2);
	if(-1 == key_02)
	{
		perror("ftok");
		return NULL;
	}

	int id=msgget(key_02,IPC_CREAT | 0666);
	if(-1 == id)
	{
		perror("msgget_p");
		return NULL;
	}

	struct msgbuf msg;
	int size=0;
	signal(SIGALRM,SIG_IGN);
	while(1)
	{
		memset(&msg,0,sizeof(msg));
		msg.type=2;
		printf("请输入：");
		scanf("%128s",msg.buf);
		while(getchar()!=10);
		size=strlen(msg.buf);
		msgsnd(id,&msg,size,0);
	}
}

int main(int argc, const char *argv[])
{
	pthread_t pid;
	int retval=pthread_create(&pid,0,run,0);
	if(-1 == retval)
	{
		perror("pthread_create");
		return 1;
	}
	pthread_detach(pid);
	key_t key_01=ftok("./ipc",1);
	if(-1 == key_01)
	{
		perror("ftok");
		return 1;
	}
	
	int id=msgget(key_01,IPC_CREAT | 0666);
	if(-1 == id)
	{
		perror("msgget");
		return 1;
	}

	struct msgbuf msg;
	int size=0;
	signal(SIGINT,handler);
	if(count/2 != 0)
	{
		while(1)
		{
			memset(&msg,0,sizeof(msg));
			msgrcv(id,&msg,128,1,020000);
			printf("读取到的消息为：%s\n",msg.buf);
		}
	}else if(count/2 == 0)
	{
		while(1)
		{
			memset(&msg,0,sizeof(msg));
			msgrcv(id,&msg,128,1,0);
			printf("读取到的消息为：%s\n",msg.buf);
		}
	}
	return 0;
}