/*
 * Copyright (C) Siddharth Bharat Purohit 2017
 * This file is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This file is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
/*
  wrappers for stdio functions

  Relies on linker wrap options

  Note that not all functions that have been wrapped are implemented
  here. The others are wrapped to ensure the function is not used
  without an implementation. If we need them then we can implement as
  needed.
 */
#include <posix.h>
#include <string.h>
#include <hal.h>
#include <memstreams.h>
#include <chprintf.h>
#include <ctype.h>
#include "stdio.h"

int vsnprintf(char *str, size_t size, const char *fmt, va_list ap)
{
  MemoryStream ms;
  BaseSequentialStream *chp;
  size_t size_wo_nul;
  int retval;

  if (size > 0)
    size_wo_nul = size - 1;
  else
    size_wo_nul = 0;

  /* Memory stream object to be used as a string writer, reserving one
     byte for the final zero.*/
  msObjectInit(&ms, (uint8_t *)str, size_wo_nul, 0);

  /* Performing the print operation using the common code.*/
  chp = (BaseSequentialStream *)(void *)&ms;

  retval = chvprintf(chp, fmt, ap);


  /* Terminate with a zero, unless size==0.*/
  if (ms.eos < size)
      str[ms.eos] = 0;

  /* Return number of bytes that would have been written.*/
  return retval;
}

int snprintf(char *str, size_t size, const char *fmt, ...)
{
   va_list arg;
   int done;
 
   va_start (arg, fmt);
   done =  vsnprintf(str, size, fmt, arg);
   va_end (arg);
 
   return done;
}

int vasprintf(char **strp, const char *fmt, va_list ap)
{
    int len = vsnprintf(NULL, 0, fmt, ap);
    if (len <= 0) {
        return -1;
    }
    char *buf = calloc(len+1, 1);
    if (!buf) {
        return -1;
    }
    vsnprintf(buf, len+1, fmt, ap);
    (*strp) = buf;
    return len;
}

int asprintf(char **strp, const char *fmt, ...)
{
    va_list ap;
    va_start(ap, fmt);
    int ret = vasprintf(strp, fmt, ap);
    va_end(ap);
    return ret;
}

int vprintf(const char *fmt, va_list arg)
{
#ifdef HAL_STDOUT_SERIAL
  return chvprintf ((BaseSequentialStream*)&HAL_STDOUT_SERIAL, fmt, arg);
#else
  (void)arg;
  return strlen(fmt);
#endif
}

int printf(const char *fmt, ...)
{
   va_list arg;
   int done;
 
   va_start (arg, fmt);
   done =  vprintf(fmt, arg);
   va_end (arg);
 
   return done;
}

//just a stub
int 
scanf (const char *fmt, ...)
{
    (void)fmt;
    return 0;
}
/*
 *  sscanf(buf,fmt,va_alist)
 */
int 
sscanf (const char *buf, const char *fmt, ...)
{
    int             count;
    va_list ap;
    
    va_start (ap, fmt);
    count = vsscanf (buf, fmt, ap);
    va_end (ap);
    return (count);
}

static char *
_getbase(char *p, int *basep)
{
  if (p[0] == '0') {
    switch (p[1]) {
    case 'x':
      *basep = 16;
      break;
    case 't': case 'n':
      *basep = 10;
      break;
    case 'o':
      *basep = 8;
      break;
    default:
      *basep = 10;
      return (p);
    }
    return (p + 2);
  }
  *basep = 10;
  return (p);
}

static int16_t
_atob (uint32_t *vp, char *p, int base)
{
  uint32_t value, v1, v2;
  char *q, tmp[20];
  int digit;

  if (p[0] == '0' && (p[1] == 'x' || p[1] == 'X')) {
    base = 16;
    p += 2;
  }

  if (base == 16 && (q = strchr (p, '.')) != 0) {
      if ((unsigned)(q - p) > (unsigned)(sizeof(tmp) - 1))
      return (0);

    strncpy (tmp, p, q - p);
    tmp[q - p] = '\0';
    if (!_atob (&v1, tmp, 16))
      return (0);

    q++;
    if (strchr (q, '.'))
      return (0);

    if (!_atob (&v2, q, 16))
      return (0);
    *vp = (v1 << 16) + v2;
    return (1);
  }

  value = *vp = 0;
  for (; *p; p++) {
    if (*p >= '0' && *p <= '9')
      digit = *p - '0';
    else if (*p >= 'a' && *p <= 'f')
      digit = *p - 'a' + 10;
    else if (*p >= 'A' && *p <= 'F')
      digit = *p - 'A' + 10;
    else
      return (0);

    if (digit >= base)
      return (0);
    value *= base;
    value += digit;
  }
  *vp = value;
  return (1);
}

/*
 *  atob(vp,p,base) 
 *      converts p to binary result in vp, rtn 1 on success
 */
int16_t
atob(uint32_t *vp, char *p, int base)
{
  uint32_t  v;

  if (base == 0)
    p = _getbase (p, &base);
  if (_atob (&v, p, base)) {
    *vp = v;
    return (1);
  }
  return (0);
}


#if defined(HAL_OS_FATFS_IO) && HAL_OS_FATFS_IO
/*
 *  vsscanf(buf,fmt,ap)
 */
int
vsscanf (const char *buf, const char *s, va_list ap)
{
    int             count, noassign, base=0, lflag;
    uint32_t width;
    const char     *tc;
    char           *t, tmp[MAXLN];

    count = noassign = width = lflag = 0;
    while (*s && *buf) {
  while (isspace ((unsigned char)(*s)))
      s++;
  if (*s == '%') {
      s++;
      for (; *s; s++) {
    if (strchr ("dibouxcsefg%", *s))
        break;
    if (*s == '*')
        noassign = 1;
    else if (*s == 'l' || *s == 'L')
        lflag = 1;
    else if (*s >= '1' && *s <= '9') {
        for (tc = s; isdigit ((unsigned)(*s)); s++);
        strncpy (tmp, tc, s - tc);
        tmp[s - tc] = '\0';
        atob (&width, tmp, 10);
        s--;
    }
      }
      if (*s == 's') {
    while (isspace ((unsigned char)(*buf)))
        buf++;
    if (!width)
        width = strcspn (buf, ISSPACE);
    if (!noassign) {
        strncpy (t = va_arg (ap, char *), buf, width);
        t[width] = '\0';
    }
    buf += width;
      } else if (*s == 'c') {
    if (!width)
        width = 1;
    if (!noassign) {
        strncpy (t = va_arg (ap, char *), buf, width);
        t[width] = '\0';
    }
    buf += width;
      } else if (strchr ("dobxu", *s)) {
    while (isspace ((unsigned char)(*buf)))
        buf++;
    if (*s == 'd' || *s == 'u')
        base = 10;
    else if (*s == 'x')
        base = 16;
    else if (*s == 'o')
        base = 8;
    else if (*s == 'b')
        base = 2;
    if (!width) {
        if (isspace ((unsigned char)(*(s + 1))) || *(s + 1) == 0)
      width = strcspn (buf, ISSPACE);
        else
      width = strchr (buf, *(s + 1)) - buf;
    }
    strncpy (tmp, buf, width);
    tmp[width] = '\0';
    buf += width;
    if (!noassign)
        atob (va_arg (ap, uint32_t *), tmp, base);
      }
      if (!noassign)
    count++;
      width = noassign = lflag = 0;
      s++;
  } else {
      while (isspace ((unsigned char)(*buf)))
    buf++;
      if (*s != *buf)
    break;
      else
    s++, buf++;
  }
    }
    return (count);
}

static int vfscanf(FILE *stream, const char *fmt, va_list ap);

/*
 *  fscanf(stream,fmt,va_alist)
 */
int fscanf (FILE *stream, const char *fmt, ...)
{
    int             count;
    va_list ap;

    va_start (ap, fmt);
    count = vfscanf (stream, fmt, ap);
    va_end (ap);
    return (count);
}

/*
 *  vfscanf(stream,fmt,ap) 
 */
static int vfscanf (FILE *stream, const char *fmt, va_list ap)
{
    int             count;
    char            buf[MAXLN + 1];

    if (fgets (buf, MAXLN, stream) == 0) {
	    return (-1);
    }
    count = vsscanf (buf, fmt, ap);
    return (count);
}
#endif // HAL_OS_FATFS_IO