libevent(五)使用例子

客户端：

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

#include <event2/event.h>
#include <event2/event_struct.h>
#include <event2/buffer.h>
#include <event2/bufferevent.h>

#define DEFAULT_HOST "127.0.0.1"
#define DEFAULT_PORT 8888
#define MAXLINE 1024

void cmd_msg_cb(int fd, short events, void *arg);
void socket_read_cb(struct bufferevent *bev, void *arg);
void event_cb(struct bufferevent *bev, short event, void *arg);

int main(int argc, char **argv) {
    char *host = DEFAULT_HOST;
    int port = DEFAULT_PORT;
    if (argc == ) {
        port = atoi(argv[]);
    } else if (argc == ){
        host = argv[];
        port = atoi(argv[]);
    }
    printf("host: %s, port: %d\n", host, port);

    struct event_base *base = event_base_new();

    struct bufferevent *bev = bufferevent_socket_new(base, -, BEV_OPT_CLOSE_ON_FREE);

    struct event *ev_cmd = event_new(base, STDIN_FILENO, EV_READ | EV_PERSIST, cmd_msg_cb, (void*) bev);
    event_add(ev_cmd, NULL);

    struct sockaddr_in servaddr;
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(port);
    inet_pton(AF_INET, host, &servaddr.sin_addr);

    bufferevent_socket_connect(bev, (struct sockaddr *)&servaddr, sizeof(servaddr));

    bufferevent_setcb(bev, socket_read_cb, NULL, event_cb, ev_cmd);
    bufferevent_enable(bev, EV_READ | EV_PERSIST);

    event_base_dispatch(base);

    return ;
}

void cmd_msg_cb(int fd, short events, void *arg) {
    char msg[MAXLINE];
    int len = read(fd, msg, MAXLINE);
    if (len <= ) {
        perror("read error");
        exit();
    }

    struct bufferevent *bev = (struct bufferevent*)arg;
    // 把终端的输入发送到服务器
    bufferevent_write(bev, msg, len);
}

void socket_read_cb(struct bufferevent *bev, void *arg) {
    char msg[MAXLINE];
    int len = bufferevent_read(bev, msg, MAXLINE);
    msg[len] = '\0';
    printf("msg from server: %s\n", msg);
}

void event_cb(struct bufferevent *bev, short event, void *arg) {
    if (event & BEV_EVENT_EOF) {
        printf("server closed\n");
    } else if (event & BEV_EVENT_ERROR) {
        printf("unknown error\n");
    } else if (event & BEV_EVENT_CONNECTED) {
        printf("Connect to the server successfully\n");
        return;
    }
    bufferevent_free(bev);
    struct event *ev = (struct event*)arg;
    event_free(ev);
}

服务器：

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

#include <event2/event.h>
#include <event2/event_struct.h>
#include <event2/buffer.h>
#include <event2/bufferevent.h>
#include <event2/bufferevent_struct.h>
#include <event2/listener.h>

#define DEFAULT_PORT 8888

#define MAXLINE 1024
#define LISTENQ 10

void listener_cb(struct evconnlistener *listener, int fd, struct sockaddr *sock,
                 int socklen, void *arg);
void socket_read_cb(struct bufferevent *bev, void *arg);
void event_cb(struct bufferevent *bev, short event, void *arg);

int main(int argc, char **argv) {
    int port = DEFAULT_PORT;
    if (argc == ) {
        port = atoi(argv[]);
    }
    printf("http port: %d\n", port);

    //
    struct sockaddr_in servaddr;
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(port);

    struct event_base *base = event_base_new();

    evconnlistener *listener = evconnlistener_new_bind(base, listener_cb, base,
                                                       LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE,
                                                       , (struct sockaddr*) &servaddr, sizeof(servaddr));

    event_base_dispatch(base);
    event_base_free(base);

    return ;
}

void listener_cb(struct evconnlistener *listener, int fd, struct sockaddr *sock,
                 int socklen, void *arg) {
    printf("accept a client: %d\n", fd);

    event_base *base = (struct event_base*)arg;

    struct bufferevent *bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE);
    bufferevent_setcb(bev, socket_read_cb, NULL, event_cb, arg);
    bufferevent_enable(bev, EV_READ | EV_PERSIST);
}

void socket_read_cb(struct bufferevent *bev, void *arg) {
    char msg[MAXLINE];
    int len = bufferevent_read(bev, msg, MAXLINE);
    msg[len] = '\0';
    printf("msg from client %d: %s\n", bev->ev_read.ev_fd, msg);
    char reply_msg[MAXLINE] = "server has got your msg";
    bufferevent_write(bev, reply_msg, strlen(reply_msg));
}

void event_cb(struct bufferevent *bev, short event, void *arg) {
    if (event & BEV_EVENT_EOF) {
        printf("client %d closed\n", bev->ev_read.ev_fd);
    } else if (event & BEV_EVENT_ERROR) {
        printf("unknown error\n");
    }
    bufferevent_free(bev);
}