Libevent 学习十二：bufferevent filter 过滤器实例

bufferevent filter 过滤器实例

本章实现一个简单的过滤器实例：在输入过滤器中将客户端发送的数据转为大写，在输出过滤器中将发送给客户端的数据打上头尾标签。代码如下

#include <iostream>
#include <string.h>
#include <errno.h>
#ifndef _WIN32
#include <signal.h>
#endif // !_WIN32

#include "event.h"
#include "event2/listener.h"

#define SERVER_PORT 8000

// 读取回调
void read_cb(struct bufferevent *bev, void *ctx)
{
    std::cout << "[R]" << std::endl;

    // 读取数据
    char data[1024] = { 0 };
    size_t len = bufferevent_read(bev, data, sizeof(data) - 1);
    std::cout << data << std::endl;

    // 回复消息，消息会经过输出过滤器
    bufferevent_write(bev, data, len);
}

// 写入回调
void write_cb(struct bufferevent *bev, void *ctx)
{
    std::cout << "[W]" << std::endl;
}

// 事件回调，错误，超时，连接断开会进入
void event_cb(struct bufferevent *bev, short what, void *ctx)
{
    std::cout << "[E]:" << what << std::endl;
}

// bufferevent filter 回调函数
enum bufferevent_filter_result filter_in_cb(
    struct evbuffer *src, struct evbuffer *dst, 
    ev_ssize_t dst_limit, enum bufferevent_flush_mode mode, void *ctx)
{
    std::cout << "filter_in_cb" << std::endl;

    // 读取并清理源数据
    char data[1024] = { 0 };
    int len = evbuffer_remove(src, data, sizeof(data) - 1);

    // 过滤器功能：将所有字母转为大写
    for (int i = 0; i < len; ++i) data[i] = toupper(data[i]);

    // 将数据添加到目标数据中
    evbuffer_add(dst, data, len);

    return BEV_OK;
}

enum bufferevent_filter_result filter_out_cb(
    struct evbuffer *src, struct evbuffer *dst, 
    ev_ssize_t dst_limit, enum bufferevent_flush_mode mode, void *ctx)
{
    std::cout << "filter_out_cb" << std::endl;

    // 读取并清理源数据
    char data[1024] = { 0 };
    int len = evbuffer_remove(src, data, sizeof(data) - 1);

    // 过滤器功能：将消息加上头和尾
    char out[1024] = { 0 };
    int out_len = snprintf(out, sizeof(out) - 1, "******\n%s\n******", data);

    // 将数据添加到目标数据中
    evbuffer_add(dst, out, out_len);

    return BEV_OK;
}

// 监听回调
void listen_cb(struct evconnlistener *ev, 
               evutil_socket_t fd, struct sockaddr *sin, int slen, void *arg)
{
    std::cout << "listen_cb" << std::endl;

    // 创建bufferevent，绑定bufferevent filter
    event_base *base = (event_base*)arg;
    bufferevent *bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE);
    bufferevent *bev_filter = bufferevent_filter_new(bev,
        filter_in_cb,               // 输入过滤函数
        filter_out_cb,              // 输出过滤函数
        BEV_OPT_CLOSE_ON_FREE,      // 关闭filter同时关闭bufferevent
        NULL, NULL);                // 清理的回调函数，回调函数参数

    // 设置bufferevent的回调函数
    bufferevent_setcb(bev_filter, read_cb, write_cb, event_cb, NULL);
    // 设置bufferevent的读写权限
    bufferevent_enable(bev_filter, EV_READ | EV_WRITE);
}

int main(int argc, char* argv[])
{
#ifdef _WIN32
    // 初始化socket库
    WSADATA wsa;
    WSAStartup(MAKEWORD(2, 2), &wsa);
#else
    // 忽略管道信号，因为发送数据给已关闭的socket会生成SIGPIPE信号，导致进程退出
    if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
        return -1;
#endif

    std::cout << "libevent bufferevent filter test" << std::endl;
    // 创建libevent上下文
    event_base *base = event_base_new();
    if (!base) {
        std::cout << "event_base_new failed" << std::endl;
        return -1;
    }
    std::cout << "event_base_new success" << std::endl;

    // 绑定IP和端口
    struct sockaddr_in sin;
    memset(&sin, 0, sizeof(sin));
    sin.sin_family = AF_INET;
    sin.sin_port = htons(SERVER_PORT);

    // 创建监听事件
    evconnlistener *ev = evconnlistener_new_bind(base,
        listen_cb,                                  // 回调函数
        base,                                       // 回调函数参数
        LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE,  // 设置地址重用和关闭同时关闭socket
        10,                                         // listen back
        (sockaddr*)&sin, sizeof(sin));              // 监听地址
    if (!ev) {
        std::cout << "evconnlistener_new_bind failed" << std::endl;
        return -1;
    }

    // 进入事件主循环
    event_base_dispatch(base);

    if (ev) evconnlistener_free(ev);
    event_base_free(base);

#ifdef _WIN32
    WSACleanup();
#endif

    return 0;
}

测试结果如下：