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@31
|
21 |
static void listener_cb(evutil_socket_t, short, 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@31
|
27 |
static char rot13_char(char);
|
franta-hg@31
|
28 |
|
franta-hg@28
|
29 |
int main(int argc, char **argv) {
|
franta-hg@28
|
30 |
struct event_base *base;
|
franta-hg@31
|
31 |
evutil_socket_t listener;
|
franta-hg@31
|
32 |
struct event *listener_event;
|
franta-hg@28
|
33 |
struct event *signal_event;
|
franta-hg@28
|
34 |
|
franta-hg@29
|
35 |
struct sockaddr_un sun;
|
franta-hg@28
|
36 |
|
franta-hg@31
|
37 |
setvbuf(stdout, NULL, _IONBF, 0);
|
franta-hg@31
|
38 |
|
franta-hg@28
|
39 |
base = event_base_new();
|
franta-hg@28
|
40 |
if (!base) {
|
franta-hg@28
|
41 |
fprintf(stderr, "Could not initialize libevent!\n");
|
franta-hg@28
|
42 |
return 1;
|
franta-hg@28
|
43 |
}
|
franta-hg@28
|
44 |
|
franta-hg@29
|
45 |
memset(&sun, 0, sizeof (sun));
|
franta-hg@29
|
46 |
sun.sun_family = AF_UNIX;
|
franta-hg@29
|
47 |
strcpy(sun.sun_path, PATH);
|
franta-hg@28
|
48 |
|
franta-hg@31
|
49 |
listener = socket(AF_UNIX, SOCK_STREAM, 0);
|
franta-hg@31
|
50 |
evutil_make_socket_nonblocking(listener);
|
franta-hg@28
|
51 |
|
franta-hg@31
|
52 |
if (bind(listener, (struct sockaddr*) &sun, sizeof (sun)) < 0) {
|
franta-hg@29
|
53 |
fprintf(stderr, "Could not create domain socket: %s!\n", PATH);
|
franta-hg@28
|
54 |
return 1;
|
franta-hg@28
|
55 |
}
|
franta-hg@28
|
56 |
|
franta-hg@31
|
57 |
if (listen(listener, 16) < 0) {
|
franta-hg@31
|
58 |
fprintf(stderr, "Could not listen\n");
|
franta-hg@31
|
59 |
return 1;
|
franta-hg@31
|
60 |
}
|
franta-hg@31
|
61 |
|
franta-hg@31
|
62 |
listener_event = event_new(base, listener, EV_READ | EV_PERSIST, listener_cb, (void*) base);
|
franta-hg@31
|
63 |
|
franta-hg@31
|
64 |
if (!listener_event) {
|
franta-hg@31
|
65 |
fprintf(stderr, "Could not do event_new\n");
|
franta-hg@31
|
66 |
return 1;
|
franta-hg@31
|
67 |
}
|
franta-hg@31
|
68 |
|
franta-hg@31
|
69 |
event_add(listener_event, NULL);
|
franta-hg@31
|
70 |
|
franta-hg@28
|
71 |
signal_event = evsignal_new(base, SIGINT, signal_cb, (void *) base);
|
franta-hg@28
|
72 |
|
franta-hg@28
|
73 |
if (!signal_event || event_add(signal_event, NULL) < 0) {
|
franta-hg@28
|
74 |
fprintf(stderr, "Could not create/add a signal event!\n");
|
franta-hg@28
|
75 |
return 1;
|
franta-hg@28
|
76 |
}
|
franta-hg@28
|
77 |
|
franta-hg@28
|
78 |
event_base_dispatch(base);
|
franta-hg@28
|
79 |
|
franta-hg@31
|
80 |
event_free(listener_event);
|
franta-hg@28
|
81 |
event_free(signal_event);
|
franta-hg@28
|
82 |
event_base_free(base);
|
franta-hg@30
|
83 |
|
franta-hg@29
|
84 |
// smažeme soket na disku / soubor -- jinak by program příště spadl na evconnlistener_new_bind()
|
franta-hg@29
|
85 |
unlink(PATH);
|
franta-hg@28
|
86 |
|
franta-hg@28
|
87 |
printf("done\n");
|
franta-hg@28
|
88 |
return 0;
|
franta-hg@28
|
89 |
}
|
franta-hg@28
|
90 |
|
franta-hg@31
|
91 |
static void listener_cb(evutil_socket_t listener, short event, void *user_data) {
|
franta-hg@28
|
92 |
struct event_base *base = (event_base *) user_data;
|
franta-hg@28
|
93 |
struct bufferevent *bev;
|
franta-hg@28
|
94 |
|
franta-hg@31
|
95 |
struct sockaddr_storage ss;
|
franta-hg@31
|
96 |
socklen_t slen = sizeof (ss);
|
franta-hg@31
|
97 |
int fd = accept(listener, (struct sockaddr*) &ss, &slen);
|
franta-hg@31
|
98 |
if (fd < 0) {
|
franta-hg@31
|
99 |
fprintf(stderr, "Unable to accept(): %d", fd);
|
franta-hg@31
|
100 |
return;
|
franta-hg@31
|
101 |
} else if (fd > FD_SETSIZE) {
|
franta-hg@31
|
102 |
close(fd);
|
franta-hg@31
|
103 |
return;
|
franta-hg@31
|
104 |
}
|
franta-hg@31
|
105 |
|
franta-hg@31
|
106 |
evutil_make_socket_nonblocking(fd);
|
franta-hg@31
|
107 |
|
franta-hg@31
|
108 |
bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE);
|
franta-hg@28
|
109 |
if (!bev) {
|
franta-hg@28
|
110 |
fprintf(stderr, "Error constructing bufferevent!");
|
franta-hg@28
|
111 |
event_base_loopbreak(base);
|
franta-hg@28
|
112 |
return;
|
franta-hg@28
|
113 |
}
|
franta-hg@30
|
114 |
bufferevent_setcb(bev, conn_read_cb, conn_write_cb, conn_event_cb, NULL);
|
franta-hg@30
|
115 |
bufferevent_setwatermark(bev, EV_READ, 0, MAX_LINE);
|
franta-hg@30
|
116 |
bufferevent_enable(bev, EV_READ | EV_WRITE);
|
franta-hg@28
|
117 |
|
franta-hg@28
|
118 |
printf("někdo se k nám připojil! / %p\n", bev);
|
franta-hg@28
|
119 |
|
franta-hg@28
|
120 |
bufferevent_write(bev, MESSAGE, strlen(MESSAGE));
|
franta-hg@28
|
121 |
}
|
franta-hg@28
|
122 |
|
franta-hg@30
|
123 |
static void conn_read_cb(struct bufferevent *bev, void *user_data) {
|
franta-hg@30
|
124 |
/* This callback is invoked when there is data to read on bev. */
|
franta-hg@30
|
125 |
struct evbuffer *input = bufferevent_get_input(bev);
|
franta-hg@30
|
126 |
struct evbuffer *output = bufferevent_get_output(bev);
|
franta-hg@30
|
127 |
struct event_base *base = (event_base *) user_data;
|
franta-hg@30
|
128 |
|
franta-hg@30
|
129 |
size_t len = evbuffer_get_length(input);
|
franta-hg@30
|
130 |
char *data;
|
franta-hg@31
|
131 |
/*
|
franta-hg@30
|
132 |
data = (char*) malloc(len);
|
franta-hg@30
|
133 |
evbuffer_copyout(input, data, len);
|
franta-hg@31
|
134 |
*/
|
franta-hg@30
|
135 |
|
franta-hg@31
|
136 |
size_t n;
|
franta-hg@31
|
137 |
int i;
|
franta-hg@31
|
138 |
while ((data = evbuffer_readln(input, &n, EVBUFFER_EOL_LF))) {
|
franta-hg@31
|
139 |
for (i = 0; i < n; ++i)
|
franta-hg@31
|
140 |
data[i] = rot13_char(data[i]);
|
franta-hg@31
|
141 |
evbuffer_add(output, data, n);
|
franta-hg@31
|
142 |
evbuffer_add(output, "\n", 1);
|
franta-hg@31
|
143 |
free(data);
|
franta-hg@31
|
144 |
}
|
franta-hg@30
|
145 |
|
franta-hg@31
|
146 |
printf("we got some data...\n");
|
franta-hg@31
|
147 |
|
franta-hg@31
|
148 |
/*
|
franta-hg@30
|
149 |
if (memcmp(data, "exit\n", len) == 0) {
|
franta-hg@30
|
150 |
struct timeval delay = {2, 123};
|
franta-hg@30
|
151 |
event_base_loopexit(base, &delay);
|
franta-hg@30
|
152 |
}
|
franta-hg@31
|
153 |
*/
|
franta-hg@31
|
154 |
|
franta-hg@31
|
155 |
if (evbuffer_get_length(input) >= MAX_LINE) {
|
franta-hg@31
|
156 |
/* Too long; just process what there is and go on so that the buffer
|
franta-hg@31
|
157 |
* doesn't grow infinitely long. */
|
franta-hg@31
|
158 |
char buf[1024];
|
franta-hg@31
|
159 |
while (evbuffer_get_length(input)) {
|
franta-hg@31
|
160 |
int n = evbuffer_remove(input, buf, sizeof (buf));
|
franta-hg@31
|
161 |
for (i = 0; i < n; ++i)
|
franta-hg@31
|
162 |
buf[i] = rot13_char(buf[i]);
|
franta-hg@31
|
163 |
evbuffer_add(output, buf, n);
|
franta-hg@31
|
164 |
}
|
franta-hg@31
|
165 |
evbuffer_add(output, "\n", 1);
|
franta-hg@31
|
166 |
}
|
franta-hg@30
|
167 |
|
franta-hg@30
|
168 |
/* Copy all the data from the input buffer to the output buffer. */
|
franta-hg@31
|
169 |
/*
|
franta-hg@30
|
170 |
evbuffer_add_buffer(output, input);
|
franta-hg@30
|
171 |
free(data);
|
franta-hg@31
|
172 |
*/
|
franta-hg@30
|
173 |
}
|
franta-hg@30
|
174 |
|
franta-hg@28
|
175 |
static void conn_write_cb(struct bufferevent *bev, void *user_data) {
|
franta-hg@28
|
176 |
struct evbuffer *output = bufferevent_get_output(bev);
|
franta-hg@28
|
177 |
if (evbuffer_get_length(output) == 0) {
|
franta-hg@28
|
178 |
printf("flushed answer / %p\n", bev);
|
franta-hg@31
|
179 |
/* nebudeme ukončovat spojení
|
franta-hg@28
|
180 |
bufferevent_free(bev);
|
franta-hg@31
|
181 |
*/
|
franta-hg@28
|
182 |
}
|
franta-hg@28
|
183 |
}
|
franta-hg@28
|
184 |
|
franta-hg@28
|
185 |
static void conn_event_cb(struct bufferevent *bev, short events, void *user_data) {
|
franta-hg@28
|
186 |
if (events & BEV_EVENT_EOF) {
|
franta-hg@28
|
187 |
printf("Connection closed.\n");
|
franta-hg@28
|
188 |
} else if (events & BEV_EVENT_ERROR) {
|
franta-hg@30
|
189 |
printf("Got an error on the connection: %s\n", strerror(errno));
|
franta-hg@28
|
190 |
}
|
franta-hg@29
|
191 |
|
franta-hg@28
|
192 |
// None of the other events can happen here, since we haven't enabled timeouts
|
franta-hg@28
|
193 |
bufferevent_free(bev);
|
franta-hg@28
|
194 |
}
|
franta-hg@28
|
195 |
|
franta-hg@28
|
196 |
static void signal_cb(evutil_socket_t sig, short events, void *user_data) {
|
franta-hg@28
|
197 |
struct event_base *base = (event_base *) user_data;
|
franta-hg@28
|
198 |
struct timeval delay = {2, 123};
|
franta-hg@28
|
199 |
|
franta-hg@28
|
200 |
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
|
201 |
|
franta-hg@28
|
202 |
event_base_loopexit(base, &delay);
|
franta-hg@28
|
203 |
}
|
franta-hg@31
|
204 |
|
franta-hg@31
|
205 |
static char rot13_char(char c) {
|
franta-hg@31
|
206 |
/* We don't want to use isalpha here; setting the locale would change
|
franta-hg@31
|
207 |
* which characters are considered alphabetical. */
|
franta-hg@31
|
208 |
if ((c >= 'a' && c <= 'm') || (c >= 'A' && c <= 'M'))
|
franta-hg@31
|
209 |
return c + 13;
|
franta-hg@31
|
210 |
else if ((c >= 'n' && c <= 'z') || (c >= 'N' && c <= 'Z'))
|
franta-hg@31
|
211 |
return c - 13;
|
franta-hg@31
|
212 |
else
|
franta-hg@31
|
213 |
return c;
|
franta-hg@31
|
214 |
}
|