select cli

Author: yangsoon

muduo/unp/select/strcliselect01.c

@@ -0,0 +1,73 @@
+// 使用select重写str_cli函数，这样服务器进程一终止，客户就能马上得到通知。
+// 之前的版本当套接字发生某些时间时，客户可能阻塞于fgets调用。新版本改为阻塞在select调用上
+#include "unp.h"
+
+
+void select_str_cli(FILE *fp, int sockfd) {
+    int maxfdp1;
+    fd_set rset;
+    int n;
+    char sendline[MAXLINE], recvline[MAXLINE];
+
+    // 用于初始化
+    FD_ZERO(&rset);
+    // 使用select来驱动，区分描述符
+    for(;;) {
+        //设置监控的文件描述符
+        //每次调用select之后 rset都会被清空 所以需要重新设置
+        FD_SET(fileno(fp), &rset);
+        FD_SET(sockfd, &rset);
+        // 注意这里的maxfd1一定设为最大值加1 因为这个传给select可以认为是描述符的个数
+        maxfdp1 = max(fileno(fp), sockfd) + 1;
+
+        // select 调用
+        // int select(int maxfdp,fd_set *readfds,fd_set *writefds,fd_set *errorfds,struct timeval*timeout);
+        // 第一个参数表示监听描述符的最大值，后面的参数依次表示 可读 可写 异常集合 以及超时时间
+        // 可读集合：sock有数据可读 对端关闭写 有新的连接
+        // 可写集合：sock有缓存空间可以写 主动关闭写端
+        // !!当套接字上发生错误时候 可写也可读
+        // 异常 只有出现带外数据的时候才出现异常
+        if((n = select(maxfdp1, &rset, NULL, NULL, NULL)) < 0) {
+            err_sys("select error");
+        }
+        if(FD_ISSET(sockfd, &rset)) {
+            if(Readline(sockfd, recvline, MAXLINE) == 0) {
+                err_quit("str_cli: server terminated prematurely");
+            }
+            Fputs(recvline, stdout);
+        }
+
+        if(FD_ISSET(fileno(fp), &rset)) {
+            if(Fgets(sendline, MAXLINE, fp) == NULL) {
+                return;
+            }
+            Write(sockfd, sendline, strlen(sendline));
+        }
+    }
+}
+
+int main(int argc, char **argv) {
+    int sockfd;
+    struct sockaddr_in servaddr;
+
+    if(argc!=2) {
+        err_quit("usage: tcpcli <IPaddress>");
+    }
+
+    if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
+        err_sys("sockfd error");
+    }
+
+    // 基本流程
+    bzero(&servaddr, sizeof(servaddr));
+    servaddr.sin_family = AF_INET;
+    servaddr.sin_port = htons(SERV_PORT);
+    Inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
+
+    if(connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0) {
+        err_sys("connect error");
+    }
+    // Connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
+    select_str_cli(stdin, sockfd);
+    exit(0);
+}

muduo/unp/select/strcliselect02.c

@@ -0,0 +1,78 @@
+#include "unp.h"
+
+void select_str_cli(FILE *fp, int sockfd) {
+    int maxfdp1, stdineof;
+    fd_set rset;
+    char buf[MAXLINE];
+    int n;
+    
+    stdineof = 0;
+    FD_ZERO(&rset);
+    for(;;) {
+        if(stdineof==0) {
+            FD_SET(sockfd, &rset);
+        }
+        FD_SET(fileno(fp), &rset);
+        maxfdp1 = max(fileno(fp), sockfd) + 1;
+
+        if((select(maxfdp1, &rset, NULL, NULL, NULL))<0) {
+            err_sys("select error");
+        }
+
+        if(FD_ISSET(sockfd, &rset)) {
+            if((n=Readline(sockfd, buf, MAXLINE)) == 0) {
+                if(stdineof==1) {
+                    return;
+                } else {
+                    err_quit("str_cli: server terminated");
+                }
+            }
+            Write(fileno(fp), buf, n);
+        }
+
+        if(FD_ISSET(fileno(fp), &rset)) {
+            if((n=Readline(fileno(fp), buf, MAXLINE)) == 0) {
+                stdineof = 1;
+                // 一般使用close来关闭一个套接字
+                // 但是使用close有2个局限性， 首先close先把描述符的引用计数减去1 只有在引用计数为0的时候才会关闭套接字
+                // 其次是close终止的时候是读写全都关闭 但是套接字是全双工的 有时候我们需要告诉对端我们完成了数据发送，还可以接受对端的数据
+                // 即关闭写的这一半 也叫半关闭
+
+                // 使用shutdown就可以解决上面的问题 使用shutdown的时候不管引用计数是否为0都照样关闭
+                // 还可以通过设置第二个参数 来选择如何关闭
+                if(shutdown(sockfd, SHUT_WR) < 0) {
+                    err_sys("shutdown error");
+                }
+                FD_CLR(fileno(fp), &rset);
+                continue;
+            }
+            Write(sockfd, buf, n);
+        }
+    }
+}
+
+int main(int argc, char **argv) {
+    int sockfd;
+    struct sockaddr_in servaddr;
+    int n;
+
+    if(argc != 2) {
+        err_quit("usage: <IPAddress>");
+    }
+
+    if((sockfd = socket(AF_INET, SOCK_STREAM, 0))<0){
+        err_sys("socket error");
+    }
+
+    bzero(&servaddr, sizeof(servaddr));
+    servaddr.sin_family = AF_INET;
+    servaddr.sin_port = htons(SERV_PORT);
+    Inet_pton(AF_INET, argv[1], &servaddr.sin_addr);
+
+    if(connect(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0) {
+        err_sys("connect error");
+    }
+
+    select_str_cli(stdin, sockfd);
+    exit(0);
+}