]>
Commit | Line | Data |
---|---|---|
9108f8d9 | 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 | |
29 | fail(const char *m) | |
30 | { | |
31 | fprintf(stderr, "snprintftest: %s\n", m); | |
32 | failed = 1; | |
33 | } | |
34 | ||
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 | |
47 | main(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 | } |