c++/domain-socket-bridge/domain-socket-bridge.c
author František Kučera <franta-hg@frantovo.cz>
Fri, 18 Nov 2016 21:41:23 +0100
changeset 28 1fec66105b8e
child 29 9c0f52aa4772
permissions -rw-r--r--
libevent: hello world
     1 #include <string.h>
     2 #include <errno.h>
     3 #include <stdio.h>
     4 #include <signal.h>
     5 
     6 #include <event2/bufferevent.h>
     7 #include <event2/buffer.h>
     8 #include <event2/listener.h>
     9 #include <event2/util.h>
    10 #include <event2/event.h>
    11 
    12 static const char MESSAGE[] = "Hello, World!\n";
    13 
    14 static const int PORT = 9995;
    15 
    16 static void listener_cb(struct evconnlistener *, evutil_socket_t,
    17 		struct sockaddr *, int socklen, void *);
    18 static void conn_write_cb(struct bufferevent *, void *);
    19 static void conn_event_cb(struct bufferevent *, short, void *);
    20 static void signal_cb(evutil_socket_t, short, void *);
    21 
    22 int main(int argc, char **argv) {
    23 	struct event_base *base;
    24 	struct evconnlistener *listener;
    25 	struct event *signal_event;
    26 
    27 	struct sockaddr_in sin;
    28 
    29 	base = event_base_new();
    30 	if (!base) {
    31 		fprintf(stderr, "Could not initialize libevent!\n");
    32 		return 1;
    33 	}
    34 
    35 	memset(&sin, 0, sizeof (sin));
    36 	sin.sin_family = AF_INET;
    37 	sin.sin_port = htons(PORT);
    38 
    39 	listener = evconnlistener_new_bind(base, listener_cb, (void *) base,
    40 			LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE, -1,
    41 			(struct sockaddr*) &sin,
    42 			sizeof (sin));
    43 
    44 	if (!listener) {
    45 		fprintf(stderr, "Could not create a listener!\n");
    46 		return 1;
    47 	}
    48 
    49 	signal_event = evsignal_new(base, SIGINT, signal_cb, (void *) base);
    50 
    51 	if (!signal_event || event_add(signal_event, NULL) < 0) {
    52 		fprintf(stderr, "Could not create/add a signal event!\n");
    53 		return 1;
    54 	}
    55 
    56 	event_base_dispatch(base);
    57 
    58 	evconnlistener_free(listener);
    59 	event_free(signal_event);
    60 	event_base_free(base);
    61 
    62 	printf("done\n");
    63 	return 0;
    64 }
    65 
    66 static void listener_cb(struct evconnlistener *listener, evutil_socket_t fd, struct sockaddr *sa, int socklen, void *user_data) {
    67 	struct event_base *base = (event_base *) user_data;
    68 	struct bufferevent *bev;
    69 
    70 	bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE); // zavírá spojení
    71 	//bev = bufferevent_socket_new(base, fd, 0); // nezavírá spojení
    72 	if (!bev) {
    73 		fprintf(stderr, "Error constructing bufferevent!");
    74 		event_base_loopbreak(base);
    75 		return;
    76 	}
    77 	bufferevent_setcb(bev, NULL, conn_write_cb, conn_event_cb, NULL);
    78 	bufferevent_enable(bev, EV_WRITE);
    79 	bufferevent_disable(bev, EV_READ);
    80 
    81 	printf("někdo se k nám připojil! / %p\n", bev);
    82 
    83 	bufferevent_write(bev, MESSAGE, strlen(MESSAGE));
    84 }
    85 
    86 static void conn_write_cb(struct bufferevent *bev, void *user_data) {
    87 	struct evbuffer *output = bufferevent_get_output(bev);
    88 	if (evbuffer_get_length(output) == 0) {
    89 		printf("flushed answer / %p\n", bev);
    90 		bufferevent_free(bev);
    91 	}
    92 }
    93 
    94 static void conn_event_cb(struct bufferevent *bev, short events, void *user_data) {
    95 	if (events & BEV_EVENT_EOF) {
    96 		printf("Connection closed.\n");
    97 	} else if (events & BEV_EVENT_ERROR) {
    98 		printf("Got an error on the connection: %s\n",
    99 				strerror(errno)); /*XXX win32*/
   100 	}
   101 	
   102 	// None of the other events can happen here, since we haven't enabled timeouts
   103 	bufferevent_free(bev);
   104 }
   105 
   106 static void signal_cb(evutil_socket_t sig, short events, void *user_data) {
   107 	struct event_base *base = (event_base *) user_data;
   108 	struct timeval delay = {2, 123};
   109 
   110 	printf("Zachycen SIGINT (Ctrl+C); ukončuji program během %ld sekund a %ld mikrosekund.\n", delay.tv_sec, delay.tv_usec);
   111 
   112 	event_base_loopexit(base, &delay);
   113 }