#include <event2/event.h>
#include <event2/bufferevent.h>
#include <event2/listener.h>
#include <event2/buffer.h>
#include <event2/util.h>
#include <iostream>
#include <signal.h>
#include <cstring>
using namespace std;
#define SPORT 5001
void listen_cb(evconnlistener* ev, evutil_socket_t s, sockaddr* sin, int slen, void* arg);
enum bufferevent_filter_result filter_out(struct evbuffer *src, struct evbuffer *dst, ev_ssize_t dst_limit, enum bufferevent_flush_mode mode, void *ctx);
enum bufferevent_filter_result filter_in(struct evbuffer *src, struct evbuffer *dst, ev_ssize_t dst_limit, enum bufferevent_flush_mode mode, void *ctx);
void read_cb(bufferevent* be, void* arg);
void write_cb(bufferevent* be, void* arg);
void event_cb(bufferevent *be, short which, void *arg);
int main() {
#ifdef _WIN32
// windows 需要初始化socket库
WSADATA wsa;
WSAStartup(MAKEWORD(2, 2), &wsa);
#else
if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) {
return -1;
}
#endif
std::cout << "Hello World!\n";
//
event_base* base = event_base_new();
if (base) {
cout << "event_base_new success " << endl;
}
// 创建网络服务器
sockaddr_in sin;
memset(&sin, 0, sizeof(sockaddr_in));
sin.sin_family = AF_INET;
sin.sin_port = htons(5001);
evconnlistener* ev = evconnlistener_new_bind(base,
listen_cb, // 回调函数
base, // 传递参数
LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE,
10,
(sockaddr*)&sin,
sizeof(sockaddr));
event_base_dispatch(base);
// 结束之后清理空间
evconnlistener_free(ev);
event_base_free(base);
return 0;
}
void listen_cb(evconnlistener* ev, evutil_socket_t s, sockaddr* sin, int slen, void* arg) {
cout << __FUNCTION__ << endl;
event_base* base = (event_base*)arg;
// 创建bufferevent,绑定bufferevent filter
bufferevent* bev = bufferevent_socket_new(base, s,BEV_OPT_CLOSE_ON_FREE);
bufferevent* bev_filter = bufferevent_filter_new(bev,
filter_in, // 输入过滤函数
filter_out, // 输出过滤函数
BEV_OPT_CLOSE_ON_FREE, // 关闭filter的时候关闭bufferevent
0, // 清理的回调函数
0 // 给清理回调函数传递的参数
);
// 设置bufferevent的回调函数
bufferevent_setcb(bev_filter, read_cb, write_cb, event_cb, 0);
bufferevent_enable(bev_filter, EV_READ | EV_WRITE); // 开启读写权限
}
// 作为接收方,数据会先经过filter_in,然后在经过read_cb,所以必须在filter_in中把处理处理好,处理之后的数据会传递给read_cb
// 作为发送方,数据先经过filter_out,在经过write_cb
enum bufferevent_filter_result filter_in(struct evbuffer *src, struct evbuffer *dst, ev_ssize_t dst_limit,enum bufferevent_flush_mode mode, void *ctx) {
cout << __FUNCTION__ << endl;
char data[1024] = { 0 };
// 读取并清理,相当于吧src中的数据转移到data中
int len = evbuffer_remove(src, data, 1023);
// 在这里我们可以对data进行我们想要的处理
// 在这里简单的做大小写转换
for (int i = 0; i < len; i++) {
data[i] = toupper(data[i]);
}
// 添加到dst中,相当于吧data中的数据转移到dst中,这部分数据就是read_cb的数据
evbuffer_add(dst, data, len);
return BEV_OK;
}
enum bufferevent_filter_result filter_out(struct evbuffer *src, struct evbuffer *dst, ev_ssize_t dst_limit, enum bufferevent_flush_mode mode, void *ctx) {
cout << __FUNCTION__ << endl;
char data[1024] = { 0 };
// 读取并清理,相当于吧src中的数据转移到data中
evbuffer_remove(src, data, 1023);
// 在这里我们可以对data进行我们想要的处理
string str = "";
str += "*******";
str += data;
str += "*******";
// 添加到dst中,相当于吧data中的数据转移到dst中,这部分数据就是read_cb的数据
evbuffer_add(dst, str.c_str(), str.size());
return BEV_OK;
}
void read_cb(bufferevent* be, void* arg) {
cout << __FUNCTION__ << endl;
char data[1024] = { 0 };
auto len = bufferevent_read(be, data, 1023);
cout << "After deal : " << data << endl;
bufferevent_write(be, data, len);
}
void write_cb(bufferevent* be, void* arg) {
cout << __FUNCTION__ << endl;
}
// 错误,超时或者连接断开的时候会进入
void event_cb(bufferevent *be, short which, void *arg) {
cout << __FUNCTION__ << endl;
}