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
26static int failed = 0;
27
28static void
29fail(const char *m)
30{
31	fprintf(stderr, "snprintftest: %s\n", m);
32	failed = 1;
33}
34
35int 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
46int
47main(void)
48{
49	char b[5];
50	char *src;
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	return failed;
73}
74