libevent笔记-evconnlistener连接监听器

//libevent笔记-evconnlistener连接监听器
//转载请注明出处: yuliying的csdn博客.

//在服务器上监听一个端口来接受新的连接.
//传统的方法是调用 socket、bind、listen、accept 等函数.
//libevent为我们提供了一个封装 evconnlistener , 只要设置好回调函数即可接受处理新连接.

//evconnlistener回调函数的原型.
//当接受到一个新连接并且帮我们accept好后,会调用这个函数.
//sock是新连接的fd , addr是该连接的地址,ptr是evconnlistener_new()传入的参数.
//typedef void (*evconnlistener_cb)(struct evconnlistener *listener,
//    evutil_socket_t sock, struct sockaddr *addr, int len, void *ptr);

//创建evconnlistener.(传入一个fd,这个fd已经绑定到一个端口).
//如果cb参数为NULL,则暂时不能接受连接,直到我们设置了回调函数.
//flag参数传入一些选项,可以控制evconnlistener的行为.
//backlog参数是listen()函数的第二个参数,如果为负数,libevent会选择一个恰当的值.
//struct evconnlistener *evconnlistener_new(struct event_base *base,
//    evconnlistener_cb cb, void *ptr, unsigned flags, int backlog,
//    evutil_socket_t fd);

//创建evconnlistener.(其他同上函数,不同的是这个函数帮我们创建fd并bind)
//struct evconnlistener *evconnlistener_new_bind(struct event_base *base,
//    evconnlistener_cb cb, void *ptr, unsigned flags, int backlog,
//    const struct sockaddr *sa, int socklen);

//销毁evconnlistener.
//void evconnlistener_free(struct evconnlistener *lev);

//创建evconnlistener时的flag可以是下列选项的组合.
//1.LEV_OPT_LEAVE_SOCKETS_BLOCKING : 默认在accept新连接后会设为非阻塞,这个选项取消非阻塞设置.
//2.LEV_OPT_CLOSE_ON_FREE : 当销毁evconnlistener的时候关闭监听的socket.
//3.LEV_OPT_CLOSE_ON_EXEC : 在监听的socket上设置 FD_CLOEXEC 选项.
//4.LEV_OPT_REUSEABLE : 监听端口关闭后端口可立即重用.
//5.LEV_OPT_THREADSAFE : 给evconnlistener加锁. (Libevent 2.0.8-rc.)

//暂时启用/禁用evconnlistener.
//int evconnlistener_disable(struct evconnlistener *lev);
//int evconnlistener_enable(struct evconnlistener *lev);

//给evconnlistener设置回调函数(如果创建的时候没有设置的话)
//void evconnlistener_set_cb(struct evconnlistener *lev,evconnlistener_cb cb, void *arg);

//获取监听fd(Libevent 2.0.3-alpha)
//evutil_socket_t evconnlistener_get_fd(struct evconnlistener *lev);

//获取event_base
//struct event_base *evconnlistener_get_base(struct evconnlistener *lev);

//设置错误回调函数,监听发生错误的时候调用.(Libevent 2.0.8-rc)
//错误回调函数的原型: typedef void (*evconnlistener_errorcb)(struct evconnlistener *lis, void *ptr);
//void evconnlistener_set_error_cb(struct evconnlistener *lev,evconnlistener_errorcb errorcb);


//下面是一个简单的echo server

#include <event2/listener.h>
#include <event2/bufferevent.h>
#include <event2/buffer.h>
#include <errno.h>
#include <string.h>

//客户端读回调函数.(从客户端读到数据,并且数据已经自动存到input buffer里面了).
static void echo_read_cb(struct bufferevent *bev, void *ctx)
{
	//分别从bufferevent里拿到 input buffer 和 output buffer.
	struct evbuffer *input = bufferevent_get_input(bev);
	struct evbuffer *output = bufferevent_get_output(bev);

	//将input buffer里面收到的数据全部搬到output buffer , libevent会为我们自动发送给客户端.
	evbuffer_add_buffer(output, input);
}

//客户端事件回调函数(发生错误或者客户端关闭).
static void echo_event_cb(struct bufferevent *bev, short events, void *ctx)
{
	if (events & BEV_EVENT_ERROR)
			perror("Error from bufferevent");
	if (events & (BEV_EVENT_EOF | BEV_EVENT_ERROR)) {
			bufferevent_free(bev);
	}
}

//接受连接回调函数(接受到一个新连接.并且accept成功)
static void accept_conn_cb(struct evconnlistener *listener,
    evutil_socket_t fd, struct sockaddr *address, int socklen,void *ctx)
{
	struct event_base *base = evconnlistener_get_base(listener);
	//我们为新连接建立一个bufferevent来处理新连接.
	struct bufferevent *bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE);
    bufferevent_setcb(bev, echo_read_cb, NULL, echo_event_cb, NULL);
    bufferevent_enable(bev, EV_READ|EV_WRITE);
}

//接受连接发生错误回调函数.
static void accept_error_cb(struct evconnlistener *listener, void *ctx)
{
	struct event_base *base = evconnlistener_get_base(listener);
	int err = EVUTIL_SOCKET_ERROR();
	fprintf(stderr, "Got an error %d (%s) on the listener. "
			"Shutting down.\n", err, evutil_socket_error_to_string(err));
	event_base_loopexit(base, NULL);
}

int main(){
	//
	struct event_base *base = event_base_new();
	//
	int port = 9999;
	struct sockaddr_in sin;
	memset(&sin, 0, sizeof(sin));
	sin.sin_family = AF_INET;
	sin.sin_addr.s_addr = htonl(0);
	sin.sin_port = htons(port);
	//
	struct evconnlistener *listener = evconnlistener_new_bind( 
		base , accept_conn_cb , NULL , LEV_OPT_CLOSE_ON_FREE|LEV_OPT_REUSEABLE , -1 
		, (struct sockaddr*)&sin, sizeof(sin)
	);
	//
	evconnlistener_set_error_cb(listener, accept_error_cb);
	//
	event_base_dispatch(base);
	//
	evconnlistener_free(listener);
	event_base_free(base);
}