1 /* Test of vasnprintf() and asnprintf() functions.
2    Copyright (C) 2007-2018 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 3 of the License, or
7    (at your option) 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, see <https://www.gnu.org/licenses/>.  */
16 
17 /* Written by Bruno Haible <bruno@clisp.org>, 2007.  */
18 
19 #include <config.h>
20 
21 #include "vasnprintf.h"
22 
23 #include <stdarg.h>
24 #include <stdlib.h>
25 #include <string.h>
26 
27 #include "macros.h"
28 
29 static void
test_function(char * (* my_asnprintf)(char *,size_t *,const char *,...))30 test_function (char * (*my_asnprintf) (char *, size_t *, const char *, ...))
31 {
32   char buf[8];
33   int size;
34 
35   for (size = 0; size <= 8; size++)
36     {
37       size_t length = size;
38       char *result = my_asnprintf (NULL, &length, "%d", 12345);
39       ASSERT (result != NULL);
40       ASSERT (strcmp (result, "12345") == 0);
41       ASSERT (length == 5);
42       free (result);
43     }
44 
45   for (size = 0; size <= 8; size++)
46     {
47       size_t length;
48       char *result;
49 
50       memcpy (buf, "DEADBEEF", 8);
51       length = size;
52       result = my_asnprintf (buf, &length, "%d", 12345);
53       ASSERT (result != NULL);
54       ASSERT (strcmp (result, "12345") == 0);
55       ASSERT (length == 5);
56       if (size < 6)
57         ASSERT (result != buf);
58       ASSERT (memcmp (buf + size, &"DEADBEEF"[size], 8 - size) == 0);
59       if (result != buf)
60         free (result);
61     }
62 }
63 
64 static char *
my_asnprintf(char * resultbuf,size_t * lengthp,const char * format,...)65 my_asnprintf (char *resultbuf, size_t *lengthp, const char *format, ...)
66 {
67   va_list args;
68   char *ret;
69 
70   va_start (args, format);
71   ret = vasnprintf (resultbuf, lengthp, format, args);
72   va_end (args);
73   return ret;
74 }
75 
76 static void
test_vasnprintf()77 test_vasnprintf ()
78 {
79   test_function (my_asnprintf);
80 }
81 
82 static void
test_asnprintf()83 test_asnprintf ()
84 {
85   test_function (asnprintf);
86 }
87 
88 int
main(int argc,char * argv[])89 main (int argc, char *argv[])
90 {
91   test_vasnprintf ();
92   test_asnprintf ();
93   return 0;
94 }
95