src/console/vsprintf.c

   1 /* SPDX-License-Identifier: GPL-2.0-only */
   2
   3 #include <console/vtxprintf.h>
   4 #include <stdarg.h>
   5 #include <string.h>
   6
   7 struct vsnprintf_context {
   8         char *str_buf;
   9         size_t buf_limit;
  10 };
  11
  12 static void str_tx_byte(unsigned char byte, void *data)
  13 {
  14         struct vsnprintf_context *ctx = data;
  15         if (ctx->buf_limit) {
  16                 *ctx->str_buf = byte;
  17                 ctx->str_buf++;
  18                 ctx->buf_limit--;
  19         }
  20 }
  21
  22 int vsnprintf(char *buf, size_t size, const char *fmt, va_list args)
  23 {
  24         int i;
  25         struct vsnprintf_context ctx;
  26
  27         ctx.str_buf = buf;
  28         ctx.buf_limit = size ? size - 1 : 0;
  29         i = vtxprintf(str_tx_byte, fmt, args, &ctx);
  30         if (size)
  31                 *ctx.str_buf = '\0';
  32
  33         return i;
  34 }
  35
  36 int snprintf(char *buf, size_t size, const char *fmt, ...)
  37 {
  38         va_list args;
  39         int i;
  40
  41         va_start(args, fmt);
  42         i = vsnprintf(buf, size, fmt, args);
  43         va_end(args);
  44
  45         return i;
  46 }