LCOV - code coverage report
Current view: top level - src/server - main.c (source / functions) Hit Total Coverage
Test: burp-coverage-clean.info Lines: 0 312 0.0 %
Date: 2016-05-01 Functions: 0 12 0.0 %

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

Generated by: LCOV version 1.10