1 #include <ccan/oserver/oserver.h>
2 #include <ccan/oserver/oserver_types.h>
3 #include <ccan/oserver/oserver_cdump.h>
4 #include <ccan/read_write_all/read_write_all.h>
5 #include <ccan/opt/opt.h>
6 #include <ccan/tevent/tevent.h>
7 #include <ccan/array_size/array_size.h>
8 #include <ccan/grab_file/grab_file.h>
10 #include <sys/socket.h>
11 #include <netinet/in.h>
12 #include <netinet/tcp.h>
22 #include <sys/types.h>
26 static uint16_t state_flag_map[] = {
27 [SENDING_GREETING] = TEVENT_FD_WRITE,
28 [RECEIVING_USER_QUESTION] = TEVENT_FD_READ,
29 [SENDING_OTHER_QUESTION_PREFIX] = TEVENT_FD_WRITE,
30 [SENDING_OTHER_QUESTION] = TEVENT_FD_WRITE,
31 [RECEIVING_OTHER_ANSWER] = TEVENT_FD_READ,
32 [SENDING_ANSWER_PREFIX] = TEVENT_FD_WRITE,
33 [SENDING_ANSWER] = TEVENT_FD_WRITE,
37 static ssize_t write_string(int fd, const char *str)
39 return write(fd, str, strlen(str));
42 static ssize_t read_string(int fd, char **buf)
44 ssize_t ret, len, maxlen;
47 maxlen = talloc_array_length(*buf);
49 if (maxlen < len + 100) {
51 *buf = talloc_realloc(NULL, *buf, char, maxlen);
54 ret = read(fd, *buf + len, maxlen - len - 1);
56 (*buf)[len + ret] = '\0';
60 static bool input_finished(const char *str)
62 return strchr(str, '\n');
65 /* Update state, and set our READ/WRITE flags appropriately. */
66 static void set_state(struct client *c, enum state state)
69 tevent_fd_set_flags(c->fde, state_flag_map[state]);
72 /* Returns false on error, increments state on finishing string. */
73 static bool send_string(struct client *c, const char *str)
75 ssize_t len = write_string(c->fd, str + c->bytes_sent);
79 if (c->bytes_sent == strlen(str)) {
81 set_state(c, c->state+1);
86 static bool get_subclient(struct client *me)
90 for (i = 0; i < me->oserver->max_clients; i++) {
91 struct client *c = me->oserver->clients[i];
94 if (c->oracle == -1 && input_finished(c->question)) {
95 me->subclient = c->id;
103 static bool get_oracle(struct client *me)
107 for (i = 0; i < me->oserver->max_clients; i++) {
108 struct client *c = me->oserver->clients[i];
111 if (c->subclient == -1 && input_finished(c->question)) {
113 c->subclient = me->id;
120 static void wakeup(struct client *c)
122 tevent_fd_set_flags(c->fde, state_flag_map[c->state]);
125 static void service_client(struct tevent_context *ev,
126 struct tevent_fd *fde, uint16_t flags, void *_c)
128 struct client *c = _c;
132 case SENDING_GREETING:
133 if (!send_string(c, "Welcome. Please ask your question.\n"))
136 case RECEIVING_USER_QUESTION:
137 len = read_string(c->fd, &c->question);
140 if (input_finished(c->question))
141 set_state(c, SENDING_OTHER_QUESTION_PREFIX);
143 case SENDING_OTHER_QUESTION_PREFIX:
144 if (c->subclient == -1)
146 if (!send_string(c, "While the Oracle ponders,"
147 " please answer the following question:\n"))
150 case SENDING_OTHER_QUESTION:
151 if (c->subclient == -1)
154 c->oserver->clients[c->subclient]->question))
157 case RECEIVING_OTHER_ANSWER:
158 if (c->subclient == -1)
160 len = read_string(c->fd,
161 &c->oserver->clients[c->subclient]->answer);
164 if (input_finished(c->oserver->clients[c->subclient]->answer)) {
165 set_state(c, SENDING_ANSWER_PREFIX);
166 wakeup(c->oserver->clients[c->subclient]);
169 case SENDING_ANSWER_PREFIX:
170 if (!input_finished(c->answer))
172 if (!send_string(c, "The Oracle spake thus:\n"))
176 if (!send_string(c, c->answer))
183 if (c->state != FINISHED)
190 if (!get_subclient(c)) {
191 /* We can't get one: go to sleep until someone find_oracle() */
192 tevent_fd_set_flags(c->fde, 0);
194 /* In case they are waiting... */
195 wakeup(c->oserver->clients[c->subclient]);
199 /* If we don't have an oracle and find one, that's OK. */
200 if (c->oracle == -1 && get_oracle(c)) {
201 /* In case they are waiting... */
202 wakeup(c->oserver->clients[c->oracle]);
206 /* Either our oracle is not finished, or we don't have one: sleep. */
207 tevent_fd_set_flags(c->fde, 0);
210 static int cleanup_client(struct client *client)
213 /* We were an oracle? */
214 if (client->subclient >= 0)
215 client->oserver->clients[client->subclient]->oracle = -1;
217 /* We had an oracle? */
218 if (client->oracle >= 0)
219 client->oserver->clients[client->oracle]->subclient = -1;
221 assert(client->oserver->clients[client->id] == client);
222 client->oserver->clients[client->id] = NULL;
226 static unsigned int find_id(struct oserver *oserver)
230 for (id = 0; id < oserver->max_clients; id++) {
231 if (oserver->clients[id] == NULL)
236 oserver->clients = talloc_realloc(oserver,
239 ++oserver->max_clients);
243 static void add_client(struct tevent_context *ev,
244 struct tevent_fd *fde, uint16_t flags, void *_oserver)
246 struct oserver *oserver = _oserver;
247 struct client *client;
249 client = talloc(oserver, struct client);
250 client->fd = accept(oserver->fd, NULL, 0);
252 err(1, "Accepting client connection");
254 client->state = SENDING_GREETING;
255 client->bytes_sent = 0;
256 client->question = talloc_strdup(client, "");
257 client->oserver = oserver;
259 client->subclient = -1;
260 client->answer = talloc_strdup(client, "");
261 client->fde = tevent_add_fd(ev, client, client->fd,
262 state_flag_map[client->state],
263 service_client, client);
264 tevent_fd_set_auto_close(client->fde);
266 client->id = find_id(oserver);
267 oserver->clients[client->id] = client;
270 static int destroy_oserver(struct oserver *oserver)
276 static void talloc_dump(struct tevent_context *ev,
277 struct tevent_signal *se,
283 struct oserver *oserver = _oserver;
286 /* Fork off a child for the report, so we aren't stopped. */
288 f = fopen("/var/run/oserver/talloc.dump", "w");
290 talloc_report_full(oserver, f);
297 static void dump(struct tevent_context *ev,
298 struct tevent_signal *se,
304 struct oserver *oserver = _oserver;
308 str = cdump_bundle(ev, cdump_struct_oserver, oserver);
309 fd = open(oserver->dumpfile, O_CREAT|O_TRUNC|O_WRONLY, 0600);
310 write(fd, str, strlen(str));
314 execvp(oserver->argv[0], oserver->argv);
317 static bool load_file(struct oserver *oserver, const char *file)
324 str = grab_file(oserver, file, NULL);
328 if (!cdump_unbundle(oserver, cdump_struct_oserver, oserver, str)) {
336 static bool complete_server(struct tevent_context *ev,
337 struct oserver *oserver, const char *dumpfile)
339 /* Re-set this even if restored from file, in case it changed. */
340 oserver->dumpfile = dumpfile;
341 if (oserver->dumpfile)
342 tevent_add_signal(ev, oserver, SIGHUP, SA_RESTART,
345 /* Don't kill us if client dies. */
346 signal(SIGPIPE, SIG_IGN);
348 /* Show talloc tree on SIGUSR1. */
349 tevent_add_signal(ev, oserver, SIGUSR1, SA_RESTART,
350 talloc_dump, oserver);
352 oserver->fde = tevent_add_fd(ev, oserver, oserver->fd,
353 TEVENT_FD_READ, add_client, oserver);
359 struct oserver *oserver_restore(struct tevent_context *ev, const char *dumpfile)
362 struct oserver *oserver = talloc(ev, struct oserver);
363 if (!load_file(oserver, dumpfile)) {
364 talloc_free(oserver);
368 /* Restore ignored fields in clients, and talloc hierarchy. */
369 for (i = 0; i < oserver->max_clients; i++) {
370 struct client *client = oserver->clients[i];
373 /* These two were marked CDUMP_IGNORE. */
374 client->oserver = oserver;
375 client->fde = tevent_add_fd(ev, client, client->fd,
376 state_flag_map[client->state],
377 service_client, client);
378 tevent_fd_set_auto_close(client->fde);
379 /* cdump knows nothing of talloc. */
380 talloc_steal(oserver, client);
381 talloc_steal(client, client->question);
382 talloc_steal(client, client->answer);
383 talloc_set_destructor(client, cleanup_client);
386 talloc_set_destructor(oserver, destroy_oserver);
387 if (!complete_server(ev, oserver, dumpfile)) {
388 talloc_free(oserver);
394 struct oserver *oserver_setup(struct tevent_context *ev, unsigned short port,
395 const char *dumpfile, char *argv[])
397 struct oserver *oserver;
400 struct sockaddr addr;
401 struct sockaddr_in in;
404 oserver = talloc(ev, struct oserver);
405 oserver->argv = argv;
407 /* Count the terminal NULL in argv_len. */
408 for (oserver->argv_len = 1;
409 argv[oserver->argv_len - 1];
410 oserver->argv_len++);
412 oserver->argv_len = 0;
413 oserver->max_clients = 0;
414 oserver->clients = NULL;
415 oserver->fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
416 if (oserver->fd < 0) {
417 talloc_free(oserver);
421 talloc_set_destructor(oserver, destroy_oserver);
423 if (setsockopt(oserver->fd, SOL_SOCKET, SO_REUSEADDR,
425 warn("Setting socket reuse");
427 u.in.sin_family = AF_INET;
428 u.in.sin_port = htons(port);
429 u.in.sin_addr.s_addr = INADDR_ANY;
430 if (bind(oserver->fd, &u.addr, sizeof(u.in)) == -1) {
431 talloc_free(oserver);
435 if (listen(oserver->fd, 0) != 0) {
436 talloc_free(oserver);
440 if (!complete_server(ev, oserver, dumpfile)) {
441 talloc_free(oserver);