1101776Stjr/*-
2129583Stjr * Copyright (c) 2002-2004 Tim J. Robbins.
3101776Stjr * All rights reserved.
4101776Stjr *
5227753Stheraven * Copyright (c) 2011 The FreeBSD Foundation
6227753Stheraven * All rights reserved.
7227753Stheraven * Portions of this software were developed by David Chisnall
8227753Stheraven * under sponsorship from the FreeBSD Foundation.
9227753Stheraven *
10101776Stjr * Redistribution and use in source and binary forms, with or without
11101776Stjr * modification, are permitted provided that the following conditions
12101776Stjr * are met:
13101776Stjr * 1. Redistributions of source code must retain the above copyright
14101776Stjr *    notice, this list of conditions and the following disclaimer.
15101776Stjr * 2. Redistributions in binary form must reproduce the above copyright
16101776Stjr *    notice, this list of conditions and the following disclaimer in the
17101776Stjr *    documentation and/or other materials provided with the distribution.
18101776Stjr *
19101776Stjr * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20101776Stjr * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21101776Stjr * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22101776Stjr * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23101776Stjr * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24101776Stjr * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25101776Stjr * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26101776Stjr * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27101776Stjr * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28101776Stjr * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29101776Stjr * SUCH DAMAGE.
30101776Stjr */
31101776Stjr
32101776Stjr#include <sys/cdefs.h>
33101776Stjr__FBSDID("$FreeBSD$");
34101776Stjr
35101776Stjr#include "namespace.h"
36101776Stjr#include <errno.h>
37149313Sstefanf#include <limits.h>
38101776Stjr#include <stdio.h>
39103523Stjr#include <stdlib.h>
40101776Stjr#include <wchar.h>
41101776Stjr#include "un-namespace.h"
42101776Stjr#include "libc_private.h"
43101776Stjr#include "local.h"
44132442Stjr#include "mblocal.h"
45227753Stheraven#include "xlocale_private.h"
46101776Stjr
47103782Stjr/*
48103782Stjr * Non-MT-safe version.
49103782Stjr */
50101776Stjrwint_t
51227753Stheraven__ungetwc(wint_t wc, FILE *fp, locale_t locale)
52101776Stjr{
53103523Stjr	char buf[MB_LEN_MAX];
54103523Stjr	size_t len;
55227753Stheraven	struct xlocale_ctype *l = XLOCALE_CTYPE(locale);
56101776Stjr
57103523Stjr	if (wc == WEOF)
58103782Stjr		return (WEOF);
59227753Stheraven	if ((len = l->__wcrtomb(buf, wc, &fp->_mbstate)) == (size_t)-1) {
60105234Stjr		fp->_flags |= __SERR;
61103782Stjr		return (WEOF);
62105234Stjr	}
63103523Stjr	while (len-- != 0)
64103677Stjr		if (__ungetc((unsigned char)buf[len], fp) == EOF)
65103782Stjr			return (WEOF);
66103523Stjr
67103523Stjr	return (wc);
68103782Stjr}
69103677Stjr
70103782Stjr/*
71103782Stjr * MT-safe version.
72103782Stjr */
73103782Stjrwint_t
74227753Stheravenungetwc_l(wint_t wc, FILE *fp, locale_t locale)
75103782Stjr{
76103782Stjr	wint_t r;
77227753Stheraven	FIX_LOCALE(locale);
78103782Stjr
79103782Stjr	FLOCKFILE(fp);
80103782Stjr	ORIENT(fp, 1);
81227753Stheraven	r = __ungetwc(wc, fp, locale);
82103677Stjr	FUNLOCKFILE(fp);
83103782Stjr
84103782Stjr	return (r);
85101776Stjr}
86227753Stheravenwint_t
87227753Stheravenungetwc(wint_t wc, FILE *fp)
88227753Stheraven{
89227753Stheraven	return ungetwc_l(wc, fp, __get_locale());
90227753Stheraven}
91