Linux -- TCP Socket_Chat

我不是女神ヾ 2021-10-29 16:12 236阅读 0赞

### 1、服务端源代码 ###

//==================================
    // FileName : tcp_server.c
    // Date : 2019/08/14
    // Author : 柚子树
    //==================================
    
    #include <signal.h>
    #include <sys/select.h>
    #include <sys/socket.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include <stdio.h>
    #include <netdb.h>
    #include <string.h>
    #include <sys/types.h>
    #include <unistd.h>
    #include <stdlib.h>
    
    int main(int argc,char* argv[])
    { 
    	if(argc!=3)
    	{ 
    		printf("error args!请输入命令行参数ip和port:\n");
    		return -1;
    	}
    	
    	int sfd;
    	sfd=socket(AF_INET,SOCK_STREAM,0); //服务器端生成一个socket描述符
    	if(-1==sfd)
    	{ 
    		perror("socket");
    		return -1;
    	}
    	struct sockaddr_in ser; //该结构体变量存储ip、port等socket信息数据
    	memset(&ser,0,sizeof(ser)); //清空结构体
    	ser.sin_family=AF_INET; //采用ipv4协议
    	ser.sin_port=htons(atoi(argv[2])); //端口号，将主机字节序的port转换为网络字节序的port
    	ser.sin_addr.s_addr=inet_addr(argv[1]); //ip，将点分十进制ip转换为网络字节序的32位二进制数值
    	int ret;
    	ret=bind(sfd,(struct sockaddr*)&ser,sizeof(ser));//将socket描述符与服务器的ip、port绑定
    
    	if(-1==ret)
    	{ 
    		perror("bind");
    		return -1;
    	}
    	
    	listen(sfd,10);
    	//使服务器的这个端口和ip处于监听状态，等待客户机的连接请求;10:同时能处理的最大连接请求数
    	int new_fd=-1;
    	struct sockaddr_in cli;
    	int len=sizeof(cli);
    	char buf[128]={ 0};
    	fd_set rdset;
    	fd_set tmpset;//tmpset记录我们要监控的描述符
    	FD_ZERO(&tmpset);//清空描述符集
    	FD_SET(0,&tmpset);//将标准输入注册为要监控的描述符
    	FD_SET(sfd,&tmpset);//将sfd注册为要监控的描述符
    	while(1)
    	{ 
    		FD_ZERO(&rdset);
    		memcpy(&rdset,&tmpset,sizeof(fd_set));//将需要监控的描述符集拷贝到rdset
    		ret=select(11,&rdset,NULL,NULL,NULL);//select函数监控rdset描述符集中相关文件的读变化
    		if(ret>0)
    		{ 
    			if(FD_ISSET(sfd,&rdset))//检查sfd是否发生读变化，即是否侦听到客户端的链接请求
    			{ 
    				new_fd=accept(sfd,(struct sockaddr*)&cli,&len);
    				//接收客户端的连接请求，并返回一个新的socket描述符，用于标识服务器与这个特定客户端的连接
    				if(-1==new_fd)
    				{ 
    					perror("accept");
    					return -1;
    				}
    				FD_SET(new_fd,&tmpset);//将该新的描述符加入监控集
    				printf("Server:已成功连接到客户端%s:%d,您可以向客户端发送信息了!\n",inet_ntoa(cli.sin_addr),ntohs(cli.sin_port));
    			}
    			if(FD_ISSET(new_fd,&rdset))//检查new_fd是否发生读变化，即客户端是否发送信息过来
    			{ 
    				memset(buf,0,sizeof(buf));
    				ret=recv(new_fd,buf,sizeof(buf),0);//接收客户端发来的信息
    				if(ret>0)
    				{ 
    					printf("Client:%s\n",buf);
    				}
    				else if(0==ret)
    				{ 
    					printf("byebye,客户端已下线!\n");
    					//close(new_fd);
    					FD_CLR(new_fd,&tmpset);//从集合当中移除该new_fd
    				}
    			}
    			if(FD_ISSET(0,&rdset))//检查服务器的标准输入是否发生读变化，即服务器是否有信息要发送给客户端
    			{ 
    				memset(buf,0,sizeof(buf));
    				ret=read(0,buf,sizeof(buf));//读取标准输入的输入信息
    				if(ret>0)
    				{ 
    					ret=send(new_fd,buf,strlen(buf)-1,0);//将标准输入的信息发送给客户端
    					if(ret<=0)
    					{ 
    						printf("Server:您未输入任何信息，请重新输入!\n");//输入为空的处理
    					}
    				}
    				else
    				{ 
    					printf("byebye,服务器准备退出!\n");
    					break;
    				}
    			}
    		}
    	}
    	close(new_fd);
    	close(sfd);
    	return 0;
    }

### 2、客户端源代码 ###

//==================================
    // FileName : tcp_client.c
    // Date : 2019/08/14
    // Author : 柚子树
    //==================================
    
    #include <signal.h>
    #include <sys/select.h>
    #include <sys/socket.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
    #include <stdio.h>
    #include <netdb.h>
    #include <string.h>
    #include <sys/types.h>
    #include <unistd.h>
    #include <stdlib.h>
    
    int main(int argc,char* argv[])
    { 
    	if(argc!=3)
    	{ 
    		printf("error args!请输入命令行参数ip和port:\n");
    		return -1;
    	}
    	int sfd;
    	sfd=socket(AF_INET,SOCK_STREAM,0);//客户端生成一个socket描述符
    	if(-1==sfd)
    	{ 
    		perror("socket");
    		return -1;
    	}
    	struct sockaddr_in ser;//该结构体变量存储ip、port等服务器端的socket信息数据
    	memset(&ser,0,sizeof(ser));//清空结构体
    	ser.sin_family=AF_INET;//采用ipv4协议
    	ser.sin_port=htons(atoi(argv[2]));//端口号，将主机字节序的port转换为网络字节序的port
    	ser.sin_addr.s_addr=inet_addr(argv[1]);//ip，将点分十进制ip转换为网络字节序的32位二进制数值
    	int ret;
    	//向服务器发出连接请求，将客户端生成的sfd描述符信息连接到ip、port指定的服务器上去
    	ret=connect(sfd,(struct sockaddr*)&ser,sizeof(ser));
    	if(-1==ret)
    	{ 
    		perror("connect");
    		return -1;
    	}
    	printf("Client:已成功连接到服务器,您可以向服务器发送消息了!\n");
    	char buf[128]={ 0};
    	fd_set rdset;//定义一个监控描述符集
    	while(1)
    	{ 
    		FD_ZERO(&rdset);//清空监控描述符集
    		FD_SET(0,&rdset);//将标准输入加入监控
    		FD_SET(sfd,&rdset);//将与服务器通信的sfd加入监控，以监控服务器是否有消息传送过来
    		ret=select(sfd+1,&rdset,NULL,NULL,NULL);//用select函数将监控描述符集rdset的读变化
    		if(ret>0)
    		{ 
    			if(FD_ISSET(sfd,&rdset))//检查sfd是否有读变化，即判断服务器是否有信息发送过来
    			{ 
    				memset(buf,0,sizeof(buf));
    				ret=recv(sfd,buf,sizeof(buf),0);//接收服务器发送的信息,存入缓冲区buf
    				if(ret>0)
    				{ 
    					printf("Server:%s\n",buf);
    				}else if(0==ret)
    				{ 
    					printf("byebye,服务器已下线,聊天终止!\n");
    					break;
    				}
    			}
    			if(FD_ISSET(0,&rdset))//检查标准输入是否有读变化，即检查客户端标准输入是否有用户输入
    			{ 
    				memset(buf,0,sizeof(buf));
    				ret=read(0,buf,sizeof(buf));//读取标准输入
    				if(ret>1)
    				{ 
    					ret=send(sfd,buf,strlen(buf)-1,0);//将用户输入信息发送给服务器
    				}else if(ret=1){ 
    					printf("Client:您未输入任何信息,请重新输入!\n");
    				}else{ 
    					printf("byebye,客户端准备退出，聊天即将关闭!\n");
    					break;
    				}
    			}
    		}
    	}
    	close(sfd);
    	return 0;
    }

### 3、测试示例 ###

![在这里插入图片描述][20190814191947406.png_pic_center]

[20190814191947406.png_pic_center]: /images/20211029/f986af4db3b84ed39567fcd4f36fc400.png