libevent 中的bufferevent做server

#include <event2/event.h>
#include <event2/bufferevent.h>
#include <event2/listener.h>
#include <iostream>
#include <signal.h>
#include <cstring>
using namespace std;

void listen_cb(evconnlistener* ev, evutil_socket_t s, sockaddr* sin, int slen, void* arg);
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的上下文
	// BEV_OPT_CLOSE_ON_FREE 清理bufferevent的時候關閉底層的socket，也就是調用bufferevent_free的時候
	bufferevent* bev = bufferevent_socket_new(base, s, BEV_OPT_CLOSE_ON_FREE);
	bufferevent_enable(bev, EV_READ | EV_WRITE);  // 啓動讀寫權限


	// 設置水位,至少收到10個數據，read的回調函數纔會被訪問，這是低水位
	// 讀的高水位指的是：如果收到的字符數大於了高水位，本次事件處理函數只處理高水位個字符，
	// 剩下的字符會再次觸發回調函數
	bufferevent_setwatermark(bev, EV_READ, 10, 0); // 0就是無限制

	// 寫操作的低水位指的是：當發送的字符數小於低水位的時候，寫操作的回調函數會被調用
	bufferevent_setwatermark(bev, EV_WRITE, 10, 0);  // 寫操作的高水位不起作用
	
	// 設置超時事件
	timeval t = { 3,0 };
	bufferevent_set_timeouts(bev, &t, 0);
	// 添加回調函數
	bufferevent_setcb(bev, read_cb, write_cb, event_cb, base);
}

void read_cb(bufferevent* be, void* arg) {
	cout << __FUNCTION__ << endl;
	char data[1024] = { 0 };
	auto len = bufferevent_read(be, data, 1023);
	if (len <= 0) {
		return;
	}
	if (strstr(data, "quit") != nullptr) {
		cout << "quit" << endl;
		// 釋放資源,會關閉socket，是因爲是前面設置了flag
		bufferevent_free(be);
	}
	cout << "client say: " << data << endl;
	// 給客戶端回覆消息，其實是寫入到了輸出緩存
	bufferevent_write(be, "ok\r\n", 4);
}
void write_cb(bufferevent* be, void* arg) {
	cout << __FUNCTION__ << endl;
}
// 錯誤，超時或者連接斷開的時候會進入
void event_cb(bufferevent *be, short which, void *arg) {
	cout << __FUNCTION__ << endl;
	if (which & BEV_EVENT_TIMEOUT) {
		
		// 需要注意的是，即便是超時也分讀超時和寫超時，可以通過如下判斷
		if (which & BEV_EVENT_READING) {
			cout << "reading timeout" << endl;
			bufferevent_free(be);
		}
	}
}