1/*-
2 * Copyright (c) 2009 David Schultz <das@FreeBSD.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27/*
28 * Tests for basic and miscellaneous printf() formats.
29 */
30
31#include <sys/cdefs.h>
32__FBSDID("$FreeBSD$");
33
34#include <err.h>
35#include <limits.h>
36#include <locale.h>
37#include <stdio.h>
38#include <stdarg.h>
39#include <stddef.h>
40#include <stdint.h>
41#include <stdlib.h>
42#include <string.h>
43#include <wchar.h>
44
45#include <atf-c.h>
46
47#define	S_UINT64MAX	"18446744073709551615"
48#define	S_UINT32MAX	"4294967295"
49#define	S_INT64MIN	"-9223372036854775808"
50#define	S_INT32MIN	"-2147483648"
51
52#define	S_SIZEMAX	(SIZE_MAX == UINT64_MAX ? S_UINT64MAX : S_UINT32MAX)
53#define	S_ULONGMAX	(ULONG_MAX == UINT64_MAX ? S_UINT64MAX : S_UINT32MAX)
54#define	S_ULLONGMAX	(ULLONG_MAX == UINT64_MAX ? S_UINT64MAX : S_UINT32MAX)
55
56static void
57smash_stack(void)
58{
59	static uint32_t junk = 0xdeadbeef;
60	uint32_t buf[512];
61	int i;
62
63	for (i = 0; i < sizeof(buf) / sizeof(buf[0]); i++)
64		buf[i] = junk;
65}
66
67#define	testfmt(result, fmt, ...)       \
68	_testfmt((result), #__VA_ARGS__, fmt, __VA_ARGS__)
69static void
70_testfmt(const char *result, const char *argstr, const char *fmt,...)
71{
72#define	BUF	100
73	wchar_t ws[BUF], wfmt[BUF], wresult[BUF];
74	char s[BUF];
75	va_list ap, ap2;
76
77	va_start(ap, fmt);
78	va_copy(ap2, ap);
79	smash_stack();
80	vsnprintf(s, sizeof(s), fmt, ap);
81	ATF_CHECK_MSG(strcmp(result, s) == 0,
82	    "printf(\"%s\", %s) ==> [%s], expected [%s]",
83	    fmt, argstr, s, result);
84
85	smash_stack();
86	mbstowcs(ws, s, BUF - 1);
87	mbstowcs(wfmt, fmt, BUF - 1);
88	mbstowcs(wresult, result, BUF - 1);
89	vswprintf(ws, sizeof(ws) / sizeof(ws[0]), wfmt, ap2);
90	ATF_CHECK_MSG(wcscmp(wresult, ws) == 0,
91	    "wprintf(\"%ls\", %s) ==> [%ls], expected [%ls]",
92	    wfmt, argstr, ws, wresult);
93
94	va_end(ap);
95	va_end(ap2);
96}
97
98ATF_TC_WITHOUT_HEAD(int_within_limits);
99ATF_TC_BODY(int_within_limits, tc)
100{
101
102	ATF_REQUIRE(setlocale(LC_NUMERIC, "C"));
103
104	/* The test requires these to be true. */
105	ATF_REQUIRE(UINTMAX_MAX == UINT64_MAX);
106	ATF_REQUIRE(UINT_MAX == UINT32_MAX);
107	ATF_REQUIRE(USHRT_MAX == 0xffff);
108	ATF_REQUIRE(UCHAR_MAX == 0xff);
109
110	/* Make sure we handle signed vs. unsigned args correctly. */
111	testfmt("-1", "%jd", (intmax_t)-1);
112	testfmt(S_UINT64MAX, "%ju", UINT64_MAX);
113
114	if (sizeof(ptrdiff_t) != sizeof(uintmax_t))
115		atf_tc_expect_fail("the %%t qualifier is broken on 32-bit "
116		    "platforms where there's a mismatch between ptrdiff_t and "
117		    "uintmax_t's type width; bug # 191674");
118
119	testfmt("-1", "%td", (ptrdiff_t)-1);
120	testfmt(S_SIZEMAX, "%tu", (size_t)-1);
121
122	testfmt("-1", "%zd", (ssize_t)-1);
123	testfmt(S_SIZEMAX, "%zu", (ssize_t)-1);
124
125	testfmt("-1", "%ld", (long)-1);
126	testfmt(S_ULONGMAX, "%lu", ULONG_MAX);
127
128	testfmt("-1", "%lld", (long long)-1);
129	testfmt(S_ULLONGMAX, "%llu", ULLONG_MAX);
130
131	testfmt("-1", "%d", -1);
132	testfmt(S_UINT32MAX, "%u", UINT32_MAX);
133
134	testfmt("-1", "%hd", -1);
135	testfmt("65535", "%hu", USHRT_MAX);
136
137	testfmt("-1", "%hhd", -1);
138	testfmt("255", "%hhu", UCHAR_MAX);
139}
140
141ATF_TC_WITHOUT_HEAD(int_limits);
142ATF_TC_BODY(int_limits, tc)
143{
144
145	ATF_REQUIRE(setlocale(LC_NUMERIC, "C"));
146
147	/*
148	 * Check that printing the largest negative number does not cause
149	 * overflow when it is negated.
150	 */
151	testfmt(S_INT32MIN, "%d", INT_MIN);
152	testfmt(S_INT64MIN, "%jd", INTMAX_MIN);
153}
154
155ATF_TP_ADD_TCS(tp)
156{
157
158	ATF_TP_ADD_TC(tp, int_within_limits);
159	ATF_TP_ADD_TC(tp, int_limits);
160
161	return (atf_no_error());
162}
163