1 /* 2 * CDDL HEADER START 3 * 4 * The contents of this file are subject to the terms of the 5 * Common Development and Distribution License, Version 1.0 only 6 * (the "License"). You may not use this file except in compliance 7 * with the License. 8 * 9 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 10 * or http://www.opensolaris.org/os/licensing. 11 * See the License for the specific language governing permissions 12 * and limitations under the License. 13 * 14 * When distributing Covered Code, include this CDDL HEADER in each 15 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 16 * If applicable, add the following below this CDDL HEADER, with the 17 * fields enclosed by brackets "[]" replaced with your own identifying 18 * information: Portions Copyright [yyyy] [name of copyright owner] 19 * 20 * CDDL HEADER END 21 */ 22 /* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */ 23 /* All Rights Reserved */ 24 25 26 27 /* 28 * Copyright 1988-2003 Sun Microsystems, Inc. All rights reserved. 29 * Use is subject to license terms. 30 */ 31 32 /* 33 * Copyright 2012 Joyent, Inc. All rights reserved. 34 * 35 * Copyright (c) 2013 Gary Mills 36 */ 37 38 #include <signal.h> 39 #include <stdio.h> 40 #include <stdlib.h> 41 #include <grp.h> 42 #include <sys/types.h> 43 #include <unistd.h> 44 #include <string.h> 45 #include <ctype.h> 46 #include <sys/stat.h> 47 #include <utmpx.h> 48 #include <sys/utsname.h> 49 #include <dirent.h> 50 #include <pwd.h> 51 #include <fcntl.h> 52 #include <time.h> 53 #include <errno.h> 54 #include <locale.h> 55 #include <syslog.h> 56 #include <sys/wait.h> 57 #include <limits.h> 58 #include <libzonecfg.h> 59 #include <zone.h> 60 #include <sys/contract/process.h> 61 #include <libcontract.h> 62 #include <sys/ctfs.h> 63 64 /* 65 * Use the full lengths from utmpx for user and line. 66 */ 67 #define NMAX (sizeof (((struct utmpx *)0)->ut_user)) 68 #define LMAX (sizeof (((struct utmpx *)0)->ut_line)) 69 70 static char mesg[3000]; 71 static char *infile; 72 static int gflag; 73 static struct group *pgrp; 74 static char *grpname; 75 static char line[MAXNAMLEN+1] = "???"; 76 static char systm[MAXNAMLEN+1]; 77 static time_t tloc; 78 static struct utsname utsn; 79 static char who[NMAX+1] = "???"; 80 static char time_buf[50]; 81 #define DATE_FMT "%a %b %e %H:%M:%S" 82 83 static void sendmes(struct utmpx *, zoneid_t); 84 static void sendmes_tozone(zoneid_t, int); 85 static int chkgrp(char *); 86 static char *copy_str_till(char *, char *, char, int); 87 88 static int init_template(void); 89 int contract_abandon_id(ctid_t); 90 91 int 92 main(int argc, char *argv[]) 93 { 94 FILE *f; 95 char *ptr, *start; 96 struct passwd *pwd; 97 char *term_name; 98 int c; 99 int aflag = 0; 100 int errflg = 0; 101 int zflg = 0; 102 int Zflg = 0; 103 104 char *zonename = NULL; 105 zoneid_t *zoneidlist = NULL; 106 uint_t nzids_saved, nzids = 0; 107 108 (void) setlocale(LC_ALL, ""); 109 110 while ((c = getopt(argc, argv, "g:az:Z")) != EOF) 111 switch (c) { 112 case 'a': 113 aflag++; 114 break; 115 case 'g': 116 if (gflag) { 117 (void) fprintf(stderr, 118 "Only one group allowed\n"); 119 return (1); 120 } 121 if ((pgrp = getgrnam(grpname = optarg)) == NULL) { 122 (void) fprintf(stderr, "Unknown group %s\n", 123 grpname); 124 return (1); 125 } 126 gflag++; 127 break; 128 case 'z': 129 zflg++; 130 zonename = optarg; 131 if (getzoneidbyname(zonename) == -1) { 132 (void) fprintf(stderr, "Specified zone %s " 133 "is invalid", zonename); 134 return (1); 135 } 136 break; 137 case 'Z': 138 Zflg++; 139 break; 140 case '?': 141 errflg++; 142 break; 143 } 144 145 if (errflg) { 146 (void) fprintf(stderr, 147 "Usage: wall [-a] [-g group] [-z zone] [-Z] [files...]\n"); 148 return (1); 149 } 150 151 if (zflg && Zflg) { 152 (void) fprintf(stderr, "Cannot use -z with -Z\n"); 153 return (1); 154 } 155 156 if (optind < argc) 157 infile = argv[optind]; 158 159 if (uname(&utsn) == -1) { 160 (void) fprintf(stderr, "wall: uname() failed, %s\n", 161 strerror(errno)); 162 return (2); 163 } 164 (void) strcpy(systm, utsn.nodename); 165 166 /* 167 * Get the name of the terminal wall is running from. 168 */ 169 170 if ((term_name = ttyname(fileno(stderr))) != NULL) { 171 /* 172 * skip the leading "/dev/" in term_name 173 */ 174 (void) strncpy(line, &term_name[5], sizeof (line) - 1); 175 } 176 177 if (who[0] == '?') { 178 if (pwd = getpwuid(getuid())) 179 (void) strncpy(&who[0], pwd->pw_name, sizeof (who)); 180 } 181 182 f = stdin; 183 if (infile) { 184 f = fopen(infile, "r"); 185 if (f == NULL) { 186 (void) fprintf(stderr, "Cannot open %s\n", infile); 187 return (1); 188 } 189 } 190 191 start = &mesg[0]; 192 ptr = start; 193 while ((ptr - start) < 3000) { 194 size_t n; 195 196 if (fgets(ptr, &mesg[sizeof (mesg)] - ptr, f) == NULL) 197 break; 198 if ((n = strlen(ptr)) == 0) 199 break; 200 ptr += n; 201 } 202 (void) fclose(f); 203 204 /* 205 * If the request is from the rwall daemon then use the caller's 206 * name and host. We determine this if all of the following is true: 207 * 1) First 5 characters are "From " 208 * 2) Next non-white characters are of the form "name@host:" 209 */ 210 if (strcmp(line, "???") == 0) { 211 char rwho[MAXNAMLEN+1]; 212 char rsystm[MAXNAMLEN+1]; 213 char *cp; 214 215 if (strncmp(mesg, "From ", 5) == 0) { 216 cp = &mesg[5]; 217 cp = copy_str_till(rwho, cp, '@', MAXNAMLEN + 1); 218 if (rwho[0] != '\0') { 219 cp = copy_str_till(rsystm, ++cp, ':', 220 MAXNAMLEN + 1); 221 if (rsystm[0] != '\0') { 222 (void) strcpy(systm, rsystm); 223 (void) strncpy(rwho, who, 224 sizeof (who)); 225 (void) strcpy(line, "rpc.rwalld"); 226 } 227 } 228 } 229 } 230 (void) time(&tloc); 231 (void) strftime(time_buf, sizeof (time_buf), 232 DATE_FMT, localtime(&tloc)); 233 234 if (zflg != 0) { 235 if ((zoneidlist = 236 malloc(sizeof (zoneid_t))) == NULL || 237 (*zoneidlist = getzoneidbyname(zonename)) == -1) 238 return (errno); 239 nzids = 1; 240 } else if (Zflg != 0) { 241 if (zone_list(NULL, &nzids) != 0) 242 return (errno); 243 again: 244 nzids *= 2; 245 if ((zoneidlist = malloc(nzids * sizeof (zoneid_t))) == NULL) 246 exit(errno); 247 nzids_saved = nzids; 248 if (zone_list(zoneidlist, &nzids) != 0) { 249 (void) free(zoneidlist); 250 return (errno); 251 } 252 if (nzids > nzids_saved) { 253 free(zoneidlist); 254 goto again; 255 } 256 } 257 if (zflg || Zflg) { 258 for (; nzids > 0; --nzids) 259 sendmes_tozone(zoneidlist[nzids-1], aflag); 260 free(zoneidlist); 261 } else 262 sendmes_tozone(getzoneid(), aflag); 263 264 return (0); 265 } 266 267 /* 268 * Copy src to destination upto but not including the delim. 269 * Leave dst empty if delim not found or whitespace encountered. 270 * Return pointer to next character (delim, whitespace, or '\0') 271 */ 272 static char * 273 copy_str_till(char *dst, char *src, char delim, int len) 274 { 275 int i = 0; 276 277 while (*src != '\0' && i < len) { 278 if (isspace(*src)) { 279 dst[0] = '\0'; 280 return (src); 281 } 282 if (*src == delim) { 283 dst[i] = '\0'; 284 return (src); 285 } 286 dst[i++] = *src++; 287 } 288 dst[0] = '\0'; 289 return (src); 290 } 291 292 static void 293 sendmes_tozone(zoneid_t zid, int aflag) { 294 int i = 0; 295 char zonename[ZONENAME_MAX], root[MAXPATHLEN]; 296 struct utmpx *p; 297 298 if (zid != getzoneid()) { 299 root[0] = '\0'; 300 (void) getzonenamebyid(zid, zonename, ZONENAME_MAX); 301 (void) zone_get_rootpath(zonename, root, sizeof (root)); 302 (void) strlcat(root, UTMPX_FILE, sizeof (root)); 303 if (!utmpxname(root)) { 304 (void) fprintf(stderr, "Cannot open %s\n", root); 305 return; 306 } 307 } else { 308 (void) utmpxname(UTMPX_FILE); 309 } 310 setutxent(); 311 while ((p = getutxent()) != NULL) { 312 if (p->ut_type != USER_PROCESS) 313 continue; 314 /* 315 * if (-a option OR NOT pty window login), send the message 316 */ 317 if (aflag || !nonuserx(*p)) 318 sendmes(p, zid); 319 } 320 endutxent(); 321 322 (void) alarm(60); 323 do { 324 i = (int)wait((int *)0); 325 } while (i != -1 || errno != ECHILD); 326 327 } 328 329 /* 330 * Note to future maintainers: with the change of wall to use the 331 * getutxent() API, the forked children (created by this function) 332 * must call _exit as opposed to exit. This is necessary to avoid 333 * unwanted fflushing of getutxent's stdio stream (caused by atexit 334 * processing). 335 */ 336 static void 337 sendmes(struct utmpx *p, zoneid_t zid) 338 { 339 int i; 340 char *s; 341 static char device[LMAX + 6]; 342 char *bp; 343 int ibp; 344 FILE *f; 345 int fd, tmpl_fd; 346 boolean_t zoneenter = B_FALSE; 347 348 if (zid != getzoneid()) { 349 zoneenter = B_TRUE; 350 tmpl_fd = init_template(); 351 if (tmpl_fd == -1) { 352 (void) fprintf(stderr, "Could not initialize " 353 "process contract"); 354 return; 355 } 356 } 357 358 while ((i = (int)fork()) == -1) { 359 (void) alarm(60); 360 (void) wait((int *)0); 361 (void) alarm(0); 362 } 363 364 if (i) 365 return; 366 367 if (zoneenter && zone_enter(zid) == -1) { 368 char zonename[ZONENAME_MAX]; 369 (void) getzonenamebyid(zid, zonename, ZONENAME_MAX); 370 (void) fprintf(stderr, "Could not enter zone " 371 "%s\n", zonename); 372 } 373 if (zoneenter) 374 (void) ct_tmpl_clear(tmpl_fd); 375 376 if (gflag) 377 if (!chkgrp(p->ut_user)) 378 _exit(0); 379 380 (void) signal(SIGHUP, SIG_IGN); 381 (void) alarm(60); 382 s = &device[0]; 383 (void) snprintf(s, sizeof (device), "/dev/%.*s", LMAX, p->ut_line); 384 385 /* check if the device is really a tty */ 386 if ((fd = open(s, O_WRONLY|O_NOCTTY|O_NONBLOCK)) == -1) { 387 (void) fprintf(stderr, "Cannot send to %.*s on %s\n", 388 NMAX, p->ut_user, s); 389 perror("open"); 390 (void) fflush(stderr); 391 _exit(1); 392 } else { 393 if (!isatty(fd)) { 394 (void) fprintf(stderr, 395 "Cannot send to device %.*s %s\n", 396 LMAX, p->ut_line, 397 "because it's not a tty"); 398 openlog("wall", 0, LOG_AUTH); 399 syslog(LOG_CRIT, "%.*s in utmpx is not a tty\n", 400 LMAX, p->ut_line); 401 closelog(); 402 (void) fflush(stderr); 403 _exit(1); 404 } 405 } 406 #ifdef DEBUG 407 (void) close(fd); 408 f = fopen("wall.debug", "a"); 409 #else 410 f = fdopen(fd, "w"); 411 #endif 412 if (f == NULL) { 413 (void) fprintf(stderr, "Cannot send to %-.*s on %s\n", 414 NMAX, &p->ut_user[0], s); 415 perror("open"); 416 (void) fflush(stderr); 417 _exit(1); 418 } 419 (void) fprintf(f, 420 "\07\07\07Broadcast Message from %s (%s) on %s %19.19s", 421 who, line, systm, time_buf); 422 if (gflag) 423 (void) fprintf(f, " to group %s", grpname); 424 (void) fprintf(f, "...\n"); 425 #ifdef DEBUG 426 (void) fprintf(f, "DEBUG: To %.*s on %s\n", NMAX, p->ut_user, s); 427 #endif 428 i = strlen(mesg); 429 for (bp = mesg; --i >= 0; bp++) { 430 ibp = (unsigned int)((unsigned char) *bp); 431 if (*bp == '\n') 432 (void) putc('\r', f); 433 if (isprint(ibp) || *bp == '\r' || *bp == '\013' || 434 *bp == ' ' || *bp == '\t' || *bp == '\n' || *bp == '\007') { 435 (void) putc(*bp, f); 436 } else { 437 if (!isascii(*bp)) { 438 (void) fputs("M-", f); 439 *bp = toascii(*bp); 440 } 441 if (iscntrl(*bp)) { 442 (void) putc('^', f); 443 (void) putc(*bp + 0100, f); 444 } 445 else 446 (void) putc(*bp, f); 447 } 448 449 if (*bp == '\n') 450 (void) fflush(f); 451 452 if (ferror(f) || feof(f)) { 453 (void) printf("\n\007Write failed\n"); 454 (void) fflush(stdout); 455 _exit(1); 456 } 457 } 458 (void) fclose(f); 459 (void) close(fd); 460 _exit(0); 461 } 462 463 464 static int 465 chkgrp(char *name) 466 { 467 int i; 468 char user[NMAX + 1]; 469 470 (void) strlcpy(user, name, sizeof (user)); 471 for (i = 0; pgrp->gr_mem[i] && pgrp->gr_mem[i][0]; i++) { 472 if (strcmp(user, pgrp->gr_mem[i]) == 0) 473 return (1); 474 } 475 476 return (0); 477 } 478 479 static int 480 init_template(void) { 481 int fd = 0; 482 int err = 0; 483 484 fd = open64(CTFS_ROOT "/process/template", O_RDWR); 485 if (fd == -1) 486 return (-1); 487 488 err |= ct_tmpl_set_critical(fd, 0); 489 err |= ct_tmpl_set_informative(fd, 0); 490 err |= ct_pr_tmpl_set_fatal(fd, CT_PR_EV_HWERR); 491 err |= ct_pr_tmpl_set_param(fd, CT_PR_PGRPONLY | CT_PR_REGENT); 492 if (err || ct_tmpl_activate(fd)) { 493 (void) close(fd); 494 return (-1); 495 } 496 497 return (fd); 498 } 499