pselect 函数

　　《unix网络环境编程》 中20-7 的示例理解。

 #include    "unp.h"

 static void    recvfrom_alarm(int);

 void
 dg_cli(FILE *fp, int sockfd, const SA *pservaddr, socklen_t servlen)
 {
     int                n;
     const int        on = ;
     char            sendline[MAXLINE], recvline[MAXLINE + ];
     fd_set            rset;
     sigset_t        sigset_alrm, sigset_empty;
     socklen_t        len;
     struct sockaddr    *preply_addr;

     preply_addr = Malloc(servlen);

     Setsockopt(sockfd, SOL_SOCKET, SO_BROADCAST, &on, sizeof(on));

     FD_ZERO(&rset);

     Sigemptyset(&sigset_empty);
     Sigemptyset(&sigset_alrm);
     Sigaddset(&sigset_alrm, SIGALRM);

     Signal(SIGALRM, recvfrom_alarm);

     while (Fgets(sendline, MAXLINE, fp) != NULL) {
         Sendto(sockfd, sendline, strlen(sendline), , pservaddr, servlen);

         Sigprocmask(SIG_BLOCK, &sigset_alrm, NULL);
         alarm();
         for ( ; ; ) {
             FD_SET(sockfd, &rset);
             n = pselect(sockfd+, &rset, NULL, NULL, NULL, &sigset_empty);
             if (n < ) {
                 if (errno == EINTR)
                     break;
                 else
                     err_sys("pselect error");
             } else if (n != )
                 err_sys("pselect error: returned %d", n);

             len = servlen;
             n = Recvfrom(sockfd, recvline, MAXLINE, , preply_addr, &len);
             recvline[n] = ;    /* null terminate */
             printf("from %s: %s",
                     Sock_ntop_host(preply_addr, len), recvline);
         }
     }
     free(preply_addr);
 }

 static void
 recvfrom_alarm(int signo)
 {
     return;        /* just interrupt the recvfrom() */
 }

#include <stdio.h>
#include <unistd.h>
#include <signal.h>

static void handle_signal(int signo)
{
    printf("come here\n");
    return ;
}

int main()
{
    sigset_t signal_usr1, signal_old;

    sigemptyset(&signal_usr1);
    sigaddset(&signal_usr1, SIGUSR1);

    sigprocmask(SIG_BLOCK, &signal_usr1, &signal_old);
    signal(SIGUSR1, handle_signal);

    sleep();

    sigprocmask(SIG_SETMASK, &signal_old, NULL);

    return ;
}

　　将上边这个程序编译好之后，运行，运行期间向这个进程发送 SIGUSR1 信号，最后发现仍然后有 “come here” 输出。

　　说明：

　　　　sigprocmask 虽然将 SIGUSR1 屏蔽掉了，但是如果在屏蔽期间收到了 SIGUSR1 信号，解除屏蔽后会统一处理该信号。

　　那么此时返回来看上边那个pselect 的例子就会发现，代码的意思是确保 SIGALRM 信号只在pselect 中处理。