使用kqueue的str_cli函数

Posted 2021-10-22 soldierback

void str_cli(FILE *fp, int sockfd)
{
    int              kq, i, n, nev, stdineof = 0, isfile;
    char             buf[MAXLINE];
    struct kevent    kev[2];
    struct timespec  ts;
    struct stat      st;

    isfile = ((fstat(fileno(fp), &st) == 0) &&
              (st.st_mode & S_IFMT) == S_IFREG);

    EV_SET(&kev[0], fileno(fp), EVFILT_READ, EV_ADD, 0, 0, NULL);
    EV_SET(&kev[1], sockfd, EVFILT_READ, EV_ADD, 0, 0, NULL);

    kq = kqueue();
    ts.tv_sec = ts.tv_nsec = 0;
    kevent(kq, kev, 2, NULL, 0, &ts);

    for ( ; ; ) {
        nev = kevent(kq, NULL, 0, kev, 2, NULL);

        for (i = 0; i < nev; i++) {
            if (kev[i].ident == sockfd) {    /* socket is readable */
                if ( (n = read(sockfd, buf, MAXLINE)) == 0) {
                    if (stdineof == 1)
                        return;              /* normal termination */
                    else
                        err_quit("str_cli: server terminated prematurely");
                }

                write(fileno(stdout), buf, n);
            }

            if (kev[i].ident == fileno(fp)) {  　　/* input is readable */
                n = read(fileno(fp), buf, MAXLINE);
                if (n > 0)
                    writen(sockfd, buf, n);

                if (n == 0 || (isfile && n == kev[i].data)) {
                    stdineof = 1;
                    shutdown(sockfd, SHUT_WR);    /* send FIN */
                    kev[i].flags = EV_DELETE;
                    kevent(kq, &kev[i], 1, NULL, 0, &ts);    /* remove kevent */
                    continue;
                }
            }
        }
    }
}