1 /*-
2 * Copyright (c) 1990, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * Copyright (c) 2011 The FreeBSD Foundation
6 * All rights reserved.
7 * Portions of this software were developed by David Chisnall
8 * under sponsorship from the FreeBSD Foundation.
9 *
10 * This code is derived from software contributed to Berkeley by
11 * Chris Torek.
12 *
13 * Redistribution and use in source and binary forms, with or without
14 * modification, are permitted provided that the following conditions
15 * are met:
16 * 1. Redistributions of source code must retain the above copyright
17 * notice, this list of conditions and the following disclaimer.
18 * 2. Redistributions in binary form must reproduce the above copyright
19 * notice, this list of conditions and the following disclaimer in the
20 * documentation and/or other materials provided with the distribution.
21 * 3. Neither the name of the University nor the names of its contributors
22 * may be used to endorse or promote products derived from this software
23 * without specific prior written permission.
24 *
25 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
28 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
29 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
30 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
31 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
32 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35 * SUCH DAMAGE.
36 *
37 * @(#)snprintf.c 8.1 (Berkeley) 6/4/93
38 * $FreeBSD: head/lib/libc/stdio/snprintf.c 249808 2013-04-23 13:33:13Z emaste $
39 */
40
41
42 #include <errno.h>
43 #include <limits.h>
44 #include <pthread.h> /* for FAKE_FILE PTHREAD_MUTEX_INITIALIZER */
45 #include <stdio.h>
46 #include <stdarg.h>
47 #include "xlocale_private.h"
48
49 #include "local.h"
50
51 int
snprintf(char * __restrict str,size_t n,char const * __restrict fmt,...)52 snprintf(char * __restrict str, size_t n, char const * __restrict fmt, ...)
53 {
54 size_t on;
55 int ret;
56 va_list ap;
57 FILE f = FAKE_FILE;
58
59 on = n;
60 if (n != 0)
61 n--;
62 if (n > INT_MAX) {
63 errno = EOVERFLOW;
64 *str = '\0';
65 return (EOF);
66 }
67 va_start(ap, fmt);
68 f.pub._flags = __SWR | __SSTR;
69 f._bf._base = f.pub._p = (unsigned char *)str;
70 f._bf._size = f.pub._w = n;
71 ret = __vfprintf(&f, __get_locale(), fmt, ap);
72 if (on > 0)
73 *f.pub._p = '\0';
74 va_end(ap);
75 return (ret);
76 }
77 int
snprintf_l(char * __restrict str,size_t n,locale_t locale,char const * __restrict fmt,...)78 snprintf_l(char * __restrict str, size_t n, locale_t locale,
79 char const * __restrict fmt, ...)
80 {
81 size_t on;
82 int ret;
83 va_list ap;
84 FILE f = FAKE_FILE;
85 FIX_LOCALE(locale);
86
87 on = n;
88 if (n != 0)
89 n--;
90 if (n > INT_MAX) {
91 errno = EOVERFLOW;
92 *str = '\0';
93 return (EOF);
94 }
95 va_start(ap, fmt);
96 f.pub._flags = __SWR | __SSTR;
97 f._bf._base = f.pub._p = (unsigned char *)str;
98 f._bf._size = f.pub._w = n;
99 ret = __vfprintf(&f, locale, fmt, ap);
100 if (on > 0)
101 *f.pub._p = '\0';
102 va_end(ap);
103 return (ret);
104 }
105