网络聊天室

服务器

#include <myhead.h>

//定义存放客户端信息的结构体
typedef struct msg
{
	char type; //L登录,Q退出,C聊天
	char name[20];
	char text[128];
}msg_t;

int main(int argc, const char *argv[])
{
	//1、创建用于通信的套接字
	int sfd=socket(AF_INET,SOCK_DGRAM,0);
	if(sfd==-1)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success\n");

	//2、绑定
	//2.1填充服务器地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(atoi(argv[2]));
	sin.sin_addr.s_addr=inet_addr(argv[1]);
	//2.2绑定服务器
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");

	//定义客户端信息结构体
	struct sockaddr_in cin;
	struct sockaddr_in save[64];
	cin.sin_family=AF_INET;
	socklen_t socklen=sizeof(cin);

	for(int i=0;i<64;i++)
	{
		save[i].sin_family=AF_INET;
	}


	msg_t msg[128];
	msg_t cli;

	//定义一个用于检测文件描述符的集合
	fd_set readfds,tempfds;

	//清空容器中的内容
	FD_ZERO(&readfds);
	//将要检测的文件描述符放入集合中
	FD_SET(sfd,&readfds); 	//将sfd文件描述符放入
	FD_SET(0,&readfds); 	//将0号文件描述符放入

	//定义一个容器
	char buf[128]="";
	int res=0; 	//接收select的返回值

	int maxi=-1;

	while(1)
	{
		//复制一份
		tempfds=readfds;

		//使用select阻塞等待集合中的文件描述符有事件产生
		res=select(sfd+1,&tempfds,NULL,NULL,NULL);
		if(res==-1)
		{
			perror("select error");
			return -1;
		}
		else if(res==0)
		{
			printf("time out\n");
			return -1;
		}

		//判断sfd是否还在集合中
		if(FD_ISSET(sfd,&tempfds))
		{
			recvfrom(sfd,&cli,sizeof(cli),0,(struct sockaddr*)&cin,&socklen);
			switch (cli.type)
			{
				case 'L':
					{
						//登录
						printf("%s[%s:%d]:登陆成功\n",cli.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
						maxi++;
						strcpy(msg[maxi].name,cli.name);
						save[maxi].sin_port=cin.sin_port;
						save[maxi].sin_addr.s_addr=cin.sin_addr.s_addr;

						for(int i=0;i<=maxi;i++)
						{
							if(cin.sin_port==save[i].sin_port)
							{
								continue;
							}
							sendto(sfd,&cli,sizeof(cli),0,(struct sockaddr *)&save[i],sizeof(save[i]));  //将登陆的消息发送给除自己外的其他用户
						}
					}
					break;
				case 'Q':
					{

						//登录
						int j=0;
						//退出
						while(1)
						{
							int t=j;
							while(1)
							{
								if(cin.sin_port==save[j].sin_port)
								{
									strcpy(msg[t].name,msg[t+1].name);
									save[t].sin_port=cin.sin_port;
									save[t].sin_addr.s_addr=cin.sin_addr.s_addr;

									t++;

									if(t==maxi-1)
									{
										break;
									}
								}
							}
							break;
							j++; //找到退出的用户的下标
						}

						for(int i=0;i<=maxi;i++)
						{
							if(cin.sin_port==save[i].sin_port)
							{
								continue;
							}

							maxi--;	
							sendto(sfd,&cli,sizeof(cli),0,(struct sockaddr *)&save[i],sizeof(save[i]));  //将信息发送给除退出用户外的其他用户
						}

					}
					break;
				case 'C':
					{

						if(strcmp(cli.text,"quit")==0)
						{
						printf("%s[%s:%d]:退出群聊\n",cli.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
						
						}
						if(strcmp(cli.text,"quit")!=0)
						{
							//登录
							printf("%s发送了一条消息\n",cli.name);
						}
						//聊天
						for(int i=0;i<=maxi;i++)
						{

							if(cin.sin_port==save[i].sin_port)
							{
								continue;
							}
							if(strcmp(cli.text,"quit")==0)
							{
								cli.type='Q';
							}
							sendto(sfd,&cli,sizeof(cli),0,(struct sockaddr *)&save[i],sizeof(save[i]));
						}
					}

					break;
				default:printf("您的协议输入有误，请重新输入!!!\n");break;

			}
		}
		if(FD_ISSET(0,&tempfds))
		{
			//终端输入
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]='\0';

			if(strcmp(buf,"quit")==0)
			{
				close(sfd);
				return 0;
			}

			cli.type='C';
			strcpy(cli.text,buf);
			strcpy(cli.name,"***system***");

			for(int i=0;i<=maxi;i++)
			{
				sendto(sfd,&cli,sizeof(cli),0,(struct sockaddr *)&save[i],sizeof(save[i]));
			}
		}

	}


	//4、关闭文件描述符
	close(sfd);

	return 0;
}

客户端

#include <myhead.h>

//定义用户信息的结构体
typedef struct cin
{
	char type;
	char name[20];
	char text[128];
}cin_t;

int main(int argc, const char *argv[])
{
	if(argc!=3)
	{
		printf("输入错误,请输入IP地址和端口号\n");
	}
	//1、创建用于通信的套接字
	int cfd=socket(AF_INET,SOCK_DGRAM,0);
	if(cfd==-1)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success\n");

	//2、绑定（可选）


	//填充服务器地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(atoi(argv[2]));
	sin.sin_addr.s_addr=inet_addr(argv[1]);
	socklen_t socklen=sizeof(sin);

	char buf[128]="";
	char rbuf[128]="";
	char name[20]="";

	//定义客户端地址信息结构体
	cin_t cin;
	cin_t save;
	//清空内容
	bzero(cin.name,sizeof(cin.name));
	bzero(cin.text,sizeof(cin.text));

	bzero(buf,sizeof(buf));
	bzero(rbuf,sizeof(rbuf));
	bzero(name,sizeof(name));

	//登录
	printf("请输入姓名>>>");
	scanf("%s",name);
	getchar();

	//比较协议
	strcpy(cin.name,name);
	cin.type='L';
	sendto(cfd,&cin,sizeof(cin),0,(struct sockaddr*)&sin,sizeof(sin));

	//定义一个用于检测文件描述符的集合
	fd_set readfds,tempfds;

	//清空容器中的内容
	FD_ZERO(&tempfds);
	FD_ZERO(&readfds);
	//将要检测的文件描述符放到集合中
	FD_SET(cfd,&readfds); 	//将cfd文件描述符放入
	FD_SET(0,&readfds); 	//将0号文件描述符放入

	int res=0; 	//用来接收select 的返回值

	while(1)
	{
		tempfds=readfds;
		res=select(cfd+1,&tempfds,NULL,NULL,NULL);
		if(res==-1)
		{
			perror("select error");
			return -1;
		}
		else if(res==0)
		{
			printf("time out\n");
			return -1;
		}

		//判断当前i是否在集合中，如果不在，则直接判断下一个
		if(FD_ISSET(cfd,&tempfds))
		{
			bzero(buf,sizeof(buf));
			bzero(rbuf,sizeof(rbuf));
			//接收从服务器发来的消息
			recvfrom(cfd,&save,sizeof(save),0,NULL,NULL);

			if(save.type=='L')
			{
				printf("------您的好友%s登陆成功------\n",save.name);
			}
			if(save.type=='Q')
			{
				printf("------您的好友%s已经下线------\n",save.name);
			}
			if(save.type=='C')
			{
				strcpy(buf,save.text);
				strcpy(rbuf,save.name);
				printf("%s:%s\n",rbuf,buf);
				if(strcmp(save.name,"***system***"))
				{
					strcpy(save.name,name);
				}
			}
		}
		if(FD_ISSET(0,&tempfds))
		{
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]='\0';

			if(strcmp(buf,"quit")==0)
			{
				save.type='C';

				strcpy(save.name,name);
				strcpy(save.text,"quit");
				sendto(cfd,&save,sizeof(save),0,(struct sockaddr*)&sin,sizeof(sin));
				close(cfd);
				return 0;	
			}
			else
			{
				strcpy(save.text,buf);
				strcpy(save.name,name);
				save.type='C';
				sendto(cfd,&save,sizeof(save),0,(struct sockaddr*)&sin,sizeof(sin));
			}
		}
	}
	//关闭文件描述符
	close(cfd);



	return 0;
}