#include <event.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>

#define SERVER_PORT 8080
int debug = 0;

struct client
{
	int fd;
	struct bufferevent * buf_ev;
};

int setnonblock(int fd)
{
	int flags;

	flags = fcntl(fd,F_GETFL);
	flags |= O_NONBLOCK;
	fcntl(fd,F_SETFL,flags);
}

/*read callback function .*/
void buf_read_callback(struct bufferevent *incoming,
					   void *arg)
{
	printf("this is called in buf_read_callback\n");
	struct evbuffer * evreturn;
	char *req;

	req = evbuffer_readline(incoming->input);

	printf("req is %s\n",req);
	if(req == NULL)
		return;

	evreturn = evbuffer_new();
	evbuffer_add_printf(evreturn,"YOU SAID %s\n",req);
	bufferevent_write_buffer(incoming,evreturn);
	evbuffer_free(evreturn);
	free(req);
}

/*write callback function.*/
void buf_write_callback(struct bufferevent *bev,
						void *arg)
{
	printf("this is called in buf_write_callback\n");
}

/*error callback function.*/
void buf_error_callback(struct bufferevent *bev,    
						short what,void *arg)
{
	struct client *client = (struct client*)arg;
	bufferevent_free(client->buf_ev);
	close(client->fd);
	free(client);
}

/*accept callback function.*/
void accept_callback(int fd,
					 short ev,void *arg)
{
	int client_fd;
	struct sockaddr_in client_addr;
	socklen_t client_len = sizeof(client_addr);
	struct client *client;

	client_fd = accept(fd,
		(struct sockaddr*)&client_addr,
		&client_len);

	if(client_fd < 0)
	{
		printf("Client:accept() faild\n");
		return;
	}

	/*set socket fd to none block.*/
	setnonblock(client_fd);

	/*new a struct client.*/
	client = calloc(1,sizeof(*client));
	if(client == NULL)
	{
		printf("calloc error\n");
		return;
	}// if
	client->fd = client_fd;

	client->buf_ev = bufferevent_new(client_fd,
		buf_read_callback,
		buf_write_callback,
		buf_error_callback,
		client);

	bufferevent_enable(client->buf_ev,EV_READ);
}

int main(int argc,
		 char **argv)
{
	int socketlisten;
	struct sockaddr_in addresslisten;
	struct event accept_event;
	int reuse = 1;

	/*initialize the event_base in this thread and if this is single-thread application,there is no need for saving the event_base.*/
	event_init();

	socketlisten = socket(AF_INET, SOCK_STREAM, 0);

	if (socketlisten < 0)
	{
		fprintf(stderr,"Failed to create listen socket");
		return 1;
	}

	memset(&addresslisten, 0, sizeof(addresslisten));

	addresslisten.sin_family = AF_INET;
	addresslisten.sin_addr.s_addr = INADDR_ANY;
	addresslisten.sin_port = htons(SERVER_PORT);

	if (bind(socketlisten,
		(struct sockaddr *)&addresslisten,
		sizeof(addresslisten)) < 0)
	{
		fprintf(stderr,"Failed to bind");
		return 1;
	}

	if (listen(socketlisten, 5) < 0)
	{
		fprintf(stderr,"Failed to listen to socket");
		return 1;
	}

	setsockopt(socketlisten,
		SOL_SOCKET,
		SO_REUSEADDR,
		&reuse,
		sizeof(reuse));

	setnonblock(socketlisten);

	/*set accept callback function including read/write/error callback function.
	  EV_READ:Wait for a socket or FD to become readable  
	  EV_WRITE:Wait for a socket or FD to become writeable 
	*/
	event_set(&accept_event,
		socketlisten,
		EV_READ|EV_PERSIST,
		accept_callback,
		NULL);

	/*add an event to the set of pending events.*/
	event_add(&accept_event,
		NULL);

	/*Loop to process events.*/
	event_dispatch();

	close(socketlisten);
	return 0;
}

daoluan@daoluan:~$ telnet 192.168.128.99 8080

root@daoluan:~# telnet 192.168.128.99 8080
Trying 192.168.128.99...
Connected to 192.168.128.99.
Escape character is '^]'.
hello world.
YOU SAID hello world.