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@40
|
6 |
#include <string>
|
franta-hg@40
|
7 |
#include <regex>
|
franta-hg@28
|
8 |
|
franta-hg@28
|
9 |
#include <event2/bufferevent.h>
|
franta-hg@28
|
10 |
#include <event2/buffer.h>
|
franta-hg@28
|
11 |
#include <event2/listener.h>
|
franta-hg@28
|
12 |
#include <event2/util.h>
|
franta-hg@28
|
13 |
#include <event2/event.h>
|
franta-hg@38
|
14 |
#include <sys/socket.h>
|
franta-hg@29
|
15 |
#include <sys/un.h>
|
franta-hg@29
|
16 |
#include <unistd.h>
|
franta-hg@28
|
17 |
|
franta-hg@40
|
18 |
using namespace std;
|
franta-hg@40
|
19 |
|
franta-hg@28
|
20 |
static const char MESSAGE[] = "Hello, World!\n";
|
franta-hg@28
|
21 |
|
franta-hg@29
|
22 |
static const char PATH[] = "./roura";
|
franta-hg@28
|
23 |
|
franta-hg@40
|
24 |
static const string COMMAND_EXIT("exit\n");
|
franta-hg@40
|
25 |
|
franta-hg@31
|
26 |
static void listener_cb(evutil_socket_t, short, void *);
|
franta-hg@30
|
27 |
static void conn_read_cb(struct bufferevent *, void *);
|
franta-hg@28
|
28 |
static void conn_write_cb(struct bufferevent *, void *);
|
franta-hg@28
|
29 |
static void conn_event_cb(struct bufferevent *, short, void *);
|
franta-hg@28
|
30 |
static void signal_cb(evutil_socket_t, short, void *);
|
franta-hg@40
|
31 |
static void print_socket_info(int, int);
|
franta-hg@31
|
32 |
|
franta-hg@28
|
33 |
int main(int argc, char **argv) {
|
franta-hg@40
|
34 |
printf("%4s %8s\n", "*", "STARTED");
|
franta-hg@28
|
35 |
struct event_base *base;
|
franta-hg@31
|
36 |
evutil_socket_t listener;
|
franta-hg@31
|
37 |
struct event *listener_event;
|
franta-hg@28
|
38 |
struct event *signal_event;
|
franta-hg@28
|
39 |
|
franta-hg@29
|
40 |
struct sockaddr_un sun;
|
franta-hg@28
|
41 |
|
franta-hg@31
|
42 |
setvbuf(stdout, NULL, _IONBF, 0);
|
franta-hg@31
|
43 |
|
franta-hg@28
|
44 |
base = event_base_new();
|
franta-hg@28
|
45 |
if (!base) {
|
franta-hg@40
|
46 |
printf("%4s %8s %s\n", "*", "ERROR", "unable to initialize libevent");
|
franta-hg@28
|
47 |
return 1;
|
franta-hg@28
|
48 |
}
|
franta-hg@28
|
49 |
|
franta-hg@29
|
50 |
memset(&sun, 0, sizeof (sun));
|
franta-hg@29
|
51 |
sun.sun_family = AF_UNIX;
|
franta-hg@29
|
52 |
strcpy(sun.sun_path, PATH);
|
franta-hg@28
|
53 |
|
franta-hg@31
|
54 |
listener = socket(AF_UNIX, SOCK_STREAM, 0);
|
franta-hg@31
|
55 |
evutil_make_socket_nonblocking(listener);
|
franta-hg@28
|
56 |
|
franta-hg@31
|
57 |
if (bind(listener, (struct sockaddr*) &sun, sizeof (sun)) < 0) {
|
franta-hg@40
|
58 |
printf("%4s %8s %s: %s\n", "*", "ERROR", "unable to create domain socket:", PATH);
|
franta-hg@28
|
59 |
return 1;
|
franta-hg@28
|
60 |
}
|
franta-hg@28
|
61 |
|
franta-hg@31
|
62 |
if (listen(listener, 16) < 0) {
|
franta-hg@40
|
63 |
printf("%4s %8s %s\n", "*", "ERROR", "unable to listen");
|
franta-hg@31
|
64 |
return 1;
|
franta-hg@31
|
65 |
}
|
franta-hg@31
|
66 |
|
franta-hg@39
|
67 |
// identifikátor serverového soketu (v současnosti číslo FD)
|
franta-hg@40
|
68 |
printf("%4s %8s listening at: sun_path = %s → socketId = %d\n", "*", "SOCKET", sun.sun_path, listener);
|
franta-hg@39
|
69 |
|
franta-hg@31
|
70 |
listener_event = event_new(base, listener, EV_READ | EV_PERSIST, listener_cb, (void*) base);
|
franta-hg@31
|
71 |
|
franta-hg@31
|
72 |
if (!listener_event) {
|
franta-hg@40
|
73 |
printf("%4s %8s %s\n", "*", "ERROR", "unable to do event_new()");
|
franta-hg@31
|
74 |
return 1;
|
franta-hg@31
|
75 |
}
|
franta-hg@31
|
76 |
|
franta-hg@31
|
77 |
event_add(listener_event, NULL);
|
franta-hg@31
|
78 |
|
franta-hg@28
|
79 |
signal_event = evsignal_new(base, SIGINT, signal_cb, (void *) base);
|
franta-hg@28
|
80 |
|
franta-hg@28
|
81 |
if (!signal_event || event_add(signal_event, NULL) < 0) {
|
franta-hg@40
|
82 |
printf("%4s %8s %s\n", "*", "ERROR", "unable to create/add a signal event");
|
franta-hg@28
|
83 |
return 1;
|
franta-hg@28
|
84 |
}
|
franta-hg@28
|
85 |
|
franta-hg@28
|
86 |
event_base_dispatch(base);
|
franta-hg@28
|
87 |
|
franta-hg@31
|
88 |
event_free(listener_event);
|
franta-hg@28
|
89 |
event_free(signal_event);
|
franta-hg@28
|
90 |
event_base_free(base);
|
franta-hg@30
|
91 |
|
franta-hg@32
|
92 |
// smažeme soket na disku / soubor -- jinak by program příště spadl na bind()
|
franta-hg@39
|
93 |
// TODO: co když soket někdo přesune a místo něj dá jiný soubor?
|
franta-hg@29
|
94 |
unlink(PATH);
|
franta-hg@28
|
95 |
|
franta-hg@40
|
96 |
printf("%4s %8s\n", "*", "FINISHED");
|
franta-hg@28
|
97 |
return 0;
|
franta-hg@28
|
98 |
}
|
franta-hg@28
|
99 |
|
franta-hg@31
|
100 |
static void listener_cb(evutil_socket_t listener, short event, void *user_data) {
|
franta-hg@28
|
101 |
struct event_base *base = (event_base *) user_data;
|
franta-hg@28
|
102 |
struct bufferevent *bev;
|
franta-hg@28
|
103 |
|
franta-hg@31
|
104 |
struct sockaddr_storage ss;
|
franta-hg@31
|
105 |
socklen_t slen = sizeof (ss);
|
franta-hg@31
|
106 |
int fd = accept(listener, (struct sockaddr*) &ss, &slen);
|
franta-hg@40
|
107 |
|
franta-hg@40
|
108 |
// identifikátor navázaného spojení (v současnosti číslo FD)
|
franta-hg@40
|
109 |
int * connectionId = (int*) malloc(sizeof (fd));
|
franta-hg@40
|
110 |
*connectionId = fd;
|
franta-hg@40
|
111 |
|
franta-hg@31
|
112 |
if (fd < 0) {
|
franta-hg@40
|
113 |
printf("%4d %8s %s\n", *connectionId, "ERROR", "unable to accept()");
|
franta-hg@31
|
114 |
return;
|
franta-hg@31
|
115 |
} else if (fd > FD_SETSIZE) {
|
franta-hg@38
|
116 |
// FD_SETSIZE = 1024 -- Proč? Co když bude spojení víc?
|
franta-hg@40
|
117 |
printf("%4d %8s fd (%d) > FD_SETSIZE (%d)\n", *connectionId, "ERROR", fd, FD_SETSIZE);
|
franta-hg@31
|
118 |
close(fd);
|
franta-hg@31
|
119 |
return;
|
franta-hg@31
|
120 |
}
|
franta-hg@31
|
121 |
|
franta-hg@31
|
122 |
evutil_make_socket_nonblocking(fd);
|
franta-hg@39
|
123 |
|
franta-hg@31
|
124 |
|
franta-hg@31
|
125 |
bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE);
|
franta-hg@28
|
126 |
if (!bev) {
|
franta-hg@40
|
127 |
printf("%4d %8s %s\n", *connectionId, "ERROR", "constructing bufferevent");
|
franta-hg@28
|
128 |
event_base_loopbreak(base);
|
franta-hg@28
|
129 |
return;
|
franta-hg@28
|
130 |
}
|
franta-hg@39
|
131 |
|
franta-hg@40
|
132 |
|
franta-hg@39
|
133 |
|
franta-hg@39
|
134 |
|
franta-hg@39
|
135 |
bufferevent_setcb(bev, conn_read_cb, conn_write_cb, conn_event_cb, (void*) connectionId);
|
franta-hg@30
|
136 |
bufferevent_enable(bev, EV_READ | EV_WRITE);
|
franta-hg@28
|
137 |
|
franta-hg@40
|
138 |
printf("%4d %8s somebody has connected: socketId = %d → connectionId = %d\n", *connectionId, "CONN", listener, *connectionId);
|
franta-hg@40
|
139 |
print_socket_info(*connectionId, fd);
|
franta-hg@28
|
140 |
|
franta-hg@28
|
141 |
bufferevent_write(bev, MESSAGE, strlen(MESSAGE));
|
franta-hg@28
|
142 |
}
|
franta-hg@28
|
143 |
|
franta-hg@30
|
144 |
static void conn_read_cb(struct bufferevent *bev, void *user_data) {
|
franta-hg@40
|
145 |
int connectionId = *((int*) user_data);
|
franta-hg@34
|
146 |
|
franta-hg@30
|
147 |
/* This callback is invoked when there is data to read on bev. */
|
franta-hg@30
|
148 |
struct evbuffer *input = bufferevent_get_input(bev);
|
franta-hg@30
|
149 |
struct evbuffer *output = bufferevent_get_output(bev);
|
franta-hg@33
|
150 |
struct event_base *base = bufferevent_get_base(bev);
|
franta-hg@30
|
151 |
|
franta-hg@30
|
152 |
size_t len = evbuffer_get_length(input);
|
franta-hg@32
|
153 |
char *data = (char*) malloc(len);
|
franta-hg@30
|
154 |
evbuffer_copyout(input, data, len);
|
franta-hg@31
|
155 |
|
franta-hg@40
|
156 |
|
franta-hg@40
|
157 |
string dataFormated(data);
|
franta-hg@40
|
158 |
dataFormated = regex_replace(dataFormated, regex("\\n"), "\\n");
|
franta-hg@40
|
159 |
printf("%4d %8s '%s'\n", connectionId, "IN", dataFormated.c_str());
|
franta-hg@40
|
160 |
|
franta-hg@40
|
161 |
if (COMMAND_EXIT.compare(data) == 0) {
|
franta-hg@30
|
162 |
struct timeval delay = {2, 123};
|
franta-hg@40
|
163 |
printf("%4d %8s client asks us to terminate; finishing in %ld sesonds and %ld microseconds\n", connectionId, "EXIT", delay.tv_sec, delay.tv_usec);
|
franta-hg@30
|
164 |
event_base_loopexit(base, &delay);
|
franta-hg@30
|
165 |
}
|
franta-hg@31
|
166 |
|
franta-hg@32
|
167 |
evbuffer_add(output, "echo: ", 6);
|
franta-hg@30
|
168 |
/* Copy all the data from the input buffer to the output buffer. */
|
franta-hg@30
|
169 |
evbuffer_add_buffer(output, input);
|
franta-hg@30
|
170 |
free(data);
|
franta-hg@30
|
171 |
}
|
franta-hg@30
|
172 |
|
franta-hg@28
|
173 |
static void conn_write_cb(struct bufferevent *bev, void *user_data) {
|
franta-hg@40
|
174 |
int connectionId = *((int*) user_data);
|
franta-hg@34
|
175 |
|
franta-hg@28
|
176 |
struct evbuffer *output = bufferevent_get_output(bev);
|
franta-hg@28
|
177 |
if (evbuffer_get_length(output) == 0) {
|
franta-hg@40
|
178 |
printf("%4d %8s\n", connectionId, "FLUSH");
|
franta-hg@31
|
179 |
/* nebudeme ukončovat spojení
|
franta-hg@28
|
180 |
bufferevent_free(bev);
|
franta-hg@31
|
181 |
*/
|
franta-hg@40
|
182 |
} else {
|
franta-hg@40
|
183 |
// FIXME: sem to nikdy nepřijde
|
franta-hg@40
|
184 |
size_t len = evbuffer_get_length(output);
|
franta-hg@40
|
185 |
char *data = (char*) malloc(len);
|
franta-hg@40
|
186 |
evbuffer_copyout(output, data, len);
|
franta-hg@40
|
187 |
printf("%4d %8s '%s'\n", connectionId, "OUT", data);
|
franta-hg@40
|
188 |
free(data);
|
franta-hg@28
|
189 |
}
|
franta-hg@28
|
190 |
}
|
franta-hg@28
|
191 |
|
franta-hg@28
|
192 |
static void conn_event_cb(struct bufferevent *bev, short events, void *user_data) {
|
franta-hg@39
|
193 |
int connectionId = *((int*) user_data);
|
franta-hg@38
|
194 |
|
franta-hg@28
|
195 |
if (events & BEV_EVENT_EOF) {
|
franta-hg@40
|
196 |
printf("%4d %8s\n", connectionId, "CLOSE");
|
franta-hg@28
|
197 |
} else if (events & BEV_EVENT_ERROR) {
|
franta-hg@40
|
198 |
printf("%4d %8s %s\n", connectionId, "ERROR", strerror(errno));
|
franta-hg@40
|
199 |
} else {
|
franta-hg@40
|
200 |
printf("%4d %8s\n", connectionId, "OTHER");
|
franta-hg@40
|
201 |
// None of the other events can happen here, since we haven't enabled timeouts
|
franta-hg@28
|
202 |
}
|
franta-hg@29
|
203 |
|
franta-hg@28
|
204 |
bufferevent_free(bev);
|
franta-hg@39
|
205 |
free(user_data);
|
franta-hg@28
|
206 |
}
|
franta-hg@28
|
207 |
|
franta-hg@28
|
208 |
static void signal_cb(evutil_socket_t sig, short events, void *user_data) {
|
franta-hg@28
|
209 |
struct event_base *base = (event_base *) user_data;
|
franta-hg@28
|
210 |
struct timeval delay = {2, 123};
|
franta-hg@28
|
211 |
|
franta-hg@40
|
212 |
printf("\n%4s %8s got SIGINT (Ctrl+C); finishing in %ld sesonds and %ld microseconds\n", "*", "SIGNAL", delay.tv_sec, delay.tv_usec);
|
franta-hg@28
|
213 |
|
franta-hg@28
|
214 |
event_base_loopexit(base, &delay);
|
franta-hg@28
|
215 |
}
|
franta-hg@38
|
216 |
|
franta-hg@40
|
217 |
static void print_socket_info(int connectionId, int fd) {
|
franta-hg@38
|
218 |
struct ucred cr;
|
franta-hg@38
|
219 |
unsigned int cl = sizeof (cr);
|
franta-hg@38
|
220 |
|
franta-hg@38
|
221 |
if (getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &cr, &cl) == 0) {
|
franta-hg@40
|
222 |
printf("%4d %8s client identification: pid=%d, uid=%d, gid=%d\n", connectionId, "CONN", cr.pid, cr.uid, cr.gid);
|
franta-hg@38
|
223 |
}
|
franta-hg@38
|
224 |
} |