xref: /freebsd/sys/kern/subr_prf.c (revision 06c3fb2749bda94cb5201f81ffdb8fa6c3161b2e)
1 /*-
2  * SPDX-License-Identifier: BSD-3-Clause
3  *
4  * Copyright (c) 1986, 1988, 1991, 1993
5  *	The Regents of the University of California.  All rights reserved.
6  * (c) UNIX System Laboratories, Inc.
7  * All or some portions of this file are derived from material licensed
8  * to the University of California by American Telephone and Telegraph
9  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
10  * the permission of UNIX System Laboratories, Inc.
11  *
12  * Redistribution and use in source and binary forms, with or without
13  * modification, are permitted provided that the following conditions
14  * are met:
15  * 1. Redistributions of source code must retain the above copyright
16  *    notice, this list of conditions and the following disclaimer.
17  * 2. Redistributions in binary form must reproduce the above copyright
18  *    notice, this list of conditions and the following disclaimer in the
19  *    documentation and/or other materials provided with the distribution.
20  * 3. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  */
36 
37 #include <sys/cdefs.h>
38 #ifdef _KERNEL
39 #include "opt_ddb.h"
40 #include "opt_printf.h"
41 #endif  /* _KERNEL */
42 
43 #include <sys/param.h>
44 #ifdef _KERNEL
45 #include <sys/systm.h>
46 #include <sys/lock.h>
47 #include <sys/kdb.h>
48 #include <sys/mutex.h>
49 #include <sys/sx.h>
50 #include <sys/kernel.h>
51 #include <sys/msgbuf.h>
52 #include <sys/malloc.h>
53 #include <sys/priv.h>
54 #include <sys/proc.h>
55 #include <sys/stddef.h>
56 #include <sys/sysctl.h>
57 #include <sys/tslog.h>
58 #include <sys/tty.h>
59 #include <sys/syslog.h>
60 #include <sys/cons.h>
61 #include <sys/uio.h>
62 #else /* !_KERNEL */
63 #include <errno.h>
64 #endif
65 #include <sys/ctype.h>
66 #include <sys/sbuf.h>
67 
68 #ifdef DDB
69 #include <ddb/ddb.h>
70 #endif
71 
72 /*
73  * Note that stdarg.h and the ANSI style va_start macro is used for both
74  * ANSI and traditional C compilers.
75  */
76 #ifdef _KERNEL
77 #include <machine/stdarg.h>
78 #else
79 #include <stdarg.h>
80 #endif
81 
82 /*
83  * This is needed for sbuf_putbuf() when compiled into userland.  Due to the
84  * shared nature of this file, it's the only place to put it.
85  */
86 #ifndef _KERNEL
87 #include <stdio.h>
88 #endif
89 
90 #ifdef _KERNEL
91 
92 #define TOCONS	0x01
93 #define TOTTY	0x02
94 #define TOLOG	0x04
95 
96 /* Max number conversion buffer length: a u_quad_t in base 2, plus NUL byte. */
97 #define MAXNBUF	(sizeof(intmax_t) * NBBY + 1)
98 
99 struct putchar_arg {
100 	int	flags;
101 	int	pri;
102 	struct	tty *tty;
103 	char	*p_bufr;
104 	size_t	n_bufr;
105 	char	*p_next;
106 	size_t	remain;
107 };
108 
109 struct snprintf_arg {
110 	char	*str;
111 	size_t	remain;
112 };
113 
114 extern	int log_open;
115 
116 static void  msglogchar(int c, int pri);
117 static void  msglogstr(char *str, int pri, int filter_cr);
118 static void  prf_putbuf(char *bufr, int flags, int pri);
119 static void  putchar(int ch, void *arg);
120 static char *ksprintn(char *nbuf, uintmax_t num, int base, int *len, int upper);
121 static void  snprintf_func(int ch, void *arg);
122 
123 static bool msgbufmapped;		/* Set when safe to use msgbuf */
124 int msgbuftrigger;
125 struct msgbuf *msgbufp;
126 
127 #ifndef BOOT_TAG_SZ
128 #define	BOOT_TAG_SZ	32
129 #endif
130 #ifndef BOOT_TAG
131 /* Tag used to mark the start of a boot in dmesg */
132 #define	BOOT_TAG	"---<<BOOT>>---"
133 #endif
134 
135 static char current_boot_tag[BOOT_TAG_SZ + 1] = BOOT_TAG;
136 SYSCTL_STRING(_kern, OID_AUTO, boot_tag, CTLFLAG_RDTUN | CTLFLAG_NOFETCH,
137     current_boot_tag, 0, "Tag added to dmesg at start of boot");
138 
139 static int log_console_output = 1;
140 SYSCTL_INT(_kern, OID_AUTO, log_console_output, CTLFLAG_RWTUN,
141     &log_console_output, 0, "Duplicate console output to the syslog");
142 
143 /*
144  * See the comment in log_console() below for more explanation of this.
145  */
146 static int log_console_add_linefeed;
147 SYSCTL_INT(_kern, OID_AUTO, log_console_add_linefeed, CTLFLAG_RWTUN,
148     &log_console_add_linefeed, 0, "log_console() adds extra newlines");
149 
150 static int always_console_output;
151 SYSCTL_INT(_kern, OID_AUTO, always_console_output, CTLFLAG_RWTUN,
152     &always_console_output, 0, "Always output to console despite TIOCCONS");
153 
154 /*
155  * Warn that a system table is full.
156  */
157 void
158 tablefull(const char *tab)
159 {
160 
161 	log(LOG_ERR, "%s: table is full\n", tab);
162 }
163 
164 /*
165  * Uprintf prints to the controlling terminal for the current process.
166  */
167 int
168 uprintf(const char *fmt, ...)
169 {
170 	va_list ap;
171 	struct putchar_arg pca;
172 	struct proc *p;
173 	struct thread *td;
174 	int retval;
175 
176 	td = curthread;
177 	if (TD_IS_IDLETHREAD(td))
178 		return (0);
179 
180 	if (td->td_proc == initproc) {
181 		/* Produce output when we fail to load /sbin/init: */
182 		va_start(ap, fmt);
183 		retval = vprintf(fmt, ap);
184 		va_end(ap);
185 		return (retval);
186 	}
187 
188 	sx_slock(&proctree_lock);
189 	p = td->td_proc;
190 	PROC_LOCK(p);
191 	if ((p->p_flag & P_CONTROLT) == 0) {
192 		PROC_UNLOCK(p);
193 		sx_sunlock(&proctree_lock);
194 		return (0);
195 	}
196 	SESS_LOCK(p->p_session);
197 	pca.tty = p->p_session->s_ttyp;
198 	SESS_UNLOCK(p->p_session);
199 	PROC_UNLOCK(p);
200 	if (pca.tty == NULL) {
201 		sx_sunlock(&proctree_lock);
202 		return (0);
203 	}
204 	pca.flags = TOTTY;
205 	pca.p_bufr = NULL;
206 	va_start(ap, fmt);
207 	tty_lock(pca.tty);
208 	sx_sunlock(&proctree_lock);
209 	retval = kvprintf(fmt, putchar, &pca, 10, ap);
210 	tty_unlock(pca.tty);
211 	va_end(ap);
212 	return (retval);
213 }
214 
215 /*
216  * tprintf and vtprintf print on the controlling terminal associated with the
217  * given session, possibly to the log as well.
218  */
219 void
220 tprintf(struct proc *p, int pri, const char *fmt, ...)
221 {
222 	va_list ap;
223 
224 	va_start(ap, fmt);
225 	vtprintf(p, pri, fmt, ap);
226 	va_end(ap);
227 }
228 
229 void
230 vtprintf(struct proc *p, int pri, const char *fmt, va_list ap)
231 {
232 	struct tty *tp = NULL;
233 	int flags = 0;
234 	struct putchar_arg pca;
235 	struct session *sess = NULL;
236 
237 	sx_slock(&proctree_lock);
238 	if (pri != -1)
239 		flags |= TOLOG;
240 	if (p != NULL) {
241 		PROC_LOCK(p);
242 		if (p->p_flag & P_CONTROLT && p->p_session->s_ttyvp) {
243 			sess = p->p_session;
244 			sess_hold(sess);
245 			PROC_UNLOCK(p);
246 			tp = sess->s_ttyp;
247 			if (tp != NULL && tty_checkoutq(tp))
248 				flags |= TOTTY;
249 			else
250 				tp = NULL;
251 		} else
252 			PROC_UNLOCK(p);
253 	}
254 	pca.pri = pri;
255 	pca.tty = tp;
256 	pca.flags = flags;
257 	pca.p_bufr = NULL;
258 	if (pca.tty != NULL)
259 		tty_lock(pca.tty);
260 	sx_sunlock(&proctree_lock);
261 	kvprintf(fmt, putchar, &pca, 10, ap);
262 	if (pca.tty != NULL)
263 		tty_unlock(pca.tty);
264 	if (sess != NULL)
265 		sess_release(sess);
266 	msgbuftrigger = 1;
267 }
268 
269 static int
270 _vprintf(int level, int flags, const char *fmt, va_list ap)
271 {
272 	struct putchar_arg pca;
273 	int retval;
274 #ifdef PRINTF_BUFR_SIZE
275 	char bufr[PRINTF_BUFR_SIZE];
276 #endif
277 
278 	TSENTER();
279 	pca.tty = NULL;
280 	pca.pri = level;
281 	pca.flags = flags;
282 #ifdef PRINTF_BUFR_SIZE
283 	pca.p_bufr = bufr;
284 	pca.p_next = pca.p_bufr;
285 	pca.n_bufr = sizeof(bufr);
286 	pca.remain = sizeof(bufr);
287 	*pca.p_next = '\0';
288 #else
289 	/* Don't buffer console output. */
290 	pca.p_bufr = NULL;
291 #endif
292 
293 	retval = kvprintf(fmt, putchar, &pca, 10, ap);
294 
295 #ifdef PRINTF_BUFR_SIZE
296 	/* Write any buffered console/log output: */
297 	if (*pca.p_bufr != '\0')
298 		prf_putbuf(pca.p_bufr, flags, level);
299 #endif
300 
301 	TSEXIT();
302 	return (retval);
303 }
304 
305 /*
306  * Log writes to the log buffer, and guarantees not to sleep (so can be
307  * called by interrupt routines).  If there is no process reading the
308  * log yet, it writes to the console also.
309  */
310 void
311 log(int level, const char *fmt, ...)
312 {
313 	va_list ap;
314 
315 	va_start(ap, fmt);
316 	vlog(level, fmt, ap);
317 	va_end(ap);
318 }
319 
320 void
321 vlog(int level, const char *fmt, va_list ap)
322 {
323 
324 	(void)_vprintf(level, log_open ? TOLOG : TOCONS | TOLOG, fmt, ap);
325 	msgbuftrigger = 1;
326 }
327 
328 #define CONSCHUNK 128
329 
330 void
331 log_console(struct uio *uio)
332 {
333 	int c, error, nl;
334 	char *consbuffer;
335 	int pri;
336 
337 	if (!log_console_output)
338 		return;
339 
340 	pri = LOG_INFO | LOG_CONSOLE;
341 	uio = cloneuio(uio);
342 	consbuffer = malloc(CONSCHUNK, M_TEMP, M_WAITOK);
343 
344 	nl = 0;
345 	while (uio->uio_resid > 0) {
346 		c = imin(uio->uio_resid, CONSCHUNK - 1);
347 		error = uiomove(consbuffer, c, uio);
348 		if (error != 0)
349 			break;
350 		/* Make sure we're NUL-terminated */
351 		consbuffer[c] = '\0';
352 		if (consbuffer[c - 1] == '\n')
353 			nl = 1;
354 		else
355 			nl = 0;
356 		msglogstr(consbuffer, pri, /*filter_cr*/ 1);
357 	}
358 	/*
359 	 * The previous behavior in log_console() is preserved when
360 	 * log_console_add_linefeed is non-zero.  For that behavior, if an
361 	 * individual console write came in that was not terminated with a
362 	 * line feed, it would add a line feed.
363 	 *
364 	 * This results in different data in the message buffer than
365 	 * appears on the system console (which doesn't add extra line feed
366 	 * characters).
367 	 *
368 	 * A number of programs and rc scripts write a line feed, or a period
369 	 * and a line feed when they have completed their operation.  On
370 	 * the console, this looks seamless, but when displayed with
371 	 * 'dmesg -a', you wind up with output that looks like this:
372 	 *
373 	 * Updating motd:
374 	 * .
375 	 *
376 	 * On the console, it looks like this:
377 	 * Updating motd:.
378 	 *
379 	 * We could add logic to detect that situation, or just not insert
380 	 * the extra newlines.  Set the kern.log_console_add_linefeed
381 	 * sysctl/tunable variable to get the old behavior.
382 	 */
383 	if (!nl && log_console_add_linefeed) {
384 		consbuffer[0] = '\n';
385 		consbuffer[1] = '\0';
386 		msglogstr(consbuffer, pri, /*filter_cr*/ 1);
387 	}
388 	msgbuftrigger = 1;
389 	free(uio, M_IOV);
390 	free(consbuffer, M_TEMP);
391 }
392 
393 int
394 printf(const char *fmt, ...)
395 {
396 	va_list ap;
397 	int retval;
398 
399 	va_start(ap, fmt);
400 	retval = vprintf(fmt, ap);
401 	va_end(ap);
402 
403 	return (retval);
404 }
405 
406 int
407 vprintf(const char *fmt, va_list ap)
408 {
409 	int retval;
410 
411 	retval = _vprintf(-1, TOCONS | TOLOG, fmt, ap);
412 
413 	if (!KERNEL_PANICKED())
414 		msgbuftrigger = 1;
415 
416 	return (retval);
417 }
418 
419 static void
420 prf_putchar(int c, int flags, int pri)
421 {
422 
423 	if (flags & TOLOG) {
424 		msglogchar(c, pri);
425 		msgbuftrigger = 1;
426 	}
427 
428 	if (flags & TOCONS) {
429 		if ((!KERNEL_PANICKED()) && (constty != NULL))
430 			msgbuf_addchar(&consmsgbuf, c);
431 
432 		if ((constty == NULL) || always_console_output)
433 			cnputc(c);
434 	}
435 }
436 
437 static void
438 prf_putbuf(char *bufr, int flags, int pri)
439 {
440 
441 	if (flags & TOLOG) {
442 		msglogstr(bufr, pri, /*filter_cr*/1);
443 		msgbuftrigger = 1;
444 	}
445 
446 	if (flags & TOCONS) {
447 		if ((!KERNEL_PANICKED()) && (constty != NULL))
448 			msgbuf_addstr(&consmsgbuf, -1,
449 			    bufr, /*filter_cr*/ 0);
450 
451 		if ((constty == NULL) || always_console_output)
452 			cnputs(bufr);
453 	}
454 }
455 
456 static void
457 putbuf(int c, struct putchar_arg *ap)
458 {
459 	/* Check if no console output buffer was provided. */
460 	if (ap->p_bufr == NULL) {
461 		prf_putchar(c, ap->flags, ap->pri);
462 	} else {
463 		/* Buffer the character: */
464 		*ap->p_next++ = c;
465 		ap->remain--;
466 
467 		/* Always leave the buffer zero terminated. */
468 		*ap->p_next = '\0';
469 
470 		/* Check if the buffer needs to be flushed. */
471 		if (ap->remain == 2 || c == '\n') {
472 			prf_putbuf(ap->p_bufr, ap->flags, ap->pri);
473 
474 			ap->p_next = ap->p_bufr;
475 			ap->remain = ap->n_bufr;
476 			*ap->p_next = '\0';
477 		}
478 
479 		/*
480 		 * Since we fill the buffer up one character at a time,
481 		 * this should not happen.  We should always catch it when
482 		 * ap->remain == 2 (if not sooner due to a newline), flush
483 		 * the buffer and move on.  One way this could happen is
484 		 * if someone sets PRINTF_BUFR_SIZE to 1 or something
485 		 * similarly silly.
486 		 */
487 		KASSERT(ap->remain > 2, ("Bad buffer logic, remain = %zd",
488 		    ap->remain));
489 	}
490 }
491 
492 /*
493  * Print a character on console or users terminal.  If destination is
494  * the console then the last bunch of characters are saved in msgbuf for
495  * inspection later.
496  */
497 static void
498 putchar(int c, void *arg)
499 {
500 	struct putchar_arg *ap = (struct putchar_arg*) arg;
501 	struct tty *tp = ap->tty;
502 	int flags = ap->flags;
503 
504 	/* Don't use the tty code after a panic or while in ddb. */
505 	if (kdb_active) {
506 		if (c != '\0')
507 			cnputc(c);
508 		return;
509 	}
510 
511 	if ((flags & TOTTY) && tp != NULL && !KERNEL_PANICKED())
512 		tty_putchar(tp, c);
513 
514 	if ((flags & (TOCONS | TOLOG)) && c != '\0')
515 		putbuf(c, ap);
516 }
517 
518 /*
519  * Scaled down version of sprintf(3).
520  */
521 int
522 sprintf(char *buf, const char *cfmt, ...)
523 {
524 	int retval;
525 	va_list ap;
526 
527 	va_start(ap, cfmt);
528 	retval = kvprintf(cfmt, NULL, (void *)buf, 10, ap);
529 	buf[retval] = '\0';
530 	va_end(ap);
531 	return (retval);
532 }
533 
534 /*
535  * Scaled down version of vsprintf(3).
536  */
537 int
538 vsprintf(char *buf, const char *cfmt, va_list ap)
539 {
540 	int retval;
541 
542 	retval = kvprintf(cfmt, NULL, (void *)buf, 10, ap);
543 	buf[retval] = '\0';
544 	return (retval);
545 }
546 
547 /*
548  * Scaled down version of snprintf(3).
549  */
550 int
551 snprintf(char *str, size_t size, const char *format, ...)
552 {
553 	int retval;
554 	va_list ap;
555 
556 	va_start(ap, format);
557 	retval = vsnprintf(str, size, format, ap);
558 	va_end(ap);
559 	return(retval);
560 }
561 
562 /*
563  * Scaled down version of vsnprintf(3).
564  */
565 int
566 vsnprintf(char *str, size_t size, const char *format, va_list ap)
567 {
568 	struct snprintf_arg info;
569 	int retval;
570 
571 	info.str = str;
572 	info.remain = size;
573 	retval = kvprintf(format, snprintf_func, &info, 10, ap);
574 	if (info.remain >= 1)
575 		*info.str++ = '\0';
576 	return (retval);
577 }
578 
579 /*
580  * Kernel version which takes radix argument vsnprintf(3).
581  */
582 int
583 vsnrprintf(char *str, size_t size, int radix, const char *format, va_list ap)
584 {
585 	struct snprintf_arg info;
586 	int retval;
587 
588 	info.str = str;
589 	info.remain = size;
590 	retval = kvprintf(format, snprintf_func, &info, radix, ap);
591 	if (info.remain >= 1)
592 		*info.str++ = '\0';
593 	return (retval);
594 }
595 
596 static void
597 snprintf_func(int ch, void *arg)
598 {
599 	struct snprintf_arg *const info = arg;
600 
601 	if (info->remain >= 2) {
602 		*info->str++ = ch;
603 		info->remain--;
604 	}
605 }
606 
607 /*
608  * Put a NUL-terminated ASCII number (base <= 36) in a buffer in reverse
609  * order; return an optional length and a pointer to the last character
610  * written in the buffer (i.e., the first character of the string).
611  * The buffer pointed to by `nbuf' must have length >= MAXNBUF.
612  */
613 static char *
614 ksprintn(char *nbuf, uintmax_t num, int base, int *lenp, int upper)
615 {
616 	char *p, c;
617 
618 	p = nbuf;
619 	*p = '\0';
620 	do {
621 		c = hex2ascii(num % base);
622 		*++p = upper ? toupper(c) : c;
623 	} while (num /= base);
624 	if (lenp)
625 		*lenp = p - nbuf;
626 	return (p);
627 }
628 
629 /*
630  * Scaled down version of printf(3).
631  *
632  * Two additional formats:
633  *
634  * The format %b is supported to decode error registers.
635  * Its usage is:
636  *
637  *	printf("reg=%b\n", regval, "<base><arg>*");
638  *
639  * where <base> is the output base expressed as a control character, e.g.
640  * \10 gives octal; \20 gives hex.  Each arg is a sequence of characters,
641  * the first of which gives the bit number to be inspected (origin 1), and
642  * the next characters (up to a control character, i.e. a character <= 32),
643  * give the name of the register.  Thus:
644  *
645  *	kvprintf("reg=%b\n", 3, "\10\2BITTWO\1BITONE");
646  *
647  * would produce output:
648  *
649  *	reg=3<BITTWO,BITONE>
650  *
651  * XXX:  %D  -- Hexdump, takes pointer and separator string:
652  *		("%6D", ptr, ":")   -> XX:XX:XX:XX:XX:XX
653  *		("%*D", len, ptr, " " -> XX XX XX XX ...
654  */
655 int
656 kvprintf(char const *fmt, void (*func)(int, void*), void *arg, int radix, va_list ap)
657 {
658 #define PCHAR(c) {int cc=(c); if (func) (*func)(cc,arg); else *d++ = cc; retval++; }
659 	char nbuf[MAXNBUF];
660 	char *d;
661 	const char *p, *percent, *q;
662 	u_char *up;
663 	int ch, n;
664 	uintmax_t num;
665 	int base, lflag, qflag, tmp, width, ladjust, sharpflag, neg, sign, dot;
666 	int cflag, hflag, jflag, tflag, zflag;
667 	int bconv, dwidth, upper;
668 	char padc;
669 	int stop = 0, retval = 0;
670 
671 	num = 0;
672 	q = NULL;
673 	if (!func)
674 		d = (char *) arg;
675 	else
676 		d = NULL;
677 
678 	if (fmt == NULL)
679 		fmt = "(fmt null)\n";
680 
681 	if (radix < 2 || radix > 36)
682 		radix = 10;
683 
684 	for (;;) {
685 		padc = ' ';
686 		width = 0;
687 		while ((ch = (u_char)*fmt++) != '%' || stop) {
688 			if (ch == '\0')
689 				return (retval);
690 			PCHAR(ch);
691 		}
692 		percent = fmt - 1;
693 		qflag = 0; lflag = 0; ladjust = 0; sharpflag = 0; neg = 0;
694 		sign = 0; dot = 0; bconv = 0; dwidth = 0; upper = 0;
695 		cflag = 0; hflag = 0; jflag = 0; tflag = 0; zflag = 0;
696 reswitch:	switch (ch = (u_char)*fmt++) {
697 		case '.':
698 			dot = 1;
699 			goto reswitch;
700 		case '#':
701 			sharpflag = 1;
702 			goto reswitch;
703 		case '+':
704 			sign = 1;
705 			goto reswitch;
706 		case '-':
707 			ladjust = 1;
708 			goto reswitch;
709 		case '%':
710 			PCHAR(ch);
711 			break;
712 		case '*':
713 			if (!dot) {
714 				width = va_arg(ap, int);
715 				if (width < 0) {
716 					ladjust = !ladjust;
717 					width = -width;
718 				}
719 			} else {
720 				dwidth = va_arg(ap, int);
721 			}
722 			goto reswitch;
723 		case '0':
724 			if (!dot) {
725 				padc = '0';
726 				goto reswitch;
727 			}
728 			/* FALLTHROUGH */
729 		case '1': case '2': case '3': case '4':
730 		case '5': case '6': case '7': case '8': case '9':
731 				for (n = 0;; ++fmt) {
732 					n = n * 10 + ch - '0';
733 					ch = *fmt;
734 					if (ch < '0' || ch > '9')
735 						break;
736 				}
737 			if (dot)
738 				dwidth = n;
739 			else
740 				width = n;
741 			goto reswitch;
742 		case 'b':
743 			ladjust = 1;
744 			bconv = 1;
745 			goto handle_nosign;
746 		case 'c':
747 			width -= 1;
748 
749 			if (!ladjust && width > 0)
750 				while (width--)
751 					PCHAR(padc);
752 			PCHAR(va_arg(ap, int));
753 			if (ladjust && width > 0)
754 				while (width--)
755 					PCHAR(padc);
756 			break;
757 		case 'D':
758 			up = va_arg(ap, u_char *);
759 			p = va_arg(ap, char *);
760 			if (!width)
761 				width = 16;
762 			while(width--) {
763 				PCHAR(hex2ascii(*up >> 4));
764 				PCHAR(hex2ascii(*up & 0x0f));
765 				up++;
766 				if (width)
767 					for (q=p;*q;q++)
768 						PCHAR(*q);
769 			}
770 			break;
771 		case 'd':
772 		case 'i':
773 			base = 10;
774 			sign = 1;
775 			goto handle_sign;
776 		case 'h':
777 			if (hflag) {
778 				hflag = 0;
779 				cflag = 1;
780 			} else
781 				hflag = 1;
782 			goto reswitch;
783 		case 'j':
784 			jflag = 1;
785 			goto reswitch;
786 		case 'l':
787 			if (lflag) {
788 				lflag = 0;
789 				qflag = 1;
790 			} else
791 				lflag = 1;
792 			goto reswitch;
793 		case 'n':
794 			/*
795 			 * We do not support %n in kernel, but consume the
796 			 * argument.
797 			 */
798 			if (jflag)
799 				(void)va_arg(ap, intmax_t *);
800 			else if (qflag)
801 				(void)va_arg(ap, quad_t *);
802 			else if (lflag)
803 				(void)va_arg(ap, long *);
804 			else if (zflag)
805 				(void)va_arg(ap, size_t *);
806 			else if (hflag)
807 				(void)va_arg(ap, short *);
808 			else if (cflag)
809 				(void)va_arg(ap, char *);
810 			else
811 				(void)va_arg(ap, int *);
812 			break;
813 		case 'o':
814 			base = 8;
815 			goto handle_nosign;
816 		case 'p':
817 			base = 16;
818 			sharpflag = (width == 0);
819 			sign = 0;
820 			num = (uintptr_t)va_arg(ap, void *);
821 			goto number;
822 		case 'q':
823 			qflag = 1;
824 			goto reswitch;
825 		case 'r':
826 			base = radix;
827 			if (sign)
828 				goto handle_sign;
829 			goto handle_nosign;
830 		case 's':
831 			p = va_arg(ap, char *);
832 			if (p == NULL)
833 				p = "(null)";
834 			if (!dot)
835 				n = strlen (p);
836 			else
837 				for (n = 0; n < dwidth && p[n]; n++)
838 					continue;
839 
840 			width -= n;
841 
842 			if (!ladjust && width > 0)
843 				while (width--)
844 					PCHAR(padc);
845 			while (n--)
846 				PCHAR(*p++);
847 			if (ladjust && width > 0)
848 				while (width--)
849 					PCHAR(padc);
850 			break;
851 		case 't':
852 			tflag = 1;
853 			goto reswitch;
854 		case 'u':
855 			base = 10;
856 			goto handle_nosign;
857 		case 'X':
858 			upper = 1;
859 			/* FALLTHROUGH */
860 		case 'x':
861 			base = 16;
862 			goto handle_nosign;
863 		case 'y':
864 			base = 16;
865 			sign = 1;
866 			goto handle_sign;
867 		case 'z':
868 			zflag = 1;
869 			goto reswitch;
870 handle_nosign:
871 			sign = 0;
872 			if (jflag)
873 				num = va_arg(ap, uintmax_t);
874 			else if (qflag)
875 				num = va_arg(ap, u_quad_t);
876 			else if (tflag)
877 				num = va_arg(ap, ptrdiff_t);
878 			else if (lflag)
879 				num = va_arg(ap, u_long);
880 			else if (zflag)
881 				num = va_arg(ap, size_t);
882 			else if (hflag)
883 				num = (u_short)va_arg(ap, int);
884 			else if (cflag)
885 				num = (u_char)va_arg(ap, int);
886 			else
887 				num = va_arg(ap, u_int);
888 			if (bconv) {
889 				q = va_arg(ap, char *);
890 				base = *q++;
891 			}
892 			goto number;
893 handle_sign:
894 			if (jflag)
895 				num = va_arg(ap, intmax_t);
896 			else if (qflag)
897 				num = va_arg(ap, quad_t);
898 			else if (tflag)
899 				num = va_arg(ap, ptrdiff_t);
900 			else if (lflag)
901 				num = va_arg(ap, long);
902 			else if (zflag)
903 				num = va_arg(ap, ssize_t);
904 			else if (hflag)
905 				num = (short)va_arg(ap, int);
906 			else if (cflag)
907 				num = (char)va_arg(ap, int);
908 			else
909 				num = va_arg(ap, int);
910 number:
911 			if (sign && (intmax_t)num < 0) {
912 				neg = 1;
913 				num = -(intmax_t)num;
914 			}
915 			p = ksprintn(nbuf, num, base, &n, upper);
916 			tmp = 0;
917 			if (sharpflag && num != 0) {
918 				if (base == 8)
919 					tmp++;
920 				else if (base == 16)
921 					tmp += 2;
922 			}
923 			if (neg)
924 				tmp++;
925 
926 			if (!ladjust && padc == '0')
927 				dwidth = width - tmp;
928 			width -= tmp + imax(dwidth, n);
929 			dwidth -= n;
930 			if (!ladjust)
931 				while (width-- > 0)
932 					PCHAR(' ');
933 			if (neg)
934 				PCHAR('-');
935 			if (sharpflag && num != 0) {
936 				if (base == 8) {
937 					PCHAR('0');
938 				} else if (base == 16) {
939 					PCHAR('0');
940 					PCHAR('x');
941 				}
942 			}
943 			while (dwidth-- > 0)
944 				PCHAR('0');
945 
946 			while (*p)
947 				PCHAR(*p--);
948 
949 			if (bconv && num != 0) {
950 				/* %b conversion flag format. */
951 				tmp = retval;
952 				while (*q) {
953 					n = *q++;
954 					if (num & (1 << (n - 1))) {
955 						PCHAR(retval != tmp ?
956 						    ',' : '<');
957 						for (; (n = *q) > ' '; ++q)
958 							PCHAR(n);
959 					} else
960 						for (; *q > ' '; ++q)
961 							continue;
962 				}
963 				if (retval != tmp) {
964 					PCHAR('>');
965 					width -= retval - tmp;
966 				}
967 			}
968 
969 			if (ladjust)
970 				while (width-- > 0)
971 					PCHAR(' ');
972 
973 			break;
974 		default:
975 			while (percent < fmt)
976 				PCHAR(*percent++);
977 			/*
978 			 * Since we ignore a formatting argument it is no
979 			 * longer safe to obey the remaining formatting
980 			 * arguments as the arguments will no longer match
981 			 * the format specs.
982 			 */
983 			stop = 1;
984 			break;
985 		}
986 	}
987 #undef PCHAR
988 }
989 
990 /*
991  * Put character in log buffer with a particular priority.
992  */
993 static void
994 msglogchar(int c, int pri)
995 {
996 	static int lastpri = -1;
997 	static int dangling;
998 	char nbuf[MAXNBUF];
999 	char *p;
1000 
1001 	if (!msgbufmapped)
1002 		return;
1003 	if (c == '\0' || c == '\r')
1004 		return;
1005 	if (pri != -1 && pri != lastpri) {
1006 		if (dangling) {
1007 			msgbuf_addchar(msgbufp, '\n');
1008 			dangling = 0;
1009 		}
1010 		msgbuf_addchar(msgbufp, '<');
1011 		for (p = ksprintn(nbuf, (uintmax_t)pri, 10, NULL, 0); *p;)
1012 			msgbuf_addchar(msgbufp, *p--);
1013 		msgbuf_addchar(msgbufp, '>');
1014 		lastpri = pri;
1015 	}
1016 	msgbuf_addchar(msgbufp, c);
1017 	if (c == '\n') {
1018 		dangling = 0;
1019 		lastpri = -1;
1020 	} else {
1021 		dangling = 1;
1022 	}
1023 }
1024 
1025 static void
1026 msglogstr(char *str, int pri, int filter_cr)
1027 {
1028 	if (!msgbufmapped)
1029 		return;
1030 
1031 	msgbuf_addstr(msgbufp, pri, str, filter_cr);
1032 }
1033 
1034 void
1035 msgbufinit(void *ptr, int size)
1036 {
1037 	char *cp;
1038 	static struct msgbuf *oldp = NULL;
1039 	bool print_boot_tag;
1040 
1041 	TSENTER();
1042 	size -= sizeof(*msgbufp);
1043 	cp = (char *)ptr;
1044 	print_boot_tag = !msgbufmapped;
1045 	/* Attempt to fetch kern.boot_tag tunable on first mapping */
1046 	if (!msgbufmapped)
1047 		TUNABLE_STR_FETCH("kern.boot_tag", current_boot_tag,
1048 		    sizeof(current_boot_tag));
1049 	msgbufp = (struct msgbuf *)(cp + size);
1050 	msgbuf_reinit(msgbufp, cp, size);
1051 	if (msgbufmapped && oldp != msgbufp)
1052 		msgbuf_copy(oldp, msgbufp);
1053 	msgbufmapped = true;
1054 	if (print_boot_tag && *current_boot_tag != '\0')
1055 		printf("%s\n", current_boot_tag);
1056 	oldp = msgbufp;
1057 	TSEXIT();
1058 }
1059 
1060 /* Sysctls for accessing/clearing the msgbuf */
1061 static int
1062 sysctl_kern_msgbuf(SYSCTL_HANDLER_ARGS)
1063 {
1064 	char buf[128], *bp;
1065 	u_int seq;
1066 	int error, len;
1067 	bool wrap;
1068 
1069 	error = priv_check(req->td, PRIV_MSGBUF);
1070 	if (error)
1071 		return (error);
1072 
1073 	/* Read the whole buffer, one chunk at a time. */
1074 	mtx_lock(&msgbuf_lock);
1075 	msgbuf_peekbytes(msgbufp, NULL, 0, &seq);
1076 	wrap = (seq != 0);
1077 	for (;;) {
1078 		len = msgbuf_peekbytes(msgbufp, buf, sizeof(buf), &seq);
1079 		mtx_unlock(&msgbuf_lock);
1080 		if (len == 0)
1081 			return (SYSCTL_OUT(req, "", 1)); /* add nulterm */
1082 		if (wrap) {
1083 			/* Skip the first line, as it is probably incomplete. */
1084 			bp = memchr(buf, '\n', len);
1085 			if (bp == NULL) {
1086 				mtx_lock(&msgbuf_lock);
1087 				continue;
1088 			}
1089 			wrap = false;
1090 			bp++;
1091 			len -= bp - buf;
1092 			if (len == 0) {
1093 				mtx_lock(&msgbuf_lock);
1094 				continue;
1095 			}
1096 		} else
1097 			bp = buf;
1098 		error = sysctl_handle_opaque(oidp, bp, len, req);
1099 		if (error)
1100 			return (error);
1101 
1102 		mtx_lock(&msgbuf_lock);
1103 	}
1104 }
1105 
1106 SYSCTL_PROC(_kern, OID_AUTO, msgbuf,
1107     CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_MPSAFE,
1108     NULL, 0, sysctl_kern_msgbuf, "A", "Contents of kernel message buffer");
1109 
1110 static int msgbuf_clearflag;
1111 
1112 static int
1113 sysctl_kern_msgbuf_clear(SYSCTL_HANDLER_ARGS)
1114 {
1115 	int error;
1116 	error = sysctl_handle_int(oidp, oidp->oid_arg1, oidp->oid_arg2, req);
1117 	if (!error && req->newptr) {
1118 		mtx_lock(&msgbuf_lock);
1119 		msgbuf_clear(msgbufp);
1120 		mtx_unlock(&msgbuf_lock);
1121 		msgbuf_clearflag = 0;
1122 	}
1123 	return (error);
1124 }
1125 
1126 SYSCTL_PROC(_kern, OID_AUTO, msgbuf_clear,
1127     CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_SECURE | CTLFLAG_MPSAFE,
1128     &msgbuf_clearflag, 0, sysctl_kern_msgbuf_clear, "I",
1129     "Clear kernel message buffer");
1130 
1131 #ifdef DDB
1132 
1133 DB_SHOW_COMMAND_FLAGS(msgbuf, db_show_msgbuf, DB_CMD_MEMSAFE)
1134 {
1135 	int i, j;
1136 
1137 	if (!msgbufmapped) {
1138 		db_printf("msgbuf not mapped yet\n");
1139 		return;
1140 	}
1141 	db_printf("msgbufp = %p\n", msgbufp);
1142 	db_printf("magic = %x, size = %d, r= %u, w = %u, ptr = %p, cksum= %u\n",
1143 	    msgbufp->msg_magic, msgbufp->msg_size, msgbufp->msg_rseq,
1144 	    msgbufp->msg_wseq, msgbufp->msg_ptr, msgbufp->msg_cksum);
1145 	for (i = 0; i < msgbufp->msg_size && !db_pager_quit; i++) {
1146 		j = MSGBUF_SEQ_TO_POS(msgbufp, i + msgbufp->msg_rseq);
1147 		db_printf("%c", msgbufp->msg_ptr[j]);
1148 	}
1149 	db_printf("\n");
1150 }
1151 
1152 #endif /* DDB */
1153 
1154 void
1155 hexdump(const void *ptr, int length, const char *hdr, int flags)
1156 {
1157 	int i, j, k;
1158 	int cols;
1159 	const unsigned char *cp;
1160 	char delim;
1161 
1162 	if ((flags & HD_DELIM_MASK) != 0)
1163 		delim = (flags & HD_DELIM_MASK) >> 8;
1164 	else
1165 		delim = ' ';
1166 
1167 	if ((flags & HD_COLUMN_MASK) != 0)
1168 		cols = flags & HD_COLUMN_MASK;
1169 	else
1170 		cols = 16;
1171 
1172 	cp = ptr;
1173 	for (i = 0; i < length; i+= cols) {
1174 		if (hdr != NULL)
1175 			printf("%s", hdr);
1176 
1177 		if ((flags & HD_OMIT_COUNT) == 0)
1178 			printf("%04x  ", i);
1179 
1180 		if ((flags & HD_OMIT_HEX) == 0) {
1181 			for (j = 0; j < cols; j++) {
1182 				k = i + j;
1183 				if (k < length)
1184 					printf("%c%02x", delim, cp[k]);
1185 				else
1186 					printf("   ");
1187 			}
1188 		}
1189 
1190 		if ((flags & HD_OMIT_CHARS) == 0) {
1191 			printf("  |");
1192 			for (j = 0; j < cols; j++) {
1193 				k = i + j;
1194 				if (k >= length)
1195 					printf(" ");
1196 				else if (cp[k] >= ' ' && cp[k] <= '~')
1197 					printf("%c", cp[k]);
1198 				else
1199 					printf(".");
1200 			}
1201 			printf("|");
1202 		}
1203 		printf("\n");
1204 	}
1205 }
1206 #endif /* _KERNEL */
1207 
1208 void
1209 sbuf_hexdump(struct sbuf *sb, const void *ptr, int length, const char *hdr,
1210 	     int flags)
1211 {
1212 	int i, j, k;
1213 	int cols;
1214 	const unsigned char *cp;
1215 	char delim;
1216 
1217 	if ((flags & HD_DELIM_MASK) != 0)
1218 		delim = (flags & HD_DELIM_MASK) >> 8;
1219 	else
1220 		delim = ' ';
1221 
1222 	if ((flags & HD_COLUMN_MASK) != 0)
1223 		cols = flags & HD_COLUMN_MASK;
1224 	else
1225 		cols = 16;
1226 
1227 	cp = ptr;
1228 	for (i = 0; i < length; i+= cols) {
1229 		if (hdr != NULL)
1230 			sbuf_printf(sb, "%s", hdr);
1231 
1232 		if ((flags & HD_OMIT_COUNT) == 0)
1233 			sbuf_printf(sb, "%04x  ", i);
1234 
1235 		if ((flags & HD_OMIT_HEX) == 0) {
1236 			for (j = 0; j < cols; j++) {
1237 				k = i + j;
1238 				if (k < length)
1239 					sbuf_printf(sb, "%c%02x", delim, cp[k]);
1240 				else
1241 					sbuf_cat(sb, "   ");
1242 			}
1243 		}
1244 
1245 		if ((flags & HD_OMIT_CHARS) == 0) {
1246 			sbuf_cat(sb, "  |");
1247 			for (j = 0; j < cols; j++) {
1248 				k = i + j;
1249 				if (k >= length)
1250 					sbuf_putc(sb, ' ');
1251 				else if (cp[k] >= ' ' && cp[k] <= '~')
1252 					sbuf_putc(sb, cp[k]);
1253 				else
1254 					sbuf_putc(sb, '.');
1255 			}
1256 			sbuf_putc(sb, '|');
1257 		}
1258 		sbuf_putc(sb, '\n');
1259 	}
1260 }
1261 
1262 #ifdef _KERNEL
1263 void
1264 counted_warning(unsigned *counter, const char *msg)
1265 {
1266 	struct thread *td;
1267 	unsigned c;
1268 
1269 	for (;;) {
1270 		c = *counter;
1271 		if (c == 0)
1272 			break;
1273 		if (atomic_cmpset_int(counter, c, c - 1)) {
1274 			td = curthread;
1275 			log(LOG_INFO, "pid %d (%s) %s%s\n",
1276 			    td->td_proc->p_pid, td->td_name, msg,
1277 			    c > 1 ? "" : " - not logging anymore");
1278 			break;
1279 		}
1280 	}
1281 }
1282 #endif
1283 
1284 #ifdef _KERNEL
1285 void
1286 sbuf_putbuf(struct sbuf *sb)
1287 {
1288 
1289 	prf_putbuf(sbuf_data(sb), TOLOG | TOCONS, -1);
1290 }
1291 #else
1292 void
1293 sbuf_putbuf(struct sbuf *sb)
1294 {
1295 
1296 	printf("%s", sbuf_data(sb));
1297 }
1298 #endif
1299 
1300 int
1301 sbuf_printf_drain(void *arg, const char *data, int len)
1302 {
1303 	size_t *retvalptr;
1304 	int r;
1305 #ifdef _KERNEL
1306 	char *dataptr;
1307 	char oldchr;
1308 
1309 	/*
1310 	 * This is allowed as an extra byte is always resvered for
1311 	 * terminating NUL byte.  Save and restore the byte because
1312 	 * we might be flushing a record, and there may be valid
1313 	 * data after the buffer.
1314 	 */
1315 	oldchr = data[len];
1316 	dataptr = __DECONST(char *, data);
1317 	dataptr[len] = '\0';
1318 
1319 	prf_putbuf(dataptr, TOLOG | TOCONS, -1);
1320 	r = len;
1321 
1322 	dataptr[len] = oldchr;
1323 
1324 #else /* !_KERNEL */
1325 
1326 	r = printf("%.*s", len, data);
1327 	if (r < 0)
1328 		return (-errno);
1329 
1330 #endif
1331 
1332 	retvalptr = arg;
1333 	if (retvalptr != NULL)
1334 		*retvalptr += r;
1335 
1336 	return (r);
1337 }
1338