前言
本文主要介绍了linux下Select的TCP通信流程，实现了客户端和服务器的通信，主要实现了消息的回发，即服务器将消息原封不动的回发给客户端。
这里主要介绍服务端代码，关于客户端代码请参考客户端代码和socket的基本使用
服务器代码
#include <errno.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/socket.h> #include <sys/types.h> #include <unistd.h> #include <sys/poll.h> #define MAXLNE 4096 #define POLL_SIZE 1024 int main(int argc, char **argv) { int listenfd, connfd, n; struct sockaddr_in servaddr; char buff[MAXLNE]; if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { printf("create socket error: %s(errno: %d)\n", strerror(errno), errno); return 0; } memset(&servaddr, 0, sizeof(servaddr)); servaddr.sin_family = AF_INET; servaddr.sin_addr.s_addr = htonl(INADDR_ANY); servaddr.sin_port = htons(9999); if (bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) == -1) { printf("bind socket error: %s(errno: %d)\n", strerror(errno), errno); return 0; } //listen接口之后会一直监听客户端的连接，每次客户端连接，都会和其创建连接（三次连接时内核完成的，不是由应用程序去控制的） //三次握手不发生在任何API中，协议栈本身被动完成的。 if (listen(listenfd, 10) == -1) { printf("listen socket error: %s(errno: %d)\n", strerror(errno), errno); return 0; } //poll 原理和select类似 struct pollfd fds[POLL_SIZE] = {0};//管理所有的fd fds[0].fd = listenfd; fds[0].events = POLLIN;//监测listenfd的连接（读）事件 int max_fd = listenfd; int i = 0; for (i = 1;i < POLL_SIZE;i ++) {//初始化其他socket fds[i].fd = -1; } while (1) { int nready = poll(fds, max_fd+1, -1); if (fds[0].revents & POLLIN) { //发生了连接 struct sockaddr_in client; socklen_t len = sizeof(client); if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) { printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno); return 0; } fds[connfd].fd = connfd; fds[connfd].events = POLLIN; if (connfd > max_fd) max_fd = connfd; if (--nready == 0) continue; } //int i = 0; for (i = listenfd+1;i <= max_fd;i ++) { if (fds[i].revents & POLLIN) { n = recv(i, buff, MAXLNE, 0); if (n > 0) { buff[n] = '\0'; printf("recv msg from client: %s\n", buff); send(i, buff, n, 0); } else if (n == 0) { //断开连接 fds[i].fd = -1; close(i); } if (--nready == 0) break; } } } close(listenfd); return 0; }
