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 } 74