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.cnhttp://www.bianceng.cn/OS/unix/

以上是小编为您精心准备的的内容，在的博客、问答、公众号、人物、课程等栏目也有的相关内容，欢迎继续使用右上角搜索按钮进行搜索socket
， struct
， include
， sizeof
， 套接字
端口
，以便于您获取更多的相关知识。