#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>

#include <event.h>


#define PORT        25341
#define BACKLOG     5
#define MEM_SIZE    10

struct event_base* base;
struct sock_ev {
    struct event* read_ev;
    struct event* write_ev;
    char* buffer;    
    
};


int setnonblock(int fd)
{
  int flags;

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

void release_sock_event(struct sock_ev* ev)
{
    event_del(ev->read_ev);
    free(ev->read_ev);
    free(ev->write_ev);
    free(ev->buffer);
    free(ev);
    ev = NULL;
}

void on_write(int sock, short event, void* arg)
{
    char* buffer = (char*)arg;
    printf("send:[%d]%s,%u\n",buffer,buffer,strlen(buffer));
    send(sock, buffer, strlen(buffer), 0);

    free(buffer);
}

void on_read(int sock, short event, void* arg)
{
    struct event* write_ev;
    int size;
    struct sock_ev* ev = (struct sock_ev*)arg;
    ev->buffer = (char*)malloc(MEM_SIZE);
    bzero(ev->buffer, MEM_SIZE);
    size = recv(sock, ev->buffer, MEM_SIZE-2, 0);
    printf("begin\n");
ev->buffer[size] = '\n';
    printf("paddr:%d\n",ev->buffer);
    printf("receive data:%s, size:%d\n", ev->buffer, size);
    if (size == 0 ) { // || size > MEM_SIZE-1 不加这个判断条件，在输入超长的时候就会报错
        release_sock_event(ev);
        close(sock);
        return;
    }
    printf("end1\n");

    event_set(ev->write_ev, sock, EV_WRITE, on_write, ev->buffer);
    event_base_set(base, ev->write_ev);
    event_add(ev->write_ev, NULL);
    
    printf("end2\n");
       
}

void on_accept(int sock, short event, void* arg)
{
    struct sockaddr_in cli_addr;
    int newfd, sin_size;
    struct sock_ev* ev = (struct sock_ev*)malloc(sizeof(struct sock_ev));
    ev->read_ev = (struct event*)malloc(sizeof(struct event));
    ev->write_ev = (struct event*)malloc(sizeof(struct event));
    sin_size = sizeof(struct sockaddr_in);
    newfd = accept(sock, (struct sockaddr*)&cli_addr, &sin_size);
    setnonblock(newfd);
    event_set(ev->read_ev, newfd, EV_READ|EV_PERSIST, on_read, ev);  //EV_PERSIST  去掉这个，只read一次就退出了
    event_set(ev->write_ev, newfd, EV_WRITE|EV_PERSIST, on_write, ev->buffer);
    event_base_set(base, ev->write_ev);
    event_base_set(base, ev->read_ev);
    event_add(ev->read_ev, NULL);
}

int main(int argc, char* argv[])
{
    struct sockaddr_in my_addr;
    int sock;

    sock = socket(AF_INET, SOCK_STREAM, 0);
    int yes = 1;
    setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int));
    memset(&my_addr, 0, sizeof(my_addr));
    my_addr.sin_family = AF_INET;
    my_addr.sin_port = htons(PORT);
    my_addr.sin_addr.s_addr = INADDR_ANY;
    bind(sock, (struct sockaddr*)&my_addr, sizeof(struct sockaddr));
    listen(sock, BACKLOG);

    struct event listen_ev;
    base = event_base_new();
    setnonblock(sock);
    event_set(&listen_ev, sock, EV_READ, on_accept, NULL); //EV_PERSIST  
    event_base_set(base, &listen_ev);
    event_add(&listen_ev, NULL);
    event_base_dispatch(base);

    return 0;
}


在学习libevent的时候，遇到一个问题，在上面的例子里，如果对服务器发送超过10个以上长度的字符串的时候，服务器就会没有反应，求牛人指点。

我自己觉得问题出在这个函数里面的buffer处理上

void on_read(int sock, short event, void* arg)
{
    struct event* write_ev;
    int size;
    struct sock_ev* ev = (struct sock_ev*)arg;
    ev->buffer = (char*)malloc(MEM_SIZE);
    bzero(ev->buffer, MEM_SIZE);
    size = recv(sock, ev->buffer, MEM_SIZE-2, 0);
    printf("begin\n");
ev->buffer[size] = '\n';
    printf("paddr:%d\n",ev->buffer);
    printf("receive data:%s, size:%d\n", ev->buffer, size);
    if (size == 0 ) { // || size > MEM_SIZE-1 不加这个判断条件，在输入超长的时候就会报错
        release_sock_event(ev);
        close(sock);
        return;
    }
    printf("end1\n");

    event_set(ev->write_ev, sock, EV_WRITE, on_write, ev->buffer);
    event_base_set(base, ev->write_ev);
    event_add(ev->write_ev, NULL);
    
    printf("end2\n");
       
}