Line data Source code
1 : #include "../burp.h"
2 : #include "../asfd.h"
3 : #include "../async.h"
4 : #include "../cntr.h"
5 : #include "../conf.h"
6 : #include "../conffile.h"
7 : #include "../cstat.h"
8 : #include "../fsops.h"
9 : #include "../handy.h"
10 : #include "../iobuf.h"
11 : #include "../lock.h"
12 : #include "../log.h"
13 : #include "auth.h"
14 : #include "ca.h"
15 : #include "child.h"
16 : #include "main.h"
17 : #include "monitor/status_server.h"
18 :
19 : static int hupreload=0;
20 : static int hupreload_logged=0;
21 : static int gentleshutdown=0;
22 : static int gentleshutdown_logged=0;
23 :
24 : // These will also be used as the exit codes of the program and are therefore
25 : // unsigned integers.
26 : // Remember to update the man page if you update these.
27 : enum serret
28 : {
29 : SERVER_OK=0,
30 : SERVER_ERROR=1
31 : };
32 :
33 0 : static void huphandler(__attribute__ ((unused)) int sig)
34 : {
35 0 : hupreload=1;
36 : // Be careful about not logging inside a signal handler.
37 0 : hupreload_logged=0;
38 0 : }
39 :
40 0 : static void usr2handler(__attribute__ ((unused)) int sig)
41 : {
42 0 : gentleshutdown=1;
43 : // Be careful about not logging inside a signal handler.
44 0 : gentleshutdown_logged=0;
45 0 : }
46 :
47 : // Remove any exiting child pids from our list.
48 0 : static void chld_check_for_exiting(struct async *mainas)
49 : {
50 : pid_t p;
51 : int status;
52 : struct asfd *asfd;
53 :
54 0 : while((p=waitpid(-1, &status, WNOHANG))>0)
55 : {
56 : // Logging a message here appeared to occasionally lock burp up
57 : // on a Ubuntu server that I used to use.
58 0 : for(asfd=mainas->asfd; asfd; asfd=asfd->next)
59 : {
60 0 : if(p!=asfd->pid) continue;
61 0 : mainas->asfd_remove(mainas, asfd);
62 0 : asfd_free(&asfd);
63 0 : break;
64 : }
65 : }
66 0 : }
67 :
68 : static void *get_in_addr(struct sockaddr *sa)
69 : {
70 : #ifdef HAVE_IPV6
71 0 : if(sa->sa_family==AF_INET6)
72 0 : return &(((struct sockaddr_in6*)sa)->sin6_addr);
73 : #endif
74 0 : return &(((struct sockaddr_in*)sa)->sin_addr);
75 : }
76 :
77 0 : static void log_listen_socket(const char *desc,
78 : struct addrinfo *rp, const char *port, int max_children)
79 : {
80 : #ifdef HAVE_IPV6
81 0 : char addr[INET6_ADDRSTRLEN]="";
82 : #else
83 : char addr[INET_ADDRSTRLEN]="";
84 : #endif
85 0 : inet_ntop(rp->ai_family, get_in_addr((struct sockaddr *)rp->ai_addr),
86 : addr, sizeof(addr));
87 0 : logp("%s %s:%s (max %d)\n",
88 : desc, addr, port, max_children);
89 0 : }
90 :
91 0 : static int init_listen_socket(struct strlist *address,
92 : struct async *mainas, enum asfd_fdtype fdtype, const char *desc)
93 : {
94 0 : int fd=-1;
95 : int gai_ret;
96 : struct addrinfo hints;
97 0 : struct addrinfo *info=NULL;
98 0 : struct asfd *newfd=NULL;
99 0 : char *a=NULL;
100 0 : char *port=NULL;
101 :
102 0 : if(!(a=strdup_w(address->path, __func__)))
103 : goto error;
104 0 : if(!(port=strrchr(a, ':')))
105 : {
106 0 : logp("Could not parse '%s'\n", address->path);
107 : goto error;
108 : }
109 0 : *port='\0';
110 0 : port++;
111 :
112 0 : memset(&hints, 0, sizeof(struct addrinfo));
113 : hints.ai_family=AF_UNSPEC;
114 0 : hints.ai_socktype=SOCK_STREAM;
115 0 : hints.ai_protocol=IPPROTO_TCP;
116 : hints.ai_flags=AI_NUMERICHOST;
117 0 : hints.ai_flags|=AI_PASSIVE;
118 :
119 0 : if((gai_ret=getaddrinfo(a, port, &hints, &info)))
120 : {
121 0 : logp("unable to getaddrinfo on %s: %s\n",
122 : address->path, gai_strerror(gai_ret));
123 : goto error;
124 : }
125 :
126 : // Just try to use the first one in info, it should be good enough.
127 0 : fd=socket(info->ai_family, info->ai_socktype, info->ai_protocol);
128 0 : if(fd<0)
129 : {
130 0 : logp("unable to create socket on %s: %s\n",
131 0 : address->path, strerror(errno));
132 : goto error;
133 : }
134 0 : set_keepalive(fd, 1);
135 : #ifdef HAVE_IPV6
136 0 : if(info->ai_family==AF_INET6)
137 : {
138 : // Attempt to say that it should not listen on IPv6
139 : // only.
140 0 : int optval=0;
141 0 : setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY,
142 : &optval, sizeof(optval));
143 : }
144 : #endif
145 0 : reuseaddr(fd);
146 0 : if(bind(fd, info->ai_addr, info->ai_addrlen))
147 : {
148 0 : logp("unable to bind socket on %s: %s\n",
149 0 : address->path, strerror(errno));
150 : goto error;
151 : }
152 :
153 : // Say that we are happy to accept connections.
154 0 : if(listen(fd, 5)<0)
155 : {
156 0 : logp("could not listen on address %s: %s\n",
157 0 : address->path, strerror(errno));
158 : goto error;
159 : }
160 :
161 0 : log_listen_socket(desc, info, port, address->flag);
162 0 : if(!(newfd=setup_asfd(mainas, desc, &fd, address->path)))
163 : goto end;
164 0 : newfd->fdtype=fdtype;
165 :
166 : goto end;
167 : error:
168 0 : free_w(&a);
169 0 : if(info)
170 0 : freeaddrinfo(info);
171 : return -1;
172 : end:
173 0 : free_w(&a);
174 0 : if(info)
175 0 : freeaddrinfo(info);
176 : return 0;
177 : }
178 :
179 : static int init_listen_sockets(struct strlist *addresses,
180 : struct async *mainas, enum asfd_fdtype fdtype, const char *desc)
181 : {
182 : struct strlist *a;
183 0 : for(a=addresses; a; a=a->next)
184 0 : if(init_listen_socket(a, mainas, fdtype, desc))
185 : return -1;
186 : return 0;
187 : }
188 :
189 0 : void setup_signals(void)
190 : {
191 : // Ignore SIGPIPE - we are careful with read and write return values.
192 0 : signal(SIGPIPE, SIG_IGN);
193 :
194 0 : setup_signal(SIGHUP, huphandler);
195 0 : setup_signal(SIGUSR2, usr2handler);
196 0 : }
197 :
198 0 : static int run_child(int *cfd, SSL_CTX *ctx, struct sockaddr_storage *addr,
199 : int status_wfd, int status_rfd, const char *conffile, int forking)
200 : {
201 0 : int ret=-1;
202 0 : int ca_ret=0;
203 0 : SSL *ssl=NULL;
204 0 : BIO *sbio=NULL;
205 0 : struct conf **confs=NULL;
206 0 : struct conf **cconfs=NULL;
207 0 : struct cntr *cntr=NULL;
208 0 : struct async *as=NULL;
209 0 : const char *cname=NULL;
210 0 : struct asfd *asfd=NULL;
211 0 : int is_status_server=0;
212 :
213 0 : if(!(confs=confs_alloc())
214 0 : || !(cconfs=confs_alloc()))
215 : goto end;
216 :
217 0 : set_peer_env_vars(addr);
218 :
219 : // Reload global config, in case things have changed. This means that
220 : // the server does not need to be restarted for most conf changes.
221 0 : confs_init(confs);
222 0 : confs_init(cconfs);
223 0 : if(conf_load_global_only(conffile, confs)) goto end;
224 :
225 : // Hack to keep forking turned off if it was specified as off on the
226 : // command line.
227 0 : if(!forking) set_int(confs[OPT_FORK], 0);
228 :
229 0 : if(!(sbio=BIO_new_socket(*cfd, BIO_NOCLOSE))
230 0 : || !(ssl=SSL_new(ctx)))
231 : {
232 0 : logp("There was a problem joining ssl to the socket\n");
233 0 : goto end;
234 : }
235 0 : SSL_set_bio(ssl, sbio, sbio);
236 :
237 : /* Do not try to check peer certificate straight away.
238 : Clients can send a certificate signing request when they have
239 : no certificate. */
240 0 : SSL_set_verify(ssl, SSL_VERIFY_PEER
241 : /* | SSL_VERIFY_FAIL_IF_NO_PEER_CERT */, 0);
242 :
243 0 : if(ssl_do_accept(ssl))
244 : goto end;
245 0 : if(!(as=async_alloc())
246 0 : || as->init(as, 0)
247 0 : || !(asfd=setup_asfd_ssl(as, "main socket", cfd, ssl)))
248 : goto end;
249 0 : asfd->set_timeout(asfd, get_int(confs[OPT_NETWORK_TIMEOUT]));
250 0 : asfd->ratelimit=get_float(confs[OPT_RATELIMIT]);
251 :
252 0 : if(authorise_server(as->asfd, confs, cconfs)
253 0 : || !(cname=get_string(cconfs[OPT_CNAME])) || !*cname)
254 : {
255 : // Add an annoying delay in case they are tempted to
256 : // try repeatedly.
257 0 : log_and_send(as->asfd, "unable to authorise on server");
258 0 : sleep(1);
259 0 : goto end;
260 : }
261 :
262 0 : if(!get_int(cconfs[OPT_ENABLED]))
263 : {
264 0 : log_and_send(as->asfd, "client not enabled on server");
265 0 : sleep(1);
266 0 : goto end;
267 : }
268 :
269 : // Set up counters. Have to wait until here to get cname.
270 0 : if(!(cntr=cntr_alloc())
271 0 : || cntr_init(cntr, cname, getpid()))
272 : goto end;
273 0 : set_cntr(confs[OPT_CNTR], cntr);
274 0 : set_cntr(cconfs[OPT_CNTR], cntr);
275 :
276 : /* At this point, the client might want to get a new certificate
277 : signed. Clients on 1.3.2 or newer can do this. */
278 0 : if((ca_ret=ca_server_maybe_sign_client_cert(as->asfd, confs, cconfs))<0)
279 : {
280 0 : logp("Error signing client certificate request for %s\n",
281 : cname);
282 0 : goto end;
283 : }
284 0 : else if(ca_ret>0)
285 : {
286 : // Certificate signed and sent back.
287 : // Everything is OK, but we will close this instance
288 : // so that the client can start again with a new
289 : // connection and its new certificates.
290 0 : logp("Signed and returned client certificate request for %s\n",
291 : cname);
292 0 : ret=0;
293 0 : goto end;
294 : }
295 :
296 : /* Now it is time to check the certificate. */
297 0 : if(ssl_check_cert(ssl, confs, cconfs))
298 : {
299 0 : log_and_send(as->asfd, "check cert failed on server");
300 0 : goto end;
301 : }
302 0 : if(status_rfd>=0)
303 : {
304 0 : is_status_server=1;
305 0 : if(!setup_asfd(as, "status server parent socket", &status_rfd,
306 : /*listen*/""))
307 : goto end;
308 : }
309 :
310 0 : ret=child(as, is_status_server, status_wfd, confs, cconfs);
311 : end:
312 0 : *cfd=-1;
313 0 : if(as && asfd_flush_asio(as->asfd))
314 0 : ret=-1;
315 0 : async_asfd_free_all(&as); // This closes cfd for us.
316 0 : logp("exit child\n");
317 0 : if(cntr) cntr_free(&cntr);
318 0 : if(confs)
319 : {
320 0 : set_cntr(confs[OPT_CNTR], NULL);
321 0 : confs_free(&confs);
322 : }
323 0 : if(cconfs)
324 : {
325 0 : set_cntr(cconfs[OPT_CNTR], NULL);
326 0 : confs_free(&cconfs);
327 : }
328 0 : return ret;
329 : }
330 :
331 0 : static struct strlist *find_listen_in_conf(struct conf **confs,
332 : enum conf_opt listen_opt, const char *listen)
333 : {
334 : struct strlist *l;
335 0 : for(l=get_strlist(confs[listen_opt]); l; l=l->next)
336 0 : if(!strcmp(listen, l->path))
337 : return l;
338 0 : logp("Could not find %s in %s confs\n",
339 0 : listen, confs[listen_opt]->field);
340 0 : return NULL;
341 : }
342 :
343 0 : static int chld_check_counts(struct conf **confs, struct asfd *asfd)
344 : {
345 0 : long count=0;
346 : struct asfd *a;
347 : struct strlist *listen;
348 : enum conf_opt listen_opt;
349 :
350 0 : switch(asfd->fdtype)
351 : {
352 : case ASFD_FD_SERVER_LISTEN_MAIN:
353 0 : listen_opt=OPT_LISTEN;
354 0 : if(!(listen=find_listen_in_conf(confs,
355 0 : listen_opt, asfd->listen)))
356 : return -1;
357 : break;
358 : case ASFD_FD_SERVER_LISTEN_STATUS:
359 0 : listen_opt=OPT_LISTEN_STATUS;
360 0 : if(!(listen=find_listen_in_conf(confs,
361 0 : listen_opt, asfd->listen)))
362 : return -1;
363 : break;
364 : default:
365 0 : logp("Unexpected fdtype in %s: %d.\n",
366 : __func__, asfd->fdtype);
367 0 : return -1;
368 : }
369 :
370 0 : for(a=asfd->as->asfd; a; a=a->next)
371 0 : if(a!=asfd
372 0 : && !strcmp(asfd->listen, a->listen))
373 0 : count++;
374 :
375 0 : logp("%d/%d child processes running on %s %s\n",
376 0 : (int)count, (int)listen->flag,
377 0 : confs[listen_opt]->field, asfd->listen);
378 0 : if(count<listen->flag)
379 0 : logp("Child %d available\n", (int)count+1);
380 : else
381 : {
382 0 : logp("No spare children available.\n");
383 0 : return -1;
384 : }
385 :
386 0 : return 0;
387 : }
388 :
389 : static struct asfd *setup_parent_child_pipe(struct async *as,
390 : const char *desc,
391 : int *fd_to_use, int *fd_to_close, pid_t childpid, const char *listen,
392 : enum asfd_fdtype fdtype)
393 : {
394 : struct asfd *newfd;
395 0 : close_fd(fd_to_close);
396 0 : if(!(newfd=setup_asfd(as, desc, fd_to_use, listen)))
397 : return NULL;
398 0 : newfd->pid=childpid;
399 0 : newfd->fdtype=fdtype;
400 : return newfd;
401 : }
402 :
403 0 : static int setup_parent_child_pipes(struct asfd *asfd,
404 : pid_t childpid, int *rfd, int *wfd)
405 : {
406 : struct asfd *newfd;
407 0 : struct async *as=asfd->as;
408 0 : switch(asfd->fdtype)
409 : {
410 : case ASFD_FD_SERVER_LISTEN_MAIN:
411 0 : logp("forked child on %s: %d\n",
412 : asfd->listen, childpid);
413 0 : if(!(newfd=setup_parent_child_pipe(as,
414 : "pipe from child",
415 0 : rfd, wfd, childpid, asfd->listen,
416 : ASFD_FD_SERVER_PIPE_READ)))
417 : return -1;
418 0 : return 0;
419 : case ASFD_FD_SERVER_LISTEN_STATUS:
420 0 : logp("forked status child on %s: %d\n",
421 : asfd->listen, childpid);
422 0 : if(!(newfd=setup_parent_child_pipe(as,
423 : "pipe to status child",
424 0 : wfd, rfd, childpid, asfd->listen,
425 : ASFD_FD_SERVER_PIPE_WRITE)))
426 : return -1;
427 0 : newfd->attempt_reads=0;
428 0 : return 0;
429 : default:
430 0 : logp("Strange fdtype after fork: %d\n",
431 : asfd->fdtype);
432 0 : return -1;
433 : }
434 :
435 : return 0;
436 : }
437 :
438 0 : static int process_incoming_client(struct asfd *asfd, SSL_CTX *ctx,
439 : const char *conffile, struct conf **confs)
440 : {
441 0 : int cfd=-1;
442 : pid_t childpid;
443 : int pipe_rfd[2];
444 : int pipe_wfd[2];
445 0 : socklen_t client_length=0;
446 : struct sockaddr_storage client_name;
447 0 : enum asfd_fdtype fdtype=asfd->fdtype;
448 0 : int forking=get_int(confs[OPT_FORK]);
449 :
450 0 : client_length=sizeof(client_name);
451 0 : if((cfd=accept(asfd->fd,
452 : (struct sockaddr *)&client_name, &client_length))==-1)
453 : {
454 : // Look out, accept will get interrupted by SIGCHLDs.
455 0 : if(errno==EINTR) return 0;
456 0 : logp("accept failed on %s (%d) in %s: %s\n", asfd->desc,
457 : asfd->fd, __func__, strerror(errno));
458 0 : return -1;
459 : }
460 0 : reuseaddr(cfd);
461 0 : if(log_peer_address(&client_name))
462 : return -1;
463 :
464 0 : if(!forking)
465 0 : return run_child(&cfd, ctx,
466 : &client_name, -1, -1, conffile, forking);
467 :
468 0 : if(chld_check_counts(confs, asfd))
469 : {
470 0 : logp("Closing new connection.\n");
471 0 : close_fd(&cfd);
472 0 : return 0;
473 : }
474 :
475 0 : if(pipe(pipe_rfd)<0 || pipe(pipe_wfd)<0)
476 : {
477 0 : logp("pipe failed: %s", strerror(errno));
478 0 : close_fd(&cfd);
479 0 : return -1;
480 : }
481 :
482 0 : switch((childpid=fork()))
483 : {
484 : case -1:
485 0 : logp("fork failed: %s\n", strerror(errno));
486 0 : return -1;
487 : case 0:
488 : {
489 : // Child.
490 : int p;
491 : int ret;
492 : struct sigaction sa;
493 0 : struct async *as=asfd->as;
494 0 : async_asfd_free_all(&as);
495 :
496 : // Close unnecessary file descriptors.
497 : // Go up to FD_SETSIZE and hope for the best.
498 : // FIX THIS: Now that async_asfd_free_all() is doing
499 : // everything, double check whether this is needed.
500 0 : for(p=3; p<(int)FD_SETSIZE; p++)
501 : {
502 0 : if(p!=pipe_rfd[1]
503 0 : && p!=pipe_wfd[0]
504 0 : && p!=cfd)
505 0 : close(p);
506 : }
507 :
508 : // Set SIGCHLD back to default, so that I
509 : // can get sensible returns from waitpid.
510 0 : memset(&sa, 0, sizeof(sa));
511 : sa.sa_handler=SIG_DFL;
512 0 : sigaction(SIGCHLD, &sa, NULL);
513 :
514 0 : close(pipe_rfd[0]); // close read end
515 0 : close(pipe_wfd[1]); // close write end
516 :
517 0 : confs_free_content(confs);
518 0 : confs_init(confs);
519 :
520 0 : ret=run_child(&cfd, ctx, &client_name, pipe_rfd[1],
521 : fdtype==ASFD_FD_SERVER_LISTEN_STATUS?pipe_wfd[0]:-1,
522 : conffile, forking);
523 :
524 0 : close(pipe_rfd[1]);
525 0 : close(pipe_wfd[0]);
526 0 : close_fd(&cfd);
527 0 : exit(ret);
528 : }
529 : default:
530 : // Parent.
531 0 : close(pipe_rfd[1]); // close write end
532 0 : close(pipe_wfd[0]); // close read end
533 0 : close_fd(&cfd);
534 :
535 0 : return setup_parent_child_pipes(asfd, childpid,
536 : &pipe_rfd[0], &pipe_wfd[1]);
537 : }
538 : }
539 :
540 0 : static int daemonise(void)
541 : {
542 : /* process ID */
543 : pid_t pid;
544 :
545 : /* session ID */
546 : pid_t sid;
547 :
548 : /* fork new child and end parent */
549 0 : pid=fork();
550 :
551 : /* did we fork? */
552 0 : if(pid<0)
553 : {
554 0 : logp("error forking\n");
555 0 : return -1;
556 : }
557 :
558 : /* parent? */
559 0 : if(pid>0)
560 0 : exit(EXIT_SUCCESS);
561 :
562 : /* now we are in the child process */
563 :
564 : /* create a session and set the process group ID */
565 0 : sid=setsid();
566 0 : if(sid<0)
567 : {
568 0 : logp("error setting sid\n");
569 0 : return -1;
570 : }
571 :
572 : /* leave and unblock current working dir */
573 0 : if(chdir("/")<0)
574 : {
575 0 : logp("error changing working dir\n");
576 0 : return -1;
577 : }
578 :
579 : /* close std* */
580 0 : close(STDIN_FILENO);
581 0 : close(STDOUT_FILENO);
582 0 : close(STDERR_FILENO);
583 :
584 0 : return 0;
585 : }
586 :
587 0 : static int extract_client_name(struct asfd *asfd)
588 : {
589 : size_t l;
590 0 : const char *cp=NULL;
591 0 : const char *dp=NULL;
592 :
593 0 : if(asfd->client)
594 : return 0;
595 0 : if(!(dp=strchr(asfd->rbuf->buf, '\t')))
596 : return 0;
597 0 : dp++;
598 0 : if(!(cp=strchr(dp, '\t')))
599 : return 0;
600 0 : cp++;
601 0 : l=cp-dp;
602 0 : if(!(asfd->client=malloc_w(l+1, __func__)))
603 : return -1;
604 0 : snprintf(asfd->client, l, "%s", dp);
605 : return 0;
606 : }
607 :
608 0 : static int write_to_status_children(struct async *mainas, struct iobuf *iobuf)
609 : {
610 : size_t wlen;
611 0 : struct asfd *scfd=NULL;
612 :
613 : // One of the child processes is giving us information.
614 : // Try to append it to any of the status child pipes.
615 0 : for(scfd=mainas->asfd; scfd; scfd=scfd->next)
616 : {
617 0 : if(scfd->fdtype!=ASFD_FD_SERVER_PIPE_WRITE)
618 0 : continue;
619 0 : wlen=iobuf->len;
620 0 : switch(scfd->append_all_to_write_buffer(scfd, iobuf))
621 : {
622 : case APPEND_OK:
623 : // Hack - the append function
624 : // will set the length to zero
625 : // on success. Set it back for
626 : // the next status child pipe.
627 0 : iobuf->len=wlen;
628 : break;
629 : case APPEND_BLOCKED:
630 : break;
631 : default:
632 : return -1;
633 : }
634 : }
635 : // Free the information, even if we did not manage to append it. That
636 : // should be OK, more will be along soon.
637 0 : iobuf_free_content(iobuf);
638 : return 0;
639 : }
640 :
641 0 : static int update_status_child_client_lists(struct async *mainas)
642 : {
643 0 : int ret=-1;
644 0 : char *buf=NULL;
645 0 : struct asfd *a=NULL;
646 : struct iobuf wbuf;
647 :
648 0 : if(!(buf=strdup_w("clients", __func__)))
649 : goto end;
650 0 : for(a=mainas->asfd; a; a=a->next)
651 : {
652 0 : if(a->fdtype!=ASFD_FD_SERVER_PIPE_READ
653 0 : || !a->client)
654 0 : continue;
655 0 : if(astrcat(&buf, "\t", __func__))
656 : goto end;
657 0 : if(astrcat(&buf, a->client, __func__))
658 : goto end;
659 : }
660 :
661 0 : iobuf_set(&wbuf, CMD_GEN, buf, strlen(buf));
662 :
663 0 : ret=write_to_status_children(mainas, &wbuf);
664 : end:
665 0 : return ret;
666 : }
667 :
668 0 : static int maybe_update_status_child_client_lists(struct async *mainas)
669 : {
670 0 : time_t now=0;
671 0 : time_t diff=0;
672 : static time_t lasttime=0;
673 0 : struct asfd *asfd=NULL;
674 :
675 : // If we have no status server child processes, do not bother.
676 0 : for(asfd=mainas->asfd; asfd; asfd=asfd->next)
677 0 : if(asfd->fdtype==ASFD_FD_SERVER_PIPE_WRITE)
678 : break;
679 0 : if(!asfd)
680 : return 0;
681 :
682 : // Only update every 5 seconds.
683 0 : now=time(NULL);
684 0 : diff=now-lasttime;
685 0 : if(diff<5)
686 : {
687 : // Might as well do this in case they fiddled their
688 : // clock back in time.
689 0 : if(diff<0) lasttime=now;
690 : return 0;
691 : }
692 0 : lasttime=now;
693 :
694 0 : return update_status_child_client_lists(mainas);
695 : }
696 :
697 0 : static int run_server(struct conf **confs, const char *conffile)
698 : {
699 0 : int ret=-1;
700 0 : SSL_CTX *ctx=NULL;
701 0 : int found_normal_child=0;
702 0 : struct asfd *asfd=NULL;
703 0 : struct async *mainas=NULL;
704 0 : struct strlist *addresses=get_strlist(confs[OPT_LISTEN]);
705 0 : struct strlist *addresses_status=get_strlist(confs[OPT_LISTEN_STATUS]);
706 :
707 0 : if(!(ctx=ssl_initialise_ctx(confs)))
708 : {
709 0 : logp("error initialising ssl ctx\n");
710 0 : goto end;
711 : }
712 0 : if((ssl_load_dh_params(ctx, confs)))
713 : {
714 0 : logp("error loading dh params\n");
715 0 : goto end;
716 : }
717 :
718 0 : if(!(mainas=async_alloc())
719 0 : || mainas->init(mainas, 0))
720 : goto end;
721 :
722 0 : if(init_listen_sockets(addresses, mainas,
723 : ASFD_FD_SERVER_LISTEN_MAIN, "server")
724 0 : || init_listen_sockets(addresses_status, mainas,
725 : ASFD_FD_SERVER_LISTEN_STATUS, "server status"))
726 : goto end;
727 :
728 0 : while(!hupreload)
729 : {
730 : int removed;
731 0 : switch(mainas->read_write(mainas))
732 : {
733 : case 0:
734 0 : for(asfd=mainas->asfd; asfd; asfd=asfd->next)
735 : {
736 0 : if(asfd->new_client)
737 : {
738 : // Incoming client.
739 0 : asfd->new_client=0;
740 0 : if(process_incoming_client(asfd,
741 : ctx, conffile, confs))
742 : goto end;
743 0 : if(!get_int(confs[OPT_FORK]))
744 : {
745 0 : gentleshutdown++;
746 0 : ret=1;
747 0 : goto end;
748 : }
749 0 : continue;
750 : }
751 : }
752 : break;
753 : default:
754 0 : removed=0;
755 : // Maybe one of the fds had a problem.
756 : // Find and remove it and carry on if possible.
757 0 : for(asfd=mainas->asfd; asfd; )
758 : {
759 : struct asfd *a;
760 0 : if(!asfd->want_to_remove)
761 : {
762 0 : asfd=asfd->next;
763 0 : continue;
764 : }
765 0 : mainas->asfd_remove(mainas, asfd);
766 0 : logp("%s: disconnected fd %d\n",
767 0 : asfd->desc, asfd->fd);
768 0 : a=asfd->next;
769 0 : asfd_free(&asfd);
770 0 : asfd=a;
771 0 : removed++;
772 : }
773 0 : if(removed) break;
774 : // If we got here, there was no fd to remove.
775 : // It is a fatal error.
776 : goto end;
777 : }
778 :
779 0 : for(asfd=mainas->asfd; asfd; asfd=asfd->next)
780 : {
781 0 : if(asfd->fdtype!=ASFD_FD_SERVER_PIPE_READ
782 0 : || !asfd->rbuf->buf)
783 0 : continue;
784 :
785 : //printf("got info from child: %s\n", asfd->rbuf->buf);
786 0 : if(extract_client_name(asfd))
787 : goto end;
788 :
789 0 : if(write_to_status_children(mainas, asfd->rbuf))
790 : goto end;
791 : }
792 :
793 0 : if(maybe_update_status_child_client_lists(mainas))
794 : goto end;
795 :
796 0 : chld_check_for_exiting(mainas);
797 :
798 : // Leave if we had a SIGUSR1 and there are no children running.
799 0 : if(gentleshutdown)
800 : {
801 0 : if(!gentleshutdown_logged)
802 : {
803 0 : logp("got SIGUSR2 gentle reload signal\n");
804 0 : logp("will shut down once children have exited\n");
805 0 : gentleshutdown_logged++;
806 : }
807 : // FIX THIS:
808 : // found_normal_child=chld_add_fd_to_normal_sets(confs, &fsr, &fse, &mfd);
809 : else if(!found_normal_child)
810 : {
811 0 : logp("all children have exited - shutting down\n");
812 0 : break;
813 : }
814 : }
815 : }
816 :
817 0 : if(hupreload) logp("got SIGHUP reload signal\n");
818 :
819 : ret=0;
820 : end:
821 0 : async_asfd_free_all(&mainas);
822 0 : if(ctx) ssl_destroy_ctx(ctx);
823 0 : return ret;
824 : }
825 :
826 0 : int server(struct conf **confs, const char *conffile,
827 : struct lock *lock, int generate_ca_only)
828 : {
829 0 : enum serret ret=SERVER_ERROR;
830 :
831 : //return champ_test(confs);
832 :
833 0 : if(ca_server_setup(confs)) goto error;
834 0 : if(generate_ca_only)
835 : {
836 0 : logp("The '-g' command line option was given. Exiting now.\n");
837 0 : goto end;
838 : }
839 :
840 0 : if(get_int(confs[OPT_FORK]) && get_int(confs[OPT_DAEMON]))
841 : {
842 0 : if(daemonise()
843 : // Need to write the new pid to the already open lock fd.
844 0 : || lock_write_pid_and_prog(lock))
845 : goto error;
846 : }
847 :
848 0 : ssl_load_globals();
849 :
850 0 : while(!gentleshutdown)
851 : {
852 0 : if(run_server(confs, conffile))
853 : goto error;
854 :
855 0 : if(hupreload && !gentleshutdown)
856 : {
857 0 : if(reload(confs, conffile,
858 : 0 // Not first time.
859 : ))
860 : goto error;
861 : }
862 0 : hupreload=0;
863 : }
864 :
865 : end:
866 : ret=SERVER_OK;
867 : error:
868 :
869 : // FIX THIS: Have an enum for a return value, so that it is more obvious what
870 : // is happening, like client.c does.
871 0 : return ret;
872 : }
|
