lib/vsnprintf.c

   1 /* Formatted output to strings.
   2    Copyright (C) 2004, 2006 Free Software Foundation, Inc.
   3    Written by Simon Josefsson and Yoann Vandoorselaere <yoann@prelude-ids.org>.
   4
   5    This program is free software; you can redistribute it and/or modify
   6    it under the terms of the GNU General Public License as published by
   7    the Free Software Foundation; either version 2, or (at your option)
   8    any later version.
   9
  10    This program is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License along
  16    with this program; if not, write to the Free Software Foundation,
  17    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
  18
  19 #ifdef HAVE_CONFIG_H
  20 # include <config.h>
  21 #endif
  22
  23 /* Specification.  */
  24 #include "vsnprintf.h"
  25
  26 #include <errno.h>
  27 #include <limits.h>
  28 #include <stdarg.h>
  29 #include <stdio.h>
  30 #include <stdlib.h>
  31 #include <string.h>
  32
  33 #include "vasnprintf.h"
  34
  35 /* Some systems, like OSF/1 4.0 and Woe32, don't have EOVERFLOW.  */
  36 #ifndef EOVERFLOW
  37 # define EOVERFLOW E2BIG
  38 #endif
  39
  40 /* Print formatted output to string STR.  Similar to vsprintf, but
  41    additional length SIZE limit how much is written into STR.  Returns
  42    string length of formatted string (which may be larger than SIZE).
  43    STR may be NULL, in which case nothing will be written.  On error,
  44    return a negative value. */
  45 int
  46 vsnprintf (char *str, size_t size, const char *format, va_list args)
  47 {
  48   char *output;
  49   size_t len;
  50   size_t lenbuf = size;
  51
  52   output = vasnprintf (str, &lenbuf, format, args);
  53   len = lenbuf;
  54
  55   if (!output)
  56     return -1;
  57
  58   if (output != str)
  59     {
  60       if (size)
  61         {
  62           size_t pruned_len = (len < size ? len : size - 1);
  63           memcpy (str, output, pruned_len);
  64           str[pruned_len] = '\0';
  65         }
  66
  67       free (output);
  68     }
  69
  70   if (len > INT_MAX)
  71     {
  72       errno = EOVERFLOW;
  73       return -1;
  74     }
  75
  76   return len;
  77 }