LCOV - code coverage report
Current view: top level - src/server - main.c (source / functions) Hit Total Coverage
Test: burp-coverage-clean.info Lines: 0 308 0.0 %
Date: 2016-01-31 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             :         // Set up counters. Have to wait until here to get cname.
     241           0 :         if(!(cntr=cntr_alloc())
     242           0 :           || cntr_init(cntr, cname))
     243             :                 goto end;
     244           0 :         set_cntr(confs[OPT_CNTR], cntr);
     245           0 :         set_cntr(cconfs[OPT_CNTR], cntr);
     246             : 
     247             :         /* At this point, the client might want to get a new certificate
     248             :            signed. Clients on 1.3.2 or newer can do this. */
     249           0 :         if((ca_ret=ca_server_maybe_sign_client_cert(as->asfd, confs, cconfs))<0)
     250             :         {
     251             :                 logp("Error signing client certificate request for %s\n",
     252           0 :                         cname);
     253           0 :                 goto end;
     254             :         }
     255           0 :         else if(ca_ret>0)
     256             :         {
     257             :                 // Certificate signed and sent back.
     258             :                 // Everything is OK, but we will close this instance
     259             :                 // so that the client can start again with a new
     260             :                 // connection and its new certificates.
     261             :                 logp("Signed and returned client certificate request for %s\n",
     262           0 :                         cname);
     263           0 :                 ret=0;
     264           0 :                 goto end;
     265             :         }
     266             : 
     267             :         /* Now it is time to check the certificate. */
     268           0 :         if(ssl_check_cert(ssl, confs, cconfs))
     269             :         {
     270           0 :                 log_and_send(as->asfd, "check cert failed on server");
     271           0 :                 goto end;
     272             :         }
     273             : 
     274           0 :         if(status_rfd>=0 && !setup_asfd(as, "status server parent socket",
     275             :                 &status_rfd, NULL,
     276           0 :                 ASFD_STREAM_STANDARD, ASFD_FD_CHILD_PIPE_READ, -1, cconfs))
     277             :                         goto end;
     278             : 
     279           0 :         ret=child(as, status_wfd, confs, cconfs);
     280             : end:
     281           0 :         *cfd=-1;
     282           0 :         if(as && asfd_flush_asio(as->asfd))
     283           0 :                 ret=-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             :                                 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             :                                 break;
     325           0 :                         logp("Too many child processes.\n");
     326             :                         return -1;
     327             :                 case ASFD_FD_SERVER_LISTEN_STATUS:
     328           0 :                         if(sc_count<get_int(confs[OPT_MAX_STATUS_CHILDREN]))
     329             :                                 break;
     330           0 :                         logp("Too many status child processes.\n");
     331             :                         return -1;
     332             :                 default:
     333             :                         logp("Unexpected fdtype in %s: %d.\n",
     334           0 :                                 __func__, asfd->fdtype);
     335             :                         return -1;
     336             :         }
     337             : 
     338             :         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             :                 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             :                         memset(&sa, 0, sizeof(sa));
     414             :                         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             :                                         break;
     467             :                                 default:
     468             :                                         logp("Strange fdtype after fork: %d\n",
     469           0 :                                                 asfd->fdtype);
     470           0 :                                         return -1;
     471             :                         }
     472             : 
     473             :                         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             :                         case GET_LOCK_GOT: return 0;
     533             :                         case GET_LOCK_NOT_GOT:
     534           0 :                                 sleep(2);
     535             :                                 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             :                 goto end;
     575             : 
     576           0 :         if(!(mainas=async_alloc())
     577           0 :           || mainas->init(mainas, 0))
     578             :                 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             :                         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             :                         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             :                                                                 goto end;
     605           0 :                                                 if(!get_int(confs[OPT_FORK]))
     606             :                                                 {
     607           0 :                                                         gentleshutdown++;
     608           0 :                                                         ret=1;
     609           0 :                                                         goto end;
     610             :                                                 }
     611             :                                                 continue;
     612             :                                         }
     613             :                                 }
     614             :                                 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             :                                 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             :                                         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             :                                                 break;
     657             :                                         default:
     658             :                                                 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             :                         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             :         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             :         init_fds(rfds);
     707             :         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             :                         goto error;
     727             : 
     728           0 :                 if(hupreload && !gentleshutdown)
     729             :                 {
     730           0 :                         if(reload(confs, conffile,
     731             :                                 0, // Not first time.
     732             :                                 get_int(confs[OPT_MAX_CHILDREN]),
     733           0 :                                 get_int(confs[OPT_MAX_STATUS_CHILDREN])))
     734             :                                         goto error;
     735             :                 }
     736           0 :                 hupreload=0;
     737             :         }
     738             : 
     739             : end:
     740             :         ret=SERVER_OK;
     741             : error:
     742             :         close_fds(rfds);
     743             :         close_fds(sfds);
     744             : 
     745             : // FIX THIS: Have an enum for a return value, so that it is more obvious what
     746             : // is happening, like client.c does.
     747           0 :         return ret;
     748             : }

Generated by: LCOV version 1.10