#include <sys/param.h>
#include <sys/systm.h>
#include <sys/reboot.h>
#include <sys/msgbuf.h>
#include <sys/proc.h>
#include <sys/tty.h>
#include <sys/tprintf.h>
#include <sys/syslog.h>
#include <sys/pool.h>
#include <sys/mutex.h>
#include <dev/cons.h>
#include <sys/stdarg.h>
#ifdef DDB
#include <ddb/db_output.h>
#include <ddb/db_var.h>
#endif
#define TOCONS 0x01
#define TOTTY 0x02
#define TOLOG 0x04
#define TOBUFONLY 0x08
#define TODDB 0x10
#define TOCOUNT 0x20
#define KPRINTF_BUFSIZE (sizeof(quad_t) * NBBY / 3 + 2)
int kprintf(const char *, int, void *, char *, va_list);
void kputchar(int, int, struct tty *);
struct mutex kprintf_mutex =
MUTEX_INITIALIZER_FLAGS(IPL_HIGH, "kprintf", MTX_NOWITNESS);
extern int log_open;
const char *panicstr;
#ifdef DDB
int db_panic = 1;
#ifdef DDB_SAFE_CONSOLE
int db_console = 1;
#else
int db_console = 0;
#endif
#endif
#ifdef SPLASSERT_WATCH
int splassert_ctl = 3;
#else
int splassert_ctl = 1;
#endif
void (*v_putc)(int) = cnputc;
#ifdef BOOT_QUIET
int printf_flags = TOLOG;
#else
int printf_flags = TOCONS | TOLOG;
#endif
void
__assert(const char *t, const char *f, int l, const char *e)
{
panic(__KASSERTSTR, t, e, f, l);
}
void
tablefull(const char *tab)
{
log(LOG_ERR, "%s: table is full\n", tab);
}
#ifdef DDB
#define panic_printf(...) db_printf(__VA_ARGS__)
#define panic_vprintf(...) db_vprintf(__VA_ARGS__)
#else
#define panic_printf(...) printf(__VA_ARGS__)
#define panic_vprintf(...) vprintf(__VA_ARGS__)
#endif
void
panic(const char *fmt, ...)
{
struct cpu_info *ci = curcpu();
int bootopt;
va_list ap;
bootopt = RB_AUTOBOOT | RB_DUMP;
if (atomic_cas_ptr(&panicstr, NULL, ci->ci_panicbuf) != NULL)
bootopt |= RB_NOSYNC;
splassert_ctl = 0;
#ifdef BOOT_QUIET
printf_flags |= TOCONS;
#endif
if (ci->ci_panicbuf[0] == '\0') {
va_start(ap, fmt);
vsnprintf(ci->ci_panicbuf, sizeof(ci->ci_panicbuf), fmt, ap);
va_end(ap);
panic_printf("panic: %s\n", ci->ci_panicbuf);
} else {
panic_printf("panic: ");
va_start(ap, fmt);
panic_vprintf(fmt, ap);
va_end(ap);
panic_printf("\n");
}
#ifdef DDB
if (db_panic)
db_enter();
else
db_stack_dump();
#endif
reboot(bootopt);
}
void
splassert_fail(int wantipl, int haveipl, const char *func)
{
if (panicstr || db_active)
return;
printf("splassert: %s: want %d have %d\n", func, wantipl, haveipl);
switch (splassert_ctl) {
case 1:
break;
case 2:
#ifdef DDB
db_stack_dump();
#endif
break;
case 3:
#ifdef DDB
db_stack_dump();
db_enter();
#endif
break;
default:
panic("spl assertion failure in %s", func);
}
}
void
log(int level, const char *fmt, ...)
{
int s;
va_list ap;
s = splhigh();
logpri(level);
va_start(ap, fmt);
kprintf(fmt, TOLOG, NULL, NULL, ap);
va_end(ap);
splx(s);
if (!log_open) {
va_start(ap, fmt);
mtx_enter(&kprintf_mutex);
kprintf(fmt, TOCONS, NULL, NULL, ap);
mtx_leave(&kprintf_mutex);
va_end(ap);
}
logwakeup();
}
void
logpri(int level)
{
char *p;
char snbuf[KPRINTF_BUFSIZE];
kputchar('<', TOLOG, NULL);
snprintf(snbuf, sizeof snbuf, "%d", level);
for (p = snbuf ; *p ; p++)
kputchar(*p, TOLOG, NULL);
kputchar('>', TOLOG, NULL);
}
int
addlog(const char *fmt, ...)
{
int s;
va_list ap;
s = splhigh();
va_start(ap, fmt);
kprintf(fmt, TOLOG, NULL, NULL, ap);
va_end(ap);
splx(s);
if (!log_open) {
va_start(ap, fmt);
mtx_enter(&kprintf_mutex);
kprintf(fmt, TOCONS, NULL, NULL, ap);
mtx_leave(&kprintf_mutex);
va_end(ap);
}
logwakeup();
return(0);
}
void
kputchar(int c, int flags, struct tty *tp)
{
extern int msgbufmapped;
if (panicstr)
constty = NULL;
if ((flags & TOCONS) && tp == NULL && constty != NULL && !db_active) {
tp = constty;
flags |= TOTTY;
}
if ((flags & TOTTY) && tp && tputchar(c, tp) < 0 &&
(flags & TOCONS) && tp == constty)
constty = NULL;
if ((flags & TOLOG) &&
c != '\0' && c != '\r' && c != 0177 && msgbufmapped)
msgbuf_putchar(msgbufp, c);
if ((flags & TOCONS) && (constty == NULL || db_active) && c != '\0')
(*v_putc)(c);
#ifdef DDB
if (flags & TODDB)
db_putchar(c);
#endif
}
void
uprintf(const char *fmt, ...)
{
struct process *pr = curproc->p_p;
va_list ap;
if (pr->ps_flags & PS_CONTROLT && pr->ps_session->s_ttyvp) {
va_start(ap, fmt);
kprintf(fmt, TOTTY, pr->ps_session->s_ttyp, NULL, ap);
va_end(ap);
}
}
#if defined(NFSSERVER) || defined(NFSCLIENT)
tpr_t
tprintf_open(struct proc *p)
{
struct process *pr = p->p_p;
if (pr->ps_flags & PS_CONTROLT && pr->ps_session->s_ttyvp) {
SESSHOLD(pr->ps_session);
return ((tpr_t)pr->ps_session);
}
return ((tpr_t) NULL);
}
void
tprintf_close(tpr_t sess)
{
if (sess)
SESSRELE((struct session *) sess);
}
void
tprintf(tpr_t tpr, const char *fmt, ...)
{
struct session *sess = (struct session *)tpr;
struct tty *tp = NULL;
int flags = TOLOG;
va_list ap;
logpri(LOG_INFO);
if (sess && sess->s_ttyvp && ttycheckoutq(sess->s_ttyp, 0)) {
flags |= TOTTY;
tp = sess->s_ttyp;
}
va_start(ap, fmt);
kprintf(fmt, flags, tp, NULL, ap);
va_end(ap);
logwakeup();
}
#endif
void
ttyprintf(struct tty *tp, const char *fmt, ...)
{
va_list ap;
va_start(ap, fmt);
kprintf(fmt, TOTTY, tp, NULL, ap);
va_end(ap);
}
#ifdef DDB
int
db_printf(const char *fmt, ...)
{
va_list ap;
int retval;
va_start(ap, fmt);
retval = db_vprintf(fmt, ap);
va_end(ap);
return(retval);
}
int
db_vprintf(const char *fmt, va_list ap)
{
int flags;
flags = TODDB;
if (db_log)
flags |= TOLOG;
return (kprintf(fmt, flags, NULL, NULL, ap));
}
#endif
int
printf(const char *fmt, ...)
{
va_list ap;
int retval;
va_start(ap, fmt);
mtx_enter(&kprintf_mutex);
retval = kprintf(fmt, printf_flags, NULL, NULL, ap);
mtx_leave(&kprintf_mutex);
va_end(ap);
if (!panicstr)
logwakeup();
return(retval);
}
int
vprintf(const char *fmt, va_list ap)
{
int retval;
mtx_enter(&kprintf_mutex);
retval = kprintf(fmt, TOCONS | TOLOG, NULL, NULL, ap);
mtx_leave(&kprintf_mutex);
if (!panicstr)
logwakeup();
return (retval);
}
int
snprintf(char *buf, size_t size, const char *fmt, ...)
{
int retval;
va_list ap;
char *p;
p = buf;
if (size > 0)
p += size - 1;
va_start(ap, fmt);
retval = kprintf(fmt, TOBUFONLY | TOCOUNT, &p, buf, ap);
va_end(ap);
if (size > 0)
*p = '\0';
return(retval);
}
int
vsnprintf(char *buf, size_t size, const char *fmt, va_list ap)
{
int retval;
char *p;
p = buf + size - 1;
if (size < 1)
p = buf;
retval = kprintf(fmt, TOBUFONLY | TOCOUNT, &p, buf, ap);
if (size > 0)
*(p) = 0;
return(retval);
}
#define to_digit(c) ((c) - '0')
#define is_digit(c) ((unsigned)to_digit(c) <= 9)
#define to_char(n) ((n) + '0')
#define ALT 0x001
#define HEXPREFIX 0x002
#define LADJUST 0x004
#define LONGDBL 0x008
#define LONGINT 0x010
#define QUADINT 0x020
#define SHORTINT 0x040
#define ZEROPAD 0x080
#define FPT 0x100
#define SIZEINT 0x200
#define SARG() \
(flags&QUADINT ? va_arg(ap, quad_t) : \
flags&LONGINT ? va_arg(ap, long) : \
flags&SIZEINT ? va_arg(ap, ssize_t) : \
flags&SHORTINT ? (long)(short)va_arg(ap, int) : \
(long)va_arg(ap, int))
#define UARG() \
(flags&QUADINT ? va_arg(ap, u_quad_t) : \
flags&LONGINT ? va_arg(ap, u_long) : \
flags&SIZEINT ? va_arg(ap, size_t) : \
flags&SHORTINT ? (u_long)(u_short)va_arg(ap, int) : \
(u_long)va_arg(ap, u_int))
#define KPRINTF_PUTCHAR(C) do { \
int chr = (C); \
ret += 1; \
if (oflags & TOBUFONLY) { \
if ((vp != NULL) && (sbuf == tailp)) { \
if (!(oflags & TOCOUNT)) \
goto overflow; \
} else \
*sbuf++ = chr; \
} else { \
kputchar(chr, oflags, (struct tty *)vp); \
} \
} while(0)
int
kprintf(const char *fmt0, int oflags, void *vp, char *sbuf, va_list ap)
{
char *fmt;
int ch;
int n;
char *cp = NULL;
int flags;
int ret;
int width;
int prec;
char sign;
u_quad_t _uquad;
enum { OCT, DEC, HEX } base;
int dprec;
int realsz;
int size = 0;
char *xdigs = NULL;
char buf[KPRINTF_BUFSIZE];
char *tailp = NULL;
if (oflags & TOCONS)
MUTEX_ASSERT_LOCKED(&kprintf_mutex);
if ((oflags & TOBUFONLY) && (vp != NULL))
tailp = *(char **)vp;
fmt = (char *)fmt0;
ret = 0;
for (;;) {
while (*fmt != '%' && *fmt) {
KPRINTF_PUTCHAR(*fmt++);
}
if (*fmt == 0)
goto done;
fmt++;
flags = 0;
dprec = 0;
width = 0;
prec = -1;
sign = '\0';
rflag: ch = *fmt++;
reswitch: switch (ch) {
case 'b': {
char *b, *z;
int tmp;
_uquad = UARG();
b = va_arg(ap, char *);
if (*b == 8)
snprintf(buf, sizeof buf, "%llo", _uquad);
else if (*b == 10)
snprintf(buf, sizeof buf, "%lld", _uquad);
else if (*b == 16)
snprintf(buf, sizeof buf, "%llx", _uquad);
else
break;
b++;
z = buf;
while (*z) {
KPRINTF_PUTCHAR(*z++);
}
if (_uquad) {
tmp = 0;
while ((n = *b++) != 0) {
if (n & 0x80)
n &= 0x7f;
else if (n <= ' ')
n = n - 1;
if (_uquad & (1LL << n)) {
KPRINTF_PUTCHAR(tmp ? ',':'<');
while (*b > ' ' &&
(*b & 0x80) == 0) {
KPRINTF_PUTCHAR(*b);
b++;
}
tmp = 1;
} else {
while (*b > ' ' &&
(*b & 0x80) == 0)
b++;
}
}
if (tmp) {
KPRINTF_PUTCHAR('>');
}
}
continue;
}
case ' ':
if (!sign)
sign = ' ';
goto rflag;
case '#':
flags |= ALT;
goto rflag;
case '*':
if ((width = va_arg(ap, int)) >= 0)
goto rflag;
width = -width;
case '-':
flags |= LADJUST;
goto rflag;
case '+':
sign = '+';
goto rflag;
case '.':
if ((ch = *fmt++) == '*') {
n = va_arg(ap, int);
prec = n < 0 ? -1 : n;
goto rflag;
}
n = 0;
while (is_digit(ch)) {
n = 10 * n + to_digit(ch);
ch = *fmt++;
}
prec = n < 0 ? -1 : n;
goto reswitch;
case '0':
flags |= ZEROPAD;
goto rflag;
case '1': case '2': case '3': case '4':
case '5': case '6': case '7': case '8': case '9':
n = 0;
do {
n = 10 * n + to_digit(ch);
ch = *fmt++;
} while (is_digit(ch));
width = n;
goto reswitch;
case 'h':
flags |= SHORTINT;
goto rflag;
case 'l':
if (*fmt == 'l') {
fmt++;
flags |= QUADINT;
} else {
flags |= LONGINT;
}
goto rflag;
case 'q':
flags |= QUADINT;
goto rflag;
case 'z':
flags |= SIZEINT;
goto rflag;
case 'c':
*(cp = buf) = va_arg(ap, int);
size = 1;
sign = '\0';
break;
case 't':
case 'D':
flags |= LONGINT;
case 'd':
case 'i':
_uquad = SARG();
if ((quad_t)_uquad < 0) {
_uquad = -_uquad;
sign = '-';
}
base = DEC;
goto number;
case 'n':
panic("no %%n support");
break;
case 'O':
flags |= LONGINT;
case 'o':
_uquad = UARG();
base = OCT;
goto nosign;
case 'p':
_uquad = (u_long)va_arg(ap, void *);
base = HEX;
xdigs = "0123456789abcdef";
flags |= HEXPREFIX;
ch = 'x';
goto nosign;
case 's':
if ((cp = va_arg(ap, char *)) == NULL)
cp = "(null)";
if (prec >= 0) {
char *p = memchr(cp, 0, prec);
if (p != NULL) {
size = p - cp;
if (size > prec)
size = prec;
} else
size = prec;
} else
size = strlen(cp);
sign = '\0';
break;
case 'U':
flags |= LONGINT;
case 'u':
_uquad = UARG();
base = DEC;
goto nosign;
case 'X':
xdigs = "0123456789ABCDEF";
goto hex;
case 'x':
xdigs = "0123456789abcdef";
hex: _uquad = UARG();
base = HEX;
if (flags & ALT && _uquad != 0)
flags |= HEXPREFIX;
nosign: sign = '\0';
number: if ((dprec = prec) >= 0)
flags &= ~ZEROPAD;
cp = buf + KPRINTF_BUFSIZE;
if (_uquad != 0 || prec != 0) {
switch (base) {
case OCT:
do {
*--cp = to_char(_uquad & 7);
_uquad >>= 3;
} while (_uquad);
if (flags & ALT && *cp != '0')
*--cp = '0';
break;
case DEC:
while (_uquad >= 10) {
*--cp = to_char(_uquad % 10);
_uquad /= 10;
}
*--cp = to_char(_uquad);
break;
case HEX:
do {
*--cp = xdigs[_uquad & 15];
_uquad >>= 4;
} while (_uquad);
break;
default:
cp = "bug in kprintf: bad base";
size = strlen(cp);
goto skipsize;
}
}
size = buf + KPRINTF_BUFSIZE - cp;
skipsize:
break;
default:
if (ch == '\0')
goto done;
cp = buf;
*cp = ch;
size = 1;
sign = '\0';
break;
}
realsz = dprec > size ? dprec : size;
if (sign)
realsz++;
else if (flags & HEXPREFIX)
realsz+= 2;
if ((flags & (LADJUST|ZEROPAD)) == 0) {
n = width - realsz;
while (n-- > 0)
KPRINTF_PUTCHAR(' ');
}
if (sign) {
KPRINTF_PUTCHAR(sign);
} else if (flags & HEXPREFIX) {
KPRINTF_PUTCHAR('0');
KPRINTF_PUTCHAR(ch);
}
if ((flags & (LADJUST|ZEROPAD)) == ZEROPAD) {
n = width - realsz;
while (n-- > 0)
KPRINTF_PUTCHAR('0');
}
n = dprec - size;
while (n-- > 0)
KPRINTF_PUTCHAR('0');
while (size--)
KPRINTF_PUTCHAR(*cp++);
if (flags & LADJUST) {
n = width - realsz;
while (n-- > 0)
KPRINTF_PUTCHAR(' ');
}
}
done:
if ((oflags & TOBUFONLY) && (vp != NULL))
*(char **)vp = sbuf;
overflow:
return (ret);
}
#if __GNUC_PREREQ__(2,96)
int puts(const char *);
int putchar(int c);
int
puts(const char *str)
{
printf("%s\n", str);
return (0);
}
int
putchar(int c)
{
printf("%c", c);
return (c);
}
#endif