1 /****************************************************************************
2  * Copyright (c) 2002-2007,2008 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.10 2008/06/07 14:50:37 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 = wcsrtombs(NULL, &tempp, 0, state);
59     } else {
60 	result = wcrtomb(target, source, state);
61     }
62     if (!isEILSEQ(result) && (result == 0))
63 	result = 1;
64     return result;
65 }
66 
67 NCURSES_EXPORT(int)
68 unget_wch(const wchar_t wch)
69 {
70     int result = OK;
71     mbstate_t state;
72     size_t length;
73     int n;
74 
75     T((T_CALLED("unget_wch(%#lx)"), (unsigned long) wch));
76 
77     init_mb(state);
78     length = _nc_wcrtomb(0, wch, &state);
79 
80     if (length != (size_t) (-1)
81 	&& length != 0) {
82 	char *string;
83 
84 	if ((string = (char *) malloc(length)) != 0) {
85 	    init_mb(state);
86 	    wcrtomb(string, wch, &state);
87 
88 	    for (n = (int) (length - 1); n >= 0; --n) {
89 		if (_nc_ungetch(SP, string[n]) != OK) {
90 		    result = ERR;
91 		    break;
92 		}
93 	    }
94 	    free(string);
95 	} else {
96 	    result = ERR;
97 	}
98     } else {
99 	result = ERR;
100     }
101 
102     returnCode(result);
103 }
104