nsz Git - musl/blob - src/stdio/vsnprintf.c

   1 #include "stdio_impl.h"
   2
   3 static size_t sn_write(FILE *f, const unsigned char *s, size_t l)
   4 {
   5         size_t k = f->wend - f->wpos;
   6         if (k > l) k = l;
   7         memcpy(f->wpos, s, k);
   8         f->wpos += k;
   9         /* pretend to succeed, but discard extra data */
  10         return l;
  11 }
  12
  13 int vsnprintf(char *restrict s, size_t n, const char *restrict fmt, va_list ap)
  14 {
  15         int r;
  16         char b;
  17         FILE f = { .lbf = EOF, .write = sn_write, .lock = -1 };
  18
  19         if (n-1 > INT_MAX-1) {
  20                 if (n) {
  21                         errno = EOVERFLOW;
  22                         return -1;
  23                 }
  24                 s = &b;
  25                 n = 1;
  26         }
  27
  28         /* Ensure pointers don't wrap if "infinite" n is passed in */
  29         if (n > (char *)0+SIZE_MAX-s-1) n = (char *)0+SIZE_MAX-s-1;
  30         f.buf_size = n;
  31         f.buf = f.wpos = (void *)s;
  32         f.wbase = f.wend = (void *)(s+n);
  33         r = vfprintf(&f, fmt, ap);
  34
  35         /* Null-terminate, overwriting last char if dest buffer is full */
  36         if (n) f.wpos[-(f.wpos == f.wend)] = 0;
  37         return r;
  38 }