条件编译 消息队列 实现的聊天程序

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

typedef struct _msg
{
	long mtype;
	char mtext[50];
	char sender[30];
}MSG;


#ifdef CC
char sender[]="CC";
#endif

#ifdef MY
char sender[]="MY";
#endif

#ifdef CCMY
char sender[]="CCMY";
#endif

int main(int argc, char *argv[])
{
	key_t key;
	int msgqid;
	MSG msg;

	key = ftok(".",2012);//以文件路径 跟一个整数值（只取低四位） 来生成一个key 两边要相同
	msgqid = msgget(key,IPC_CREAT); //消息队列id  用key跟创建方式生成 
	if(msgqid == -1)
	{
		perror("msgget");
		exit(-1);
	}

#ifdef CC
	printf("hello, I am CC!\n");
#endif
	
#ifdef MY
	printf("hello, I am MY!\n");
#endif
	
#ifdef CCMY
	printf("hello, I am CCMY!\n");
#endif
	
	pid_t pid;
	
	pid = fork();
	if(pid==0)
	{
		while(1)//接受用户输入 往消息队列里写
		{
			char bufw[256] = "";
			
			printf("%s:", sender);
			fflush(stdout);
			bzero(bufw, sizeof(bufw));
			if(fgets(bufw, sizeof(bufw), stdin) == NULL)
				exit(-1);
			bufw[strlen(bufw)-1] = '\0';
			if(bufw[0]=='C')//判断输入的首字母 标示
				msg.mtype = 8;
			else if(bufw[0]=='M')
				msg.mtype = 9;
			else
				msg.mtype = 10;
			
		//	printf("----------1-------------\n");
			
		//	printf("bufw[0] is %c ,msg.mtype is %d bufw+2 is %s \n",bufw[0],msg.mtype,bufw+2);
			strcpy(msg.mtext,bufw+2);//后移两位开始才是真正的发送内容
			strcpy(msg.sender,sender);
	
			msgsnd(msgqid,&msg,(sizeof(msg.mtext)+sizeof(msg.sender)),0);
			
			
		}
	}
	else
	{
		while(1)//从消息队列里读
		{
			char bufr[256] = "";

		//	long recno = 0;//接收号   CC8 MY9 CCMY10
			
		
			#ifdef CC
			long	recno = 8;
			#endif
				
			#ifdef MY
			long	recno = 9;
			#endif
				
			#ifdef CCMY
			long	recno = 10;
			#endif
			

		//	printf("recno is %d \n",recno);
			int retu;
			retu = msgrcv(msgqid,&msg,(sizeof(msg.mtext)+sizeof(msg.sender)),recno,0);//参数含义
			//消息队列id   消息结构体指针  传递数据大小（已去除long mtype长度 如有多个 如有多个sizeof+sizeof。。。） 消息id 
			//最后一个0  对于读来说：如果队列为空 堵塞等待 直到有数据可读  
			//           对于写来说：如果队列已满 堵塞等待 直到有空间可写
		//	if(retu>0)
			printf("\rmessage from %s: msg.mtext = %s \n",msg.sender,msg.mtext);
		//	msgctl(msgqid,IPC_RMID,NULL);//删除队列
			printf("%s:", sender);
			fflush(stdout);
		}
	}
	return 0;
}



all: 
	gcc threechat.c -o cc -DCC
	gcc threechat.c -o my -DMY
	gcc threechat.c -o ccmy -DCCMY
clean:
	rm cc my ccmy -f