1 /*
2  * Copyright (c) 2005 Darren Tucker
3  * Copyright (c) 2005 Damien Miller
4  *
5  * Permission to use, copy, modify, and distribute this software for any
6  * purpose with or without fee is hereby granted, provided that the above
7  * copyright notice and this permission notice appear in all copies.
8  *
9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16  */
17 
18 #define BUFSZ 2048
19 
20 #include <sys/types.h>
21 #include <stdlib.h>
22 #include <stdio.h>
23 #include <stdarg.h>
24 #include <string.h>
25 
26 static int failed = 0;
27 
28 static void
fail(const char * m)29 fail(const char *m)
30 {
31 	fprintf(stderr, "snprintftest: %s\n", m);
32 	failed = 1;
33 }
34 
x_snprintf(char * str,size_t count,const char * fmt,...)35 int x_snprintf(char *str, size_t count, const char *fmt, ...)
36 {
37 	size_t ret;
38 	va_list ap;
39 
40 	va_start(ap, fmt);
41 	ret = vsnprintf(str, count, fmt, ap);
42 	va_end(ap);
43 	return ret;
44 }
45 
46 int
main(void)47 main(void)
48 {
49 	char b[5];
50 	char *src = NULL;
51 
52 	snprintf(b,5,"123456789");
53 	if (b[4] != '\0')
54 		fail("snprintf does not correctly terminate long strings");
55 
56 	/* check for read overrun on unterminated string */
57 	if ((src = malloc(BUFSZ)) == NULL) {
58 		fail("malloc failed");
59 	} else {
60 		memset(src, 'a', BUFSZ);
61 		snprintf(b, sizeof(b), "%.*s", 1, src);
62 		if (strcmp(b, "a") != 0)
63 			fail("failed with length limit '%%.s'");
64 	}
65 
66 	/* check that snprintf and vsnprintf return sane values */
67 	if (snprintf(b, 1, "%s %d", "hello", 12345) != 11)
68 		fail("snprintf does not return required length");
69 	if (x_snprintf(b, 1, "%s %d", "hello", 12345) != 11)
70 		fail("vsnprintf does not return required length");
71 
72 	free(src);
73 	return failed;
74 }
75