基于UDP的网络聊天室

项目需求：

1. 如果有用户登录，其他用户可以收到这个人的登录信息
2. 如果有人发送信息，其他用户可以收到这个人的群聊信息
3. 如果有人下线，其他用户可以收到这个人的下线信息
4. 服务器可以发送系统信息
    

流程图：

 

 服务器代码：

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>                                               
#include <stdlib.h>

typedef char datatype;
#define ERR_MSG(msg) do{\
	fprintf(stderr, "line:%d ", __LINE__);\
	perror(msg);\
}while(0)

#define PORT 6666               //1024~49151
#define IP  "192.168.31.126"    //IP地址，本机IP ifconfig 
struct cli_msg
{
	char type; 		//L C Q
	char name[20];
	char text[128];
};

typedef struct node
{
	union{
		int len;
		struct sockaddr_in sin;
	};
	struct node *next;
}*Linklist,Node;

Linklist LinklistCreateHead()
{
	Linklist L = (Linklist)malloc(sizeof(Node));
	if(L == NULL)
		return NULL;
	L->len = 0;
	L->next = NULL;
	return L;
}

Linklist LinklistCreateNode()
{
	Linklist p = (Linklist)malloc(sizeof(Node));
	if(p == NULL)
	{
		return NULL;
	}
	p->next = NULL;
	return p;
}


/*void LinklistShow(Linklist L)
  {
  while(L->next)
  {
  L=L->next;
  printf("%c\t",L->sin.name);
  }

  }*/
int do_login(Linklist L,struct cli_msg rcvbuf,int sfd,Linklist p);
int do_chat(Linklist L,Linklist p,struct cli_msg rcvbuf,int sfd);
//int do_quit(Linklist L,Linklist p,struct cli_msg rcvbuf,int sfd);
int main(int argc, const char *argv[])
{

	Linklist L=LinklistCreateHead();
	if(L==NULL)
		return -1;
	struct cli_msg rcvbuf;

	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success  sfd=%d __%d__\n", sfd, __LINE__);




	//填充服务器自身的地址信息结构体 AF_INET : man 7 IP
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET; 	//必须填AF_INET;
	sin.sin_port 		= htons(PORT); 	//服务器绑定的端口，网络字节序
	sin.sin_addr.s_addr = inet_addr(IP); 	//服务器绑定的IP，本机IP ifconfig

	//绑定--->必须绑定
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n", __LINE__);

	char buf[128] = "";
	ssize_t res = 0;

	struct sockaddr_in cin; 	//存储数据包从谁哪里过来



	pid_t cpid = fork();

	if(cpid > 0)
	{
		//接收数据
		struct cli_msg rcvbuf;
		while(1)
		{
			Linklist p = LinklistCreateNode();
			if(p == NULL)
				return -1;
			socklen_t addrlen = sizeof(p->sin);

			bzero(rcvbuf.name,sizeof(rcvbuf.name));
			bzero(rcvbuf.text,sizeof(rcvbuf.text));

			res = recvfrom(sfd, &rcvbuf, sizeof(rcvbuf), 0, (struct sockaddr*)&(p->sin), &addrlen);
			if(res < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
			printf("type = %c\n", rcvbuf.type);
			switch(rcvbuf.type)
			{
			case 'L': 	
				do_login(L,rcvbuf,sfd,p);
				break;
			case 'C':
				do_chat(L,p,rcvbuf,sfd);
				break;
				/*			case 'Q':
							do_quit(L,p,rcvbuf,sfd);
							break;*/
			default:
				printf("协议 %c 错误 __%s__ __%d__\n", rcvbuf.type, __FILE__, __LINE__);
			}

		}
	}
	else if(0 == cpid)
	{
		//发送数据
		while(1)

		{

			rcvbuf.type = 'C';
			bzero(rcvbuf.name,sizeof(rcvbuf.name));
			strcpy(rcvbuf.name,"system");
			bzero(rcvbuf.text,sizeof(rcvbuf.text));
			printf("请输入:");
			fgets(rcvbuf.text,sizeof(rcvbuf.text),stdin);
			rcvbuf.text[strlen(rcvbuf.text) - 1] = 0;
			//服务器收到的消息转发给所有人
			if(sendto(sfd,&rcvbuf,sizeof(rcvbuf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				return -1;
			}
			printf("sendto success\n");

		}
	}


	else 
	{
		ERR_MSG("fork");
		return -1;
	}


	//关闭所有文件描述符
	close(sfd);
	return 0;
}


int do_login(Linklist L,struct cli_msg rcvbuf,int sfd,Linklist p)
{
	//转发登录信息
	Linklist q = L;
	while(q->next != NULL)
	{
		q=q->next;
	}
	q->next = p;
	L->len++;
	printf("%s:登录成功\n",rcvbuf.name);
	Linklist s = L;
	while(s->next != NULL)
	{
		s=s->next;
		if(ntohs((s->sin).sin_port)!=ntohs((p->sin).sin_port))
		{
			if(sendto(sfd,&rcvbuf,sizeof(rcvbuf),\
						0,(struct sockaddr*)&(s->sin),sizeof(s->sin)) < 0)
			{
				ERR_MSG("sendto");
				return -1;
			}
		}
	}
	//将当前登录的用户的地址信息结构体存储起来
	return 0;
}

int do_chat(Linklist L,Linklist p,struct cli_msg rcvbuf,int sfd)
{
	//转发下线信息
	Linklist s = L;
	if(strcmp(rcvbuf.text,"quit")==0)
	{
		rcvbuf.type = 'Q';
		printf("%s [%s:%d]:quit\n",rcvbuf.name,inet_ntoa((p->sin).sin_addr),ntohs((p->sin).sin_port));
		while(s->next != NULL)
		{
			s=s->next;
			if(ntohs((s->sin).sin_port)!=ntohs((p->sin).sin_port))
			{
				if(sendto(sfd,&rcvbuf,sizeof(rcvbuf),\
							0,(struct sockaddr*)&(s->sin),sizeof(s->sin)) < 0)
				{
					ERR_MSG("sendto");
					return -1;
				}
			}
		}
	}
	//将当前下线客户端的地址信息结构体从存储位置删除
	//转发群聊信息
	printf("%s [%s:%d]:%s \n",rcvbuf.name,inet_ntoa((p->sin).sin_addr),ntohs((p->sin).sin_port),rcvbuf.text);
	while(s->next!=NULL)
	{
		s=s->next;
		if(ntohs((s->sin).sin_port)!=ntohs((p->sin).sin_port))
		{
			if(sendto(sfd,&rcvbuf,sizeof(rcvbuf),0,(struct sockaddr*)&(s->sin),sizeof(s->sin)) < 0)
			{
				ERR_MSG("sendto");
				return -1;
			}		
		}
	}
	return 0;
}

/*int do_quit(Linklist L,Linklist p,struct cli_msg rcvbuf,int sfd)
  {
//转发下线信息
Linklist s = L;
if(strcmp(rcvbuf.text,"quit")==0)
{	
printf("%s [%s:%d]:quit\n",rcvbuf.name,inet_ntoa((p->sin).sin_addr),ntohs((p->sin).sin_port));
while(s->next!=NULL);
{
s=s->next;
if(ntohs((s->sin).sin_port)!=ntohs((p->sin).sin_port))
{
if(sendto(sfd,&rcvbuf,sizeof(rcvbuf),0,(struct sockaddr*)&(s->sin),sizeof(s->sin)) < 0)
{
ERR_MSG("sendto");
return -1;
}
}
}
}
//将当前下线客户端的地址信息结构体从存储位置删除
return 0;
}*/

 客户端代码

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h> 
#include <stdlib.h>
#include <semaphore.h>
#include <signal.h>

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

#define PORT 6666               //1024~49151
#define IP  "192.168.31.126"    //IP地址，本机IP ifconfig 

struct cli_msg
{
	char type;      //L C Q
	char name[20];
	char text[128];
};


int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success  sfd=%d __%d__\n", sfd, __LINE__);
	//填充服务器的地址信息结构体
	//供给下面的sendto使用，
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET; 	//必须填AF_INET;
	sin.sin_port 		= htons(PORT); 	//服务器绑定的端口，网络字节序
	sin.sin_addr.s_addr = inet_addr(IP); 	//服务器绑定的IP，本机IP ifconfig 

	printf("请输入名字>>>");
	char name[20] = "";
	scanf("%s", name);
	getchar();

	struct cli_msg sndbuf;
	sndbuf.type = 'L';
	strcpy(sndbuf.name, name);
	//发送登录请求

	if(sendto(sfd,&sndbuf,sizeof(sndbuf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("--------登陆成功------------\n");

	pid_t cpid = fork();

	if(0 == cpid)
	{
		//接收数据
		while(1)
		{
			bzero(sndbuf.text,sizeof(sndbuf.text));
			if(recvfrom(sfd,&sndbuf,sizeof(sndbuf),0,NULL,NULL) < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
			if('L' == sndbuf.type)
			{
				printf("------------%s上线-----------\n",sndbuf.name);

			}
			else if('Q' == sndbuf.type)
			{
				printf("------------%s下线-----------\n",sndbuf.name);
			}
			else if('C' == sndbuf.type)
			{
				printf("%s:%s\n",sndbuf.name,sndbuf.text);
			}
				
		}
	}
	if(cpid > 0)
	{
		//发送数据

		while(1)
		{
			bzero(sndbuf.text,sizeof(sndbuf.text));
			//printf("请输入>>>");
			fgets(sndbuf.text,sizeof(sndbuf.text),stdin);
			sndbuf.text[strlen(sndbuf.text)-1] = 0;
			sndbuf.type = 'C';
			strcpy(sndbuf.name,name);
			if(sendto(sfd,&sndbuf,sizeof(sndbuf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				break;
			}
			//printf("sendto success\n");	
			if(strcmp(sndbuf.text,"quit")==0)
				exit(0);
		}
	}
	else
	{
		ERR_MSG("fork");
		return -1;
	}

	//关闭所有文件描述符
	close(sfd);
	return 0;
}

输出结果：