git.sur5r.net Git - bacula/bacula/blob - bacula/src/lib/bnet_server.c

   1 /*
   2    Copyright (C) 2000-2004 Kern Sibbald and John Walker
   3
   4    This program is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU General Public License as
   6    published by the Free Software Foundation; either version 2 of
   7    the License, or (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12    General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public
  15    License along with this program; if not, write to the Free
  16    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  17    MA 02111-1307, USA.
  18
  19  */
  20  /*
  21   * Originally written by Kern Sibbald for inclusion in apcupsd,
  22   *  but heavily modified for Bacula
  23   *
  24   *   Version $Id$
  25   */
  26
  27 #include "bacula.h"
  28 #undef DEV_BSIZE
  29 #include <netinet/in.h>
  30 #include <sys/socket.h>
  31 #include <stdlib.h>
  32 #include <arpa/inet.h>
  33 #include <netdb.h>
  34 #ifdef HAVE_ARPA_NAMESER_H
  35 #include <arpa/nameser.h>
  36 #endif
  37 #ifdef HAVE_RESOLV_H
  38 #include <resolv.h>
  39 #endif
  40
  41
  42 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
  43
  44 #ifdef HAVE_LIBWRAP
  45 #include "tcpd.h"
  46 int allow_severity = LOG_NOTICE;
  47 int deny_severity = LOG_WARNING;
  48 #endif
  49
  50 static bool quit = false;
  51
  52 void bnet_stop_thread_server(pthread_t tid)
  53 {
  54    quit = true;
  55    pthread_kill(tid, TIMEOUT_SIGNAL);
  56 }
  57
  58 /*
  59         Become Threaded Network Server
  60     This function is able to handle multiple server ips in
  61     ipv4 and ipv6 style. The Addresse are give in a comma
  62     seperated string in bind_addr
  63     In the moment it is inpossible to bind different ports.
  64 */
  65 void
  66 bnet_thread_server(dlist *addrs, int max_clients, workq_t *client_wq,
  67                    void *handle_client_request(void *bsock))
  68 {
  69    int newsockfd, stat;
  70    socklen_t clilen;
  71    struct sockaddr cli_addr;       /* client's address */
  72    int tlog;
  73    int turnon = 1;
  74 #ifdef HAVE_LIBWRAP
  75    struct request_info request;
  76 #endif
  77    IPADDR *p;
  78    struct s_sockfd {
  79       dlink link;                     /* this MUST be the first item */
  80       int fd;
  81       int port;
  82    } *fd_ptr = NULL;
  83    char buf[128];
  84    dlist sockfds;
  85
  86    char allbuf[256 * 10];
  87    Dmsg1(100, "Addresses %s\n", build_addresses_str(addrs, allbuf, sizeof(allbuf)));
  88
  89    foreach_dlist(p, addrs) {
  90       /* Allocate on stack frame -- no need to free */
  91       fd_ptr = (s_sockfd *)alloca(sizeof(s_sockfd));
  92       fd_ptr->port = p->get_port_net_order();
  93       /*
  94        * Open a TCP socket
  95        */
  96       for (tlog= 60; (fd_ptr->fd=socket(p->get_family(), SOCK_STREAM, 0)) < 0; tlog -= 10) {
  97          if (tlog <= 0) {
  98             berrno be;
  99             char curbuf[256];
 100             Emsg3(M_ABORT, 0, _("Cannot open stream socket. ERR=%s. Current %s All %s\n"),
 101                        be.strerror(),
 102                        p->build_address_str(curbuf, sizeof(curbuf)),
 103                        build_addresses_str(addrs, allbuf, sizeof(allbuf)));
 104          }
 105          bmicrosleep(10, 0);
 106       }
 107       /*
 108        * Reuse old sockets
 109        */
 110       if (setsockopt(fd_ptr->fd, SOL_SOCKET, SO_REUSEADDR, (sockopt_val_t) & turnon,
 111            sizeof(turnon)) < 0) {
 112          berrno be;
 113          Emsg1(M_WARNING, 0, _("Cannot set SO_REUSEADDR on socket: %s\n"),
 114                be.strerror());
 115       }
 116
 117       int tmax = 30 * (60 / 5);    /* wait 30 minutes max */
 118       /* FIXME i can go to a endless loop i get a invalid address */
 119       for (tlog = 0; bind(fd_ptr->fd, p->get_sockaddr(), p->get_sockaddr_len()) < 0; tlog -= 5) {
 120          berrno be;
 121          if (tlog <= 0) {
 122             tlog = 2 * 60;         /* Complain every 2 minutes */
 123             Emsg2(M_WARNING, 0, _("Cannot bind port %d: ERR=%s. Retrying ...\n"),
 124                   ntohs(fd_ptr->port), be.strerror());
 125          }
 126          bmicrosleep(5, 0);
 127          if (--tmax <= 0) {
 128             Emsg2(M_ABORT, 0, _("Cannot bind port %d: ERR=%s.\n"), ntohs(fd_ptr->port),
 129                   be.strerror());
 130          }
 131       }
 132       listen(fd_ptr->fd, 5);       /* tell system we are ready */
 133       sockfds.append(fd_ptr);
 134    }
 135    /* Start work queue thread */
 136    if ((stat = workq_init(client_wq, max_clients, handle_client_request)) != 0) {
 137       berrno be;
 138       be.set_errno(stat);
 139       Emsg1(M_ABORT, 0, _("Could not init client queue: ERR=%s\n"), be.strerror());
 140    }
 141    /*
 142     * Wait for a connection from the client process.
 143     */
 144    for (; !quit;) {
 145       unsigned int maxfd = 0;
 146       fd_set sockset;
 147       FD_ZERO(&sockset);
 148       foreach_dlist(fd_ptr, &sockfds) {
 149          FD_SET((unsigned)fd_ptr->fd, &sockset);
 150          maxfd = maxfd > (unsigned)fd_ptr->fd ? maxfd : fd_ptr->fd;
 151       }
 152       errno = 0;
 153       if ((stat = select(maxfd + 1, &sockset, NULL, NULL, NULL)) < 0) {
 154          berrno be;                   /* capture errno */
 155          if (errno == EINTR || errno == EAGAIN) {
 156             continue;
 157          }
 158          /* Error, get out */
 159          foreach_dlist(fd_ptr, &sockfds) {
 160             close(fd_ptr->fd);
 161             free((void *)fd_ptr);
 162          }
 163          Emsg1(M_FATAL, 0, _("Error in select: %s\n"), be.strerror());
 164          break;
 165       }
 166
 167       foreach_dlist(fd_ptr, &sockfds) {
 168          if (FD_ISSET(fd_ptr->fd, &sockset)) {
 169             /* Got a connection, now accept it. */
 170             do {
 171                clilen = sizeof(cli_addr);
 172                newsockfd = accept(fd_ptr->fd, &cli_addr, &clilen);
 173             } while (newsockfd < 0 && (errno == EINTR || errno == EAGAIN));
 174             if (newsockfd < 0) {
 175                continue;
 176             }
 177 #ifdef HAVE_LIBWRAP
 178             P(mutex);              /* hosts_access is not thread safe */
 179             request_init(&request, RQ_DAEMON, my_name, RQ_FILE, newsockfd, 0);
 180             fromhost(&request);
 181             if (!hosts_access(&request)) {
 182                V(mutex);
 183 #ifndef HAVE_INET_NTOP
 184                Jmsg2(NULL, M_SECURITY, 0,
 185                      _("Connection from %s:%d refused by hosts.access\n"),
 186                      inet_ntoa(((sockaddr_in *)&cli_addr)->sin_addr),
 187                      ntohs(((sockaddr_in *)&cli_addr)->sin_port));
 188 #else
 189                Jmsg2(NULL, M_SECURITY, 0,
 190                      _("Connection from %s:%d refused by hosts.access\n"),
 191                      inet_ntop(clilen == sizeof(sockaddr_in) ? AF_INET : AF_INET6,
 192                                &clilen, buf, clilen),
 193                      ntohs(clilen == sizeof(sockaddr_in) ?
 194                            ((sockaddr_in *)&cli_addr)->sin_port :
 195                             ((sockaddr_in6 *)&cli_addr)->sin6_port));
 196 #endif
 197                close(newsockfd);
 198                continue;
 199             }
 200             V(mutex);
 201 #endif
 202
 203             /*
 204              * Receive notification when connection dies.
 205              */
 206             if (setsockopt(newsockfd, SOL_SOCKET, SO_KEEPALIVE, (sockopt_val_t) & turnon,
 207                  sizeof(turnon)) < 0) {
 208                berrno be;
 209                Emsg1(M_WARNING, 0, _("Cannot set SO_KEEPALIVE on socket: %s\n"),
 210                      be.strerror());
 211             }
 212
 213             /* see who client is. i.e. who connected to us. */
 214             P(mutex);
 215 #ifdef HAVE_INET_NTOP
 216             inet_ntop(clilen == sizeof(sockaddr_in) ? AF_INET : AF_INET6, &clilen,
 217                       buf, sizeof(buf));
 218 #else
 219             bstrncpy(buf, inet_ntoa(((sockaddr_in *)&cli_addr)->sin_addr), sizeof(buf));      /* NOT thread safe, use mutex */
 220 #endif
 221             V(mutex);
 222             BSOCK *bs;
 223             bs = init_bsock(NULL, newsockfd, "client", buf, fd_ptr->port, &cli_addr);
 224             if (bs == NULL) {
 225                Jmsg0(NULL, M_ABORT, 0, _("Could not create client BSOCK.\n"));
 226             }
 227
 228             /* Queue client to be served */
 229             if ((stat = workq_add(client_wq, (void *)bs, NULL, 0)) != 0) {
 230                berrno be;
 231                be.set_errno(stat);
 232                Jmsg1(NULL, M_ABORT, 0, _("Could not add job to client queue: ERR=%s\n"),
 233                      be.strerror());
 234             }
 235          }
 236       }
 237    }
 238
 239    /* Stop work queue thread */
 240    if ((stat = workq_destroy(client_wq)) != 0) {
 241       berrno be;
 242       be.set_errno(stat);
 243       Emsg1(M_FATAL, 0, _("Could not destroy client queue: ERR=%s\n"),
 244             be.strerror());
 245    }
 246 }
 247
 248
 249 #ifdef REALLY_USED
 250 /*
 251  * Bind an address so that we may accept connections
 252  * one at a time.
 253  */
 254 BSOCK *bnet_bind(int port)
 255 {
 256    int sockfd;
 257    struct sockaddr_in serv_addr;   /* our address */
 258    int tlog;
 259    int turnon = 1;
 260
 261    /*
 262     * Open a TCP socket
 263     */
 264    for (tlog = 0; (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0; tlog -= 10) {
 265       if (errno == EINTR || errno == EAGAIN) {
 266          continue;
 267       }
 268       if (tlog <= 0) {
 269          tlog = 2 * 60;
 270          Emsg1(M_ERROR, 0, _("Cannot open stream socket: %s\n"), strerror(errno));
 271       }
 272       bmicrosleep(60, 0);
 273    }
 274
 275    /*
 276     * Reuse old sockets
 277     */
 278    if (setsockopt
 279        (sockfd, SOL_SOCKET, SO_REUSEADDR, (sockopt_val_t) & turnon,
 280         sizeof(turnon)) < 0) {
 281       Emsg1(M_WARNING, 0, _("Cannot set SO_REUSEADDR on socket: %s\n"),
 282             strerror(errno));
 283    }
 284
 285    /*
 286     * Bind our local address so that the client can send to us.
 287     */
 288    bzero((char *)&serv_addr, sizeof(serv_addr));
 289    serv_addr.sin_family = AF_INET;
 290    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
 291    serv_addr.sin_port = htons(port);
 292
 293    for (tlog = 0; bind(sockfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0;
 294         tlog -= 5) {
 295       berrno be;
 296       if (errno == EINTR || errno == EAGAIN) {
 297          continue;
 298       }
 299       if (tlog <= 0) {
 300          tlog = 2 * 60;
 301          Emsg2(M_WARNING, 0, _("Cannot bind port %d: ERR=%s: retrying ...\n"), port,
 302                be.strerror());
 303       }
 304       bmicrosleep(5, 0);
 305    }
 306    listen(sockfd, 1);              /* tell system we are ready */
 307    return init_bsock(NULL, sockfd, _("Server socket"), _("client"), port,
 308                      &serv_addr);
 309 }
 310
 311 /*
 312  * Accept a single connection
 313  */
 314 BSOCK *bnet_accept(BSOCK * bsock, char *who)
 315 {
 316    fd_set ready, sockset;
 317    int newsockfd, stat, len;
 318    socklen_t clilen;
 319    struct sockaddr_in cli_addr;    /* client's address */
 320    char *caller, *buf;
 321    BSOCK *bs;
 322    int turnon = 1;
 323 #ifdef HAVE_LIBWRAP
 324    struct request_info request;
 325 #endif
 326
 327    /*
 328     * Wait for a connection from the client process.
 329     */
 330    FD_ZERO(&sockset);
 331    FD_SET((unsigned)bsock->fd, &sockset);
 332
 333    for (;;) {
 334       /*
 335        * Wait for a connection from a client process.
 336        */
 337       ready = sockset;
 338       if ((stat = select(bsock->fd + 1, &ready, NULL, NULL, NULL)) < 0) {
 339          if (errno == EINTR || errno = EAGAIN) {
 340             errno = 0;
 341             continue;
 342          }
 343          Emsg1(M_FATAL, 0, _("Error in select: %s\n"), strerror(errno));
 344          newsockfd = -1;
 345          break;
 346       }
 347       do {
 348          clilen = sizeof(cli_addr);
 349          newsockfd = accept(bsock->fd, (struct sockaddr *)&cli_addr, &clilen);
 350       } while (newsockfd < 0 && (errno == EINTR || errno = EAGAIN));
 351       if (newsockfd >= 0) {
 352          break;
 353       }
 354    }
 355
 356 #ifdef HAVE_LIBWRAP
 357    P(mutex);
 358    request_init(&request, RQ_DAEMON, my_name, RQ_FILE, newsockfd, 0);
 359    fromhost(&request);
 360    if (!hosts_access(&request)) {
 361       V(mutex);
 362       Emsg2(M_SECURITY, 0, _("Connection from %s:%d refused by hosts.access\n"),
 363             inet_ntoa(cli_addr.sin_addr), ntohs(cli_addr.sin_port));
 364       close(newsockfd);
 365       return NULL;
 366    }
 367    V(mutex);
 368 #endif
 369
 370    /*
 371     * Receive notification when connection dies.
 372     */
 373    if (setsockopt
 374        (newsockfd, SOL_SOCKET, SO_KEEPALIVE, (sockopt_val_t) & turnon,
 375         sizeof(turnon)) < 0) {
 376       Emsg1(M_WARNING, 0, _("Cannot set SO_KEEPALIVE on socket: %s\n"),
 377             strerror(errno));
 378    }
 379
 380    /* see who client is. I.e. who connected to us.
 381     * return it in the input message buffer.
 382     */
 383    if ((caller = inet_ntoa(cli_addr.sin_addr)) != NULL) {
 384       pm_strcpy(&bsock->msg, caller);
 385    } else {
 386       bsock->msg[0] = 0;
 387    }
 388    bsock->msglen = strlen(bsock->msg);
 389
 390    if (newsockfd < 0) {
 391       Emsg2(M_FATAL, 0, _("Socket accept error for %s. ERR=%s\n"), who,
 392             strerror(errno));
 393       return NULL;
 394    } else {
 395       if (caller == NULL) {
 396          caller = "unknown";
 397       }
 398       len = strlen(caller) + strlen(who) + 3;
 399       buf = (char *)malloc(len);
 400       bstrncpy(buf, len, who);
 401       bstrncat(buf, len, ": ");
 402       bstrncat(buf, len, caller);
 403       bs = init_bsock(NULL, newsockfd, "client", buf, bsock->port, &cli_addr);
 404       free(buf);
 405       return bs;                   /* return new BSOCK */
 406    }
 407 }
 408
 409 #endif