TinyWebServer:一个Linux下C++轻量级Web服务器(上)

入门学习了Linux的系统编程和网路编程,应该写一个小项目来练练手啦。这里模仿的是Github上一个开源项目:TinyWebServer。

项目地址:https://github.com/qinguoyi/TinyWebServer

非常感谢社长(TinyWebServer项目owner)的项目,项目代码量不算多,但是麻雀虽小五脏俱全,是一个非常好的把学过的各个知识点串在一起的小项目!!这里想讲一下实现过程以及其中的一些坑和收获(我学习的是项目的raw version)。这篇(上)就先讲项目最最最重要那些功能。阅读本文或者说要完全理解这个项目需要有Linux系统编程基础和网络编程基础,以及对计算机网路和http有一定了解。

博主水平十分有限(不是谦虚呀qaq),文章很可能有遗漏/错误,请大家也自行思考,欢迎指出讨论。

(更新中~~~~~)

概括

项目分成Main模块,epoll模块,http模块,lock模块,threadpool模块,log模块,timer模块,mysql模块。先简单讲一下各个模块的功能。main模块就是服务器主函数,主要是负责把各个模块组合协调工作。epoll模块是写epoll相关的函数。http模块是最关键的,提供处理http请求以及处理后返回http响应的所有函数。后面的lock就是提供同步工具(互斥锁/条件锁/信号量),threadpool当然就是线程池提供工作线程来处理http请求,log日记模块,mysql数据库模块提供数据库服务,timer定时器模块主要任务是负责定时清理长时间无反应的连接。

main模块

首先由main开始,主函数要协调各个模块进行工作,主要任务就是调用epoll函数监听各种事件并对事件调用相应的处理模块,接下啦详细讲一下:

在开始监听之前我们先创建listenfd并用epoll对其监听,然后我们开始正式工作,不断while知道WebServer停止服务,在while中我们调用epoll_wait函数得到所有有反应的事件,我们对这些事件分成几类来进行处理(根据事件的sockfd来判断):

①新连接请求,判断条件是sockfd == listenfd,那么没什么好说的我们accept接受请求就是了,并且保存好用户连接数据(ip,端口,connfd等),然后就把connfd挂到eoll上监听它的读事件

②对端关闭,判断条件是EPOLLRDHUP | EPOLLHUP | EPOLLERR,那么我们这边也关闭该连接相关资源。

③读事件且是信号事件,判断条件是(sockfd==pipefd[0]) && (whatopt & EPOLLIN),这里为什么信号会当成epoll的读事件?是因为统一了信号源,在信号的回调函数想管道里写东西,然后epoll监听管道,所以会监听到读事件,这里也许需要配合信号模块的讲解才容易理解。

④读事件且是客户端发送请求报文,判断条件是EPOLLIN,那么这里是读又不是信号,那么就是客户端的请求报文啦,那么我们把这次的读数据全部读到我们提前为该客户的准备的读数据缓冲区,然后我们把一个任务插入到线程池的任务队列中。于是我们就不用管了,工作线程自然会处理。

⑤写事件,判断条件是EPOLLOUT。WebServer只用一种写事件就是我们的工作线程做好了请求报文的处理,并且已经搓好了响应报文放在了客户的写数据缓冲区中,那么我们的任务就是调用函数把响应报文发送给浏览器。

这里直接讲main函数,可以看到整个服务器是怎么工作的,当然也有可能还没有接触到响应的各个模块所以有些懵逼,但是这里先有个大概流程了解后面了解清除所有模块工作之后回来会更加清晰。

while (!stop_server)
    {
        int total = epoll_wait(epollfd,events,MAX_EVENT_NUMBER,-1);
        if (total < 0 && errno != EINTR) {
                //把错误记录到日志
            break;
        }

        for (int i = 0; i < total; i++) {
            int sockfd = events[i].data.fd;
            int whatopt = events[i].events;

            //有新连接请求事件
            if (sockfd == listenfd) {
                struct sockaddr_in client_address;
                socklen_t client_address_len = sizeof(client_address);
                
                //LT
                int connfd = accept(listenfd, (struct sockaddr*)&client_address, &client_address_len);
                if (connfd < 0) {
                    //accept错误
                    continue;
                }
                if (http_conn::m_user_count >= MAX_FD) {
                    //用户数量超过最大描述符了
                    continue;
                }

                in_addr client_ip;
                memcpy(&client_ip, &client_address.sin_addr.s_addr, 4);
                printf("ip:%s connect\n", inet_ntoa(client_ip) );
                 
                // 初始化新客户,并在这里面把课后挂到epoll监听树上
                users[connfd].init(connfd, client_address);    

                //创造timer和client_data
                user_timer[connfd].address = client_address;
                user_timer[connfd].sockfd = connfd;
                
                util_timer* timer = new util_timer;
                timer->user_data = &user_timer[connfd];
                timer->cb_func = cb_func;
                timer->expire = time(NULL) + 6 * TIMESLOT;

                user_timer[connfd].timer = timer;
                //上面创造好了timer,加入到链表中
                timer_lst.add_timer(timer);
            }
            //对端关闭连接事件(EPOLLRDHUP | EPOLLHUP 这两个是关闭)
            else if (whatopt & (EPOLLRDHUP | EPOLLHUP | EPOLLERR)) {
                //对端关闭了,我们这边也关闭然后取消定时器
                util_timer* timer = user_timer[sockfd].timer;
                timer->cb_func(&user_timer[sockfd]);
                if (timer) timer_lst.del_timer(timer);
            }
            //因为统一了事件源,信号处理当成读事件来处理
            //怎么统一?就是信号回调函数哪里不立即处理而是写到:pipe的写端
            else if ((sockfd==pipefd[0]) && (whatopt & EPOLLIN)) {
                int sig;
                char signals[1024];
                int ret = recv(pipefd[0], signals, sizeof(signals), 0);
                if (ret == -1) continue;
                if (ret == 0) continue;
                //在这里处理信号
                for (int i = 0; i < ret; i++) {
                    switch (signals[i])
                    {
                    case SIGALRM:
                        timeout = true;
                        break;
                    case SIGTERM:
                        stop_server = true;
                    default:
                        break;
                    }
                }
            }
            /*输入事件,理想步骤是:
            process->porcess_read(不断parse_line->parse_status_line/parse_headers/parse_content
            ->do_request)->process_write(add_line/heads/content...)
            ->把报文搓到输出缓冲区
           */
            else if (whatopt & EPOLLIN) {
                //开始处理这个浏览器请求
                util_timer* timer = user_timer[sockfd].timer;
                if (users[sockfd].read_once()) {        //1,把所有数据读进来
                    pool->append(users + sockfd);   //2,读完之后把往线程池任务队列放入一个任务,这里process函数最后会添加监听写时间
                
                    //因为有了新请求,所以把这个客户的不活跃事件延后
                    //延后时间之后做出位置调整
                    if (timer) {
                        timer->expire = time(NULL) + 6 * TIMESLOT;
                        timer_lst.adjust_timer(timer);
                    }
                }
                //read_once()失败,关闭连接吧
                else {
                    timer->cb_func(&user_timer[sockfd]);
                    if (timer) timer_lst.del_timer(timer);
                }
            }
            //输出事件
            else if (whatopt & EPOLLOUT) {
                util_timer* timer = user_timer[sockfd].timer;
                //在上面读事件已经搓好响应报文就等这里write把输出缓冲区发送给浏览器
                if (users[sockfd].write()) {        //write函数最后会重新监听读事件
                    //跟读事件一样,延后这个客户的不活跃事件
                    if (timer) {
                        timer->expire = time(NULL) + 6 * TIMESLOT;
                        timer_lst.adjust_timer(timer);
                    }
                }
                else {      //这里的话就是write发送给浏览器失败,关闭连接
                    timer->cb_func(&user_timer[sockfd]);
                    if (timer) timer_lst.del_timer(timer);
                }
            }
        }

        if (timeout) {
            timer_handler();
            printf("Now %d clients connect\n", http_conn::m_user_count);
            timeout = false;
        }
    }

main模块

epoll模块

epoll模块就是和epoll相关的我们需要的函数在这里定义,这里比较简答就不细讲了,看代码注释肯定能懂。

int epoll_myinit() {
    int epollfd = epoll_create(5);
    assert(epollfd != -1);
    return epollfd;
}

//把文件描述符fd设为非阻塞
int setnoblocking(int fd) {
    int old_option = fcntl(fd, F_GETFL);
    int new_option = old_option | O_NONBLOCK;
    fcntl(fd, F_SETFL, new_option);
    return old_option;
}

//把fd添加到监听红黑树epollfd上
void addfd(int epollfd, int fd, bool oneshot) {
    epoll_event event;
    event.data.fd = fd;

    event.events = EPOLLIN | EPOLLRDHUP;        //读事件
    //EPOLLRDHUP 表示读关闭。
    //1 对端发送 FIN (对端调用close 或者 shutdown(SHUT_WR)).
    //2 本端调用 shutdown(SHUT_RD). 当然,关闭 SHUT_RD 的场景很少。

    if (oneshot)
        event.events |= EPOLLONESHOT;
    /*eppll 即使使用ET模式,一个socket上的某个事件还是可能被触发多次,采用线程城池的方式来处理事件,可能一个socket同时被多个线程处理
       如果对描述符socket注册了EPOLLONESHOT事件,那么操作系统最多触发其上注册的一个可读、可写或者异常事件,且只触发一次。。想要下次再触发则必须使用epoll_ctl重置该描述符上注册的事件,包括EPOLLONESHOT 事件。
       EPOLLONESHOT:只监听一次事件,当监听完这次事件之后,如果还需要继续监听这个socket的话,需要再次把这个socket加入到EPOLL队列里 
       */

    epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &event);
    setnoblocking(fd);              //设置为非阻塞,如果ET是必须的
}
//把fd从监听红黑树epollfd上摘下来
void removefd(int epollfd, int fd) {
    epoll_ctl(epollfd, EPOLL_CTL_DEL, fd,0);
    close(fd);
}
//将事件重置为EPOLLONESHOT
void modfd(int epollfd, int fd, int ev) {
    epoll_event event;
    event.data.fd = fd;
    //LT
    event.events = ev | EPOLLONESHOT | EPOLLRDHUP;
    epoll_ctl(epollfd, EPOLL_CTL_MOD, fd, &event);
}

epoll模块

locker模块

locker模块包装了一些我们做线程同步的工具类,具体有:信号量类,互斥锁,条件变量类。没有什么太特别的,就是把这些工具函数将他们的错误处理包装起来方便使用。

#ifndef LOCKER_H
#define LOCKER_H

#include<exception>
#include<pthread.h>
#include<semaphore.h>

//一个简单的信号量类
class sem
{
public:
    sem() {
        if (sem_init(&m_sem, 0, 0) != 0)
            throw std::exception();
    }
    sem(int num) {
        if (sem_init(&m_sem, 0, num) != 0)
            throw std::exception();
    }
    ~sem() {
        sem_destroy(&m_sem);
    }

    //加锁与解锁
    bool wait() {
        return sem_wait(&m_sem) == 0;
    }
    bool post() {
        return sem_post(&m_sem) == 0;
    }

private:
    sem_t m_sem;
};


//简单的互斥锁
class locker
{
public:
    locker() {
        if (pthread_mutex_init(&m_mutex,NULL) != 0)
            throw std::exception();
    }
    ~locker() {
        pthread_mutex_destroy(&m_mutex);
    }

    bool lock() {
        return pthread_mutex_lock(&m_mutex) == 0;
    }
    bool unlock() {
        return pthread_mutex_unlock(&m_mutex) == 0;
    }
    pthread_mutex_t* get() {
        return &m_mutex;
    }

private:
    pthread_mutex_t m_mutex;
};


//简单的条件变量
class cond
{
public:
    cond() {
        if (pthread_cond_init(&m_cond, NULL) != 0)
            throw std::exception();
    }
    ~cond() {
        pthread_cond_destroy(&m_cond);
    }

    //设置条件变量
    bool wait(pthread_mutex_t* m_mutex) {
        return pthread_cond_wait(&m_cond, m_mutex);
    }
    bool timewait(pthread_mutex_t* m_mutex, struct timespec t) {
        return pthread_cond_timedwait(&m_cond, m_mutex, &t)==0;
    }
    //条件变量满足,唤醒阻塞在m_mutex互斥量的线程
    bool signal() {
        return pthread_cond_signal(&m_cond) == 0;
    }
    bool broadcast() {
        return pthread_cond_broadcast(&m_cond) == 0;
    }

private:
    pthread_cond_t m_cond;
};

#endif // !LOCKER_H

locker模块

threadpool模块

一个web服务器几乎离不开多线程了,在main那里我们说到main把所有读时间能读到的数据都存放在客户读缓冲区中,然后就插入任务到任务队列等待线程去完成。我们来仔细讲一下:

线程池类有两个最为关键的成员:

pthread_t* m_threads; //线程池数组
  std::list<T*> m_workqueue; //请求队列

什么是请求队列,就是一个存储任务的list,我们在main函数把新任务放到list的尾部,然后所有线程争夺list中的任务(这里要使用条件变量),争夺到的线程先对任务队列加互斥锁然后从list头取出任务结构体,在这个任务结构体内有一个回调函数,这个函数就是真正的工作(包括解析http请求报文,对请求资源的检查,搓响应报文一条龙)当然这个函数我们放在http模块以更加模块化,从线程的角度就是我们拿到任务然后调用这个函数,线程就是在漫长的这个函数中度过了。

然后是线程池数组,这个线程池还是比较简单的线程池(没有对线程的动态删减等等),那么我们就是在线程池构造的时候就创建好约定个数的线程储存在线程数组里,并且把线程detach掉,这样我们就不需要对线程进行回收等等操作。线程的关键是线程的工作函数run(),这个函数不断while循环直到被条件变量唤醒然后上锁从list尾取出任务,开始执行任务(函数是porcess,看下个模块)

#ifndef THREADPOOL_H
#define THREADPOOL_H

#include<cstdio>
#include<pthread.h>
#include<exception>
#include<list>

#include "../lock/locker.h"
#include"../CGImysql/sql_connection_pool.h"

template<typename T>
class threadpool
{
public:
    /*thread_number是线程池中线程的数量,max_requests是请求队列中最多允许的、等待处理的请求的数量*/
    threadpool(connection_pool* connPool, int thread_number = 8, int max_request = 10000);
    ~threadpool();
    bool append(T* request);

private:
    /*工作线程运行的函数,它不断从工作队列中取出任务并执行之*/
    static void* worker(void* arg);
    void run();

private:
    int m_thread_number;        //线程池线程数
    int m_max_requests;            //请求队列的最大请求数

    pthread_t* m_threads;        //线程池  数组
    std::list<T*> m_workqueue;        //请求    队列

    locker m_queuelocker;                //请求队列的互斥锁
    sem m_queuestat;                    //请求队列的信号量(可以看出要处理的任务数)

    bool m_stop;            //线程池结束标志
    connection_pool* m_connPool;            //数据库连接池
};

//线程池构造函数
template<typename T>
threadpool<T>::threadpool(connection_pool* connPool, int thread_number, int max_request) :
    m_thread_number(thread_number), m_max_requests(max_request), m_stop(false), m_threads(NULL), m_connPool(connPool) {
    if (thread_number <= 0 || max_request <= 0)    //不合理的线程数量和请求队列数量
        throw std::exception();
    m_threads = new pthread_t[m_thread_number];
    if (!m_threads)
        throw std::exception();
    //创造thread_number个线程并且存储起来
    for (int i = 0; i < thread_number; i++) {
        if (pthread_create(m_threads + i, NULL, worker, this) != 0) {
            delete[] m_threads;        //失败
            throw std::exception();
        }
        if (pthread_detach(m_threads[i])) {
            delete[] m_threads;        //失败
            throw std::exception();
        }
    }
}

//线程池析构函数
template<typename T>
threadpool<T>::~threadpool() {
    delete[] m_threads;
    m_stop = true;
}

//将“待办工作”加入到请求队列
template<typename T>
bool threadpool<T>::append(T *request) {
    m_queuelocker.lock();
    if (m_workqueue.size() > m_max_requests) {
        m_queuelocker.unlock();
        return false;
    }
    m_workqueue.push_back(request);
    m_queuelocker.unlock();
    m_queuestat.post();
    return true;
}

//线程回调函数/工作函数,arg其实是this
template<typename T>
void* threadpool<T>::worker(void *arg) {
    threadpool* pool = (threadpool*)arg;
    pool->run();
    return pool;
}

//回调函数会调用这个函数工作
//工作线程就是不断地等任务队列有新任务,然后就加锁取任务->取到任务解锁->执行任务
template<typename T>
void threadpool<T>::run() {
    while (!m_stop) {
        //请求队列长度--,互斥锁锁住
        m_queuestat.wait();
        m_queuelocker.lock();
        
        if (m_workqueue.empty()) {
            m_queuelocker.unlock();
            continue;
        }

        T* request = m_workqueue.front();
        m_workqueue.pop_front();
        
        m_queuelocker.unlock();

        if (!request) continue;

        //
        connectionRAII mysqlcon(&request->mysql, m_connPool);

        request->process();
    }
}


#endif // THREADPOOL_H

threadpool模块

http模块

好的到了这里我们终于到了最关键的http模块了,在http模块我们将完成最关键的处理http请求报文和搓http响应报文的工作。如果一个个函数细细讲这一块可以讲很久很久。。。

我们先看看主要函数以及它们的协调合作:

 TinyWebServer:一个Linux下C++轻量级Web服务器(上)

 我们先理解这一块使用到的“状态机设计模式”,我们学过UML里的状态图,那么我们应该很容易理解这个设计模式。简单理解就是,我们经常会遇到需要根据不同的情况作出不同的处理的情况,这时候我们写出大量的if else使得逻辑十分混乱。那么我们可以这样设计:我们在类里面设计一个状态,并且允许一个对象在其内部状态改变时改变它的行为,对象看起来似乎修改了它的类。感觉说起来还是比较抽象,看代码会比较容易理解,其实就是看状态调用不同的函数。

在http中我们如何使用状态机,我们有两个状态机:主状态机和从状态机。

//主状态机的状态:解析请求行  解析请求头  解析消息体(仅用于解析POST请求)
enum CHECK_STATE { CHECK_STATE_REQUESTLINE = 0, CHECK_STATE_HEADER, CHECK_STATE_CONTENT };
//报文解析的结果:请求不完整需要继续读取请求报文数据  获得了完整的HTTP请求  HTTP请求报文有语法错误  服务器内部错误,该结果在主状态机逻辑switch的default下,一般不会触发
enum HTTP_CODE { NO_REQUEST, GET_REQUEST, BAD_REQUEST, NO_RESOURCE, FORBIDDEN_REQUEST, FILE_REQUEST, INTERNAL_ERROR, CLOSED_CONNECTION };
//从状态机的状态:完整读取一行  报文语法有误  读取的行不完整
enum LINE_STATUS { LINE_OK = 0, LINE_BAD, LINE_OPEN };

这一部分知识强烈建议看一下社长公众号系列文章,我们需要重点理解主状态机和从状态机,我的个人理解是主状态机是更为宏观一点的他主要关注当前解析到 请求行 / 请求头 / 请求主体 ?那么我们的代码就要根据这个主状态机的状态判断当前的解析进度从而判断当前进度下一步要做的动作(举个例子:比如我现在是请求头状态并且也发现现在已经把最后一行请求头解析完了,那么我们就变换主状态机状态,这时下一次代码就能判断到当前解析到请求主体了所以调用相关的函数)。

那么从状态机又怎么理解呢?从状态机就是更加聚焦于一行行,着眼更加细致,他关心的是当前这一行读完整了/不完整/格式有误,亦即从状态机关注一行的解析状态。(那么显然这里可以想到其实每一个主状态机状态可能对应多轮的从状态改变,类似与包含关系),所以从状态机的函数会关注当前字符是什么,根据这个字符判断当前是读完了吗是格式错误吗等等。

我们从状态机的角度看看这个函数调用:

 TinyWebServer:一个Linux下C++轻量级Web服务器(上)

http模块就讲到这里了,具体每个函数实现还是得认真看代码,有很多很多细节值得学习,这里怕是讲不完。 

 OK到这里我们就讲了TinyWebServer的前五个模块了,写成这样其实webserver已经能够基础工作了,监听/请求/响应都可以完成了。后面我们要对这个服务器增加更多的模块,使得他的性能上升和提供更多的功能。

参考资料:

TinyWebServer项目地址:https://github.com/qinguoyi/TinyWebServer

社长的系列讲解博文:http://mp.weixin.qq.com/mp/homepage?__biz=MzAxNzU2MzcwMw==&hid=6&sn=353ef6eadc7a8daf9c82d005c15adcd2&scene=18#wechat_redirect