UNIX網絡編程：如何使用select函數的TCP和UDP回射服務器程序

服務器程序：

#include <sys/wait.h>  
#include <string.h>  
#include <string.h>  
#include <errno.h>  
#include <stdio.h>  
#include <stdlib.h>  
#include <sys/types.h>  
#include <sys/socket.h>  
#include <signal.h>  
#include <arpa/inet.h>  
#include <sys/select.h>  
#include <sys/time.h>  
#include <unistd.h>  
#define SERV_PORT 3334  
#define LISTENQ 5  
#define MAXLINE 100  
      
      
void str_echo(int sockfd)  
{  
    ssize_t     n;  
    char        buf[MAXLINE];  
      
again:  
    while ( (n = read(sockfd, buf, MAXLINE)) > 0)  
        write(sockfd, buf, n);  
      
    if (n < 0 && errno == EINTR)  
        goto again;  
    else if (n < 0)  
        perror("read");  
}  
      
      
void sig_chld(int signo)  
{  
    pid_t   pid;  
    int     stat;  
      
    while ( (pid = waitpid(-1, &stat, WNOHANG)) > 0)  
        printf("child %d terminated\n", pid);  
    return;  
}  
int main(int argc, char **argv)  
{  
    int             listenfd, connfd, udpfd, nready, maxfdp1;  
    char                mesg[MAXLINE];  
    pid_t               childpid;  
    fd_set              rset;  
    ssize_t             n;  
    socklen_t           len;  
    const int           on = 1;  
    struct sockaddr_in  cliaddr, servaddr;  
    void                sig_chld(int);  
      
        /* 4create listening TCP socket */
    listenfd = socket(AF_INET, SOCK_STREAM, 0);  
      
    bzero(&servaddr, sizeof(servaddr));  
    servaddr.sin_family      = AF_INET;  
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);  
    servaddr.sin_port        = htons(SERV_PORT);  
      
    setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));  
    bind(listenfd, (struct sockaddr *) &servaddr, sizeof(servaddr));  
      
    listen(listenfd, LISTENQ);  
      
        /* 4create UDP socket */
    udpfd = socket(AF_INET, SOCK_DGRAM, 0);  
      
    bzero(&servaddr, sizeof(servaddr));  
    servaddr.sin_family      = AF_INET;  
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);  
    servaddr.sin_port        = htons(SERV_PORT);  
      
    bind(udpfd, (struct sockaddr *) &servaddr, sizeof(servaddr));  
    signal(SIGCHLD, sig_chld);  /* must call waitpid() */
      
    FD_ZERO(&rset);  
    maxfdp1 =( (listenfd>udpfd)?listenfd : udpfd ) + 1;  
    for ( ; ; ) {  
        FD_SET(listenfd, &rset);  
        FD_SET(udpfd, &rset);  
        if ( (nready = select(maxfdp1, &rset, NULL, NULL, NULL)) < 0) {  
            if (errno == EINTR)  
                continue;       /* back to for() */
            else
                perror("select");  
        }  
      
        if (FD_ISSET(listenfd, &rset)) {  
            len = sizeof(cliaddr);  
            connfd = accept(listenfd, (struct sockaddr *) &cliaddr, &len);  
          
            if ( (childpid = fork()) == 0) {    /* child process */
                close(listenfd);    /* close listening socket */
                str_echo(connfd);   /* process the request */
                exit(0);  
            }  
            close(connfd);          /* parent closes connected socket */
        }  
      
        if (FD_ISSET(udpfd, &rset)) {  
            len = sizeof(cliaddr);  
            n = recvfrom(udpfd, mesg, MAXLINE, 0, (struct sockaddr*) &cliaddr, &len);  
      
            sendto(udpfd, mesg, n, 0, (struct sockaddr *) &cliaddr, len);  
        }  
    }  
}

57-67 創建一個監聽TCP套接字並捆綁服務器的眾所周知的端口，設置SO_REUSEADDR套接字選項以防止該端口上已有連接存在。

70-77 還創建一個UDP套接字並捆綁與TCP套接字相同的端口。這裡無需在調用bind之前設置SO_REUSEADDR套接字選項，因為TCP端口是獨立於UDP端口的。

78 給SIGCHLD建立信號處理程序，因為TCP連接將由某個子進程處理。

83-90 我們調用select只是為了等待監聽TCP套接字的可讀條件或UDP套接字的可讀條件。既然我們的sig_chld信號處理函數可能中斷我們對select的調用，於是我們需要處理EINTR錯誤。

作者：csdn博客 ctthuangcheng

查看本欄目更多精彩內容：http://www.bianceng.cn/OS/unix/