1212525Simp/*-
2212525Simp * Copyright (c) 2002-2004 Tim J. Robbins.
3212525Simp * All rights reserved.
4212525Simp *
5212525Simp * Copyright (c) 2011 The FreeBSD Foundation
6212525Simp * All rights reserved.
7212525Simp * Portions of this software were developed by David Chisnall
8212525Simp * under sponsorship from the FreeBSD Foundation.
9212525Simp *
10212525Simp * Redistribution and use in source and binary forms, with or without
11212525Simp * modification, are permitted provided that the following conditions
12212525Simp * are met:
13245652Sneel * 1. Redistributions of source code must retain the above copyright
14245652Sneel *    notice, this list of conditions and the following disclaimer.
15245652Sneel * 2. Redistributions in binary form must reproduce the above copyright
16212525Simp *    notice, this list of conditions and the following disclaimer in the
17212525Simp *    documentation and/or other materials provided with the distribution.
18212525Simp *
19212525Simp * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20212525Simp * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21212525Simp * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22212525Simp * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23252356Sdavide * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24212525Simp * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25212525Simp * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26212525Simp * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27212525Simp * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28212525Simp * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29212525Simp * SUCH DAMAGE.
30212525Simp */
31
32#include <sys/cdefs.h>
33__FBSDID("$FreeBSD: stable/10/lib/libc/stdio/ungetwc.c 321074 2017-07-17 14:09:34Z kib $");
34
35#include "namespace.h"
36#include <errno.h>
37#include <limits.h>
38#include <stdio.h>
39#include <stdlib.h>
40#include <wchar.h>
41#include "un-namespace.h"
42#include "libc_private.h"
43#include "local.h"
44#include "mblocal.h"
45#include "xlocale_private.h"
46
47/*
48 * Non-MT-safe version.
49 */
50wint_t
51__ungetwc(wint_t wc, FILE *fp, locale_t locale)
52{
53	char buf[MB_LEN_MAX];
54	size_t len;
55	struct xlocale_ctype *l = XLOCALE_CTYPE(locale);
56
57	if (wc == WEOF)
58		return (WEOF);
59	if ((len = l->__wcrtomb(buf, wc, &fp->_mbstate)) == (size_t)-1) {
60		fp->_flags |= __SERR;
61		return (WEOF);
62	}
63	while (len-- != 0)
64		if (__ungetc((unsigned char)buf[len], fp) == EOF)
65			return (WEOF);
66
67	return (wc);
68}
69
70/*
71 * MT-safe version.
72 */
73wint_t
74ungetwc_l(wint_t wc, FILE *fp, locale_t locale)
75{
76	wint_t r;
77	FIX_LOCALE(locale);
78
79	FLOCKFILE_CANCELSAFE(fp);
80	ORIENT(fp, 1);
81	r = __ungetwc(wc, fp, locale);
82	FUNLOCKFILE_CANCELSAFE();
83
84	return (r);
85}
86wint_t
87ungetwc(wint_t wc, FILE *fp)
88{
89	return ungetwc_l(wc, fp, __get_locale());
90}
91