libevent 之 hello world

0x00 基于 libevent 的 hello world

• 今晚刚看了一下 libevent 的示例代码，简洁记录一下自己对这份 hello world 代码的理解
• 这份代码是在 ubuntu 16.04 里编译生成， 下面用 netcat 测试

#include <string.h>
#include <errno.h>
#include <stdio.h>
#include <signal.h>
#include <netinet/in.h>

#include <arpa/inet.h>

#include <sys/socket.h>


#include <event2/bufferevent.h>
#include <event2/buffer.h>
#include <event2/listener.h>
#include <event2/util.h>
#include <event2/event.h>

static const char MESSAGE[] = "Hello, World!\n";

static const int PORT = 19994;

static void listener_cb(struct evconnlistener *, evutil_socket_t,
    struct sockaddr *, int socklen, void *);
static void conn_writecb(struct bufferevent *, void *);
static void conn_eventcb(struct bufferevent *, short, void *);
static void signal_cb(evutil_socket_t, short, void *);

int
main(int argc, char **argv)
{
    struct event_base *base;
    struct evconnlistener *listener;
    struct event *signal_event;

    struct sockaddr_in sin;

    //创建一个 base
    base = event_base_new();
    if (!base) {
        fprintf(stderr, "Could not initialize libevent!\n");
        return 1;
    }

    memset(&sin, 0, sizeof(sin));
    sin.sin_family = AF_INET;
    sin.sin_port = htons(PORT);

    //监听事件    LEV_OPT_REUSEABLE ----> 如果把服务器关闭，再启动服务器，会起不来，因为服务器处于一个 timewait 状态 去掉 timewait 状态      LEV_OPT_CLOSE_ON_FREE ----> 当把 listener 关闭的时候 同时把 绑定的 fd 也一起关闭
    // -1  ---->  一般填 -1
    listener = evconnlistener_new_bind(base, listener_cb, (void *)base,
        LEV_OPT_REUSEABLE|LEV_OPT_CLOSE_ON_FREE, -1,
        (struct sockaddr*)&sin,
        sizeof(sin));

    if (!listener) {
        fprintf(stderr, "Could not create a listener!\n");
        return 1;
    }

    //创建一个 监听信号事件， 当服务器接受到系统发送的信号时调用 回调函数
    signal_event = evsignal_new(base, SIGINT, signal_cb, (void *)base);

    if (!signal_event || event_add(signal_event, NULL)<0) {
        fprintf(stderr, "Could not create/add a signal event!\n");
        return 1;
    }


    //进入循环等待
    event_base_dispatch(base);

    evconnlistener_free(listener);
    event_free(signal_event);
    event_base_free(base);

    printf("done\n");
    return 0;
}


// 监听事件回调函数
/*

@pro listener    监听的事件
@pro   fd    监听的 fd
@pro   sa        客户端的地址信息
@pro   socklen   客户端地址信息长度
@pro   出发回调函数时传入的参数
*/
static void
listener_cb(struct evconnlistener *listener, evutil_socket_t fd,
    struct sockaddr *sa, int socklen, void *user_data)
{
    //传入的参数
    struct event_base *base = (struct event_base *)user_data;
    struct bufferevent *bev;

    //创建一个 bufferevent 事件    ------>>>>>    初始化 bufferevnet 事件
    bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE);
    if (!bev) {
        fprintf(stderr, "Error constructing bufferevent!");
        event_base_loopbreak(base);
        return;
    }
    // 设置 bufferevnet 事件的回调函数        1、bufferevnet   2、读回调函数    3、写回调函数   4、错误回调函数   5、前面三个回调函数传入的参数
    bufferevent_setcb(bev, NULL, conn_writecb, conn_eventcb, NULL);
    // 打开写事件
    bufferevent_enable(bev, EV_WRITE);
    //关闭读事件
    bufferevent_disable(bev, EV_READ);

    // 发送数据
    bufferevent_write(bev, MESSAGE, strlen(MESSAGE));
}

static void
conn_writecb(struct bufferevent *bev, void *user_data)
{
    //当成功发送数据后调用该函数
    struct evbuffer *output = (struct evbuffer *)bufferevent_get_output(bev);
    if (evbuffer_get_length(output) == 0) {
        printf("flushed answer\n");
        //发送完数据后释放 bufferevnet 和 关闭监听事件 和关闭监听 fd
        bufferevent_free(bev);
    }
}

// 监听事件 错误回调函数
static void
conn_eventcb(struct bufferevent *bev, short events, void *user_data)
{
    if (events & BEV_EVENT_EOF) {
        printf("Connection closed.\n");
    } else if (events & BEV_EVENT_ERROR) {
        printf("Got an error on the connection: %s\n",
            strerror(errno));
    }

    bufferevent_free(bev);
}


// 信号事件回调函数
static void
signal_cb(evutil_socket_t sig, short events, void *user_data)
{
    struct event_base *base = (struct event_base *)user_data;
    struct timeval delay = { 2, 0 };

    printf("Caught an interrupt signal; exiting cleanly in two seconds.\n");

    event_base_loopexit(base, &delay);
}

测试：
    netcat 127.0.0.1 19994

服务器返回数据：
    hello world