term-utils/wall.c

   1 /*
   2  * Copyright (c) 1988, 1990, 1993
   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  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  *
  33  * Modified Sun Mar 12 10:34:34 1995, faith@cs.unc.edu, for Linux
  34  */
  35
  36 /*
  37  * This program is not related to David Wall, whose Stanford Ph.D. thesis
  38  * is entitled "Mechanisms for Broadcast and Selective Broadcast".
  39  *
  40  * 1999-02-22 Arkadiusz Miśkiewicz <misiek@pld.ORG.PL>
  41  * - added Native Language Support
  42  *
  43  */
  44
  45 #include <sys/param.h>
  46 #include <sys/stat.h>
  47 #include <sys/time.h>
  48 #include <sys/uio.h>
  49
  50 #include <errno.h>
  51 #include <paths.h>
  52 #include <ctype.h>
  53 #include <pwd.h>
  54 #include <stdio.h>
  55 #include <stdlib.h>
  56 #include <string.h>
  57 #include <time.h>
  58 #include <unistd.h>
  59 #include <utmp.h>
  60 #include <getopt.h>
  61
  62 #include "nls.h"
  63 #include "xalloc.h"
  64 #include "strutils.h"
  65 #include "ttymsg.h"
  66 #include "pathnames.h"
  67 #include "carefulputc.h"
  68 #include "c.h"
  69 #include "fileutils.h"
  70 #include "closestream.h"
  71
  72 #define IGNOREUSER      "sleeper"
  73 #define WRITE_TIME_OUT  300             /* in seconds */
  74
  75 /* Function prototypes */
  76 static char *makemsg(char *fname, char **mvec, int mvecsz,
  77                     size_t *mbufsize, int print_banner);
  78
  79 static void __attribute__((__noreturn__)) usage(FILE *out)
  80 {
  81         fputs(_("\nUsage:\n"), out);
  82         fprintf(out,
  83                 _(" %s [options] [<file> | <message>]\n"),
  84                 program_invocation_short_name);
  85
  86         fputs(_("\nOptions:\n"), out);
  87         fputs(_(" -n, --nobanner          do not print banner, works only for root\n"
  88                 " -t, --timeout <timeout> write timeout in seconds\n"
  89                 " -V, --version           output version information and exit\n"
  90                 " -h, --help              display this help and exit\n\n"), out);
  91
  92         exit(out == stderr ? EXIT_FAILURE : EXIT_SUCCESS);
  93 }
  94
  95 int
  96 main(int argc, char **argv) {
  97         int ch;
  98         struct iovec iov;
  99         struct utmp *utmpptr;
 100         char *p;
 101         char line[sizeof(utmpptr->ut_line) + 1];
 102         int print_banner = TRUE;
 103         char *mbuf, *fname = NULL;
 104         size_t mbufsize;
 105         unsigned timeout = WRITE_TIME_OUT;
 106         char **mvec = NULL;
 107         int mvecsz = 0;
 108
 109         static const struct option longopts[] = {
 110                 { "nobanner",   no_argument,            0, 'n' },
 111                 { "timeout",    required_argument,      0, 't' },
 112                 { "version",    no_argument,            0, 'V' },
 113                 { "help",       no_argument,            0, 'h' },
 114                 { NULL, 0, 0, 0 }
 115         };
 116
 117         setlocale(LC_ALL, "");
 118         bindtextdomain(PACKAGE, LOCALEDIR);
 119         textdomain(PACKAGE);
 120         atexit(close_stdout);
 121
 122         while ((ch = getopt_long(argc, argv, "nt:Vh", longopts, NULL)) != -1) {
 123                 switch (ch) {
 124                 case 'n':
 125                         if (geteuid() == 0)
 126                                 print_banner = FALSE;
 127                         else
 128                                 warnx(_("--nobanner is available only for root"));
 129                         break;
 130                 case 't':
 131                         timeout = strtou32_or_err(optarg, _("invalid timeout argument"));
 132                         if (timeout < 1)
 133                                 errx(EXIT_FAILURE, _("invalid timeout argument: %s"), optarg);
 134                         break;
 135                 case 'V':
 136                         printf(_("%s from %s\n"), program_invocation_short_name,
 137                                                   PACKAGE_STRING);
 138                         exit(EXIT_SUCCESS);
 139                 case 'h':
 140                         usage(stdout);
 141                 default:
 142                         usage(stderr);
 143                 }
 144         }
 145         argc -= optind;
 146         argv += optind;
 147
 148         if (argc == 1 && access(argv[0], F_OK) == 0)
 149                 fname = argv[0];
 150         else if (argc >= 1) {
 151                 mvec = argv;
 152                 mvecsz = argc;
 153         }
 154
 155         mbuf = makemsg(fname, mvec, mvecsz, &mbufsize, print_banner);
 156
 157         iov.iov_base = mbuf;
 158         iov.iov_len = mbufsize;
 159         while((utmpptr = getutent())) {
 160                 if (!utmpptr->ut_name[0] ||
 161                     !strncmp(utmpptr->ut_name, IGNOREUSER,
 162                              sizeof(utmpptr->ut_name)))
 163                         continue;
 164 #ifdef USER_PROCESS
 165                 if (utmpptr->ut_type != USER_PROCESS)
 166                         continue;
 167 #endif
 168
 169                 /* Joey Hess reports that use-sessreg in /etc/X11/wdm/
 170                    produces ut_line entries like :0, and a write
 171                    to /dev/:0 fails. */
 172                 if (utmpptr->ut_line[0] == ':')
 173                         continue;
 174
 175                 xstrncpy(line, utmpptr->ut_line, sizeof(utmpptr->ut_line));
 176                 if ((p = ttymsg(&iov, 1, line, timeout)) != NULL)
 177                         warnx("%s", p);
 178         }
 179         endutent();
 180         free(mbuf);
 181         exit(EXIT_SUCCESS);
 182 }
 183
 184 static char *makemsg(char *fname, char **mvec, int mvecsz, size_t *mbufsize, int print_banner)
 185 {
 186         register int ch, cnt;
 187         struct tm *lt;
 188         struct passwd *pw;
 189         struct stat sbuf;
 190         time_t now;
 191         FILE *fp;
 192         char *p, *whom, *where, *lbuf, *tmpname, *mbuf;
 193         long line_max;
 194
 195         line_max = sysconf(_SC_LINE_MAX);
 196         lbuf = xmalloc(line_max);
 197
 198         if ((fp = xfmkstemp(&tmpname, NULL)) == NULL)
 199                 err(EXIT_FAILURE, _("can't open temporary file"));
 200         unlink(tmpname);
 201         free(tmpname);
 202
 203         if (print_banner == TRUE) {
 204                 char *hostname = xgethostname();
 205                 if (!(whom = getlogin()) || !*whom)
 206                         whom = (pw = getpwuid(getuid())) ? pw->pw_name : "???";
 207                 if (!whom) {
 208                         whom = "someone";
 209                         warn(_("cannot get passwd uid"));
 210                 }
 211                 where = ttyname(STDOUT_FILENO);
 212                 if (!where) {
 213                         where = "somewhere";
 214                         warn(_("cannot get tty name"));
 215                 }
 216                 time(&now);
 217                 lt = localtime(&now);
 218
 219                 /*
 220                  * all this stuff is to blank out a square for the message;
 221                  * we wrap message lines at column 79, not 80, because some
 222                  * terminals wrap after 79, some do not, and we can't tell.
 223                  * Which means that we may leave a non-blank character
 224                  * in column 80, but that can't be helped.
 225                  */
 226                 /* snprintf is not always available, but the sprintf's here
 227                    will not overflow as long as %d takes at most 100 chars */
 228                 fprintf(fp, "\r%79s\r\n", " ");
 229                 sprintf(lbuf, _("Broadcast Message from %s@%s"),
 230                               whom, hostname);
 231                 free(hostname);
 232                 fprintf(fp, "%-79.79s\007\007\r\n", lbuf);
 233                 sprintf(lbuf, "        (%s) at %d:%02d ...",
 234                               where, lt->tm_hour, lt->tm_min);
 235                 fprintf(fp, "%-79.79s\r\n", lbuf);
 236         }
 237         fprintf(fp, "%79s\r\n", " ");
 238
 239          if (mvec) {
 240                 /*
 241                  * Read message from argv[]
 242                  */
 243                 int i;
 244
 245                 for (i = 0; i < mvecsz; i++) {
 246                         fputs(mvec[i], fp);
 247                         if (i < mvecsz - 1)
 248                                 fputc(' ', fp);
 249                 }
 250                 fputc('\r', fp);
 251                 fputc('\n', fp);
 252
 253         } else {
 254                 /*
 255                  * read message from <file>
 256                  */
 257                 if (fname) {
 258                         /*
 259                          * When we are not root, but suid or sgid, refuse to read files
 260                          * (e.g. device files) that the user may not have access to.
 261                          * After all, our invoker can easily do "wall < file"
 262                          * instead of "wall file".
 263                          */
 264                         uid_t uid = getuid();
 265                         if (uid && (uid != geteuid() || getgid() != getegid()))
 266                                 errx(EXIT_FAILURE, _("will not read %s - use stdin."),
 267                                      fname);
 268
 269                         if (!freopen(fname, "r", stdin))
 270                                 err(EXIT_FAILURE, _("cannot open %s"), fname);
 271
 272                 }
 273
 274                 /*
 275                  * Read message from stdin.
 276                  */
 277                 while (fgets(lbuf, line_max, stdin)) {
 278                         for (cnt = 0, p = lbuf; (ch = *p) != '\0'; ++p, ++cnt) {
 279                                 if (cnt == 79 || ch == '\n') {
 280                                         for (; cnt < 79; ++cnt)
 281                                                 putc(' ', fp);
 282                                         putc('\r', fp);
 283                                         putc('\n', fp);
 284                                         cnt = 0;
 285                                 }
 286                                 if (ch != '\n')
 287                                         carefulputc(ch, fp);
 288                         }
 289                 }
 290         }
 291         fprintf(fp, "%79s\r\n", " ");
 292
 293         free(lbuf);
 294         rewind(fp);
 295
 296         if (fstat(fileno(fp), &sbuf))
 297                 err(EXIT_FAILURE, _("stat failed"));
 298
 299         *mbufsize = (size_t) sbuf.st_size;
 300         mbuf = xmalloc(*mbufsize);
 301
 302         if (fread(mbuf, 1, *mbufsize, fp) != *mbufsize)
 303                 err(EXIT_FAILURE, _("fread failed"));
 304
 305         if (close_stream(fp) != 0)
 306                 errx(EXIT_FAILURE, _("write error"));
 307         return mbuf;
 308 }