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;
}
测试结果如下: