// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Copyright (c) 2008-2014 Travis Geiselbrecht

// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <limits.h>
#include <printf.h>
#include <stdarg.h>
#include <string.h>
#include <sys/types.h>
#include <xefi.h>

int sprintf(char *str, const char *fmt, ...) {
  int err;

  va_list ap;
  va_start(ap, fmt);
  err = vsprintf(str, fmt, ap);
  va_end(ap);

  return err;
}

int snprintf(char *str, size_t len, const char *fmt, ...) {
  int err;

  va_list ap;
  va_start(ap, fmt);
  err = vsnprintf(str, len, fmt, ap);
  va_end(ap);

  return err;
}

#define LONGFLAG 0x00000001
#define LONGLONGFLAG 0x00000002
#define HALFFLAG 0x00000004
#define HALFHALFFLAG 0x00000008
#define SIZETFLAG 0x00000010
#define INTMAXFLAG 0x00000020
#define PTRDIFFFLAG 0x00000040
#define ALTFLAG 0x00000080
#define CAPSFLAG 0x00000100
#define SHOWSIGNFLAG 0x00000200
#define SIGNEDFLAG 0x00000400
#define LEFTFORMATFLAG 0x00000800
#define LEADZEROFLAG 0x00001000
#define BLANKPOSFLAG 0x00002000

__NO_INLINE static char *longlong_to_string(char *buf, unsigned long long n, size_t len,
                                            unsigned int flag, char *signchar) {
  size_t pos = len;
  int negative = 0;

  if ((flag & SIGNEDFLAG) && (long long)n < 0) {
    negative = 1;
    n = -n;
  }

  buf[--pos] = 0;

  /* only do the math if the number is >= 10 */
  while (n >= 10) {
    unsigned long long digit = n % 10;

    n /= 10;

    buf[--pos] = (char)digit + '0';
  }
  buf[--pos] = (char)n + '0';

  if (negative)
    *signchar = '-';
  else if ((flag & SHOWSIGNFLAG))
    *signchar = '+';
  else if ((flag & BLANKPOSFLAG))
    *signchar = ' ';
  else
    *signchar = '\0';

  return &buf[pos];
}

static const char hextable[] = {'0', '1', '2', '3', '4', '5', '6', '7',
                                '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
static const char hextable_caps[] = {'0', '1', '2', '3', '4', '5', '6', '7',
                                     '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'};

__NO_INLINE static char *longlong_to_hexstring(char *buf, unsigned long long u, size_t len,
                                               unsigned int flag) {
  size_t pos = len;
  const char *table = (flag & CAPSFLAG) ? hextable_caps : hextable;

  buf[--pos] = 0;
  do {
    unsigned int digit = u % 16;
    u /= 16;

    buf[--pos] = table[digit];
  } while (u != 0);

  return &buf[pos];
}

int vsprintf(char *str, const char *fmt, va_list ap) { return vsnprintf(str, INT_MAX, fmt, ap); }

struct _output_args {
  char *outstr;
  size_t len;
  size_t pos;
};

static int _vsnprintf_output(const char *str, size_t len, void *state) {
  struct _output_args *args = state;

  size_t count = 0;
  while (count < len) {
    if (args->pos < args->len) {
      args->outstr[args->pos++] = *str;
    }

    str++;
    count++;
  }

  return (int)count;
}

int vsnprintf(char *str, size_t len, const char *fmt, va_list ap) {
  struct _output_args args;
  int wlen;

  args.outstr = str;
  args.len = len;
  args.pos = 0;

  wlen = _printf_engine(&_vsnprintf_output, (void *)&args, fmt, ap);
  if (args.pos >= len)
    str[len - 1] = '\0';
  else
    str[wlen] = '\0';
  return wlen;
}

int _printf_engine(_printf_engine_output_func out, void *state, const char *fmt, va_list ap) {
  int err = 0;
  char c;
  unsigned char uc;
  const char *s;
  size_t string_len;
  unsigned long long n;
  void *ptr;
  int flags;
  unsigned int format_num;
  char signchar;
  size_t chars_written = 0;
  char num_buffer[32];

#define OUTPUT_STRING(str, len) \
  do {                          \
    err = out(str, len, state); \
    if (err < 0) {              \
      goto exit;                \
    } else {                    \
      chars_written += err;     \
    }                           \
  } while (0)
#define OUTPUT_CHAR(c)        \
  do {                        \
    char __temp[1] = {c};     \
    OUTPUT_STRING(__temp, 1); \
  } while (0)

  for (;;) {
    /* reset the format state */
    flags = 0;
    format_num = 0;
    signchar = '\0';

    /* handle regular chars that aren't format related */
    s = fmt;
    string_len = 0;
    while ((c = *fmt++) != 0) {
      if (c == '%')
        break; /* we saw a '%', break and start parsing format */
      string_len++;
    }

    /* output the string we've accumulated */
    OUTPUT_STRING(s, string_len);

    /* make sure we haven't just hit the end of the string */
    if (c == 0)
      break;

  next_format:
    /* grab the next format character */
    c = *fmt++;
    if (c == 0)
      break;

    switch (c) {
      case '0' ... '9':
        if (c == '0' && format_num == 0)
          flags |= LEADZEROFLAG;
        format_num *= 10;
        format_num += c - '0';
        goto next_format;
      case '.':
        /* XXX for now eat numeric formatting */
        goto next_format;
      case '%':
        OUTPUT_CHAR('%');
        break;
      case 'c':
        uc = (unsigned char)va_arg(ap, unsigned int);
        OUTPUT_CHAR(uc);
        break;
      case 's':
        s = va_arg(ap, const char *);
        if (s == 0)
          s = "<null>";
        flags &= ~LEADZEROFLAG; /* doesn't make sense for strings */
        goto _output_string;
      case '-':
        flags |= LEFTFORMATFLAG;
        goto next_format;
      case '+':
        flags |= SHOWSIGNFLAG;
        goto next_format;
      case ' ':
        flags |= BLANKPOSFLAG;
        goto next_format;
      case '#':
        flags |= ALTFLAG;
        goto next_format;
      case 'l':
        if (flags & LONGFLAG)
          flags |= LONGLONGFLAG;
        flags |= LONGFLAG;
        goto next_format;
      case 'h':
        if (flags & HALFFLAG)
          flags |= HALFHALFFLAG;
        flags |= HALFFLAG;
        goto next_format;
      case 'z':
        flags |= SIZETFLAG;
        goto next_format;
      case 'j':
        flags |= INTMAXFLAG;
        goto next_format;
      case 't':
        flags |= PTRDIFFFLAG;
        goto next_format;
      case 'i':
      case 'd':
        n = (flags & LONGLONGFLAG)   ? va_arg(ap, long long)
            : (flags & LONGFLAG)     ? va_arg(ap, long)
            : (flags & HALFHALFFLAG) ? (signed char)va_arg(ap, int)
            : (flags & HALFFLAG)     ? (short)va_arg(ap, int)
            : (flags & SIZETFLAG)    ? va_arg(ap, ssize_t)
            : (flags & INTMAXFLAG)   ? va_arg(ap, intmax_t)
            : (flags & PTRDIFFFLAG)  ? va_arg(ap, ptrdiff_t)
                                     : va_arg(ap, int);
        flags |= SIGNEDFLAG;
        s = longlong_to_string(num_buffer, n, sizeof(num_buffer), flags, &signchar);
        goto _output_string;
      case 'u':
        n = (flags & LONGLONGFLAG)   ? va_arg(ap, unsigned long long)
            : (flags & LONGFLAG)     ? va_arg(ap, unsigned long)
            : (flags & HALFHALFFLAG) ? (unsigned char)va_arg(ap, unsigned int)
            : (flags & HALFFLAG)     ? (unsigned short)va_arg(ap, unsigned int)
            : (flags & SIZETFLAG)    ? va_arg(ap, size_t)
            : (flags & INTMAXFLAG)   ? va_arg(ap, uintmax_t)
            : (flags & PTRDIFFFLAG)  ? (uintptr_t)va_arg(ap, ptrdiff_t)
                                     : va_arg(ap, unsigned int);
        s = longlong_to_string(num_buffer, n, sizeof(num_buffer), flags, &signchar);
        goto _output_string;
      case 'p':
        flags |= LONGFLAG | ALTFLAG;
        goto hex;
      case 'X':
        flags |= CAPSFLAG;
        /* fallthrough */
      hex:
      case 'x':
        n = (flags & LONGLONGFLAG)   ? va_arg(ap, unsigned long long)
            : (flags & LONGFLAG)     ? va_arg(ap, unsigned long)
            : (flags & HALFHALFFLAG) ? (unsigned char)va_arg(ap, unsigned int)
            : (flags & HALFFLAG)     ? (unsigned short)va_arg(ap, unsigned int)
            : (flags & SIZETFLAG)    ? va_arg(ap, size_t)
            : (flags & INTMAXFLAG)   ? va_arg(ap, uintmax_t)
            : (flags & PTRDIFFFLAG)  ? (uintptr_t)va_arg(ap, ptrdiff_t)
                                     : va_arg(ap, unsigned int);
        s = longlong_to_hexstring(num_buffer, n, sizeof(num_buffer), flags);
        if (flags & ALTFLAG) {
          OUTPUT_CHAR('0');
          OUTPUT_CHAR((flags & CAPSFLAG) ? 'X' : 'x');
        }
        goto _output_string;
      case 'n':
        ptr = va_arg(ap, void *);
        if (flags & LONGLONGFLAG)
          *(long long *)ptr = (long long)chars_written;
        else if (flags & LONGFLAG)
          *(long *)ptr = (long)chars_written;
        else if (flags & HALFHALFFLAG)
          *(signed char *)ptr = (signed char)chars_written;
        else if (flags & HALFFLAG)
          *(short *)ptr = (short)chars_written;
        else if (flags & SIZETFLAG)
          *(size_t *)ptr = chars_written;
        else
          *(int *)ptr = (int)chars_written;
        break;
      default:
        OUTPUT_CHAR('%');
        OUTPUT_CHAR(c);
        break;
    }

    /* move on to the next field */
    continue;

    /* shared output code */
  _output_string:
    string_len = strlen(s);

    if (flags & LEFTFORMATFLAG) {
      /* left justify the text */
      OUTPUT_STRING(s, string_len);
      unsigned int written = err;

      /* pad to the right (if necessary) */
      for (; format_num > written; format_num--)
        OUTPUT_CHAR(' ');
    } else {
      /* right justify the text (digits) */

      /* if we're going to print a sign digit,
         it'll chew up one byte of the format size */
      if (signchar != '\0' && format_num > 0)
        format_num--;

      /* output the sign char before the leading zeros */
      if (flags & LEADZEROFLAG && signchar != '\0')
        OUTPUT_CHAR(signchar);

      /* pad according to the format string */
      for (; format_num > string_len; format_num--)
        OUTPUT_CHAR(flags & LEADZEROFLAG ? '0' : ' ');

      /* if not leading zeros, output the sign char just before the number */
      if (!(flags & LEADZEROFLAG) && signchar != '\0')
        OUTPUT_CHAR(signchar);

      /* output the string */
      OUTPUT_STRING(s, string_len);
    }
    continue;
  }

#undef OUTPUT_STRING
#undef OUTPUT_CHAR

exit:
  return (err < 0) ? err : (int)chars_written;
}

int write_to_serial(char16_t *buffer, uint64_t len) {
  if (gSerial == NULL) {
    return 0;
  }
  len *= sizeof(char16_t);
  return gSerial->Write(gSerial, &len, buffer) == EFI_SUCCESS ? (int)len : -1;
}

int puts16(char16_t *str) {
  int r1 = write_to_serial(str, strlen_16(str));
  int r2 = (gConOut->OutputString(gConOut, str) == EFI_SUCCESS ? 0 : -1);
  return (r1 < 0 || r2 < 0) ? -1 : 0;
}