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/ungetwc.c,v 1.11 2008/04/17 22:17:54 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 */
46wint_t
47__ungetwc(wint_t wc, FILE *fp, locale_t loc)
48{
49	char buf[MB_LEN_MAX];
50	size_t len;
51
52	if (wc == WEOF)
53		return (WEOF);
54	if ((len = loc->__lc_ctype->__wcrtomb(buf, wc, &fp->_mbstate, loc)) == (size_t)-1) {
55		fp->_flags |= __SERR;
56		return (WEOF);
57	}
58	while (len-- != 0)
59		if (__ungetc((unsigned char)buf[len], fp) == EOF)
60			return (WEOF);
61
62	return (wc);
63}
64
65/*
66 * MT-safe version.
67 */
68wint_t
69ungetwc(wint_t wc, FILE *fp)
70{
71	wint_t r;
72
73	FLOCKFILE(fp);
74	ORIENT(fp, 1);
75	r = __ungetwc(wc, fp, __current_locale());
76	FUNLOCKFILE(fp);
77
78	return (r);
79}
80
81wint_t
82ungetwc_l(wint_t wc, FILE *fp, locale_t loc)
83{
84	wint_t r;
85
86	NORMALIZE_LOCALE(loc);
87	FLOCKFILE(fp);
88	ORIENT(fp, 1);
89	r = __ungetwc(wc, fp, loc);
90	FUNLOCKFILE(fp);
91
92	return (r);
93}
94