Mongoose源码之master_thread

转自于：https://blog.csdn.net/larryliuqing/article/details/7544374

 

    本来要写一篇Mongoose源码分析，发现用一篇很难尽述其详，今天先来分析一下master_thread函数的工作原理。

    Master_thread函数是Mongoose启动服务时，启动的第一个监听线程，它的主要工作是监听所有的client连接的请求。

    启动master_thread后，就相当于启动了一个web server，master_thread将server的socket加入到监听集合中，然后一直监听该端口，只要有来自客户端的请求，就会调用accept_new_connection()去处理连接请求。具体实现如下：

 

static void
master_thread(struct mg_context *ctx)
{
    fd_set        read_set;
    struct timeval    tv;
    int        i, max_fd;

    while (ctx->stop_flag == 0) {
        FD_ZERO(&read_set);
        max_fd = -1;

        /* Add listening sockets to the read set */
        lock_option(ctx, OPT_PORTS);
        for (i = 0; i < ctx->num_listeners; i++)
            add_to_set(ctx->listeners[i].sock, &read_set, &max_fd);
        unlock_option(ctx, OPT_PORTS);

        tv.tv_sec = 1;
        tv.tv_usec = 0;

        if (select(max_fd + 1, &read_set, NULL, NULL, &tv) < 0) {
#ifdef _WIN32
            /*
             * On windows, if read_set and write_set are empty,
             * select() returns “Invalid parameter” error
             * (at least on my Windows XP Pro). So in this case,
             * we sleep here.
             */
            sleep(1);
#endif /* _WIN32 */
        } else {
            lock_option(ctx, OPT_PORTS);
            for (i = 0; i < ctx->num_listeners; i++)
                if (FD_ISSET(ctx->listeners[i].sock, &read_set))
                    accept_new_connection(
                        ctx->listeners + i, ctx);
            unlock_option(ctx, OPT_PORTS);
        }
    }

    /* Stop signal received: somebody called mg_stop. Quit. */
    mg_fini(ctx);
}

    master_thread内主要是一个while循环，判断条件是ctx->stop_flag，ctx是一个mg_context类型的结构体，它用来保存mongoose服务器的上下文；stop_flag分别为0、1、2三种状态，当stop_flag为0时，标志着mongoose正常运行，当它为1时，表面mg_stop函数被调用，mongoose的资源并没有被释放，需要等待mg_fini函数返回，当为2时，server结束。

    master_thread中维护着一个fd_set集合，这个集合用于存储所有来自客户端请求的socket。先后使用了FD_ZERO、FD_SET、select、FD_ISSET等用于同步多路I/O复用的函数。

    首先，FD_ZERO将fd_set这个文件描述符集(socket描述符也是文件描述符)清空，在对此集合进行设置前，必须对它进行初始化，如果不清空，原有的内存内容并不进行清理。

    然后，将所有的监听的socket加入fd_set中，即调用add_to_set。在这个函数中，调用FD_SET方法，在fd_set中增加一个新的文件描述符，被增加的描述符由mg_context的num_listeners决定。

    调用select方法判断tv.tv_sec = 1 即一秒时间内，被监听的描述符是否有事件发生，如果有则返回一个非0，否则返回0。这里的select方法，有必要讲讲，比如，fd_set长度为1字节，fd_set中的每一bit可以对应一个文件描述符fd。则1字节长的fd_set最大可以对应8个fd。

（1）执行fd_set set; FD_ZERO(&set); 则set用位表示是0000,0000。
（2）若fd＝5，执行FD_SET(fd,&set);后set变为0001,0000(第5位置为1)

（3）若再加入fd＝2，fd=1,则set变为0001,0011
（4）执行select(6,&set,0,0,0)阻塞等待
（5）若fd=1,fd=2上都发生可读事件，则select返回，此时set变为0000,0011。注意：没有事件发生的fd=5被清空。

    随后，判断select的返回值，如果小于0，说明出错，否则，用判断FD_ISSET是否文件描述符是否在该集合中，如果在，用accept_new_connection处理请求。