Linux C 基于tcp和epoll在线聊天室

说明

  服务端：实现了验证用户是否已经存在（支持最大64用户连接）支持广播用户进入退出聊天室以及用户聊天内容。
  这里只提供里服务端代码，如果想要看客户端代码点击这里。

服务端代码

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include<sys/types.h>
#include<sys/socket.h>
#include <arpa/inet.h>
#include <time.h>
#include <sys/epoll.h>

typedef struct sockaddr  SA;
typedef struct sockaddr_in  SIN;
#define MAXBACKLOG   100

int Socket(int domain,int type,int protocol);
int Bind(int sockfd,struct sockaddr * my_addr,int addrlen);
int Listen(int s,int backlog);
int Accept(int s,struct sockaddr * addr,int * addrlen);
int is_exist(char * username);
void broadcast(char *r,char *n);

char Userlist[64][20] = {0};
int Userfdlist[64] = {0};

//./app 192.168.5.166  8888
int main(int argc,char *argv[])
{	
	int opt = 1;
	//建立监听套接字
	int socketfd = Socket(AF_INET,SOCK_STREAM,0);
	//需要进行重用地址及其端口号
	setsockopt(socketfd,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));
	//绑定信息编写服务器信息
	SIN   serverinfo;
	serverinfo.sin_family =AF_INET;
	serverinfo.sin_port   =htons(atoi(argv[2])); 
	serverinfo.sin_addr.s_addr=  inet_addr(argv[1]);
	int addrlen = sizeof(SIN);
	Bind(socketfd,(SA*)&serverinfo,addrlen);
	//监听
	Listen(socketfd,MAXBACKLOG);
	//epoll创建根节点
	int epollfd = epoll_create(1024);
	
	//添加socketfd文件描述符至内核 红黑树
	struct epoll_event event;
	event.events = EPOLLIN;			//事件成员
	event.data.fd = socketfd;		//数据
	epoll_ctl(epollfd,EPOLL_CTL_ADD,socketfd, &event);
	//读写
	while(1)
	{
		struct epoll_event events[10];
		int count = epoll_wait(epollfd,events,10,-1);
		for(int i = 0; i< count;i++)
		{
			if(events[i].events == EPOLLIN)
			{
				if(events[i].data.fd == socketfd)
				{
					//wait client connect
					SIN clientinfo;
					struct epoll_event event;
					int  clientaddrlen =sizeof(SA);
					int newfd = Accept(socketfd,(SA*)&clientinfo,&clientaddrlen);
					printf("客户端地址:%s 端口号:%d\n",inet_ntoa(clientinfo.sin_addr),ntohs(clientinfo.sin_port));
					
					//read username
					char namebuff[20];
					read(newfd,namebuff,sizeof(namebuff));
					if(is_exist(namebuff))
					{
						write(newfd,"已存在",sizeof("已存在"));
						close(newfd);
					}
					else
					{
						//put newfd into Red-Black Tree
						event.events = EPOLLIN;			//事件成员
						event.data.fd = newfd;			//数据
						epoll_ctl(epollfd,EPOLL_CTL_ADD,newfd, &event);
						write(newfd,"登录成功",sizeof("登录成功"));	
						//save userfd & username
						for(int j=0;j<64;j++)
							if(strlen(Userlist[j])==0)
							{Userfdlist[j]=event.data.fd;strcpy(Userlist[j],namebuff);break;}	
						char r[50];
						sprintf(r,"%s %s",namebuff,"进入聊天室");
						printf("%s\n",r);
						broadcast(r,namebuff);
	
					}
				}
				else
				{
					//read
					char readbuff[512] = {0};
					int len = read(events[i].data.fd,readbuff,sizeof(readbuff));
					
					//get name and pos
					char namebuff[20] = {0};
					int key = 0;
					for(int i=0;i<10;i++)if(strstr(readbuff,Userlist[i])){strcpy(namebuff,Userlist[i]);key=i;break;}
					
					if(len > 0 && strlen(readbuff)!=0)
					{
						printf("%s\n",readbuff);
						broadcast(readbuff,namebuff);
						if(strstr(readbuff,"退出"))
						{
							strcpy(Userlist[key],"\0");
							Userfdlist[key] = 0;
							epoll_ctl(epollfd,EPOLL_CTL_DEL,events[i].data.fd,NULL);
							close(events[i].data.fd);
						}
					}
					else if(len <= 0)
					{
						epoll_ctl(epollfd,EPOLL_CTL_DEL,events[i].data.fd,NULL);
						close(events[i].data.fd);
					}
				}
			}
		}
	}
	//关闭
	close(socketfd);
	return 0;
}
int is_exist(char * username)
{
	for(int i = 0 ; i < 10; i++)
		if(strcmp(username,Userlist[i]) == 0)
			return 1;
	return 0;
}
void broadcast(char *r,char *n)
{
	for(int i=0 ; i<64 ;i++)
		//if it is a user and not himself
		if(strcmp(Userlist[i],n)!=0 && strlen(Userlist[i])!=0)
			write(Userfdlist[i],r,strlen(r));
}
int Socket(int domain,int type,int protocol)
{
	int socketFd = socket(domain,type,protocol);
	if(socketFd == -1)
	{
		perror("socket");
		exit(1);
	}
	return socketFd;
}
int Bind(int sockfd,struct sockaddr * my_addr,int addrlen)
{
	int val = bind(sockfd,my_addr,addrlen);
	if(val)
	{
		perror("bind");
		exit(1);
	}
	return 0;
}
int Listen(int s,int backlog)
{
	int val = listen(s,backlog);
	if(val == -1)
	{
		perror("listen");
		exit(1);
	}
	return val;
}
int Accept(int s,struct sockaddr * addr,int * addrlen)
{
	int NEWfd = accept(s,addr,addrlen);
	if(NEWfd == -1)
	{
		perror("listen");
		exit(1);
	}
	return NEWfd;
}