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

Generated by: LCOV version 1.10