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


    基于tcp和epoll在线聊天室

    说明

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

    服务端代码

    #include 
    #include 
    #include 
    #include 
    #include 
    #include 
    #include 
    #include 
    #include 
    #include
    #include
    #include 
    #include 
    #include 
    
    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;
    }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28
    • 29
    • 30
    • 31
    • 32
    • 33
    • 34
    • 35
    • 36
    • 37
    • 38
    • 39
    • 40
    • 41
    • 42
    • 43
    • 44
    • 45
    • 46
    • 47
    • 48
    • 49
    • 50
    • 51
    • 52
    • 53
    • 54
    • 55
    • 56
    • 57
    • 58
    • 59
    • 60
    • 61
    • 62
    • 63
    • 64
    • 65
    • 66
    • 67
    • 68
    • 69
    • 70
    • 71
    • 72
    • 73
    • 74
    • 75
    • 76
    • 77
    • 78
    • 79
    • 80
    • 81
    • 82
    • 83
    • 84
    • 85
    • 86
    • 87
    • 88
    • 89
    • 90
    • 91
    • 92
    • 93
    • 94
    • 95
    • 96
    • 97
    • 98
    • 99
    • 100
    • 101
    • 102
    • 103
    • 104
    • 105
    • 106
    • 107
    • 108
    • 109
    • 110
    • 111
    • 112
    • 113
    • 114
    • 115
    • 116
    • 117
    • 118
    • 119
    • 120
    • 121
    • 122
    • 123
    • 124
    • 125
    • 126
    • 127
    • 128
    • 129
    • 130
    • 131
    • 132
    • 133
    • 134
    • 135
    • 136
    • 137
    • 138
    • 139
    • 140
    • 141
    • 142
    • 143
    • 144
    • 145
    • 146
    • 147
    • 148
    • 149
    • 150
    • 151
    • 152
    • 153
    • 154
    • 155
    • 156
    • 157
    • 158
    • 159
    • 160
    • 161
    • 162
    • 163
    • 164
    • 165
    • 166
    • 167
    • 168
    • 169
    • 170
    • 171
    • 172
    • 173
    • 174
    • 175
    • 176
    • 177
    • 178
    • 179
    • 180
    • 181
    • 182
    • 183
    • 184
    • 185
    • 186
    • 187
    • 188
  • 相关阅读:
    金仓数据库KingbaseES客户端应用参考手册--5. dropdb
    fcitx5 中文输入在 chrome/vscode 等应用中的问题及解决
    SpringCloudAlibaba 相关组件的学习一
    报错:AttributeError: module ‘tensorflow‘ has no attribute ‘flags‘
    python,满分,砝码称重【第十二届】【省赛】【研究生组】
    Redis Redis的数据结构 - 通用命令 - String类型命令 - Hash类型命令
    PaddleOCR ‘could not create a primitive descriptor for a reorder primitive‘异常处理
    全流量安全分析的重要性
    Python11-正则表达式
    禁用鼠标的侧边按键
  • 原文地址:https://blog.csdn.net/weixin_52604835/article/details/134561795