0812 练习

头文件：

#ifndef __CHAT_H__
#define __CHAT_H__
 
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <signal.h>
#define IP "192.168.31.117"

//打印错误新的宏函数
#define ERR_MSG(msg1)  do{\
	fprintf(stderr, " __%d__ ", __LINE__);\
	perror(msg1);\
}while(0)
 
 
//存储客户端地址信息的结构体说明
typedef struct Node
{
	struct sockaddr_in cin; 
    struct Node *next;         //指针域，记录下个结点的地址
}linkList;

//定义数据包的结构体
struct USER_MSG
{
	char type;    // 1.登录　２．聊天　３．下线
	char username[20];
	char text[516];
};

//遍历链表并且发送登录信息
void list_login(linkList*user,struct USER_MSG msg,int sfd);


//将客户端信息头插进链表
int list_insert_head(linkList *user,struct sockaddr_in cin);

//遍历并发送聊天信息
void list_msg(linkList*user,struct USER_MSG msg,int sfd,struct sockaddr_in cin);


//遍历并发送下线信息
void list_exit(linkList*user,struct USER_MSG msg,int sfd,struct sockaddr_in cin);


#endif

调用函数：

#include"chat.h"


//将客户端信息头插进链表
int list_insert_head(linkList *user,struct sockaddr_in cin)
{
	linkList *p=(linkList*)malloc(sizeof(linkList));
	if(NULL==p)
	{
		printf("节点申请失败\n");
		return -1;
	}
	p->cin=cin;
	p->next=NULL;
	//头插
	p->next=user->next;
	user->next=p;
	return 0;
}



//遍历链表并且发送登录信息
void list_login(linkList*user,struct USER_MSG msg,int sfd)
{
	linkList *q=user;
	sprintf(msg.text,"%s","登录成功");
	while(q->next!=NULL)	
	{
		q=q->next;
		if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&(q->cin),sizeof(q->cin))<0)
		{
			ERR_MSG("sendto");
			return;
		}
	}
}


//遍历并发送聊天信息
void list_msg(linkList*user,struct USER_MSG msg,int sfd,struct sockaddr_in cin)
{
	linkList*q=user;
	while(q->next!=NULL)
	{
		q=q->next;
		if((q->cin).sin_port==cin.sin_port)
		{
			if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&(q->cin),sizeof(q->cin))<0)
			{
				ERR_MSG("sendto");
				return;
			}
		}
	
	}
}

//遍历并发送下线信息
void list_exit(linkList*user,struct USER_MSG msg,int sfd,struct sockaddr_in cin)
{
	linkList*q=user;
	sprintf(msg.text,"%s","退出聊天");
	while(q->next!=NULL)
	{
		q=q->next;
		if((q->cin).sin_port==cin.sin_port)
		{
			if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&(q->cin),sizeof(q->cin))<0)
			{
				ERR_MSG("sendto");
				return;
			}		
		}
		else
		{
			
			if(q==user)
			{
				user=user->next;
				free(q);	
				q=NULL;
			}
			else 
			{
				linkList*p=q->next;
				q->cin=p->cin;
				q->next=p->next;
				free(p);
				p=NULL;
			}

		}

	}
}

客户端：

#include"chat.h"


int main(int argc, const char *argv[])
{
	
	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	
	//填充服务器的IP地址以及端口号
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(8888);
	sin.sin_addr.s_addr = inet_addr(IP);


	
	
	

	//用户登录
	struct USER_MSG msg;
	msg.type=1;
	printf("请输入用户名:");
	fgets(msg.username,20,stdin);
	msg.username[strlen(msg.username)-1]=0;
	
	//发送登录请求
	if(sendto(sfd,&msg,sizeof(msg),0,NULL,0)<0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	
	//创建父子进程
	pid_t pid=fork();
	if(0==pid)
	{
		//子进程负责发送请求
		while(1)
		{
			fgets(msg.text,516,stdin);
			msg.text[strlen(msg.text)-1]=0; //向数据包写入信息
			
			//判断是否下线
			if(strcasecmp(msg.text,"quit"))
			{
				msg.type=2;
				if(sendto(sfd,&msg,sizeof(msg),0,NULL,0)<0)
				{
					ERR_MSG("sendto");     //不下线执行聊天
					return -1;
				}
			}

			else
			{
				msg.type=3;
				exit(0);        //下线直接退出
			}
		}
	
	}
	else if(pid>0)
	{
		//父进程接收
		while(1)
		{
			bzero(&msg,sizeof(msg));
			if(recvfrom(sfd,&msg,sizeof(msg),0,NULL,NULL)<0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
			printf("%s:%s\n",msg.username,msg.text);
		}
	}
	else
	{
		ERR_MSG("fork");
		return -1;
	}
	close(sfd);
	return 0;
}

服务端：

#include"chat.h"


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

	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
 
 
	//填充服务器的IP地址以及端口号
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(8878);
	sin.sin_addr.s_addr = inet_addr(IP);
 
 
	//绑定服务器的地址信息结构体
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)        
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//定义客户端的网络信息结构体
	struct sockaddr_in cin;
	socklen_t addrlen=sizeof(cin);
	//创建父子进程
	pid_t pid=fork();
	if(pid>0)   //父进程负责接收客户端的请求
	{  
		//创建存储客户端信息的链表						
		linkList* user=(linkList*)malloc(sizeof(linkList));
		if(NULL==user)
		{
			printf("创建失败\n");
			return -1;
		}

		user->next=NULL;
		printf("create success\n");
		struct USER_MSG msg;
		while(1)
		{
			bzero(&msg,sizeof(msg));
			if(recvfrom(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&cin,&addrlen)<0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
			if(msg.type==1)
			{
				list_login(user,msg,sfd);
				list_insert_head(user,cin);
			}
			else if(msg.type==2)
			{
				list_msg(user,msg,sfd,cin);
			}
			else if(msg.type==3)
			{
				list_exit(user,msg,sfd,cin);
			}
		}
	
	}	
	
	return 0;

}