1/*-
2 * Copyright (c) 2002-2004 Tim J. Robbins.
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD: src/lib/libc/stdio/fputwc.c,v 1.11 2008/04/17 22:17:53 jhb Exp $");
29
30#include "xlocale_private.h"
31
32#include "namespace.h"
33#include <errno.h>
34#include <limits.h>
35#include <stdio.h>
36#include <stdlib.h>
37#include <wchar.h>
38#include "un-namespace.h"
39#include "libc_private.h"
40#include "local.h"
41#include "mblocal.h"
42
43/*
44 * Non-MT-safe version.
45 */
46__private_extern__ wint_t
47__fputwc(wchar_t wc, FILE *fp, locale_t loc)
48{
49	char buf[MB_LEN_MAX];
50	size_t i, len;
51	struct __xlocale_st_runelocale *xrl = loc->__lc_ctype;
52
53	if (xrl->__mb_cur_max == 1 && wc > 0 && wc <= UCHAR_MAX) {
54		/*
55		 * Assume single-byte locale with no special encoding.
56		 * A more careful test would be to check
57		 * _CurrentRuneLocale->encoding.
58		 */
59		*buf = (unsigned char)wc;
60		len = 1;
61	} else {
62		if ((len = loc->__lc_ctype->__wcrtomb(buf, wc, &fp->_mbstate, loc)) == (size_t)-1) {
63			fp->_flags |= __SERR;
64			return (WEOF);
65		}
66	}
67
68	for (i = 0; i < len; i++)
69		if (__sputc((unsigned char)buf[i], fp) == EOF)
70			return (WEOF);
71
72	return ((wint_t)wc);
73}
74
75/*
76 * MT-safe version.
77 */
78wint_t
79fputwc(wchar_t wc, FILE *fp)
80{
81	wint_t r;
82
83	FLOCKFILE(fp);
84	ORIENT(fp, 1);
85	r = __fputwc(wc, fp, __current_locale());
86	FUNLOCKFILE(fp);
87
88	return (r);
89}
90
91wint_t
92fputwc_l(wchar_t wc, FILE *fp, locale_t loc)
93{
94	wint_t r;
95
96	NORMALIZE_LOCALE(loc);
97	FLOCKFILE(fp);
98	ORIENT(fp, 1);
99	r = __fputwc(wc, fp, loc);
100	FUNLOCKFILE(fp);
101
102	return (r);
103}
104