libevent 過濾器

#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;
}