Select實作并發伺服器

并發伺服器除了可以用多線程和多程序實作以外，還可以用select實作單線程并發，下面用select實作簡單的示例，伺服器接收用戶端的連接配接，并将客戶發的消息傳回，代碼如下：

伺服器端代碼：

main.c

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <stdarg.h>

#include <unistd.h>
#include <string.h>
#include <stdio.h>
#include <errno.h>

#define PORT 8888
#define BUFF_LEN 1024

int client_sock;
int errexit(const char *format, ...);
int echo(int fd);

int main(){
    int ss = create_tcp_server(PORT);
    if(- == ss)
        exit(-);

    struct sockaddr_in fsin;
    unsigned int alen;
    int fd;
    fd_set rfds;/* read file descriptor set */
    fd_set afds;/* active file descriptor set   */
    int nfds = getdtablesize();
    FD_ZERO(&afds);
    FD_SET(ss, &afds);

    while () {
        memcpy(&rfds, &afds, sizeof(rfds));
        if (select(nfds, &rfds, (fd_set *), (fd_set *), (struct timeval *)) < ){
            errexit("select: %s\n", strerror(errno));
        }

        if (FD_ISSET(ss, &rfds)) {
            alen = sizeof(fsin);
            int ssock = accept(ss, (struct sockaddr *)&fsin, &alen);
            if (ssock < )
                errexit("accept: %s\n", strerror(errno));
            else
                printf("accept clinet %d\n", ssock);
            FD_SET(ssock, &afds);
        }
        for (fd = ; fd < nfds; ++fd){
            if (fd != ss && FD_ISSET(fd, &rfds)){
                if (echo(fd) <= ) {
                    (void) close(fd);
                    FD_CLR(fd, &afds);
                }
            }
        }
    }
}

int echo(int fd){
    char buf[BUFF_LEN];
    int cc = read(fd, buf, sizeof(buf));
    if(cc > ){
        printf("recv msg from client %d : %s\n", fd, buf);
        if(write(fd, buf, cc) < )
            printf("write to client %d error, close!\n", fd);
    } else if(cc == )
        printf("client %d disconnect\n", fd);
    else
        printf("read from client %d error, close!\n", fd);
    return cc;
}

int errexit(const char *format, ...){
    va_list args;
    va_start(args, format);
    vfprintf(stderr, format, args);
    va_end(args);
    exit();
}
           

CreateServer.h//封裝了建立伺服器代碼

#ifndef CREATESERVER_H
#define CREATESERVER_H

int create_tcp_server(int);
int create_udp_server(int);

#endif // CREATESERVER_H
           

CreateServer.c//封裝方法的具體實作

#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#define LISTEN_SIZE 20

int start_server(int port, int type){
    //建立伺服器套接字
    int ss = socket(AF_INET, type, );
    if(ss < ){
        printf("create socket error\n");
        return -;
    }

    //設定伺服器位址
    struct sockaddr_in server_addr; //伺服器位址結構
    bzero(&server_addr, sizeof(struct sockaddr_in)); //清零
    server_addr.sin_family = AF_INET; //協定族
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY); //ip位址
    server_addr.sin_port = htons(port); //端口
    //綁定位址結構到套接字描述符
    if(bind(ss, (struct sockaddr*)&server_addr, sizeof(server_addr)) < ){
        printf("bind error\n");
        return -;
    }
    //TCP
    if(SOCK_STREAM == type){
        //設定偵聽
        if(listen(ss, LISTEN_SIZE) < ){
            printf("listen error\n");
            return -;
        }
        printf("tcp server start\n");
    }
    else
        printf("udp server start\n");
    return ss;
}

int create_tcp_server(int port){
    start_server(port, SOCK_STREAM);
}

int create_udp_server(int port){
    start_server(port, SOCK_DGRAM);
}
           

用戶端測試代碼：

代碼同之前的部落格，參見：http://blog.csdn.net/wanna_wsl/article/details/53712066

源碼下載下傳位址：

https://github.com/Wushaoling/Linux-C-Socket/tree/master/TCP/TCP%E7%AE%80%E5%8D%95%E6%95%B0%E6%8D%AE%E5%8F%91%E9%80%81