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