ネットワークプログラミング-6日目

することが

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0);

#define PORT 8888           //1024-49151
#define IP "192.168.114.110"   //本机IP ifconfig
 
//用户信息结构体
typedef struct Node
{
	struct sockaddr_in cin;
	struct Node *next;
}Linklist;

struct _msg
{
	char type;  //登录L,群聊C,退出Q
	char name[20];
	char text[128];
};
 
//登录信息更新
int  do_login(int sfd, struct _msg msg, Linklist* L, struct sockaddr_in cin)
{
	printf("[%s:%d] %s登录成功\n",(char *)inet_ntoa(cin.sin_addr),ntohs(cin.sin_port), msg.name);
	sprintf(msg.text,"-----%s登录成功-----",msg.name);
	//先遍历链表,将新用户加入群聊的消息发送给所有人
	while(L->next != NULL)
	{
		L = L->next;
		if(sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&(L->cin), sizeof(L->cin))<0)
		{
			ERR_MSG("sendto");
			return -1 ;
		}
	}
	
	//将新用户的网络信息结构体,头插入链表
	Linklist* temp = (Linklist*)malloc(sizeof(Linklist));
	temp->cin = cin;
	temp->next = NULL;
	L->next = temp;
	return 0 ;
}

//接收并打印群聊
int do_chat(int sfd, struct _msg msg, Linklist* L, struct sockaddr_in cin)
{
	printf("%s[%s:%d]chat成功\n", msg.name,(char*)inet_ntoa(cin.sin_addr),\
			ntohs(cin.sin_port));
 
	//遍历链表,将群聊发给除自己以外的所有人
	while(L->next != NULL)
	{
		L = L->next;
		if(memcmp(&cin,&L->cin,sizeof(cin)) != 0)    //判断是是否为自己
		{
			if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&(L->cin),sizeof(L->cin))<0)
			{
				perror("sendto");
				return -1;
			}
		}
	}
	return 0;
 
}
 
//退出下线操作
int do_quit(int sfd,struct _msg msg,Linklist* L,struct sockaddr_in cin)
{
	//遍历链表,是自己就把自己在链表中删除,不是自己,就发生退出群聊的数据
	while(L->next!=NULL)
	{
		if(memcmp(&L->next->cin,&cin,sizeof(cin))!=0)
		{
			L = L->next;
			if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&L->cin,sizeof(L->cin))<0)
			{
				perror("sendto");
				return -1;
			}
		}
		else
		{
			Linklist * ptemp = L->next;
			L->next=ptemp->next;
			free(ptemp);
			ptemp = NULL;
		}
	}
	return 0 ;
}

int do_recv(int sfd,struct _msg msg,Linklist *L)
{
	struct sockaddr_in cin; //存储连接成功的客户端地址信息
	socklen_t addrlen=sizeof(cin);
 
	while(1)
	{
		//接收消息
		memset(&msg,0,sizeof(msg));
		memset(&cin,0,sizeof(cin));
		int res = 0;
		res = recvfrom(sfd,&msg,sizeof(msg),0,(struct sockaddr *)&cin,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		//提取出协议
		switch(msg.type)
		{
		case 'L':
			do_login(sfd,msg,L,cin);
			break;
		case 'C':
			do_chat(sfd,msg,L,cin);
			break;
		case 'Q':
			do_quit(sfd,msg,L,cin);
			break;
		}
 
	}
	
	return 0;
}


int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("sfd = %d\n", sfd);
 
	//填充地址信息结构体
	//真实的地址信息结构体根据地址族指定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;        //必须填AF_INEF
	sin.sin_port   = htons(PORT);     //端口号,1024-49151
	sin.sin_addr.s_addr = inet_addr(IP); //本机IP
 
	//将IP和端口绑定到套接字上
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
		
 
	ssize_t res = 0;
	struct _msg msg;   //定义接收信息的变量
	
	//创建子进程
	int cpid = 0;
	cpid= fork();
	if(cpid < 0)
	{
		ERR_MSG("fork");
		return -1;
	}
	//父进程接受recvfrom
	if(cpid > 0)
	{
		Linklist* L = (Linklist*)malloc(sizeof(Linklist));
		L->next = NULL;
		do_recv(sfd, msg, L);
 
	}
	//子进程做sendto
	else if(0 == cpid)
	{
		//子进程,用来发送系统消息
		strcpy(msg.name,"系统信息");
      	msg.type = 'C';
		while(1)
		{
			bzero(msg.text, sizeof(msg.text));
			fgets(msg.text, sizeof(msg.text), stdin);
			msg.text[strlen(msg.text)-1]='\0';
			if(sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&sin, sizeof(sin))<0)
			{
				ERR_MSG("sendto");
				return -1;
			}
		}
		printf("系统消息发送成功\n");
	}
 
	close(sfd);
	return 0;
}

クリ

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"line:%d ",__LINE__);\
	perror(msg);\
}while(0);
#define SER_IP "192.168.114.110"   //本机IP ifconfig
#define SER_PORT 6665           //1024-49151
 
struct node
{
	char type;   //协议'L'登录,'C'发送群聊,'Q'下线
	char name[32];
	char text[128];
};

void handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0);
	exit(0);
}
int main(int argc, const char *argv[])
{
	__sighandler_t s=signal(SIGCHLD,handler);
	if(s == SIG_ERR)
	{
		perror("signal");
		return -1;
	}
	
	//创建报式套接字
	int cfd = socket(AF_INET,SOCK_DGRAM,0);
	if(cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("cfd = %d\n",cfd);
 
	//填充地址信息结构体
	//真实的地址信息结构体根据地址族指定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;        //必须填AF_INEF
	sin.sin_port   =htons(SER_PORT);     //端口号,1024-49151
	sin.sin_addr.s_addr =inet_addr(SER_IP); //本机IP
	
	struct node argee;
	//清空
	memset(&argee,0,sizeof(argee));

	//登录
	printf("请输入登录用户>>>");
	fgets(argee.name,sizeof(argee.name),stdin);
	argee.name[strlen(argee.name)-1]='\0';
	
	argee.type = 'L';
	strcpy(argee.text,"加入群聊");
	if(sendto(cfd,&argee,sizeof(argee),0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		return -1;
	}
 
	int cpid=0;
	cpid = fork();
	if(cpid <0)
	{
		perror("fork");
		return -1;
	}

	//父进程接收recvfrom
	if(cpid > 0)
	{
		while(1)
		{
			if(recvfrom(cfd, &argee, sizeof(argee), 0, NULL, NULL)<0)
			{
				perror("recvfrom");
				return -1;
			}
			printf("[%s]:%s\n",argee.name,argee.text);
		}
		
	}
	
	//子进程发送sendto
	else if(cpid == 0)
	{
		while(1)
		{
			bzero(argee.text,sizeof(argee.text));
			//发送数据包
			fgets(argee.text,sizeof(argee.text),stdin);
			argee.text[strlen(argee.text)-1]='\0';

			//退出群聊
			if(!strcmp(argee.text,"quit"))
			{
				argee.type='Q';
				strcpy(argee.text,"退出群聊");
			}
			else
			{
				//加入群聊
				argee.type='C';
			}

			if(sendto(cfd,&argee,sizeof(argee),0,(struct sockaddr*)&sin,sizeof(sin))<0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			if(!strcmp(argee.text,"退出群聊"))
			{
				break;
			}
 
		}
 
	}

	close(cfd);
	return 0;
}

マインドマッピング: