ncurses 5.9 - patch 20111022
[ncurses.git] / ncurses / widechar / lib_get_wch.c
1 /****************************************************************************
2  * Copyright (c) 2002-2010,2011 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-on                                        *
31  ****************************************************************************/
32
33 /*
34 **      lib_get_wch.c
35 **
36 **      The routine get_wch().
37 **
38 */
39
40 #include <curses.priv.h>
41 #include <ctype.h>
42
43 MODULE_ID("$Id: lib_get_wch.c,v 1.23 2011/05/28 23:00:29 tom Exp $")
44
45 NCURSES_EXPORT(int)
46 wget_wch(WINDOW *win, wint_t *result)
47 {
48     SCREEN *sp;
49     int code;
50     char buffer[(MB_LEN_MAX * 9) + 1];  /* allow some redundant shifts */
51     int status;
52     size_t count = 0;
53     int value = 0;
54     wchar_t wch;
55 #ifndef state_unused
56     mbstate_t state;
57 #endif
58
59     T((T_CALLED("wget_wch(%p)"), (void *) win));
60
61     /*
62      * We can get a stream of single-byte characters and KEY_xxx codes from
63      * _nc_wgetch(), while we want to return a wide character or KEY_xxx code.
64      */
65     _nc_lock_global(curses);
66     sp = _nc_screen_of(win);
67     if (sp != 0) {
68         for (;;) {
69             T(("reading %d of %d", (int) count + 1, (int) sizeof(buffer)));
70             code = _nc_wgetch(win, &value, TRUE EVENTLIST_2nd((_nc_eventlist
71                                                                *) 0));
72             if (code == ERR) {
73                 break;
74             } else if (code == KEY_CODE_YES) {
75                 /*
76                  * If we were processing an incomplete multibyte character,
77                  * return an error since we have a KEY_xxx code which
78                  * interrupts it.  For some cases, we could improve this by
79                  * writing a new version of lib_getch.c(!), but it is not clear
80                  * whether the improvement would be worth the effort.
81                  */
82                 if (count != 0) {
83                     safe_ungetch(SP_PARM, value);
84                     code = ERR;
85                 }
86                 break;
87             } else if (count + 1 >= sizeof(buffer)) {
88                 safe_ungetch(SP_PARM, value);
89                 code = ERR;
90                 break;
91             } else {
92                 buffer[count++] = (char) UChar(value);
93                 reset_mbytes(state);
94                 status = count_mbytes(buffer, count, state);
95                 if (status >= 0) {
96                     reset_mbytes(state);
97                     if (check_mbytes(wch, buffer, count, state) != status) {
98                         code = ERR;     /* the two calls should match */
99                         safe_ungetch(SP_PARM, value);
100                     }
101                     value = wch;
102                     break;
103                 }
104             }
105         }
106     } else {
107         code = ERR;
108     }
109
110     if (result != 0)
111         *result = (wint_t) value;
112
113     _nc_unlock_global(curses);
114     T(("result %#o", value));
115     returnCode(code);
116 }