]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - usr.sbin/inetd/builtins.c
This commit was generated by cvs2svn to compensate for changes in r156373,
[FreeBSD/FreeBSD.git] / usr.sbin / inetd / builtins.c
1 /*-
2  * Copyright (c) 1983, 1991, 1993, 1994
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  */
26
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
29
30 #include <sys/filio.h>
31 #include <sys/ioccom.h>
32 #include <sys/param.h>
33 #include <sys/stat.h>
34 #include <sys/socket.h>
35 #include <sys/sysctl.h>
36 #include <sys/ucred.h>
37 #include <sys/uio.h>
38 #include <sys/utsname.h>
39
40 #include <ctype.h>
41 #include <err.h>
42 #include <errno.h>
43 #include <fcntl.h>
44 #include <limits.h>
45 #include <pwd.h>
46 #include <signal.h>
47 #include <stdlib.h>
48 #include <string.h>
49 #include <sysexits.h>
50 #include <syslog.h>
51 #include <unistd.h>
52
53 #include "inetd.h"
54
55 void            chargen_dg(int, struct servtab *);
56 void            chargen_stream(int, struct servtab *);
57 void            daytime_dg(int, struct servtab *);
58 void            daytime_stream(int, struct servtab *);
59 void            discard_dg(int, struct servtab *);
60 void            discard_stream(int, struct servtab *);
61 void            echo_dg(int, struct servtab *);
62 void            echo_stream(int, struct servtab *);
63 static int      getline(int, char *, int);
64 void            iderror(int, int, int, const char *);
65 void            ident_stream(int, struct servtab *);
66 void            initring(void);
67 unsigned long   machtime(void);
68 void            machtime_dg(int, struct servtab *);
69 void            machtime_stream(int, struct servtab *);
70
71 char ring[128];
72 char *endring;
73
74
75 struct biltin biltins[] = {
76         /* Echo received data */
77         { "echo",       SOCK_STREAM,    1, -1,  echo_stream },
78         { "echo",       SOCK_DGRAM,     0, 1,   echo_dg },
79
80         /* Internet /dev/null */
81         { "discard",    SOCK_STREAM,    1, -1,  discard_stream },
82         { "discard",    SOCK_DGRAM,     0, 1,   discard_dg },
83
84         /* Return 32 bit time since 1900 */
85         { "time",       SOCK_STREAM,    0, -1,  machtime_stream },
86         { "time",       SOCK_DGRAM,     0, 1,   machtime_dg },
87
88         /* Return human-readable time */
89         { "daytime",    SOCK_STREAM,    0, -1,  daytime_stream },
90         { "daytime",    SOCK_DGRAM,     0, 1,   daytime_dg },
91
92         /* Familiar character generator */
93         { "chargen",    SOCK_STREAM,    1, -1,  chargen_stream },
94         { "chargen",    SOCK_DGRAM,     0, 1,   chargen_dg },
95
96         { "tcpmux",     SOCK_STREAM,    1, -1,  (bi_fn_t *)tcpmux },
97
98         { "auth",       SOCK_STREAM,    1, -1,  ident_stream },
99
100         { NULL,         0,              0, 0,   NULL }
101 };
102
103 /*
104  * RFC864 Character Generator Protocol. Generates character data without
105  * any regard for input.
106  */
107
108 void
109 initring(void)
110 {
111         int i;
112
113         endring = ring;
114
115         for (i = 0; i <= 128; ++i)
116                 if (isprint(i))
117                         *endring++ = i;
118 }
119
120 /* Character generator */
121 /* ARGSUSED */
122 void
123 chargen_dg(int s, struct servtab *sep)
124 {
125         struct sockaddr_storage ss;
126         static char *rs;
127         int len;
128         socklen_t size;
129         char text[LINESIZ+2];
130
131         if (endring == 0) {
132                 initring();
133                 rs = ring;
134         }
135
136         size = sizeof(ss);
137         if (recvfrom(s, text, sizeof(text), 0,
138                      (struct sockaddr *)&ss, &size) < 0)
139                 return;
140
141         if (check_loop((struct sockaddr *)&ss, sep))
142                 return;
143
144         if ((len = endring - rs) >= LINESIZ)
145                 memmove(text, rs, LINESIZ);
146         else {
147                 memmove(text, rs, len);
148                 memmove(text + len, ring, LINESIZ - len);
149         }
150         if (++rs == endring)
151                 rs = ring;
152         text[LINESIZ] = '\r';
153         text[LINESIZ + 1] = '\n';
154         (void) sendto(s, text, sizeof(text), 0, (struct sockaddr *)&ss, size);
155 }
156
157 /* Character generator */
158 /* ARGSUSED */
159 void
160 chargen_stream(int s, struct servtab *sep)
161 {
162         int len;
163         char *rs, text[LINESIZ+2];
164
165         inetd_setproctitle(sep->se_service, s);
166
167         if (!endring) {
168                 initring();
169                 rs = ring;
170         }
171
172         text[LINESIZ] = '\r';
173         text[LINESIZ + 1] = '\n';
174         for (rs = ring;;) {
175                 if ((len = endring - rs) >= LINESIZ)
176                         memmove(text, rs, LINESIZ);
177                 else {
178                         memmove(text, rs, len);
179                         memmove(text + len, ring, LINESIZ - len);
180                 }
181                 if (++rs == endring)
182                         rs = ring;
183                 if (write(s, text, sizeof(text)) != sizeof(text))
184                         break;
185         }
186         exit(0);
187 }
188
189 /*
190  * RFC867 Daytime Protocol. Sends the current date and time as an ascii
191  * character string without any regard for input.
192  */
193
194 /* Return human-readable time of day */
195 /* ARGSUSED */
196 void
197 daytime_dg(int s, struct servtab *sep)
198 {
199         char buffer[256];
200         time_t now;
201         struct sockaddr_storage ss;
202         socklen_t size;
203
204         now = time((time_t *) 0);
205
206         size = sizeof(ss);
207         if (recvfrom(s, buffer, sizeof(buffer), 0,
208                      (struct sockaddr *)&ss, &size) < 0)
209                 return;
210
211         if (check_loop((struct sockaddr *)&ss, sep))
212                 return;
213
214         (void) sprintf(buffer, "%.24s\r\n", ctime(&now));
215         (void) sendto(s, buffer, strlen(buffer), 0,
216                       (struct sockaddr *)&ss, size);
217 }
218
219 /* Return human-readable time of day */
220 /* ARGSUSED */
221 void
222 daytime_stream(int s, struct servtab *sep __unused)
223 {
224         char buffer[256];
225         time_t now;
226
227         now = time((time_t *) 0);
228
229         (void) sprintf(buffer, "%.24s\r\n", ctime(&now));
230         (void) send(s, buffer, strlen(buffer), MSG_EOF);
231 }
232
233 /*
234  * RFC863 Discard Protocol. Any data received is thrown away and no response
235  * is sent.
236  */
237
238 /* Discard service -- ignore data */
239 /* ARGSUSED */
240 void
241 discard_dg(int s, struct servtab *sep __unused)
242 {
243         char buffer[BUFSIZE];
244
245         (void) read(s, buffer, sizeof(buffer));
246 }
247
248 /* Discard service -- ignore data */
249 /* ARGSUSED */
250 void
251 discard_stream(int s, struct servtab *sep)
252 {
253         int ret;
254         char buffer[BUFSIZE];
255
256         inetd_setproctitle(sep->se_service, s);
257         while (1) {
258                 while ((ret = read(s, buffer, sizeof(buffer))) > 0)
259                         ;
260                 if (ret == 0 || errno != EINTR)
261                         break;
262         }
263         exit(0);
264 }
265
266 /*
267  * RFC862 Echo Protocol. Any data received is sent back to the sender as
268  * received.
269  */
270
271 /* Echo service -- echo data back */
272 /* ARGSUSED */
273 void
274 echo_dg(int s, struct servtab *sep)
275 {
276         char buffer[65536]; /* Should be sizeof(max datagram). */
277         int i;
278         socklen_t size;
279         struct sockaddr_storage ss;
280
281         size = sizeof(ss);
282         if ((i = recvfrom(s, buffer, sizeof(buffer), 0,
283                           (struct sockaddr *)&ss, &size)) < 0)
284                 return;
285
286         if (check_loop((struct sockaddr *)&ss, sep))
287                 return;
288
289         (void) sendto(s, buffer, i, 0, (struct sockaddr *)&ss, size);
290 }
291
292 /* Echo service -- echo data back */
293 /* ARGSUSED */
294 void
295 echo_stream(int s, struct servtab *sep)
296 {
297         char buffer[BUFSIZE];
298         int i;
299
300         inetd_setproctitle(sep->se_service, s);
301         while ((i = read(s, buffer, sizeof(buffer))) > 0 &&
302             write(s, buffer, i) > 0)
303                 ;
304         exit(0);
305 }
306
307 /*
308  * RFC1413 Identification Protocol. Given a TCP port number pair, return a
309  * character string which identifies the owner of that connection on the
310  * server's system. Extended to allow for ~/.fakeid support and ~/.noident
311  * support.
312  */
313
314 /* RFC 1413 says the following are the only errors you can return. */
315 #define ID_INVALID      "INVALID-PORT"  /* Port number improperly specified. */
316 #define ID_NOUSER       "NO-USER"       /* Port not in use/not identifable. */
317 #define ID_HIDDEN       "HIDDEN-USER"   /* Hiden at user's request. */
318 #define ID_UNKNOWN      "UNKNOWN-ERROR" /* Everything else. */
319
320 /* Generic ident_stream error-sending func */
321 /* ARGSUSED */
322 void
323 iderror(int lport, int fport, int s, const char *er)
324 {
325         char *p;
326
327         asprintf(&p, "%d , %d : ERROR : %s\r\n", lport, fport, er);
328         if (p == NULL) {
329                 syslog(LOG_ERR, "asprintf: %m");
330                 exit(EX_OSERR);
331         }
332         send(s, p, strlen(p), MSG_EOF);
333         free(p);
334
335         exit(0);
336 }
337
338 /* Ident service (AKA "auth") */
339 /* ARGSUSED */
340 void
341 ident_stream(int s, struct servtab *sep)
342 {
343         struct utsname un;
344         struct stat sb;
345         struct sockaddr_in sin4[2];
346 #ifdef INET6
347         struct sockaddr_in6 sin6[2];
348 #endif
349         struct sockaddr_storage ss[2];
350         struct xucred uc;
351         struct timeval tv = {
352                 10,
353                 0
354         }, to;
355         struct passwd *pw = NULL;
356         fd_set fdset;
357         char buf[BUFSIZE], *p, **av, *osname = NULL, e;
358         char idbuf[MAXLOGNAME] = ""; /* Big enough to hold uid in decimal. */
359         socklen_t socklen;
360         ssize_t ssize;
361         size_t size, bufsiz;
362         int c, fflag = 0, nflag = 0, rflag = 0, argc = 0;
363         int gflag = 0, iflag = 0, Fflag = 0, getcredfail = 0, onreadlen;
364         u_short lport, fport;
365
366         inetd_setproctitle(sep->se_service, s);
367         /*
368          * Reset getopt() since we are a fork() but not an exec() from
369          * a parent which used getopt() already.
370          */
371         optind = 1;
372         optreset = 1;
373         /*
374          * Take the internal argument vector and count it out to make an
375          * argument count for getopt. This can be used for any internal
376          * service to read arguments and use getopt() easily.
377          */
378         for (av = sep->se_argv; *av; av++)
379                 argc++;
380         if (argc) {
381                 int sec, usec;
382                 size_t i;
383                 u_int32_t rnd32;
384
385                 while ((c = getopt(argc, sep->se_argv, "d:fFgino:rt:")) != -1)
386                         switch (c) {
387                         case 'd':
388                                 if (!gflag)
389                                         strlcpy(idbuf, optarg, sizeof(idbuf));
390                                 break;
391                         case 'f':
392                                 fflag = 1;
393                                 break;
394                         case 'F':
395                                 fflag = 1;
396                                 Fflag=1;
397                                 break;
398                         case 'g':
399                                 gflag = 1;
400                                 rnd32 = 0;      /* Shush, compiler. */
401                                 /*
402                                  * The number of bits in "rnd32" divided
403                                  * by the number of bits needed per iteration
404                                  * gives a more optimal way to reload the
405                                  * random number only when necessary.
406                                  *
407                                  * 32 bits from arc4random corresponds to
408                                  * about 6 base-36 digits, so we reseed evey 6.
409                                  */
410                                 for (i = 0; i < sizeof(idbuf) - 1; i++) {
411                                         static const char *const base36 =
412                                             "0123456789"
413                                             "abcdefghijklmnopqrstuvwxyz";
414                                         if (i % 6 == 0)
415                                                 rnd32 = arc4random();
416                                         idbuf[i] = base36[rnd32 % 36];
417                                         rnd32 /= 36;
418                                 }
419                                 idbuf[i] = '\0';
420                                 break;
421                         case 'i':
422                                 iflag = 1;
423                                 break;
424                         case 'n':
425                                 nflag = 1;
426                                 break;
427                         case 'o':
428                                 osname = optarg;
429                                 break;
430                         case 'r':
431                                 rflag = 1;
432                                 break;
433                         case 't':
434                                 switch (sscanf(optarg, "%d.%d", &sec, &usec)) {
435                                 case 2:
436                                         tv.tv_usec = usec;
437                                         /* FALLTHROUGH */
438                                 case 1:
439                                         tv.tv_sec = sec;
440                                         break;
441                                 default:
442                                         if (debug)
443                                                 warnx("bad -t argument");
444                                         break;
445                                 }
446                                 break;
447                         default:
448                                 break;
449                         }
450         }
451         if (osname == NULL) {
452                 if (uname(&un) == -1)
453                         iderror(0, 0, s, ID_UNKNOWN);
454                 osname = un.sysname;
455         }
456
457         /*
458          * We're going to prepare for and execute reception of a
459          * packet of data from the user. The data is in the format
460          * "local_port , foreign_port\r\n" (with local being the
461          * server's port and foreign being the client's.)
462          */
463         gettimeofday(&to, NULL);
464         to.tv_sec += tv.tv_sec;
465         to.tv_usec += tv.tv_usec;
466         if (to.tv_usec >= 1000000) {
467                 to.tv_usec -= 1000000;
468                 to.tv_sec++;
469         }
470
471         size = 0;
472         bufsiz = sizeof(buf) - 1;
473         FD_ZERO(&fdset);
474         while (bufsiz > 0) {
475                 gettimeofday(&tv, NULL);
476                 tv.tv_sec = to.tv_sec - tv.tv_sec;
477                 tv.tv_usec = to.tv_usec - tv.tv_usec;
478                 if (tv.tv_usec < 0) {
479                         tv.tv_usec += 1000000;
480                         tv.tv_sec--;
481                 }
482                 if (tv.tv_sec < 0)
483                         break;
484                 FD_SET(s, &fdset);
485                 if (select(s + 1, &fdset, NULL, NULL, &tv) == -1)
486                         iderror(0, 0, s, ID_UNKNOWN);
487                 if (ioctl(s, FIONREAD, &onreadlen) == -1)
488                         iderror(0, 0, s, ID_UNKNOWN);
489                 if ((size_t)onreadlen > bufsiz)
490                         onreadlen = bufsiz;
491                 ssize = read(s, &buf[size], (size_t)onreadlen);
492                 if (ssize == -1)
493                         iderror(0, 0, s, ID_UNKNOWN);
494                 else if (ssize == 0)
495                         break;
496                 bufsiz -= ssize;
497                 size += ssize;
498                 if (memchr(&buf[size - ssize], '\n', ssize) != NULL)
499                         break;
500         }
501         buf[size] = '\0';
502         /* Read two characters, and check for a delimiting character */
503         if (sscanf(buf, "%hu , %hu%c", &lport, &fport, &e) != 3 || isdigit(e))
504                 iderror(0, 0, s, ID_INVALID);
505
506         /* Send garbage? */
507         if (gflag)
508                 goto printit;
509
510         /*
511          * If not "real" (-r), send a HIDDEN-USER error for everything.
512          * If -d is used to set a fallback username, this is used to
513          * override it, and the fallback is returned instead.
514          */
515         if (!rflag) {
516                 if (*idbuf == '\0')
517                         iderror(lport, fport, s, ID_HIDDEN);
518                 goto printit;
519         }
520
521         /*
522          * We take the input and construct an array of two sockaddr_ins
523          * which contain the local address information and foreign
524          * address information, respectively, used to look up the
525          * credentials for the socket (which are returned by the
526          * sysctl "net.inet.tcp.getcred" when we call it.)
527          */
528         socklen = sizeof(ss[0]);
529         if (getsockname(s, (struct sockaddr *)&ss[0], &socklen) == -1)
530                 iderror(lport, fport, s, ID_UNKNOWN);
531         socklen = sizeof(ss[1]);
532         if (getpeername(s, (struct sockaddr *)&ss[1], &socklen) == -1)
533                 iderror(lport, fport, s, ID_UNKNOWN);
534         if (ss[0].ss_family != ss[1].ss_family)
535                 iderror(lport, fport, s, ID_UNKNOWN);
536         size = sizeof(uc);
537         switch (ss[0].ss_family) {
538         case AF_INET:
539                 sin4[0] = *(struct sockaddr_in *)&ss[0];
540                 sin4[0].sin_port = htons(lport);
541                 sin4[1] = *(struct sockaddr_in *)&ss[1];
542                 sin4[1].sin_port = htons(fport);
543                 if (sysctlbyname("net.inet.tcp.getcred", &uc, &size, sin4,
544                                  sizeof(sin4)) == -1)
545                         getcredfail = errno;
546                 break;
547 #ifdef INET6
548         case AF_INET6:
549                 sin6[0] = *(struct sockaddr_in6 *)&ss[0];
550                 sin6[0].sin6_port = htons(lport);
551                 sin6[1] = *(struct sockaddr_in6 *)&ss[1];
552                 sin6[1].sin6_port = htons(fport);
553                 if (sysctlbyname("net.inet6.tcp6.getcred", &uc, &size, sin6,
554                                  sizeof(sin6)) == -1)
555                         getcredfail = errno;
556                 break;
557 #endif
558         default: /* should not reach here */
559                 getcredfail = EAFNOSUPPORT;
560                 break;
561         }
562         if (getcredfail != 0 || uc.cr_version != XUCRED_VERSION) {
563                 if (*idbuf == '\0')
564                         iderror(lport, fport, s,
565                             getcredfail == ENOENT ? ID_NOUSER : ID_UNKNOWN);
566                 goto printit;
567         }
568
569         /* Look up the pw to get the username and home directory*/
570         errno = 0;
571         pw = getpwuid(uc.cr_uid);
572         if (pw == NULL)
573                 iderror(lport, fport, s, errno == 0 ? ID_NOUSER : ID_UNKNOWN);
574
575         if (iflag)
576                 snprintf(idbuf, sizeof(idbuf), "%u", (unsigned)pw->pw_uid);
577         else
578                 strlcpy(idbuf, pw->pw_name, sizeof(idbuf));
579
580         /*
581          * If enabled, we check for a file named ".noident" in the user's
582          * home directory. If found, we return HIDDEN-USER.
583          */
584         if (nflag) {
585                 if (asprintf(&p, "%s/.noident", pw->pw_dir) == -1)
586                         iderror(lport, fport, s, ID_UNKNOWN);
587                 if (lstat(p, &sb) == 0) {
588                         free(p);
589                         iderror(lport, fport, s, ID_HIDDEN);
590                 }
591                 free(p);
592         }
593
594         /*
595          * Here, if enabled, we read a user's ".fakeid" file in their
596          * home directory. It consists of a line containing the name
597          * they want.
598          */
599         if (fflag) {
600                 int fakeid_fd;
601
602                 /*
603                  * Here we set ourself to effectively be the user, so we don't
604                  * open any files we have no permission to open, especially
605                  * symbolic links to sensitive root-owned files or devices.
606                  */
607                 if (initgroups(pw->pw_name, pw->pw_gid) == -1)
608                         iderror(lport, fport, s, ID_UNKNOWN);
609                 if (seteuid(pw->pw_uid) == -1)
610                         iderror(lport, fport, s, ID_UNKNOWN);
611                 /*
612                  * We can't stat() here since that would be a race
613                  * condition.
614                  * Therefore, we open the file we have permissions to open
615                  * and if it's not a regular file, we close it and end up
616                  * returning the user's real username.
617                  */
618                 if (asprintf(&p, "%s/.fakeid", pw->pw_dir) == -1)
619                         iderror(lport, fport, s, ID_UNKNOWN);
620                 fakeid_fd = open(p, O_RDONLY | O_NONBLOCK);
621                 free(p);
622                 if (fakeid_fd == -1 || fstat(fakeid_fd, &sb) == -1 ||
623                     !S_ISREG(sb.st_mode))
624                         goto fakeid_fail;
625
626                 if ((ssize = read(fakeid_fd, buf, sizeof(buf) - 1)) < 0)
627                         goto fakeid_fail;
628                 buf[ssize] = '\0';
629
630                 /*
631                  * Usually, the file will have the desired identity
632                  * in the form "identity\n". Allow for leading white
633                  * space and trailing white space/end of line.
634                  */
635                 p = buf;
636                 p += strspn(p, " \t");
637                 p[strcspn(p, " \t\r\n")] = '\0';
638                 if (strlen(p) > MAXLOGNAME - 1) /* Too long (including nul)? */
639                         p[MAXLOGNAME - 1] = '\0';
640
641                 /*
642                  * If the name is a zero-length string or matches it
643                  * the id or name of another user (unless permitted by -F)
644                  * then it is invalid.
645                  */
646                 if (*p == '\0')
647                         goto fakeid_fail;
648                 if (!Fflag) {
649                         if (iflag) {
650                                 if (p[strspn(p, "0123456789")] == '\0' &&
651                                     getpwuid(atoi(p)) != NULL)
652                                         goto fakeid_fail;
653                         } else {
654                                 if (getpwnam(p) != NULL)
655                                         goto fakeid_fail;
656                         }
657                 }
658
659                 strlcpy(idbuf, p, sizeof(idbuf));
660
661 fakeid_fail:
662                 if (fakeid_fd != -1)
663                         close(fakeid_fd);
664         }
665
666 printit:
667         /* Finally, we make and send the reply. */
668         if (asprintf(&p, "%d , %d : USERID : %s : %s\r\n", lport, fport, osname,
669             idbuf) == -1) {
670                 syslog(LOG_ERR, "asprintf: %m");
671                 exit(EX_OSERR);
672         }
673         send(s, p, strlen(p), MSG_EOF);
674         free(p);
675         
676         exit(0);
677 }
678
679 /*
680  * RFC738 Time Server.
681  * Return a machine readable date and time, in the form of the
682  * number of seconds since midnight, Jan 1, 1900.  Since gettimeofday
683  * returns the number of seconds since midnight, Jan 1, 1970,
684  * we must add 2208988800 seconds to this figure to make up for
685  * some seventy years Bell Labs was asleep.
686  */
687
688 unsigned long
689 machtime(void)
690 {
691         struct timeval tv;
692
693         if (gettimeofday(&tv, (struct timezone *)NULL) < 0) {
694                 if (debug)
695                         warnx("unable to get time of day");
696                 return (0L);
697         }
698 #define OFFSET ((u_long)25567 * 24*60*60)
699         return (htonl((long)(tv.tv_sec + OFFSET)));
700 #undef OFFSET
701 }
702
703 /* ARGSUSED */
704 void
705 machtime_dg(int s, struct servtab *sep)
706 {
707         unsigned long result;
708         struct sockaddr_storage ss;
709         socklen_t size;
710
711         size = sizeof(ss);
712         if (recvfrom(s, (char *)&result, sizeof(result), 0,
713                      (struct sockaddr *)&ss, &size) < 0)
714                 return;
715
716         if (check_loop((struct sockaddr *)&ss, sep))
717                 return;
718
719         result = machtime();
720         (void) sendto(s, (char *) &result, sizeof(result), 0,
721                       (struct sockaddr *)&ss, size);
722 }
723
724 /* ARGSUSED */
725 void
726 machtime_stream(int s, struct servtab *sep __unused)
727 {
728         unsigned long result;
729
730         result = machtime();
731         (void) send(s, (char *) &result, sizeof(result), MSG_EOF);
732 }
733
734 /*
735  * RFC1078 TCP Port Service Multiplexer (TCPMUX). Service connections to
736  * services based on the service name sent.
737  *
738  *  Based on TCPMUX.C by Mark K. Lottor November 1988
739  *  sri-nic::ps:<mkl>tcpmux.c
740  */
741
742 #define MAX_SERV_LEN    (256+2)         /* 2 bytes for \r\n */
743 #define strwrite(fd, buf)       (void) write(fd, buf, sizeof(buf)-1)
744
745 static int              /* # of characters upto \r,\n or \0 */
746 getline(int fd, char *buf, int len)
747 {
748         int count = 0, n;
749         struct sigaction sa;
750
751         sa.sa_flags = 0;
752         sigemptyset(&sa.sa_mask);
753         sa.sa_handler = SIG_DFL;
754         sigaction(SIGALRM, &sa, (struct sigaction *)0);
755         do {
756                 alarm(10);
757                 n = read(fd, buf, len-count);
758                 alarm(0);
759                 if (n == 0)
760                         return (count);
761                 if (n < 0)
762                         return (-1);
763                 while (--n >= 0) {
764                         if (*buf == '\r' || *buf == '\n' || *buf == '\0')
765                                 return (count);
766                         count++;
767                         buf++;
768                 }
769         } while (count < len);
770         return (count);
771 }
772
773 struct servtab *
774 tcpmux(int s)
775 {
776         struct servtab *sep;
777         char service[MAX_SERV_LEN+1];
778         int len;
779
780         /* Get requested service name */
781         if ((len = getline(s, service, MAX_SERV_LEN)) < 0) {
782                 strwrite(s, "-Error reading service name\r\n");
783                 return (NULL);
784         }
785         service[len] = '\0';
786
787         if (debug)
788                 warnx("tcpmux: someone wants %s", service);
789
790         /*
791          * Help is a required command, and lists available services,
792          * one per line.
793          */
794         if (!strcasecmp(service, "help")) {
795                 for (sep = servtab; sep; sep = sep->se_next) {
796                         if (!ISMUX(sep))
797                                 continue;
798                         (void)write(s,sep->se_service,strlen(sep->se_service));
799                         strwrite(s, "\r\n");
800                 }
801                 return (NULL);
802         }
803
804         /* Try matching a service in inetd.conf with the request */
805         for (sep = servtab; sep; sep = sep->se_next) {
806                 if (!ISMUX(sep))
807                         continue;
808                 if (!strcasecmp(service, sep->se_service)) {
809                         if (ISMUXPLUS(sep)) {
810                                 strwrite(s, "+Go\r\n");
811                         }
812                         return (sep);
813                 }
814         }
815         strwrite(s, "-Service not available\r\n");
816         return (NULL);
817 }