1 /* $NetBSD: t_humanize_number.c,v 1.8 2012/03/18 07:14:08 jruoho Exp $ */
4 * Copyright (c) 2010, 2011 The NetBSD Foundation, Inc.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
46 const char *ho_suffix;
49 int ho_retval; /* expected return value */
50 const char *ho_retstr; /* expected string in buffer */
53 * Rev. 1.6 produces "10.0".
55 { 5, 10737418236ULL * 1024, "",
56 HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL, 3, "10T" },
59 HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL, 3, "10M" },
60 { 5, 10500000, "", /* just for reference */
61 HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL, 3, "10M" },
64 * Trailing space. Rev. 1.7 produces "1 ".
66 { 5, 1, "", 0, HN_NOSPACE, 1, "1" },
68 { 5, 1, "", 0, 0, 2, "1 " }, /* just for reference */
69 { 5, 1, "", 0, HN_B, 3, "1 B" }, /* and more ... */
70 { 5, 1, "", 0, HN_DECIMAL, 2, "1 " },
71 { 5, 1, "", 0, HN_NOSPACE | HN_B, 2, "1B" },
72 { 5, 1, "", 0, HN_B | HN_DECIMAL, 3, "1 B" },
73 { 5, 1, "", 0, HN_NOSPACE | HN_B | HN_DECIMAL, 2, "1B" },
76 * Space and HN_B. Rev. 1.7 produces "1B".
78 { 5, 1, "", HN_AUTOSCALE, HN_B, 3, "1 B" },
79 { 5, 1000, "", /* just for reference */
80 HN_AUTOSCALE, HN_B, 3, "1 K" },
83 * Truncated output. Rev. 1.7 produces "1.0 K".
86 { 6, 1000, "A", HN_AUTOSCALE, HN_DECIMAL, -1, "" },
89 * Failure case reported by Greg Troxel <gdt@NetBSD.org>.
90 * Rev. 1.11 incorrectly returns 5 with filling the buffer
94 HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL, 4, "1.0G" },
95 /* Similar case it prints 1000 where it shouldn't */
97 HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL, 4, "1.0M" },
100 HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL, 4, "1.0M" },
108 const struct hnflags scale_flags[] = {
109 { HN_GETSCALE, "HN_GETSCALE" },
110 { HN_AUTOSCALE, "HN_AUTOSCALE" },
112 const struct hnflags normal_flags[] = {
113 { HN_DECIMAL, "HN_DECIMAL" },
114 { HN_NOSPACE, "HN_NOSPACE" },
116 { HN_DIVISOR_1000, "HN_DIVISOR_1000" },
119 const char *formatflags(char *, size_t, const struct hnflags *, size_t, int);
121 void w_printf(const char *, ...) __printflike(1, 2);
122 int main(int, char *[]);
125 formatflags(char *buf, size_t buflen, const struct hnflags *hfs,
126 size_t hfslen, int flags)
128 const struct hnflags *hf;
130 ssize_t len = buflen;
131 unsigned int i, found;
135 snprintf(buf, buflen, "0");
138 for (i = found = 0; i < hfslen && flags & ~found; i++) {
140 if (flags & hf->hf_flags) {
141 found |= hf->hf_flags;
142 n = snprintf(p, len, "|%s", hf->hf_name);
146 /* Print `flags' as number */
156 snprintf(p, len, "|0x%x", flags);
157 return (*buf == '|' ? buf + 1 : buf);
160 static int col, bol = 1;
165 fprintf(stderr, "\n");
171 w_printf(const char *fmt, ...)
179 n = vsnprintf(buf, sizeof(buf), fmt, ap);
180 if (n >= (int)sizeof(buf)) {
188 fprintf(stderr, "\n "), col = 4;
190 fprintf(stderr, " "), col++;
192 fprintf(stderr, "%s", buf);
197 vfprintf(stderr, fmt, ap);
203 ATF_TC(humanize_number_basic);
204 ATF_TC_HEAD(humanize_number_basic, tc)
207 atf_tc_set_md_var(tc, "descr", "Test humanize_number(3)");
210 ATF_TC_BODY(humanize_number_basic, tc)
213 const struct hnopts *ho;
219 for (i = 0; i < __arraycount(hnopts); i++) {
221 if (buflen < ho->ho_len) {
223 buf = realloc(buf, buflen);
225 atf_tc_fail("realloc(..., %zu) failed", buflen);
228 rv = humanize_number(buf, ho->ho_len, ho->ho_num,
229 ho->ho_suffix, ho->ho_scale, ho->ho_flags);
231 if (rv == ho->ho_retval &&
232 (rv == -1 || strcmp(buf, ho->ho_retstr) == 0))
235 w_printf("humanize_number(\"%s\", %zu, %" PRId64 ",",
236 ho->ho_retstr, ho->ho_len, ho->ho_num);
237 w_printf("\"%s\",", ho->ho_suffix);
238 w_printf("%s,", formatflags(fbuf, sizeof(fbuf), scale_flags,
239 sizeof(scale_flags) / sizeof(scale_flags[0]),
241 w_printf("%s)", formatflags(fbuf, sizeof(fbuf), normal_flags,
242 sizeof(normal_flags) / sizeof(normal_flags[0]),
244 w_printf("= %d,", ho->ho_retval);
246 w_printf("%d/[%s]", rv, rv == -1 ? "" : buf);
248 atf_tc_fail_nonfatal("Failed for table entry %d", i);
252 ATF_TC(humanize_number_big);
253 ATF_TC_HEAD(humanize_number_big, tc)
256 atf_tc_set_md_var(tc, "descr", "Test humanize "
257 "big numbers (PR lib/44097)");
260 ATF_TC_BODY(humanize_number_big, tc)
268 (void)memset(buf, 0, sizeof(buf));
270 rv = humanize_number(buf, 10, 10000, "", HN_AUTOSCALE, HN_NOSPACE);
272 ATF_REQUIRE(rv != -1);
273 ATF_CHECK_STREQ(buf, "10000");
276 * A bogus value with large number.
278 (void)memset(buf, 0, sizeof(buf));
280 rv = humanize_number(buf, 10, INT64_MAX, "", HN_AUTOSCALE, HN_NOSPACE);
282 ATF_REQUIRE(rv != -1);
283 ATF_REQUIRE(strcmp(buf, "0") != 0);
286 * Large buffer with HN_AUTOSCALE. Entirely bogus.
288 (void)memset(buf, 0, sizeof(buf));
290 rv = humanize_number(buf, sizeof(buf), 10000, "",
291 HN_AUTOSCALE, HN_NOSPACE);
293 ATF_REQUIRE(rv != -1);
294 ATF_REQUIRE(strcmp(buf, "0%d%s%d%s%s%s") != 0);
299 * The man page says that len must be at least 4.
300 * 3 works, but anything less that will not. This
301 * is because baselen starts with 2 for positive
304 (void)memset(buf, 0, sizeof(buf));
306 rv = humanize_number(buf, 3, 1, "", HN_AUTOSCALE, HN_NOSPACE);
308 ATF_REQUIRE(rv != -1);
314 ATF_TP_ADD_TC(tp, humanize_number_basic);
315 ATF_TP_ADD_TC(tp, humanize_number_big);
317 return atf_no_error();