tests/test-vasnprintf.c

   1 /* Test of vasnprintf() and asnprintf() functions.
   2    Copyright (C) 2007 Free Software Foundation, Inc.
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2, or (at your option)
   7    any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program; if not, write to the Free Software Foundation,
  16    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
  17
  18 /* Written by Bruno Haible <bruno@clisp.org>, 2007.  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 # include <config.h>
  22 #endif
  23
  24 #include "vasnprintf.h"
  25
  26 #include <stdarg.h>
  27 #include <stdlib.h>
  28 #include <string.h>
  29
  30 #define ASSERT(expr) if (!(expr)) abort ();
  31
  32 static char *
  33 my_asnprintf (char *resultbuf, size_t *lengthp, const char *format, ...)
  34 {
  35   va_list args;
  36   char *ret;
  37
  38   va_start (args, format);
  39   ret = vasnprintf (resultbuf, lengthp, format, args);
  40   va_end (args);
  41   return ret;
  42 }
  43
  44 static void
  45 test_vasnprintf ()
  46 {
  47   char buf[8];
  48   int size;
  49
  50   for (size = 0; size <= 8; size++)
  51     {
  52       size_t length = size;
  53       char *result = my_asnprintf (NULL, &length, "%d", 12345);
  54       ASSERT (result != NULL);
  55       ASSERT (strcmp (result, "12345") == 0);
  56       ASSERT (length == 5);
  57       free (result);
  58     }
  59
  60   for (size = 0; size <= 8; size++)
  61     {
  62       size_t length;
  63       char *result;
  64
  65       memcpy (buf, "DEADBEEF", 8);
  66       length = size;
  67       result = my_asnprintf (buf, &length, "%d", 12345);
  68       ASSERT (result != NULL);
  69       ASSERT (strcmp (result, "12345") == 0);
  70       ASSERT (length == 5);
  71       if (size < 6)
  72         ASSERT (result != buf);
  73       ASSERT (memcmp (buf + size, "DEADBEEF" + size, 8 - size) == 0);
  74       if (result != buf)
  75         free (result);
  76     }
  77 }
  78
  79 static void
  80 test_asnprintf ()
  81 {
  82   char buf[8];
  83   int size;
  84
  85   for (size = 0; size <= 8; size++)
  86     {
  87       size_t length = size;
  88       char *result = asnprintf (NULL, &length, "%d", 12345);
  89       ASSERT (result != NULL);
  90       ASSERT (strcmp (result, "12345") == 0);
  91       ASSERT (length == 5);
  92       free (result);
  93     }
  94
  95   for (size = 0; size <= 8; size++)
  96     {
  97       size_t length;
  98       char *result;
  99
 100       memcpy (buf, "DEADBEEF", 8);
 101       length = size;
 102       result = asnprintf (buf, &length, "%d", 12345);
 103       ASSERT (result != NULL);
 104       ASSERT (strcmp (result, "12345") == 0);
 105       ASSERT (length == 5);
 106       if (size < 6)
 107         ASSERT (result != buf);
 108       ASSERT (memcmp (buf + size, "DEADBEEF" + size, 8 - size) == 0);
 109       if (result != buf)
 110         free (result);
 111     }
 112 }
 113
 114 int
 115 main (int argc, char *argv[])
 116 {
 117   test_vasnprintf ();
 118   test_asnprintf ();
 119   return 0;
 120 }