xref: /titanic_44/usr/src/cmd/zoneadmd/zcons.c (revision 55434c770c89aa1b84474f2559a106803511aba0)
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 (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  */
26 
27 #pragma ident	"%Z%%M%	%I%	%E% SMI"
28 
29 /*
30  * Console support for zones requires a significant infrastructure.  The
31  * core pieces are contained in this file, but other portions of note
32  * are in the zlogin(1M) command, the zcons(7D) driver, and in the
33  * devfsadm(1M) misc_link generator.
34  *
35  * Care is taken to make the console behave in an "intuitive" fashion for
36  * administrators.  Essentially, we try as much as possible to mimic the
37  * experience of using a system via a tip line and system controller.
38  *
39  * The zone console architecture looks like this:
40  *
41  *                                      Global Zone | Non-Global Zone
42  *                        .--------------.          |
43  *        .-----------.   | zoneadmd -z  |          | .--------. .---------.
44  *        | zlogin -C |   |     myzone   |          | | ttymon | | syslogd |
45  *        `-----------'   `--------------'          | `--------' `---------'
46  *                  |       |       | |             |      |       |
47  *  User            |       |       | |             |      V       V
48  * - - - - - - - - -|- - - -|- - - -|-|- - - - - - -|- - /dev/zconsole - - -
49  *  Kernel          V       V       | |                        |
50  *               [AF_UNIX Socket]   | `--------. .-------------'
51  *                                  |          | |
52  *                                  |          V V
53  *                                  |     +-----------+
54  *                                  |     |  ldterm,  |
55  *                                  |     |   etc.    |
56  *                                  |     +-----------+
57  *                                  |     +-[Anchor]--+
58  *                                  |     |   ptem    |
59  *                                  V     +-----------+
60  *                           +---master---+---slave---+
61  *                           |                        |
62  *                           |      zcons driver      |
63  *                           |    zonename="myzone"   |
64  *                           +------------------------+
65  *
66  * There are basically three major tasks which the console subsystem in
67  * zoneadmd accomplishes:
68  *
69  * - Setup and teardown of zcons driver instances.  One zcons instance
70  *   is maintained per zone; we take advantage of the libdevice APIs
71  *   to online new instances of zcons as needed.  Care is taken to
72  *   prune and manage these appropriately; see init_console_dev() and
73  *   destroy_console_dev().  The end result is the creation of the
74  *   zcons(7D) instance and an open file descriptor to the master side.
75  *   zcons instances are associated with zones via their zonename device
76  *   property.  This the console instance to persist across reboots,
77  *   and while the zone is halted.
78  *
79  * - Initialization of the slave side of the console.  This is
80  *   accomplished by pushing various STREAMS modules onto the console.
81  *   The ptem(7M) module gets special treatment, and is anchored into
82  *   place using the I_ANCHOR facility.  This is so that the zcons driver
83  *   always has terminal semantics, as would a real hardware terminal.
84  *   This means that ttymon(1M) works unmodified;  at boot time, ttymon
85  *   will do its own plumbing of the console stream, and will even
86  *   I_POP modules off.  Hence the anchor, which assures that ptem will
87  *   never be I_POP'd.
88  *
89  * - Acting as a server for 'zlogin -C' instances.  When zlogin -C is
90  *   run, zlogin connects to zoneadmd via unix domain socket.  zoneadmd
91  *   functions as a two-way proxy for console I/O, relaying user input
92  *   to the master side of the console, and relaying output from the
93  *   zone to the user.
94  */
95 
96 #include <sys/types.h>
97 #include <sys/socket.h>
98 #include <sys/stat.h>
99 #include <sys/termios.h>
100 #include <sys/zcons.h>
101 #include <sys/mkdev.h>
102 
103 #include <assert.h>
104 #include <ctype.h>
105 #include <errno.h>
106 #include <fcntl.h>
107 #include <stdarg.h>
108 #include <stdio.h>
109 #include <stdlib.h>
110 #include <strings.h>
111 #include <stropts.h>
112 #include <thread.h>
113 #include <ucred.h>
114 #include <unistd.h>
115 #include <zone.h>
116 
117 #include <libdevinfo.h>
118 #include <libdevice.h>
119 #include <libzonecfg.h>
120 
121 #include <syslog.h>
122 #include <sys/modctl.h>
123 #include <sys/fs/sdev_node.h>
124 
125 #include "zoneadmd.h"
126 
127 #define	ZCONSNEX_DEVTREEPATH	"/pseudo/zconsnex@1"
128 #define	ZCONSNEX_FILEPATH	"/devices/pseudo/zconsnex@1"
129 
130 #define	CONSOLE_SOCKPATH	ZONES_TMPDIR "/%s.console_sock"
131 
132 static int	slavefd = -1;	/* slave side of console */
133 static int	serverfd = -1;	/* console server unix domain socket fd */
134 char boot_args[BOOTARGS_MAX];
135 char bad_boot_arg[BOOTARGS_MAX];
136 
137 static struct termios base_termios = {	/* from init.c */
138 	BRKINT|ICRNL|IXON|IMAXBEL,			/* iflag */
139 	OPOST|ONLCR|TAB3,				/* oflag */
140 	CS8|CREAD|B9600,				/* cflag */
141 	ISIG|ICANON|ECHO|ECHOE|ECHOK|ECHOCTL|ECHOKE|IEXTEN,	/* lflag */
142 	CINTR, CQUIT, CERASE, CKILL, CEOF, 0, 0, 0,
143 	0, 0, 0, 0, 0, 0, 0, 0,
144 	0, 0, 0
145 };
146 
147 /*
148  * The eventstream is a simple one-directional flow of messages from the
149  * door server to the console subsystem, implemented with a pipe.
150  * It is used to wake up the console poller when it needs to take action,
151  * message the user, die off, etc.
152  */
153 static int eventstream[2];
154 
155 
156 
157 int
158 eventstream_init()
159 {
160 	if (pipe(eventstream) == -1)
161 		return (-1);
162 	return (0);
163 }
164 
165 void
166 eventstream_write(zone_evt_t evt)
167 {
168 	(void) write(eventstream[0], &evt, sizeof (evt));
169 }
170 
171 static zone_evt_t
172 eventstream_read(void)
173 {
174 	zone_evt_t evt = Z_EVT_NULL;
175 
176 	(void) read(eventstream[1], &evt, sizeof (evt));
177 	return (evt);
178 }
179 
180 /*
181  * count_console_devs() and its helper count_cb() do a walk of the
182  * subtree of the device tree where zone console nodes are represented.
183  * The goal is to count zone console instances already setup for a zone
184  * with the given name.  More than 1 is anomolous, and our caller will
185  * have to deal with that if we find that's the case.
186  *
187  * Note: this algorithm is a linear search of nodes in the zconsnex subtree
188  * of the device tree, and could be a scalability problem, but I don't see
189  * how to avoid it.
190  */
191 
192 /*
193  * cb_data is shared by count_cb and destroy_cb for simplicity.
194  */
195 struct cb_data {
196 	zlog_t *zlogp;
197 	int found;
198 	int killed;
199 };
200 
201 static int
202 count_cb(di_node_t node, void *arg)
203 {
204 	struct cb_data *cb = (struct cb_data *)arg;
205 	char *prop_data;
206 
207 	if (di_prop_lookup_strings(DDI_DEV_T_ANY, node, "zonename",
208 	    &prop_data) != -1) {
209 		assert(prop_data != NULL);
210 		if (strcmp(prop_data, zone_name) == 0) {
211 			cb->found++;
212 			return (DI_WALK_CONTINUE);
213 		}
214 	}
215 	return (DI_WALK_CONTINUE);
216 }
217 
218 static int
219 count_console_devs(zlog_t *zlogp)
220 {
221 	di_node_t root;
222 	struct cb_data cb;
223 
224 	bzero(&cb, sizeof (cb));
225 	cb.zlogp = zlogp;
226 
227 	if ((root = di_init(ZCONSNEX_DEVTREEPATH, DINFOCPYALL)) ==
228 	    DI_NODE_NIL) {
229 		zerror(zlogp, B_TRUE, "%s failed", "di_init");
230 		return (-1);
231 	}
232 
233 	(void) di_walk_node(root, DI_WALK_CLDFIRST, (void *)&cb, count_cb);
234 	di_fini(root);
235 	return (cb.found);
236 }
237 
238 /*
239  * destroy_console_devs() and its helper destroy_cb() tears down any console
240  * instances associated with this zone.  If things went very wrong, we
241  * might have more than one console instance hanging around.  This routine
242  * hunts down and tries to remove all of them.  Of course, if the console
243  * is open, the instance will not detach, which is a potential issue.
244  */
245 static int
246 destroy_cb(di_node_t node, void *arg)
247 {
248 	struct cb_data *cb = (struct cb_data *)arg;
249 	char *prop_data;
250 	char *tmp;
251 	char devpath[MAXPATHLEN];
252 	devctl_hdl_t hdl;
253 
254 	if (di_prop_lookup_strings(DDI_DEV_T_ANY, node, "zonename",
255 	    &prop_data) == -1)
256 		return (DI_WALK_CONTINUE);
257 
258 	assert(prop_data != NULL);
259 	if (strcmp(prop_data, zone_name) != 0) {
260 		/* this is the console for a different zone */
261 		return (DI_WALK_CONTINUE);
262 	}
263 
264 	cb->found++;
265 	tmp = di_devfs_path(node);
266 	(void) snprintf(devpath, sizeof (devpath), "/devices/%s", tmp);
267 	di_devfs_path_free(tmp);
268 
269 	if ((hdl = devctl_device_acquire(devpath, 0)) == NULL) {
270 		zerror(cb->zlogp, B_TRUE, "WARNING: console %s found, "
271 		    "but it could not be controlled.", devpath);
272 		return (DI_WALK_CONTINUE);
273 	}
274 	if (devctl_device_remove(hdl) == 0) {
275 		cb->killed++;
276 	} else {
277 		zerror(cb->zlogp, B_TRUE, "WARNING: console %s found, "
278 		    "but it could not be removed.", devpath);
279 	}
280 	devctl_release(hdl);
281 	return (DI_WALK_CONTINUE);
282 }
283 
284 static int
285 destroy_console_devs(zlog_t *zlogp)
286 {
287 	di_node_t root;
288 	struct cb_data cb;
289 
290 	bzero(&cb, sizeof (cb));
291 	cb.zlogp = zlogp;
292 
293 	if ((root = di_init(ZCONSNEX_DEVTREEPATH, DINFOCPYALL)) ==
294 	    DI_NODE_NIL) {
295 		zerror(zlogp, B_TRUE, "%s failed", "di_init");
296 		return (-1);
297 	}
298 
299 	(void) di_walk_node(root, DI_WALK_CLDFIRST, (void *)&cb, destroy_cb);
300 	if (cb.found > 1) {
301 		zerror(zlogp, B_FALSE, "WARNING: multiple zone console "
302 		    "instances detected for zone '%s'; %d of %d "
303 		    "successfully removed.",
304 		    zone_name, cb.killed, cb.found);
305 	}
306 
307 	di_fini(root);
308 	return (0);
309 }
310 
311 /*
312  * init_console_dev() drives the device-tree configuration of the zone
313  * console device.  The general strategy is to use the libdevice (devctl)
314  * interfaces to instantiate a new zone console node.  We do a lot of
315  * sanity checking, and are careful to reuse a console if one exists.
316  *
317  * Once the device is in the device tree, we kick devfsadm via di_init_devs()
318  * to ensure that the appropriate symlinks (to the master and slave console
319  * devices) are placed in /dev in the global zone.
320  */
321 static int
322 init_console_dev(zlog_t *zlogp)
323 {
324 	devctl_hdl_t bus_hdl = NULL, dev_hdl = NULL;
325 	devctl_ddef_t ddef_hdl = NULL;
326 	di_devlink_handle_t dl = NULL;
327 	int rv = -1, ndevs;
328 
329 	/*
330 	 * Don't re-setup console if it is working and ready already; just
331 	 * skip ahead to making devlinks, which we do for sanity's sake.
332 	 */
333 	ndevs = count_console_devs(zlogp);
334 	if (ndevs == 1) {
335 		goto devlinks;
336 	} else if (ndevs > 1 || ndevs == -1) {
337 		/*
338 		 * For now, this seems like a reasonable but harsh punishment.
339 		 * If needed, we could try to get clever and delete all but
340 		 * the console which is pointed at by the current symlink.
341 		 */
342 		if (destroy_console_devs(zlogp) == -1) {
343 			goto error;
344 		}
345 	}
346 
347 	/*
348 	 * Time to make the consoles!
349 	 */
350 	if ((bus_hdl = devctl_bus_acquire(ZCONSNEX_FILEPATH, 0)) == NULL) {
351 		zerror(zlogp, B_TRUE, "%s failed", "devctl_bus_acquire");
352 		goto error;
353 	}
354 	if ((ddef_hdl = devctl_ddef_alloc("zcons", 0)) == NULL) {
355 		zerror(zlogp, B_TRUE, "failed to allocate ddef handle");
356 		goto error;
357 	}
358 	/*
359 	 * Set three properties on this node; the first is the name of the
360 	 * zone; the second is a flag which lets pseudo know that it is
361 	 * OK to automatically allocate an instance # for this device;
362 	 * the third tells the device framework not to auto-detach this
363 	 * node-- we need the node to still be there when we ask devfsadmd
364 	 * to make links, and when we need to open it.
365 	 */
366 	if (devctl_ddef_string(ddef_hdl, "zonename", zone_name) == -1) {
367 		zerror(zlogp, B_TRUE, "failed to create zonename property");
368 		goto error;
369 	}
370 	if (devctl_ddef_int(ddef_hdl, "auto-assign-instance", 1) == -1) {
371 		zerror(zlogp, B_TRUE, "failed to create auto-assign-instance "
372 		    "property");
373 		goto error;
374 	}
375 	if (devctl_ddef_int(ddef_hdl, "ddi-no-autodetach", 1) == -1) {
376 		zerror(zlogp, B_TRUE, "failed to create ddi-no-auto-detach "
377 		    "property");
378 		goto error;
379 	}
380 	if (devctl_bus_dev_create(bus_hdl, ddef_hdl, 0, &dev_hdl) == -1) {
381 		zerror(zlogp, B_TRUE, "failed to create console node");
382 		goto error;
383 	}
384 
385 devlinks:
386 	if ((dl = di_devlink_init("zcons", DI_MAKE_LINK)) != NULL) {
387 		(void) di_devlink_fini(&dl);
388 	} else {
389 		zerror(zlogp, B_TRUE, "failed to create devlinks");
390 		goto error;
391 	}
392 
393 	rv = 0;
394 error:
395 	if (ddef_hdl)
396 		devctl_ddef_free(ddef_hdl);
397 	if (bus_hdl)
398 		devctl_release(bus_hdl);
399 	if (dev_hdl)
400 		devctl_release(dev_hdl);
401 	return (rv);
402 }
403 
404 /*
405  * init_console_slave() sets up the console slave device; the device node
406  * itself has already been set up in the device tree; the primary job
407  * here is to do some STREAMS plumbing.
408  *
409  * The slave side of the console is opened and the appropriate STREAMS
410  * modules are pushed on.  A wrinkle is that 'ptem' must be anchored
411  * in place (see streamio(7i) since we always want the console to
412  * have terminal semantics.)
413  */
414 int
415 init_console_slave(zlog_t *zlogp)
416 {
417 	char zconspath[MAXPATHLEN];
418 
419 	if (slavefd != -1)
420 		return (0);
421 
422 	(void) snprintf(zconspath, sizeof (zconspath),
423 	    "/dev/zcons/%s/%s", zone_name, ZCONS_SLAVE_NAME);
424 
425 	if ((slavefd = open(zconspath, O_RDWR | O_NOCTTY)) < 0) {
426 		zerror(zlogp, B_TRUE, "failed to open %s", zconspath);
427 		goto error;
428 	}
429 
430 	/*
431 	 * Just to make sure the whole stream is pristine.
432 	 */
433 	(void) ioctl(slavefd, I_FLUSH, FLUSHRW);
434 
435 	/*
436 	 * Push hardware emulation (ptem) module; we would rather not, but
437 	 * are forced to push ldterm and ttcompat here.  Ultimately ttymon
438 	 * will pop them off, so we ANCHOR only ptem.
439 	 *
440 	 * We need to use __I_PUSH_NOCTTY instead of I_PUSH here, otherwise
441 	 * we'll end up having the slave device as *our* controling terminal.
442 	 */
443 	if (ioctl(slavefd, __I_PUSH_NOCTTY, "ptem") == -1) {
444 		zerror(zlogp, B_TRUE, "failed to push ptem module");
445 		goto error;
446 	}
447 
448 	/*
449 	 * Anchor the stream to prevent malicious or accidental I_POP of ptem.
450 	 */
451 	if (ioctl(slavefd, I_ANCHOR) == -1) {
452 		zerror(zlogp, B_TRUE, "failed to set stream anchor");
453 		goto error;
454 	}
455 
456 	if (ioctl(slavefd, __I_PUSH_NOCTTY, "ldterm") == -1) {
457 		zerror(zlogp, B_TRUE, "failed to push ldterm module");
458 		goto error;
459 	}
460 	if (ioctl(slavefd, __I_PUSH_NOCTTY, "ttcompat") == -1) {
461 		zerror(zlogp, B_TRUE, "failed to push ttcompat module");
462 		goto error;
463 	}
464 
465 	/*
466 	 * Setup default terminal settings
467 	 */
468 	if (tcsetattr(slavefd, TCSAFLUSH, &base_termios) == -1) {
469 		zerror(zlogp, B_TRUE, "failed to set base terminal settings");
470 		goto error;
471 	}
472 
473 	return (0);
474 
475 error:
476 	if (slavefd != -1)
477 		(void) close(slavefd);
478 	slavefd = -1;
479 	zerror(zlogp, B_FALSE, "could not initialize console slave");
480 	return (-1);
481 }
482 
483 void
484 destroy_console_slave(void)
485 {
486 	(void) close(slavefd);
487 	slavefd = -1;
488 }
489 
490 /*
491  * Restore initial terminal attributes to the zone console.
492  */
493 void
494 reset_slave_terminal(zlog_t *zlogp)
495 {
496 	assert(slavefd != -1);
497 
498 	/* Probably not fatal, so we drive on if it fails */
499 	if (tcsetattr(slavefd, TCSAFLUSH, &base_termios) == -1) {
500 		zerror(zlogp, B_TRUE, "WARNING: failed to set terminal "
501 		    "settings.");
502 	}
503 }
504 
505 static int
506 init_console_sock(zlog_t *zlogp)
507 {
508 	int servfd;
509 	struct sockaddr_un servaddr;
510 
511 	bzero(&servaddr, sizeof (servaddr));
512 	servaddr.sun_family = AF_UNIX;
513 	(void) snprintf(servaddr.sun_path, sizeof (servaddr.sun_path),
514 	    CONSOLE_SOCKPATH, zone_name);
515 
516 	if ((servfd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
517 		zerror(zlogp, B_TRUE, "console setup: could not create socket");
518 		return (-1);
519 	}
520 	(void) unlink(servaddr.sun_path);
521 
522 	if (bind(servfd, (struct sockaddr *)&servaddr,
523 	    sizeof (servaddr)) == -1) {
524 		zerror(zlogp, B_TRUE,
525 		    "console setup: could not bind to socket");
526 		goto out;
527 	}
528 
529 	if (listen(servfd, 4) == -1) {
530 		zerror(zlogp, B_TRUE,
531 		    "console setup: could not listen on socket");
532 		goto out;
533 	}
534 	return (servfd);
535 
536 out:
537 	(void) unlink(servaddr.sun_path);
538 	(void) close(servfd);
539 	return (-1);
540 }
541 
542 static void
543 destroy_console_sock(int servfd)
544 {
545 	char path[MAXPATHLEN];
546 
547 	(void) snprintf(path, sizeof (path), CONSOLE_SOCKPATH, zone_name);
548 	(void) unlink(path);
549 	(void) shutdown(servfd, SHUT_RDWR);
550 	(void) close(servfd);
551 }
552 
553 /*
554  * Read the "ident" string from the client's descriptor; this routine also
555  * tolerates being called with pid=NULL, for times when you want to "eat"
556  * the ident string from a client without saving it.
557  */
558 static int
559 get_client_ident(int clifd, pid_t *pid, char *locale, size_t locale_len)
560 {
561 	char buf[BUFSIZ], *bufp;
562 	size_t buflen = sizeof (buf);
563 	char c = '\0';
564 	int i = 0, r;
565 
566 	/* "eat up the ident string" case, for simplicity */
567 	if (pid == NULL) {
568 		assert(locale == NULL && locale_len == 0);
569 		while (read(clifd, &c, 1) == 1) {
570 			if (c == '\n')
571 				return (0);
572 		}
573 	}
574 
575 	bzero(buf, sizeof (buf));
576 	while ((buflen > 1) && (r = read(clifd, &c, 1)) == 1) {
577 		buflen--;
578 		if (c == '\n')
579 			break;
580 
581 		buf[i] = c;
582 		i++;
583 	}
584 	if (r == -1)
585 		return (-1);
586 
587 	/*
588 	 * We've filled the buffer, but still haven't seen \n.  Keep eating
589 	 * until we find it; we don't expect this to happen, but this is
590 	 * defensive.
591 	 */
592 	if (c != '\n') {
593 		while ((r = read(clifd, &c, sizeof (c))) > 0)
594 			if (c == '\n')
595 				break;
596 	}
597 
598 	/*
599 	 * Parse buffer for message of the form: IDENT <pid> <locale>
600 	 */
601 	bufp = buf;
602 	if (strncmp(bufp, "IDENT ", 6) != 0)
603 		return (-1);
604 	bufp += 6;
605 	errno = 0;
606 	*pid = strtoll(bufp, &bufp, 10);
607 	if (errno != 0)
608 		return (-1);
609 
610 	while (*bufp != '\0' && isspace(*bufp))
611 		bufp++;
612 	(void) strlcpy(locale, bufp, locale_len);
613 
614 	return (0);
615 }
616 
617 static int
618 accept_client(int servfd, pid_t *pid, char *locale, size_t locale_len)
619 {
620 	int connfd;
621 	struct sockaddr_un cliaddr;
622 	socklen_t clilen;
623 
624 	clilen = sizeof (cliaddr);
625 	connfd = accept(servfd, (struct sockaddr *)&cliaddr, &clilen);
626 	if (connfd == -1)
627 		return (-1);
628 	if (get_client_ident(connfd, pid, locale, locale_len) == -1) {
629 		(void) shutdown(connfd, SHUT_RDWR);
630 		(void) close(connfd);
631 		return (-1);
632 	}
633 	(void) write(connfd, "OK\n", 3);
634 	return (connfd);
635 }
636 
637 static void
638 reject_client(int servfd, pid_t clientpid)
639 {
640 	int connfd;
641 	struct sockaddr_un cliaddr;
642 	socklen_t clilen;
643 	char nak[MAXPATHLEN];
644 
645 	clilen = sizeof (cliaddr);
646 	connfd = accept(servfd, (struct sockaddr *)&cliaddr, &clilen);
647 
648 	/*
649 	 * After hear its ident string, tell client to get lost.
650 	 */
651 	if (get_client_ident(connfd, NULL, NULL, 0) == 0) {
652 		(void) snprintf(nak, sizeof (nak), "%lu\n",
653 		    clientpid);
654 		(void) write(connfd, nak, strlen(nak));
655 	}
656 	(void) shutdown(connfd, SHUT_RDWR);
657 	(void) close(connfd);
658 }
659 
660 static void
661 event_message(int clifd, char *clilocale, zone_evt_t evt)
662 {
663 	char *str, *lstr = NULL;
664 	char lmsg[BUFSIZ];
665 	char outbuf[BUFSIZ];
666 
667 	if (clifd == -1)
668 		return;
669 
670 	switch (evt) {
671 	case Z_EVT_ZONE_BOOTING:
672 		if (*boot_args == '\0') {
673 			str = "NOTICE: Zone booting up";
674 			break;
675 		}
676 		/*LINTED*/
677 		(void) snprintf(lmsg, sizeof (lmsg), localize_msg(clilocale,
678 		    "NOTICE: Zone booting up with arguments: %s"), boot_args);
679 		lstr = lmsg;
680 		break;
681 	case Z_EVT_ZONE_READIED:
682 		str = "NOTICE: Zone readied";
683 		break;
684 	case Z_EVT_ZONE_HALTED:
685 		str = "NOTICE: Zone halted";
686 		break;
687 	case Z_EVT_ZONE_REBOOTING:
688 		if (*boot_args == '\0') {
689 			str = "NOTICE: Zone rebooting";
690 			break;
691 		}
692 		/*LINTED*/
693 		(void) snprintf(lmsg, sizeof (lmsg), localize_msg(clilocale,
694 		    "NOTICE: Zone rebooting with arguments: %s"), boot_args);
695 		lstr = lmsg;
696 		break;
697 	case Z_EVT_ZONE_UNINSTALLING:
698 		str = "NOTICE: Zone is being uninstalled.  Disconnecting...";
699 		break;
700 	case Z_EVT_ZONE_BOOTFAILED:
701 		str = "NOTICE: Zone boot failed";
702 		break;
703 	case Z_EVT_ZONE_BADARGS:
704 		/*LINTED*/
705 		(void) snprintf(lmsg, sizeof (lmsg),
706 		    localize_msg(clilocale,
707 		    "WARNING: Ignoring invalid boot arguments: %s"),
708 		    bad_boot_arg);
709 		lstr = lmsg;
710 		break;
711 	default:
712 		return;
713 	}
714 
715 	if (lstr == NULL)
716 		lstr = localize_msg(clilocale, str);
717 	(void) snprintf(outbuf, sizeof (outbuf), "\r\n[%s]\r\n", lstr);
718 	(void) write(clifd, outbuf, strlen(outbuf));
719 }
720 
721 /*
722  * Check to see if the client at the other end of the socket is still
723  * alive; we know it is not if it throws EPIPE at us when we try to write
724  * an otherwise harmless 0-length message to it.
725  */
726 static int
727 test_client(int clifd)
728 {
729 	if ((write(clifd, "", 0) == -1) && errno == EPIPE)
730 		return (-1);
731 	return (0);
732 }
733 
734 /*
735  * This routine drives the console I/O loop.  It polls for input from the
736  * master side of the console (output to the console), and from the client
737  * (input from the console user).  Additionally, it polls on the server fd,
738  * and disconnects any clients that might try to hook up with the zone while
739  * the console is in use.
740  *
741  * When the client first calls us up, it is expected to send a line giving
742  * its "identity"; this consists of the string 'IDENT <pid> <locale>'.
743  * This is so that we can report that the console is busy along with
744  * some diagnostics about who has it busy; the locale is used so that
745  * asynchronous messages about zone state (like the NOTICE: zone halted
746  * messages) can be output in the user's locale.
747  */
748 static void
749 do_console_io(zlog_t *zlogp, int consfd, int servfd)
750 {
751 	struct pollfd pollfds[4];
752 	char ibuf[BUFSIZ];
753 	int cc, ret;
754 	int clifd = -1;
755 	int pollerr = 0;
756 	char clilocale[MAXPATHLEN];
757 	pid_t clipid = 0;
758 
759 	/* console side, watch for read events */
760 	pollfds[0].fd = consfd;
761 	pollfds[0].events = POLLIN | POLLRDNORM | POLLRDBAND |
762 	    POLLPRI | POLLERR | POLLHUP | POLLNVAL;
763 
764 	/* client side, watch for read events */
765 	pollfds[1].fd = clifd;
766 	pollfds[1].events = pollfds[0].events;
767 
768 	/* the server socket; watch for events (new connections) */
769 	pollfds[2].fd = servfd;
770 	pollfds[2].events = pollfds[0].events;
771 
772 	/* the eventstram; watch for events (e.g.: zone halted) */
773 	pollfds[3].fd = eventstream[1];
774 	pollfds[3].events = pollfds[0].events;
775 
776 	for (;;) {
777 		pollfds[0].revents = pollfds[1].revents = 0;
778 		pollfds[2].revents = pollfds[3].revents = 0;
779 
780 		ret = poll(pollfds,
781 		    sizeof (pollfds) / sizeof (struct pollfd), -1);
782 		if (ret == -1 && errno != EINTR) {
783 			zerror(zlogp, B_TRUE, "poll failed");
784 			/* we are hosed, close connection */
785 			break;
786 		}
787 
788 		/* event from console side */
789 		if (pollfds[0].revents) {
790 			if (pollfds[0].revents &
791 			    (POLLIN | POLLRDNORM | POLLRDBAND | POLLPRI)) {
792 				errno = 0;
793 				cc = read(consfd, ibuf, BUFSIZ);
794 				if (cc <= 0 && (errno != EINTR) &&
795 				    (errno != EAGAIN))
796 					break;
797 				/*
798 				 * Lose I/O if no one is listening
799 				 */
800 				if (clifd != -1 && cc > 0)
801 					(void) write(clifd, ibuf, cc);
802 			} else {
803 				pollerr = pollfds[0].revents;
804 				zerror(zlogp, B_FALSE,
805 				    "closing connection with (console) "
806 				    "pollerr %d\n", pollerr);
807 				break;
808 			}
809 		}
810 
811 		/* event from client side */
812 		if (pollfds[1].revents) {
813 			if (pollfds[1].revents &
814 			    (POLLIN | POLLRDNORM | POLLRDBAND | POLLPRI)) {
815 				errno = 0;
816 				cc = read(clifd, ibuf, BUFSIZ);
817 				if (cc <= 0 && (errno != EINTR) &&
818 				    (errno != EAGAIN))
819 					break;
820 				(void) write(consfd, ibuf, cc);
821 			} else {
822 				pollerr = pollfds[1].revents;
823 				zerror(zlogp, B_FALSE,
824 				    "closing connection with (client) "
825 				    "pollerr %d\n", pollerr);
826 				break;
827 			}
828 		}
829 
830 		/* event from server socket */
831 		if (pollfds[2].revents &&
832 		    (pollfds[2].revents & (POLLIN | POLLRDNORM))) {
833 			if (clifd != -1) {
834 				/*
835 				 * Test the client to see if it is really
836 				 * still alive.  If it has died but we
837 				 * haven't yet detected that, we might
838 				 * deny a legitimate connect attempt.  If it
839 				 * is dead, we break out; once we tear down
840 				 * the old connection, the new connection
841 				 * will happen.
842 				 */
843 				if (test_client(clifd) == -1) {
844 					break;
845 				}
846 				/* we're already handling a client */
847 				reject_client(servfd, clipid);
848 
849 
850 			} else if ((clifd = accept_client(servfd, &clipid,
851 			    clilocale, sizeof (clilocale))) != -1) {
852 				pollfds[1].fd = clifd;
853 
854 			} else {
855 				break;
856 			}
857 		}
858 
859 		/*
860 		 * Watch for events on the eventstream.  This is how we get
861 		 * notified of the zone halting, etc.  It provides us a
862 		 * "wakeup" from poll when important things happen, which
863 		 * is good.
864 		 */
865 		if (pollfds[3].revents) {
866 			int evt = eventstream_read();
867 			/*
868 			 * After we drain out the event, if we aren't servicing
869 			 * a console client, we hop back out to our caller,
870 			 * which will check to see if it is time to shutdown
871 			 * the daemon, or if we should take another console
872 			 * service lap.
873 			 */
874 			if (clifd == -1) {
875 				break;
876 			}
877 			event_message(clifd, clilocale, evt);
878 			/*
879 			 * Special handling for the message that the zone is
880 			 * uninstalling; we boot the client, then break out
881 			 * of this function.  When we return to the
882 			 * serve_console loop, we will see that the zone is
883 			 * in a state < READY, and so zoneadmd will shutdown.
884 			 */
885 			if (evt == Z_EVT_ZONE_UNINSTALLING) {
886 				break;
887 			}
888 		}
889 
890 	}
891 
892 	if (clifd != -1) {
893 		(void) shutdown(clifd, SHUT_RDWR);
894 		(void) close(clifd);
895 	}
896 }
897 
898 int
899 init_console(zlog_t *zlogp)
900 {
901 	if (init_console_dev(zlogp) == -1) {
902 		zerror(zlogp, B_FALSE,
903 		    "console setup: device initialization failed");
904 		return (-1);
905 	}
906 
907 	if ((serverfd = init_console_sock(zlogp)) == -1) {
908 		zerror(zlogp, B_FALSE,
909 		    "console setup: socket initialization failed");
910 		return (-1);
911 	}
912 	return (0);
913 }
914 
915 /*
916  * serve_console() is the master loop for driving console I/O.  It is also the
917  * routine which is ultimately responsible for "pulling the plug" on zoneadmd
918  * when it realizes that the daemon should shut down.
919  *
920  * The rules for shutdown are: there must be no console client, and the zone
921  * state must be < ready.  However, we need to give things a chance to actually
922  * get going when the daemon starts up-- otherwise the daemon would immediately
923  * exit on startup if the zone was in the installed state, so we first drop
924  * into the do_console_io() loop in order to give *something* a chance to
925  * happen.
926  */
927 void
928 serve_console(zlog_t *zlogp)
929 {
930 	int masterfd;
931 	zone_state_t zstate;
932 	char conspath[MAXPATHLEN];
933 
934 	(void) snprintf(conspath, sizeof (conspath),
935 	    "/dev/zcons/%s/%s", zone_name, ZCONS_MASTER_NAME);
936 
937 	for (;;) {
938 		masterfd = open(conspath, O_RDWR|O_NONBLOCK|O_NOCTTY);
939 		if (masterfd == -1) {
940 			zerror(zlogp, B_TRUE, "failed to open console master");
941 			(void) mutex_lock(&lock);
942 			goto death;
943 		}
944 
945 		/*
946 		 * Setting RPROTDIS on the stream means that the control
947 		 * portion of messages received (which we don't care about)
948 		 * will be discarded by the stream head.  If we allowed such
949 		 * messages, we wouldn't be able to use read(2), as it fails
950 		 * (EBADMSG) when a message with a control element is received.
951 		 */
952 		if (ioctl(masterfd, I_SRDOPT, RNORM|RPROTDIS) == -1) {
953 			zerror(zlogp, B_TRUE, "failed to set options on "
954 			    "console master");
955 			(void) mutex_lock(&lock);
956 			goto death;
957 		}
958 
959 		do_console_io(zlogp, masterfd, serverfd);
960 
961 		/*
962 		 * We would prefer not to do this, but hostile zone processes
963 		 * can cause the stream to become tainted, and reads will
964 		 * fail.  So, in case something has gone seriously ill,
965 		 * we dismantle the stream and reopen the console when we
966 		 * take another lap.
967 		 */
968 		(void) close(masterfd);
969 
970 		(void) mutex_lock(&lock);
971 		/*
972 		 * We need to set death_throes (see below) atomically with
973 		 * respect to noticing that (a) we have no console client and
974 		 * (b) the zone is not installed.  Otherwise we could get a
975 		 * request to boot during this time.  Once we set death_throes,
976 		 * any incoming door stuff will be turned away.
977 		 */
978 		if (zone_get_state(zone_name, &zstate) == Z_OK) {
979 			if (zstate < ZONE_STATE_READY)
980 				goto death;
981 		} else {
982 			zerror(zlogp, B_FALSE,
983 			    "unable to determine state of zone");
984 			goto death;
985 		}
986 		/*
987 		 * Even if zone_get_state() fails, stay conservative, and
988 		 * take another lap.
989 		 */
990 		(void) mutex_unlock(&lock);
991 	}
992 
993 death:
994 	assert(MUTEX_HELD(&lock));
995 	in_death_throes = B_TRUE;
996 	(void) mutex_unlock(&lock);
997 
998 	destroy_console_sock(serverfd);
999 	(void) destroy_console_devs(zlogp);
1000 }
1001