/*-
 * Copyright (c) 1986, 1988, 1991, 1993
 *	The Regents of the University of California.  All rights reserved.
 * (c) UNIX System Laboratories, Inc.
 * All or some portions of this file are derived from material licensed
 * to the University of California by American Telephone and Telegraph
 * Co. or Unix System Laboratories, Inc. and are reproduced herein with
 * the permission of UNIX System Laboratories, Inc.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 4. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 *	@(#)subr_prf.c	8.3 (Berkeley) 1/21/94
 */

/* __FBSDID("$FreeBSD: stable/8/sys/kern/subr_prf.c 210305 2010-07-20 18:55:13Z
 * jkim $"); */

#include <cdefs.h>
#include <cheri-builtins.h>
#include <compartment.h>
#include <cstdint>
#include <function_wrapper.hh>
#include <inttypes.h>
#include <platform-uart.hh>
#include <stdarg.h>
#include <stdio.h>
#include <string.h>

/*
 * Definitions snarfed from various parts of the FreeBSD headers:
 */
namespace
{
	__always_inline char toupper(char c)
	{
		return ((c)-0x20 * (((c) >= 'a') && ((c) <= 'z')));
	}

	static char hex2ascii(uintmax_t in)
	{
		return (in < 10) ? '0' + in : 'a' + in - 10;
	}

/* Max number conversion buffer length: a u_quad_t in base 2, plus NUL byte. */
#define MAXNBUF (sizeof(intmax_t) * CHAR_BIT + 1)

	/*
	 * Put a NUL-terminated ASCII number (base <= 36) in a buffer in reverse
	 * order; return an optional length and a pointer to the last character
	 * written in the buffer (i.e., the first character of the string).
	 * The buffer pointed to by `nbuf' must have length >= MAXNBUF.
	 */
	// FIXME: Using `unsigned` for `num` instead of `uintmax_t` means that we
	// are going to truncate large numbers, but it avoids needing a library
	// routine to handle division.
	static char *
	ksprintn(char *nbuf, unsigned num, int base, int *lenp, int upper)
	{
		char *p, c;

		p  = nbuf;
		*p = '\0';
		do
		{
			c    = hex2ascii(num % base);
			*++p = upper ? toupper(c) : c;
		} while (num /= base);
		if (lenp)
			*lenp = p - nbuf;
		return (p);
	}

	/*
	 * Scaled down version of printf(3).
	 *
	 * Two additional formats:
	 *
	 * The format %b is supported to decode error registers.
	 * Its usage is:
	 *
	 *	printf("reg=%b\n", regval, "<base><arg>*");
	 *
	 * where <base> is the output base expressed as a control character, e.g.
	 * \10 gives octal; \20 gives hex.  Each arg is a sequence of characters,
	 * the first of which gives the bit number to be inspected (origin 1), and
	 * the next characters (up to a control character, i.e. a character <= 32),
	 * give the name of the register.  Thus:
	 *
	 *	kvprintf("reg=%b\n", 3, "\10\2BITTWO\1BITONE\n");
	 *
	 * would produce output:
	 *
	 *	reg=3<BITTWO,BITONE>
	 *
	 * %D  -- Hexdump, takes pointer and separator string:
	 *		("%6D", ptr, ":")   -> XX:XX:XX:XX:XX:XX
	 *		("%*D", len, ptr, " " -> XX XX XX XX ...
	 */
	__noinline int kvprintf(char const                *fmt,
	                        FunctionWrapper<void(int)> func,
	                        void                      *arg,
	                        int                        radix,
	                        va_list                    ap)
	{
		char           nbuf[MAXNBUF];
		const char    *p, *percent, *q;
		unsigned char *up;
		int            ch, n;
		uintmax_t      num;
		int  base, lflag, qflag, tmp, width, ladjust, sharpflag, neg, sign, dot;
		int  cflag, hflag, jflag, tflag, zflag;
		int  dwidth, upper;
		char padc;
		int  stop = 0, retval = 0;
		const char *emptyString = "";

		auto putchar = [&](int c) {
			func(c);
			retval++;
		};

		if (fmt == nullptr)
		{
			fmt = emptyString;
		}

		if (radix < 2 || radix > 36)
		{
			radix = 10;
		}

		for (;;)
		{
			padc  = ' ';
			width = 0;
			while ((ch = static_cast<unsigned char>(*fmt++)) != '%' || stop)
			{
				if (ch == '\0')
					return (retval);
				putchar(ch);
			}
			percent   = fmt - 1;
			qflag     = 0;
			lflag     = 0;
			ladjust   = 0;
			sharpflag = 0;
			neg       = 0;
			sign      = 0;
			dot       = 0;
			dwidth    = 0;
			upper     = 0;
			cflag     = 0;
			hflag     = 0;
			jflag     = 0;
			tflag     = 0;
			zflag     = 0;
		reswitch:
			switch (ch = static_cast<unsigned char>(*fmt++))
			{
				case '.':
					dot = 1;
					goto reswitch; // NOLINT
				case '#':
					sharpflag = 1;
					goto reswitch; // NOLINT
				case '+':
					sign = 1;
					goto reswitch; // NOLINT
				case '-':
					ladjust = 1;
					goto reswitch; // NOLINT
				case '%':
					putchar(ch);
					break;
				case '*':
					if (!dot)
					{
						width = va_arg(ap, int);
						if (width < 0)
						{
							ladjust = !ladjust;
							width   = -width;
						}
					}
					else
					{
						dwidth = va_arg(ap, int);
					}
					goto reswitch; // NOLINT
				case '0':
					if (!dot)
					{
						padc = '0';
						goto reswitch; // NOLINT
					}
				case '1':
				case '2':
				case '3':
				case '4':
				case '5':
				case '6':
				case '7':
				case '8':
				case '9':
					for (n = 0;; ++fmt)
					{
						n  = n * 10 + ch - '0';
						ch = *fmt;
						if (ch < '0' || ch > '9')
							break;
					}
					if (dot)
						dwidth = n;
					else
						width = n;
					goto reswitch; // NOLINT
				case 'b':
					num = static_cast<unsigned int>(va_arg(ap, int));
					p   = va_arg(ap, char *);
					for (q = ksprintn(nbuf, num, *p++, nullptr, 0); *q;)
						putchar(*q--);

					if (num == 0)
						break;

					for (tmp = 0; *p;)
					{
						n = *p++;
						if (num & (1 << (n - 1)))
						{
							putchar(tmp ? ',' : '<');
							for (; (n = *p) > ' '; ++p)
								putchar(n);
							tmp = 1;
						}
						else
							for (; *p > ' '; ++p)
								continue;
					}
					if (tmp)
						putchar('>');
					break;
				case 'c':
					putchar(va_arg(ap, int));
					break;
				case 'D':
					up = va_arg(ap, unsigned char *);
					p  = va_arg(ap, char *);
					if (!width)
						width = 16;
					while (width--)
					{
						putchar(hex2ascii(*up >> 4));
						putchar(hex2ascii(*up & 0x0f));
						up++;
						if (width)
							for (q = p; *q; q++)
								putchar(*q);
					}
					break;
				case 'd':
				case 'i':
					base = 10;
					sign = 1;
					goto handle_sign; // NOLINT
				case 'h':
					if (hflag)
					{
						hflag = 0;
						cflag = 1;
					}
					else
						hflag = 1;
					goto reswitch; // NOLINT
				case 'j':
					jflag = 1;
					goto reswitch; // NOLINT
				case 'l':
					if (lflag)
					{
						lflag = 0;
						qflag = 1;
					}
					else
						lflag = 1;
					goto reswitch; // NOLINT
				case 'n':
					if (jflag)
						*(va_arg(ap, intmax_t *)) = retval;
					else if (qflag)
						*(va_arg(ap, long long *)) = retval;
					else if (lflag)
						*(va_arg(ap, long *)) = retval;
					else if (zflag)
						*(va_arg(ap, size_t *)) = retval;
					else if (hflag)
						*(va_arg(ap, short *)) = static_cast<short>(retval);
					else if (cflag)
						*(va_arg(ap, char *)) = retval;
					else
						*(va_arg(ap, int *)) = retval;
					break;
				case 'o':
					base = 8;
					goto handle_nosign; // NOLINT
				case 'p':
					base      = 16;
					sharpflag = (width == 0);
					sign      = 0;
					num       = static_cast<size_t>(
                      reinterpret_cast<uintptr_t>(va_arg(ap, void *)));
					goto number; // NOLINT
				case 'q':
					qflag = 1;
					goto reswitch; // NOLINT
				case 'r':
					base = radix;
					if (sign)
						goto handle_sign; // NOLINT
					goto handle_nosign;   // NOLINT
				case 's':
					p = va_arg(ap, char *);
					if (p == nullptr)
					{
						p = emptyString;
					}
					if (!dot)
					{
						n = strlen(p);
					}
					else
					{
						for (n = 0; n < dwidth && p[n]; n++)
						{
							continue;
						}
					}

					width -= n;

					if (!ladjust && width > 0)
					{
						while (width--)
						{
							putchar(padc);
						}
					}
					while (n--)
					{
						putchar(*p++);
					}
					if (ladjust && width > 0)
					{
						while (width--)
						{
							putchar(padc);
						}
					}
					break;
				case 't':
					tflag = 1;
					goto reswitch; // NOLINT
				case 'u':
					base = 10;
					goto handle_nosign; // NOLINT
				case 'X':
					upper = 1;
				case 'x':
					base = 16;
					goto handle_nosign; // NOLINT
				case 'y':
					base = 16;
					sign = 1;
					goto handle_sign; // NOLINT
				case 'z':
					zflag = 1;
					goto reswitch; // NOLINT
				handle_nosign:
					sign = 0;
					if (jflag)
						num = va_arg(ap, uintmax_t);
					else if (qflag)
						num = va_arg(ap, unsigned long long);
					else if (tflag)
						num = va_arg(ap, ptrdiff_t);
					else if (lflag)
						num = va_arg(ap, unsigned long);
					else if (zflag)
						num = va_arg(ap, size_t);
					else if (hflag)
						num = static_cast<unsigned short>(va_arg(ap, int));
					else if (cflag)
						num = static_cast<unsigned char>(va_arg(ap, int));
					else
						num = va_arg(ap, unsigned int);
					goto number; // NOLINT
				handle_sign:
					if (jflag)
						num = va_arg(ap, intmax_t);
					else if (qflag)
						num = va_arg(ap, long long);
					else if (tflag)
						num = va_arg(ap, ptrdiff_t);
					else if (lflag)
						num = va_arg(ap, long);
					else if (zflag)
						num = va_arg(ap, ssize_t);
					else if (hflag)
						num = static_cast<short>(va_arg(ap, int));
					else if (cflag)
						num = static_cast<char>(va_arg(ap, int));
					else
						num = va_arg(ap, int);
				number:
					if (sign && static_cast<intmax_t>(num) < 0)
					{
						neg = 1;
						num = -static_cast<intmax_t>(num);
					}
					p   = ksprintn(nbuf, num, base, &n, upper);
					tmp = 0;
					if (sharpflag && num != 0)
					{
						if (base == 8)
							tmp++;
						else if (base == 16)
							tmp += 2;
					}
					if (neg)
						tmp++;

					if (!ladjust && padc == '0')
						dwidth = width - tmp;
					width -= tmp + (dwidth > n ? dwidth : n);
					dwidth -= n;
					if (!ladjust)
						while (width-- > 0)
							putchar(' ');
					if (neg)
						putchar('-');
					if (sharpflag && num != 0)
					{
						if (base == 8)
						{
							putchar('0');
						}
						else if (base == 16)
						{
							putchar('0');
							putchar('x');
						}
					}
					while (dwidth-- > 0)
						putchar('0');

					while (*p)
						putchar(*p--);

					if (ladjust)
						while (width-- > 0)
							putchar(' ');

					break;
				default:
					while (percent < fmt)
						putchar(*percent++);
					/*
					 * Since we ignore an formatting argument it is no
					 * longer safe to obey the remaining formatting
					 * arguments as the arguments will no longer match
					 * the format specs.
					 */
					stop = 1;
					break;
			}
		}
	}

} // namespace

/*
 * Scaled down version of vsnprintf(3).
 */
int __cheri_libcall
vsnprintf(char *str, // NOLINT (clang-tidy spuriously thinks this should be
                     // const, even though it's being written to)
          size_t      size,
          const char *format,
          va_list     ap)
{
	struct Buffer
	{
		char  *str;
		size_t remain;
	} info        = {str, size};
	auto callback = [&](int ch) {
		if (info.remain >= 2)
		{
			*info.str++ = ch;
			info.remain--;
		}
	};
	int retval = kvprintf(format, callback, &info, 10, ap);
	if (info.remain >= 1)
		*info.str++ = '\0';
	return (retval);
}

[[cheri::interrupt_state(disabled)]] int __cheri_libcall
vfprintf(FILE *stream, const char *fmt, va_list ap)
{
	return kvprintf(
	  fmt,
	  [=](int ch) { static_cast<volatile Uart *>(stream)->blocking_write(ch); },
	  nullptr,
	  10,
	  ap);
}

int __cheri_libcall snprintf(char *str, size_t size, const char *format, ...)
{
	int     rv;
	va_list ap;

	va_start(ap, format);
	rv = vsnprintf(str, size, format, ap);
	va_end(ap);

	return rv;
}