Linux下socket实现TCP网络通讯（多个客户端之间的通讯）

/*服务器端程序：一个服务器和多个客户端通讯（复用）*/ #include <stdio.h> #include <netinet/in.h> //for souockaddr_in #include <sys/types.h> #include <sys/socket.h> #include <errno.h> #include <stdlib.h> #include <arpa/inet.h> //for select #include <sys/time.h> #include <sys/types.h> #include <unistd.h> #include <sys/select.h> #include <strings.h> //for bzero #include <string.h> #define BUFF_SIZE 1024 #define backlog 7 #define ser_port 11277 #define CLI_NUM 3 int client_fds[CLI_NUM]; //创建套接字描述符数组 int main(int agrc,char **argv) { int ser_souck_fd; int i; char input_message[BUFF_SIZE]; char resv_message[BUFF_SIZE]; //======初始化套接字地址结构======= struct sockaddr_in ser_addr; ser_addr.sin_family= AF_INET; //IPV4 ser_addr.sin_port = htons(ser_port); ser_addr.sin_addr.s_addr = INADDR_ANY; //指定的是所有地址 //creat socket if( (ser_souck_fd = socket(AF_INET,SOCK_STREAM,0)) < 0 ) { perror("creat failure"); return -1; } //bind soucket if(bind(ser_souck_fd, (const struct sockaddr *)&ser_addr,sizeof(ser_addr)) < 0) { perror("bind failure"); return -1; } //listen if(listen(ser_souck_fd, backlog) < 0) { perror("listen failure"); return -1; } //fd_set为描述符集，用作select函数的参数 fd_set ser_fdset; //这个集合中存放的是文件描述符(file descriptor)，即文件句柄 int max_fd=1; //集合中所有文件描述符的范围，即所有文件描述符的最大值加1 struct timeval mytime; //用于指定等待时间，也是select函数的参数 printf("wait for client connnect!\n"); while(1) { mytime.tv_sec=27; //秒数 mytime.tv_usec=0; //微秒数 FD_ZERO(&ser_fdset); //初始化描述符集 //add standard input FD_SET(0,&ser_fdset); //打开描述符0的对应位 if(max_fd < 0) { max_fd=0; } //add serverce FD_SET(ser_souck_fd,&ser_fdset); //打开描述符ser_souck_fd的对应位 if(max_fd < ser_souck_fd) { max_fd = ser_souck_fd; } //add client for(i=0;i<CLI_NUM;i++) //用数组定义多个客户端fd { if(client_fds[i]!=0) { FD_SET(client_fds[i],&ser_fdset); //将各个客户端的描述符ser_souck_fd加入到集合中 if(max_fd < client_fds[i]) { max_fd = client_fds[i]; } } } ////==============select多路复用================= int ret = select(max_fd + 1, &ser_fdset, NULL, NULL, &mytime); if(ret < 0) //返回值小于0，select出错 { perror("select failure\n"); continue; } else if(ret == 0) //返回值等于0，等待超时 { printf("time out!\n"); continue; } else { //标准输入是否存在于ser_fdset集合中（也就是说，检测到输入时，做如下事情） if(FD_ISSET(0,&ser_fdset)) //检查fdset联系的文件句柄fd是否可读写，>0表示可读写。 { printf("=========send all these message to clients=========== \n"); bzero(input_message,BUFF_SIZE); fgets(input_message,BUFF_SIZE,stdin); for(i=0;i<CLI_NUM;i++) { if(client_fds[i] != 0) { printf("clients information: client_fds[%d]=%d\n", i, client_fds[i]); send(client_fds[i], input_message, BUFF_SIZE, 0); //遍历已连接客户端，群发消息 } } } if(FD_ISSET(ser_souck_fd, &ser_fdset)) //检查与fdset联系的文件句柄(监听套接字描述符)ser_souck_fd是否可读写，>0表示可读写 { struct sockaddr_in client_address; socklen_t address_len; int client_sock_fd = accept(ser_souck_fd,(struct sockaddr *)&client_address, &address_len); if(client_sock_fd > 0) { int flags=-1; //一个客户端到来分配一个fd，CLI_NUM=3，则最多只能有三个客户端，超过4以后跳出for循环，flags重新被赋值为-1 for(i=0;i<CLI_NUM;i++) { if(client_fds[i] == 0) { flags=i; client_fds[i] = client_sock_fd; //将空余的描述符分配给新的客户端 break; } } if (flags >= 0) { printf("new user client[%d] add sucessfully!\n",flags); } else //flags=-1 { char full_message[]="the client is full!can't join!\n"; bzero(input_message,BUFF_SIZE); strncpy(input_message, full_message,100); send(client_sock_fd, input_message, BUFF_SIZE, 0); } } } } //=============处理客户端数据deal with the message=================== for(i=0; i<CLI_NUM; i++) { if(client_fds[i] != 0) { if(FD_ISSET(client_fds[i],&ser_fdset)) //判断客户端client_fds的文件句柄是否可读写 { bzero(resv_message,BUFF_SIZE); int byte_num=read(client_fds[i],resv_message,BUFF_SIZE); //int byte_num=recv(client_fds[i],resv_message,BUFF_SIZE,0); //resv_message[byte_num]='\0'; if(byte_num > 0) { printf("\n========receive message========\n"); printf("message from client[%d]:%s", i, resv_message); } else if(byte_num < 0) { printf("rescessed error!"); } //某个客户端退出 else //cancel fdset and set fd=0 { printf("clien[%d] exit!\n",i); FD_CLR(client_fds[i], &ser_fdset); client_fds[i] = 0; continue; //这里如果用break的话一个客户端退出会造成服务器也退出。 } } } } } return 0; }