franta-hg@28
|
1 |
#include <string.h>
|
franta-hg@28
|
2 |
#include <errno.h>
|
franta-hg@28
|
3 |
#include <stdio.h>
|
franta-hg@28
|
4 |
#include <signal.h>
|
franta-hg@30
|
5 |
#include <stdlib.h>
|
franta-hg@28
|
6 |
|
franta-hg@28
|
7 |
#include <event2/bufferevent.h>
|
franta-hg@28
|
8 |
#include <event2/buffer.h>
|
franta-hg@28
|
9 |
#include <event2/listener.h>
|
franta-hg@28
|
10 |
#include <event2/util.h>
|
franta-hg@28
|
11 |
#include <event2/event.h>
|
franta-hg@29
|
12 |
#include <sys/un.h>
|
franta-hg@29
|
13 |
#include <unistd.h>
|
franta-hg@28
|
14 |
|
franta-hg@30
|
15 |
#define MAX_LINE 16384
|
franta-hg@30
|
16 |
|
franta-hg@28
|
17 |
static const char MESSAGE[] = "Hello, World!\n";
|
franta-hg@28
|
18 |
|
franta-hg@29
|
19 |
static const char PATH[] = "./roura";
|
franta-hg@28
|
20 |
|
franta-hg@30
|
21 |
static void listener_cb(struct evconnlistener *, evutil_socket_t, struct sockaddr *, int socklen, void *);
|
franta-hg@30
|
22 |
static void conn_read_cb(struct bufferevent *, void *);
|
franta-hg@28
|
23 |
static void conn_write_cb(struct bufferevent *, void *);
|
franta-hg@28
|
24 |
static void conn_event_cb(struct bufferevent *, short, void *);
|
franta-hg@28
|
25 |
static void signal_cb(evutil_socket_t, short, void *);
|
franta-hg@28
|
26 |
|
franta-hg@28
|
27 |
int main(int argc, char **argv) {
|
franta-hg@28
|
28 |
struct event_base *base;
|
franta-hg@28
|
29 |
struct evconnlistener *listener;
|
franta-hg@28
|
30 |
struct event *signal_event;
|
franta-hg@28
|
31 |
|
franta-hg@29
|
32 |
struct sockaddr_un sun;
|
franta-hg@28
|
33 |
|
franta-hg@28
|
34 |
base = event_base_new();
|
franta-hg@28
|
35 |
if (!base) {
|
franta-hg@28
|
36 |
fprintf(stderr, "Could not initialize libevent!\n");
|
franta-hg@28
|
37 |
return 1;
|
franta-hg@28
|
38 |
}
|
franta-hg@28
|
39 |
|
franta-hg@29
|
40 |
memset(&sun, 0, sizeof (sun));
|
franta-hg@29
|
41 |
sun.sun_family = AF_UNIX;
|
franta-hg@29
|
42 |
strcpy(sun.sun_path, PATH);
|
franta-hg@28
|
43 |
|
franta-hg@28
|
44 |
listener = evconnlistener_new_bind(base, listener_cb, (void *) base,
|
franta-hg@28
|
45 |
LEV_OPT_REUSEABLE | LEV_OPT_CLOSE_ON_FREE, -1,
|
franta-hg@29
|
46 |
(struct sockaddr*) &sun,
|
franta-hg@29
|
47 |
sizeof (sun));
|
franta-hg@28
|
48 |
|
franta-hg@28
|
49 |
if (!listener) {
|
franta-hg@29
|
50 |
fprintf(stderr, "Could not create domain socket: %s!\n", PATH);
|
franta-hg@28
|
51 |
return 1;
|
franta-hg@28
|
52 |
}
|
franta-hg@28
|
53 |
|
franta-hg@28
|
54 |
signal_event = evsignal_new(base, SIGINT, signal_cb, (void *) base);
|
franta-hg@28
|
55 |
|
franta-hg@28
|
56 |
if (!signal_event || event_add(signal_event, NULL) < 0) {
|
franta-hg@28
|
57 |
fprintf(stderr, "Could not create/add a signal event!\n");
|
franta-hg@28
|
58 |
return 1;
|
franta-hg@28
|
59 |
}
|
franta-hg@28
|
60 |
|
franta-hg@28
|
61 |
event_base_dispatch(base);
|
franta-hg@28
|
62 |
|
franta-hg@28
|
63 |
evconnlistener_free(listener);
|
franta-hg@28
|
64 |
event_free(signal_event);
|
franta-hg@28
|
65 |
event_base_free(base);
|
franta-hg@30
|
66 |
|
franta-hg@29
|
67 |
// smažeme soket na disku / soubor -- jinak by program příště spadl na evconnlistener_new_bind()
|
franta-hg@29
|
68 |
unlink(PATH);
|
franta-hg@28
|
69 |
|
franta-hg@28
|
70 |
printf("done\n");
|
franta-hg@28
|
71 |
return 0;
|
franta-hg@28
|
72 |
}
|
franta-hg@28
|
73 |
|
franta-hg@28
|
74 |
static void listener_cb(struct evconnlistener *listener, evutil_socket_t fd, struct sockaddr *sa, int socklen, void *user_data) {
|
franta-hg@28
|
75 |
struct event_base *base = (event_base *) user_data;
|
franta-hg@28
|
76 |
struct bufferevent *bev;
|
franta-hg@28
|
77 |
|
franta-hg@30
|
78 |
//bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE); // zavírá spojení
|
franta-hg@30
|
79 |
bev = bufferevent_socket_new(base, fd, 0); // nezavírá spojení
|
franta-hg@28
|
80 |
if (!bev) {
|
franta-hg@28
|
81 |
fprintf(stderr, "Error constructing bufferevent!");
|
franta-hg@28
|
82 |
event_base_loopbreak(base);
|
franta-hg@28
|
83 |
return;
|
franta-hg@28
|
84 |
}
|
franta-hg@30
|
85 |
bufferevent_setcb(bev, conn_read_cb, conn_write_cb, conn_event_cb, NULL);
|
franta-hg@30
|
86 |
bufferevent_setwatermark(bev, EV_READ, 0, MAX_LINE);
|
franta-hg@30
|
87 |
bufferevent_enable(bev, EV_READ | EV_WRITE);
|
franta-hg@28
|
88 |
|
franta-hg@28
|
89 |
printf("někdo se k nám připojil! / %p\n", bev);
|
franta-hg@28
|
90 |
|
franta-hg@28
|
91 |
bufferevent_write(bev, MESSAGE, strlen(MESSAGE));
|
franta-hg@28
|
92 |
}
|
franta-hg@28
|
93 |
|
franta-hg@30
|
94 |
static void conn_read_cb(struct bufferevent *bev, void *user_data) {
|
franta-hg@30
|
95 |
/* This callback is invoked when there is data to read on bev. */
|
franta-hg@30
|
96 |
struct evbuffer *input = bufferevent_get_input(bev);
|
franta-hg@30
|
97 |
struct evbuffer *output = bufferevent_get_output(bev);
|
franta-hg@30
|
98 |
struct event_base *base = (event_base *) user_data;
|
franta-hg@30
|
99 |
|
franta-hg@30
|
100 |
size_t len = evbuffer_get_length(input);
|
franta-hg@30
|
101 |
char *data;
|
franta-hg@30
|
102 |
data = (char*) malloc(len);
|
franta-hg@30
|
103 |
evbuffer_copyout(input, data, len);
|
franta-hg@30
|
104 |
|
franta-hg@30
|
105 |
printf("we got some data: %s / %d\n", data, memcmp(data, "exit", len));
|
franta-hg@30
|
106 |
|
franta-hg@30
|
107 |
if (memcmp(data, "exit\n", len) == 0) {
|
franta-hg@30
|
108 |
struct timeval delay = {2, 123};
|
franta-hg@30
|
109 |
event_base_loopexit(base, &delay);
|
franta-hg@30
|
110 |
}
|
franta-hg@30
|
111 |
|
franta-hg@30
|
112 |
/* Copy all the data from the input buffer to the output buffer. */
|
franta-hg@30
|
113 |
evbuffer_add_buffer(output, input);
|
franta-hg@30
|
114 |
free(data);
|
franta-hg@30
|
115 |
}
|
franta-hg@30
|
116 |
|
franta-hg@28
|
117 |
static void conn_write_cb(struct bufferevent *bev, void *user_data) {
|
franta-hg@28
|
118 |
struct evbuffer *output = bufferevent_get_output(bev);
|
franta-hg@28
|
119 |
if (evbuffer_get_length(output) == 0) {
|
franta-hg@28
|
120 |
printf("flushed answer / %p\n", bev);
|
franta-hg@28
|
121 |
bufferevent_free(bev);
|
franta-hg@28
|
122 |
}
|
franta-hg@28
|
123 |
}
|
franta-hg@28
|
124 |
|
franta-hg@28
|
125 |
static void conn_event_cb(struct bufferevent *bev, short events, void *user_data) {
|
franta-hg@28
|
126 |
if (events & BEV_EVENT_EOF) {
|
franta-hg@28
|
127 |
printf("Connection closed.\n");
|
franta-hg@28
|
128 |
} else if (events & BEV_EVENT_ERROR) {
|
franta-hg@30
|
129 |
printf("Got an error on the connection: %s\n", strerror(errno));
|
franta-hg@28
|
130 |
}
|
franta-hg@29
|
131 |
|
franta-hg@28
|
132 |
// None of the other events can happen here, since we haven't enabled timeouts
|
franta-hg@28
|
133 |
bufferevent_free(bev);
|
franta-hg@28
|
134 |
}
|
franta-hg@28
|
135 |
|
franta-hg@28
|
136 |
static void signal_cb(evutil_socket_t sig, short events, void *user_data) {
|
franta-hg@28
|
137 |
struct event_base *base = (event_base *) user_data;
|
franta-hg@28
|
138 |
struct timeval delay = {2, 123};
|
franta-hg@28
|
139 |
|
franta-hg@28
|
140 |
printf("Zachycen SIGINT (Ctrl+C); ukončuji program během %ld sekund a %ld mikrosekund.\n", delay.tv_sec, delay.tv_usec);
|
franta-hg@28
|
141 |
|
franta-hg@28
|
142 |
event_base_loopexit(base, &delay);
|
franta-hg@28
|
143 |
}
|