xref: /freebsd/lib/libc/stdio/snprintf.c (revision 666d00d34aa1b075412f216f191def6b6f3e2c88)
158f0484fSRodney W. Grimes /*-
258f0484fSRodney W. Grimes  * Copyright (c) 1990, 1993
358f0484fSRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
458f0484fSRodney W. Grimes  *
53c87aa1dSDavid Chisnall  * Copyright (c) 2011 The FreeBSD Foundation
63c87aa1dSDavid Chisnall  * All rights reserved.
73c87aa1dSDavid Chisnall  * Portions of this software were developed by David Chisnall
83c87aa1dSDavid Chisnall  * under sponsorship from the FreeBSD Foundation.
93c87aa1dSDavid Chisnall  *
1058f0484fSRodney W. Grimes  * This code is derived from software contributed to Berkeley by
1158f0484fSRodney W. Grimes  * Chris Torek.
1258f0484fSRodney W. Grimes  *
1358f0484fSRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
1458f0484fSRodney W. Grimes  * modification, are permitted provided that the following conditions
1558f0484fSRodney W. Grimes  * are met:
1658f0484fSRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
1758f0484fSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
1858f0484fSRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
1958f0484fSRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
2058f0484fSRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
2158f0484fSRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
2258f0484fSRodney W. Grimes  *    may be used to endorse or promote products derived from this software
2358f0484fSRodney W. Grimes  *    without specific prior written permission.
2458f0484fSRodney W. Grimes  *
2558f0484fSRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
2658f0484fSRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
2758f0484fSRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2858f0484fSRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
2958f0484fSRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
3058f0484fSRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
3158f0484fSRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
3258f0484fSRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
3358f0484fSRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
3458f0484fSRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
3558f0484fSRodney W. Grimes  * SUCH DAMAGE.
3658f0484fSRodney W. Grimes  */
3758f0484fSRodney W. Grimes 
3858f0484fSRodney W. Grimes #if defined(LIBC_SCCS) && !defined(lint)
3958f0484fSRodney W. Grimes static char sccsid[] = "@(#)snprintf.c	8.1 (Berkeley) 6/4/93";
4058f0484fSRodney W. Grimes #endif /* LIBC_SCCS and not lint */
41333fc21eSDavid E. O'Brien #include <sys/cdefs.h>
42333fc21eSDavid E. O'Brien __FBSDID("$FreeBSD$");
4358f0484fSRodney W. Grimes 
44*666d00d3SDavid Schultz #include <errno.h>
458c6d2f42SAndrey A. Chernov #include <limits.h>
4658f0484fSRodney W. Grimes #include <stdio.h>
4758f0484fSRodney W. Grimes #include <stdarg.h>
483c87aa1dSDavid Chisnall #include "xlocale_private.h"
4958f0484fSRodney W. Grimes 
5029ac6bd2SDaniel Eischen #include "local.h"
5129ac6bd2SDaniel Eischen 
52f70177e7SJulian Elischer int
5371a00a44SRobert Drehmel snprintf(char * __restrict str, size_t n, char const * __restrict fmt, ...)
5458f0484fSRodney W. Grimes {
558c6d2f42SAndrey A. Chernov 	size_t on;
5658f0484fSRodney W. Grimes 	int ret;
5758f0484fSRodney W. Grimes 	va_list ap;
581b0181dfSJohn Baldwin 	FILE f = FAKE_FILE;
5958f0484fSRodney W. Grimes 
606e690ad4SAndrey A. Chernov 	on = n;
618c6d2f42SAndrey A. Chernov 	if (n != 0)
626e690ad4SAndrey A. Chernov 		n--;
63*666d00d3SDavid Schultz 	if (n > INT_MAX) {
64*666d00d3SDavid Schultz 		errno = EOVERFLOW;
65*666d00d3SDavid Schultz 		*str = '\0';
66*666d00d3SDavid Schultz 		return (EOF);
67*666d00d3SDavid Schultz 	}
6858f0484fSRodney W. Grimes 	va_start(ap, fmt);
6958f0484fSRodney W. Grimes 	f._flags = __SWR | __SSTR;
7058f0484fSRodney W. Grimes 	f._bf._base = f._p = (unsigned char *)str;
714ecaf220SAndrey A. Chernov 	f._bf._size = f._w = n;
723c87aa1dSDavid Chisnall 	ret = __vfprintf(&f, __get_locale(), fmt, ap);
733c87aa1dSDavid Chisnall 	if (on > 0)
743c87aa1dSDavid Chisnall 		*f._p = '\0';
753c87aa1dSDavid Chisnall 	va_end(ap);
763c87aa1dSDavid Chisnall 	return (ret);
773c87aa1dSDavid Chisnall }
783c87aa1dSDavid Chisnall int
793c87aa1dSDavid Chisnall snprintf_l(char * __restrict str, size_t n, locale_t locale,
803c87aa1dSDavid Chisnall 		char const * __restrict fmt, ...)
813c87aa1dSDavid Chisnall {
823c87aa1dSDavid Chisnall 	size_t on;
833c87aa1dSDavid Chisnall 	int ret;
843c87aa1dSDavid Chisnall 	va_list ap;
853c87aa1dSDavid Chisnall 	FILE f = FAKE_FILE;
863c87aa1dSDavid Chisnall 	FIX_LOCALE(locale);
873c87aa1dSDavid Chisnall 
883c87aa1dSDavid Chisnall 	on = n;
893c87aa1dSDavid Chisnall 	if (n != 0)
903c87aa1dSDavid Chisnall 		n--;
91*666d00d3SDavid Schultz 	if (n > INT_MAX) {
92*666d00d3SDavid Schultz 		errno = EOVERFLOW;
93*666d00d3SDavid Schultz 		*str = '\0';
94*666d00d3SDavid Schultz 		return (EOF);
95*666d00d3SDavid Schultz 	}
963c87aa1dSDavid Chisnall 	va_start(ap, fmt);
973c87aa1dSDavid Chisnall 	f._flags = __SWR | __SSTR;
983c87aa1dSDavid Chisnall 	f._bf._base = f._p = (unsigned char *)str;
993c87aa1dSDavid Chisnall 	f._bf._size = f._w = n;
1003c87aa1dSDavid Chisnall 	ret = __vfprintf(&f, locale, fmt, ap);
1016e690ad4SAndrey A. Chernov 	if (on > 0)
1026e690ad4SAndrey A. Chernov 		*f._p = '\0';
10358f0484fSRodney W. Grimes 	va_end(ap);
1046e690ad4SAndrey A. Chernov 	return (ret);
10558f0484fSRodney W. Grimes }
106