libexec/rbootd/rbootd.c

   1 /*
   2  * Copyright (c) 1988, 1992 The University of Utah and the Center
   3  *      for Software Science (CSS).
   4  * Copyright (c) 1992, 1993
   5  *      The Regents of the University of California.  All rights reserved.
   6  *
   7  * This code is derived from software contributed to Berkeley by
   8  * the Center for Software Science of the University of Utah Computer
   9  * Science Department.  CSS requests users of this software to return
  10  * to css-dist@cs.utah.edu any improvements that they make and grant
  11  * CSS redistribution rights.
  12  *
  13  * Redistribution and use in source and binary forms, with or without
  14  * modification, are permitted provided that the following conditions
  15  * are met:
  16  * 1. Redistributions of source code must retain the above copyright
  17  *    notice, this list of conditions and the following disclaimer.
  18  * 2. Redistributions in binary form must reproduce the above copyright
  19  *    notice, this list of conditions and the following disclaimer in the
  20  *    documentation and/or other materials provided with the distribution.
  21  * 3. All advertising materials mentioning features or use of this software
  22  *    must display the following acknowledgement:
  23  *      This product includes software developed by the University of
  24  *      California, Berkeley and its contributors.
  25  * 4. Neither the name of the University nor the names of its contributors
  26  *    may be used to endorse or promote products derived from this software
  27  *    without specific prior written permission.
  28  *
  29  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  30  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  31  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  32  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  33  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  34  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  35  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  36  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  37  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  38  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  39  * SUCH DAMAGE.
  40  *
  41  *      from: @(#)rbootd.c      8.1 (Berkeley) 6/4/93
  42  *
  43  * From: Utah Hdr: rbootd.c 3.1 92/07/06
  44  * Author: Jeff Forys, University of Utah CSS
  45  *
  46  * @(#) Copyright (c) 1992, 1993 The Regents of the University of California.  All rights reserved.
  47  * @(#)rbootd.c 8.1 (Berkeley) 6/4/93
  48  * $FreeBSD: src/libexec/rbootd/rbootd.c,v 1.11.2.1 2001/02/18 02:54:11 kris Exp $
  49  * $DragonFly: src/libexec/rbootd/rbootd.c,v 1.3 2003/11/14 03:54:30 dillon Exp $
  50  */
  51
  52 #include <sys/param.h>
  53 #include <sys/time.h>
  54 #include <ctype.h>
  55 #include <err.h>
  56 #include <errno.h>
  57 #include <fcntl.h>
  58 #include <signal.h>
  59 #include <stdio.h>
  60 #include <stdlib.h>
  61 #include <string.h>
  62 #include <syslog.h>
  63 #include <unistd.h>
  64 #include "defs.h"
  65
  66 static void usage (void);
  67
  68 int
  69 main(argc, argv)
  70         int argc;
  71         char *argv[];
  72 {
  73         int c, fd, omask, maxfds;
  74         fd_set rset;
  75
  76         /*
  77          *  Close any open file descriptors.
  78          *  Temporarily leave stdin & stdout open for `-d',
  79          *  and stderr open for any pre-syslog error messages.
  80          */
  81         {
  82                 int i, nfds = getdtablesize();
  83
  84                 for (i = 0; i < nfds; i++)
  85                         if (i != fileno(stdin) && i != fileno(stdout) &&
  86                             i != fileno(stderr))
  87                                 (void) close(i);
  88         }
  89
  90         /*
  91          *  Parse any arguments.
  92          */
  93         while ((c = getopt(argc, argv, "adi:")) != -1)
  94                 switch(c) {
  95                     case 'a':
  96                         BootAny++;
  97                         break;
  98                     case 'd':
  99                         DebugFlg++;
 100                         break;
 101                     case 'i':
 102                         IntfName = optarg;
 103                         break;
 104                     default:
 105                         usage();
 106                 }
 107         for (; optind < argc; optind++) {
 108                 if (ConfigFile == NULL)
 109                         ConfigFile = argv[optind];
 110                 else {
 111                         warnx("too many config files (`%s' ignored)",
 112                             argv[optind]);
 113                 }
 114         }
 115
 116         if (ConfigFile == NULL)                 /* use default config file */
 117                 ConfigFile = DfltConfig;
 118
 119         if (DebugFlg) {
 120                 DbgFp = stdout;                         /* output to stdout */
 121
 122                 (void) signal(SIGUSR1, SIG_IGN);        /* dont muck w/DbgFp */
 123                 (void) signal(SIGUSR2, SIG_IGN);
 124                 (void) fclose(stderr);                  /* finished with it */
 125         } else {
 126                 if (daemon(0, 0))
 127                         err(1, "can't detach from terminal");
 128
 129                 (void) signal(SIGUSR1, DebugOn);
 130                 (void) signal(SIGUSR2, DebugOff);
 131         }
 132
 133         openlog("rbootd", LOG_PID, LOG_DAEMON);
 134
 135         /*
 136          *  If no interface was specified, get one now.
 137          *
 138          *  This is convoluted because we want to get the default interface
 139          *  name for the syslog("restarted") message.  If BpfGetIntfName()
 140          *  runs into an error, it will return a syslog-able error message
 141          *  (in `errmsg') which will be displayed here.
 142          */
 143         if (IntfName == NULL) {
 144                 char *errmsg;
 145
 146                 if ((IntfName = BpfGetIntfName(&errmsg)) == NULL) {
 147                         syslog(LOG_NOTICE, "restarted (??)");
 148                         /* BpfGetIntfName() returns safe names, using %m */
 149                         syslog(LOG_ERR, "%s", errmsg);
 150                         Exit(0);
 151                 }
 152         }
 153
 154         syslog(LOG_NOTICE, "restarted (%s)", IntfName);
 155
 156         (void) signal(SIGHUP, ReConfig);
 157         (void) signal(SIGINT, Exit);
 158         (void) signal(SIGTERM, Exit);
 159
 160         /*
 161          *  Grab our host name and pid.
 162          */
 163         if (gethostname(MyHost, MAXHOSTNAMELEN - 1) < 0) {
 164                 syslog(LOG_ERR, "gethostname: %m");
 165                 Exit(0);
 166         }
 167         MyHost[MAXHOSTNAMELEN - 1] = '\0';
 168
 169         MyPid = getpid();
 170
 171         /*
 172          *  Write proc's pid to a file.
 173          */
 174         {
 175                 FILE *fp;
 176
 177                 if ((fp = fopen(PidFile, "w")) != NULL) {
 178                         (void) fprintf(fp, "%d\n", (int) MyPid);
 179                         (void) fclose(fp);
 180                 } else {
 181                         syslog(LOG_WARNING, "fopen: failed (%s)", PidFile);
 182                 }
 183         }
 184
 185         /*
 186          *  All boot files are relative to the boot directory, we might
 187          *  as well chdir() there to make life easier.
 188          */
 189         if (chdir(BootDir) < 0) {
 190                 syslog(LOG_ERR, "chdir: %m (%s)", BootDir);
 191                 Exit(0);
 192         }
 193
 194         /*
 195          *  Initial configuration.
 196          */
 197         omask = sigblock(sigmask(SIGHUP));      /* prevent reconfig's */
 198         if (GetBootFiles() == 0)                /* get list of boot files */
 199                 Exit(0);
 200         if (ParseConfig() == 0)                 /* parse config file */
 201                 Exit(0);
 202
 203         /*
 204          *  Open and initialize a BPF device for the appropriate interface.
 205          *  If an error is encountered, a message is displayed and Exit()
 206          *  is called.
 207          */
 208         fd = BpfOpen();
 209
 210         (void) sigsetmask(omask);               /* allow reconfig's */
 211
 212         /*
 213          *  Main loop: receive a packet, determine where it came from,
 214          *  and if we service this host, call routine to handle request.
 215          */
 216         maxfds = fd + 1;
 217         FD_ZERO(&rset);
 218         FD_SET(fd, &rset);
 219         for (;;) {
 220                 struct timeval timeout;
 221                 fd_set r;
 222                 int nsel;
 223
 224                 r = rset;
 225
 226                 if (RmpConns == NULL) {         /* timeout isnt necessary */
 227                         nsel = select(maxfds, &r, NULL, NULL, NULL);
 228                 } else {
 229                         timeout.tv_sec = RMP_TIMEOUT;
 230                         timeout.tv_usec = 0;
 231                         nsel = select(maxfds, &r, NULL, NULL, &timeout);
 232                 }
 233
 234                 if (nsel < 0) {
 235                         if (errno == EINTR)
 236                                 continue;
 237                         syslog(LOG_ERR, "select: %m");
 238                         Exit(0);
 239                 } else if (nsel == 0) {         /* timeout */
 240                         DoTimeout();                    /* clear stale conns */
 241                         continue;
 242                 }
 243
 244                 if (FD_ISSET(fd, &r)) {
 245                         RMPCONN rconn;
 246                         CLIENT *client, *FindClient();
 247                         int doread = 1;
 248
 249                         while (BpfRead(&rconn, doread)) {
 250                                 doread = 0;
 251
 252                                 if (DbgFp != NULL)      /* display packet */
 253                                         DispPkt(&rconn,DIR_RCVD);
 254
 255                                 omask = sigblock(sigmask(SIGHUP));
 256
 257                                 /*
 258                                  *  If we do not restrict service, set the
 259                                  *  client to NULL (ProcessPacket() handles
 260                                  *  this).  Otherwise, check that we can
 261                                  *  service this host; if not, log a message
 262                                  *  and ignore the packet.
 263                                  */
 264                                 if (BootAny) {
 265                                         client = NULL;
 266                                 } else if ((client=FindClient(&rconn))==NULL) {
 267                                         syslog(LOG_INFO,
 268                                                "%s: boot packet ignored",
 269                                                EnetStr(&rconn));
 270                                         (void) sigsetmask(omask);
 271                                         continue;
 272                                 }
 273
 274                                 ProcessPacket(&rconn,client);
 275
 276                                 (void) sigsetmask(omask);
 277                         }
 278                 }
 279         }
 280 }
 281
 282 static void
 283 usage()
 284 {
 285         fprintf(stderr, "usage: rbootd [-ad] [-i interface] [config_file]\n");
 286         exit (1);
 287 }
 288
 289 /*
 290 **  DoTimeout -- Free any connections that have timed out.
 291 **
 292 **      Parameters:
 293 **              None.
 294 **
 295 **      Returns:
 296 **              Nothing.
 297 **
 298 **      Side Effects:
 299 **              - Timed out connections in `RmpConns' will be freed.
 300 */
 301 void
 302 DoTimeout()
 303 {
 304         RMPCONN *rtmp;
 305         struct timeval now;
 306
 307         (void) gettimeofday(&now, (struct timezone *)0);
 308
 309         /*
 310          *  For each active connection, if RMP_TIMEOUT seconds have passed
 311          *  since the last packet was sent, delete the connection.
 312          */
 313         for (rtmp = RmpConns; rtmp != NULL; rtmp = rtmp->next)
 314                 if ((rtmp->tstamp.tv_sec + RMP_TIMEOUT) < now.tv_sec) {
 315                         syslog(LOG_WARNING, "%s: connection timed out (%u)",
 316                                EnetStr(rtmp), rtmp->rmp.r_type);
 317                         RemoveConn(rtmp);
 318                 }
 319 }
 320
 321 /*
 322 **  FindClient -- Find client associated with a packet.
 323 **
 324 **      Parameters:
 325 **              rconn - the new packet.
 326 **
 327 **      Returns:
 328 **              Pointer to client info if found, NULL otherwise.
 329 **
 330 **      Side Effects:
 331 **              None.
 332 **
 333 **      Warnings:
 334 **              - This routine must be called with SIGHUP blocked since
 335 **                a reconfigure can invalidate the information returned.
 336 */
 337
 338 CLIENT *
 339 FindClient(rconn)
 340         RMPCONN *rconn;
 341 {
 342         CLIENT *ctmp;
 343
 344         for (ctmp = Clients; ctmp != NULL; ctmp = ctmp->next)
 345                 if (bcmp((char *)&rconn->rmp.hp_hdr.saddr[0],
 346                          (char *)&ctmp->addr[0], RMP_ADDRLEN) == 0)
 347                         break;
 348
 349         return(ctmp);
 350 }
 351
 352 /*
 353 **  Exit -- Log an error message and exit.
 354 **
 355 **      Parameters:
 356 **              sig - caught signal (or zero if not dying on a signal).
 357 **
 358 **      Returns:
 359 **              Does not return.
 360 **
 361 **      Side Effects:
 362 **              - This process ceases to exist.
 363 */
 364 void
 365 Exit(sig)
 366         int sig;
 367 {
 368         if (sig > 0)
 369                 syslog(LOG_ERR, "going down on signal %d", sig);
 370         else
 371                 syslog(LOG_ERR, "going down with fatal error");
 372         BpfClose();
 373         exit(1);
 374 }
 375
 376 /*
 377 **  ReConfig -- Get new list of boot files and reread config files.
 378 **
 379 **      Parameters:
 380 **              None.
 381 **
 382 **      Returns:
 383 **              Nothing.
 384 **
 385 **      Side Effects:
 386 **              - All active connections are dropped.
 387 **              - List of boot-able files is changed.
 388 **              - List of clients is changed.
 389 **
 390 **      Warnings:
 391 **              - This routine must be called with SIGHUP blocked.
 392 */
 393 void
 394 ReConfig(signo)
 395         int signo;
 396 {
 397         syslog(LOG_NOTICE, "reconfiguring boot server");
 398
 399         FreeConns();
 400
 401         if (GetBootFiles() == 0)
 402                 Exit(0);
 403
 404         if (ParseConfig() == 0)
 405                 Exit(0);
 406 }
 407
 408 /*
 409 **  DebugOff -- Turn off debugging.
 410 **
 411 **      Parameters:
 412 **              None.
 413 **
 414 **      Returns:
 415 **              Nothing.
 416 **
 417 **      Side Effects:
 418 **              - Debug file is closed.
 419 */
 420 void
 421 DebugOff(signo)
 422         int signo;
 423 {
 424         if (DbgFp != NULL)
 425                 (void) fclose(DbgFp);
 426
 427         DbgFp = NULL;
 428 }
 429
 430 /*
 431 **  DebugOn -- Turn on debugging.
 432 **
 433 **      Parameters:
 434 **              None.
 435 **
 436 **      Returns:
 437 **              Nothing.
 438 **
 439 **      Side Effects:
 440 **              - Debug file is opened/truncated if not already opened,
 441 **                otherwise do nothing.
 442 */
 443 void
 444 DebugOn(signo)
 445         int signo;
 446 {
 447         if (DbgFp == NULL) {
 448                 if ((DbgFp = fopen(DbgFile, "w")) == NULL)
 449                         syslog(LOG_ERR, "can't open debug file (%s)", DbgFile);
 450         }
 451 }