ncurses 6.1 - patch 20200118
[ncurses.git] / ncurses / widechar / lib_unget_wch.c
1 /****************************************************************************
2  * Copyright (c) 2002-2011,2016 Free Software Foundation, Inc.              *
3  *                                                                          *
4  * Permission is hereby granted, free of charge, to any person obtaining a  *
5  * copy of this software and associated documentation files (the            *
6  * "Software"), to deal in the Software without restriction, including      *
7  * without limitation the rights to use, copy, modify, merge, publish,      *
8  * distribute, distribute with modifications, sublicense, and/or sell       *
9  * copies of the Software, and to permit persons to whom the Software is    *
10  * furnished to do so, subject to the following conditions:                 *
11  *                                                                          *
12  * The above copyright notice and this permission notice shall be included  *
13  * in all copies or substantial portions of the Software.                   *
14  *                                                                          *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS  *
16  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF               *
17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.   *
18  * IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,   *
19  * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR    *
20  * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR    *
21  * THE USE OR OTHER DEALINGS IN THE SOFTWARE.                               *
22  *                                                                          *
23  * Except as contained in this notice, the name(s) of the above copyright   *
24  * holders shall not be used in advertising or otherwise to promote the     *
25  * sale, use or other dealings in this Software without prior written       *
26  * authorization.                                                           *
27  ****************************************************************************/
28
29 /****************************************************************************
30  *  Author: Thomas E. Dickey 2002                                           *
31  ****************************************************************************/
32
33 /*
34 **      lib_unget_wch.c
35 **
36 **      The routine unget_wch().
37 **
38 */
39
40 #include <curses.priv.h>
41
42 MODULE_ID("$Id: lib_unget_wch.c,v 1.16 2016/05/28 23:36:34 tom Exp $")
43
44 /*
45  * Wrapper for wcrtomb() which obtains the length needed for the given
46  * wide-character 'source'.
47  */
48 NCURSES_EXPORT(size_t)
49 _nc_wcrtomb(char *target, wchar_t source, mbstate_t * state)
50 {
51     int result;
52
53     if (target == 0) {
54         wchar_t temp[2];
55         const wchar_t *tempp = temp;
56         temp[0] = source;
57         temp[1] = 0;
58         result = (int) wcsrtombs(NULL, &tempp, (size_t) 0, state);
59     } else {
60         result = (int) wcrtomb(target, source, state);
61     }
62     if (!isEILSEQ(result) && (result == 0))
63         result = 1;
64     return (size_t) result;
65 }
66
67 NCURSES_EXPORT(int)
68 NCURSES_SP_NAME(unget_wch) (NCURSES_SP_DCLx const wchar_t wch)
69 {
70     int result = OK;
71     mbstate_t state;
72     size_t length;
73
74     T((T_CALLED("unget_wch(%p, %#lx)"), (void *) SP_PARM, (unsigned long) wch));
75
76     init_mb(state);
77     length = _nc_wcrtomb(0, wch, &state);
78
79     if (length != (size_t) (-1)
80         && length != 0) {
81         char *string;
82
83         if ((string = (char *) malloc(length)) != 0) {
84             int n;
85
86             init_mb(state);
87             /* ignore the result, since we already validated the character */
88             IGNORE_RC((int) wcrtomb(string, wch, &state));
89
90             for (n = (int) (length - 1); n >= 0; --n) {
91                 if (NCURSES_SP_NAME(ungetch) (NCURSES_SP_ARGx
92                                               UChar(string[n])) !=OK) {
93                     result = ERR;
94                     break;
95                 }
96             }
97             free(string);
98         } else {
99             result = ERR;
100         }
101     } else {
102         result = ERR;
103     }
104
105     returnCode(result);
106 }
107
108 #if NCURSES_SP_FUNCS
109 NCURSES_EXPORT(int)
110 unget_wch(const wchar_t wch)
111 {
112     return NCURSES_SP_NAME(unget_wch) (CURRENT_SCREEN, wch);
113 }
114 #endif