1  | 
     | 
     | 
    /* $OpenBSD: lib_get_wch.c,v 1.1 2010/09/06 17:26:17 nicm Exp $ */  | 
    
    
    2  | 
     | 
     | 
     | 
    
    
    3  | 
     | 
     | 
    /****************************************************************************  | 
    
    
    4  | 
     | 
     | 
     * Copyright (c) 2002-2007,2008 Free Software Foundation, Inc.              *  | 
    
    
    5  | 
     | 
     | 
     *                                                                          *  | 
    
    
    6  | 
     | 
     | 
     * Permission is hereby granted, free of charge, to any person obtaining a  *  | 
    
    
    7  | 
     | 
     | 
     * copy of this software and associated documentation files (the            *  | 
    
    
    8  | 
     | 
     | 
     * "Software"), to deal in the Software without restriction, including      *  | 
    
    
    9  | 
     | 
     | 
     * without limitation the rights to use, copy, modify, merge, publish,      *  | 
    
    
    10  | 
     | 
     | 
     * distribute, distribute with modifications, sublicense, and/or sell       *  | 
    
    
    11  | 
     | 
     | 
     * copies of the Software, and to permit persons to whom the Software is    *  | 
    
    
    12  | 
     | 
     | 
     * furnished to do so, subject to the following conditions:                 *  | 
    
    
    13  | 
     | 
     | 
     *                                                                          *  | 
    
    
    14  | 
     | 
     | 
     * The above copyright notice and this permission notice shall be included  *  | 
    
    
    15  | 
     | 
     | 
     * in all copies or substantial portions of the Software.                   *  | 
    
    
    16  | 
     | 
     | 
     *                                                                          *  | 
    
    
    17  | 
     | 
     | 
     * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS  *  | 
    
    
    18  | 
     | 
     | 
     * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF               *  | 
    
    
    19  | 
     | 
     | 
     * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.   *  | 
    
    
    20  | 
     | 
     | 
     * IN NO EVENT SHALL THE ABOVE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,   *  | 
    
    
    21  | 
     | 
     | 
     * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR    *  | 
    
    
    22  | 
     | 
     | 
     * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR    *  | 
    
    
    23  | 
     | 
     | 
     * THE USE OR OTHER DEALINGS IN THE SOFTWARE.                               *  | 
    
    
    24  | 
     | 
     | 
     *                                                                          *  | 
    
    
    25  | 
     | 
     | 
     * Except as contained in this notice, the name(s) of the above copyright   *  | 
    
    
    26  | 
     | 
     | 
     * holders shall not be used in advertising or otherwise to promote the     *  | 
    
    
    27  | 
     | 
     | 
     * sale, use or other dealings in this Software without prior written       *  | 
    
    
    28  | 
     | 
     | 
     * authorization.                                                           *  | 
    
    
    29  | 
     | 
     | 
     ****************************************************************************/  | 
    
    
    30  | 
     | 
     | 
     | 
    
    
    31  | 
     | 
     | 
    /****************************************************************************  | 
    
    
    32  | 
     | 
     | 
     *  Author: Thomas E. Dickey 2002-on                                        *  | 
    
    
    33  | 
     | 
     | 
     ****************************************************************************/  | 
    
    
    34  | 
     | 
     | 
     | 
    
    
    35  | 
     | 
     | 
    /*  | 
    
    
    36  | 
     | 
     | 
    **	lib_get_wch.c  | 
    
    
    37  | 
     | 
     | 
    **  | 
    
    
    38  | 
     | 
     | 
    **	The routine get_wch().  | 
    
    
    39  | 
     | 
     | 
    **  | 
    
    
    40  | 
     | 
     | 
    */  | 
    
    
    41  | 
     | 
     | 
     | 
    
    
    42  | 
     | 
     | 
    #include <curses.priv.h>  | 
    
    
    43  | 
     | 
     | 
    #include <ctype.h>  | 
    
    
    44  | 
     | 
     | 
     | 
    
    
    45  | 
     | 
     | 
    MODULE_ID("$Id: lib_get_wch.c,v 1.1 2010/09/06 17:26:17 nicm Exp $") | 
    
    
    46  | 
     | 
     | 
     | 
    
    
    47  | 
     | 
     | 
    #if HAVE_MBTOWC && HAVE_MBLEN  | 
    
    
    48  | 
     | 
     | 
    #define reset_mbytes(state) mblen(NULL, 0), mbtowc(NULL, NULL, 0)  | 
    
    
    49  | 
     | 
     | 
    #define count_mbytes(buffer,length,state) mblen(buffer,length)  | 
    
    
    50  | 
     | 
     | 
    #define check_mbytes(wch,buffer,length,state) \  | 
    
    
    51  | 
     | 
     | 
    	(int) mbtowc(&wch, buffer, length)  | 
    
    
    52  | 
     | 
     | 
    #define state_unused  | 
    
    
    53  | 
     | 
     | 
    #elif HAVE_MBRTOWC && HAVE_MBRLEN  | 
    
    
    54  | 
     | 
     | 
    #define reset_mbytes(state) init_mb(state)  | 
    
    
    55  | 
     | 
     | 
    #define count_mbytes(buffer,length,state) mbrlen(buffer,length,&state)  | 
    
    
    56  | 
     | 
     | 
    #define check_mbytes(wch,buffer,length,state) \  | 
    
    
    57  | 
     | 
     | 
    	(int) mbrtowc(&wch, buffer, length, &state)  | 
    
    
    58  | 
     | 
     | 
    #else  | 
    
    
    59  | 
     | 
     | 
    make an error  | 
    
    
    60  | 
     | 
     | 
    #endif  | 
    
    
    61  | 
     | 
     | 
     | 
    
    
    62  | 
     | 
     | 
    NCURSES_EXPORT(int)  | 
    
    
    63  | 
     | 
     | 
    wget_wch(WINDOW *win, wint_t *result)  | 
    
    
    64  | 
     | 
     | 
    { | 
    
    
    65  | 
     | 
     | 
        SCREEN *sp;  | 
    
    
    66  | 
     | 
     | 
        int code;  | 
    
    
    67  | 
     | 
     | 
        char buffer[(MB_LEN_MAX * 9) + 1];	/* allow some redundant shifts */  | 
    
    
    68  | 
     | 
     | 
        int status;  | 
    
    
    69  | 
     | 
     | 
        size_t count = 0;  | 
    
    
    70  | 
     | 
     | 
        unsigned long value;  | 
    
    
    71  | 
     | 
     | 
        wchar_t wch;  | 
    
    
    72  | 
     | 
     | 
    #ifndef state_unused  | 
    
    
    73  | 
     | 
     | 
        mbstate_t state;  | 
    
    
    74  | 
     | 
     | 
    #endif  | 
    
    
    75  | 
     | 
     | 
     | 
    
    
    76  | 
     | 
     | 
        T((T_CALLED("wget_wch(%p)"), win)); | 
    
    
    77  | 
     | 
     | 
     | 
    
    
    78  | 
     | 
     | 
        /*  | 
    
    
    79  | 
     | 
     | 
         * We can get a stream of single-byte characters and KEY_xxx codes from  | 
    
    
    80  | 
     | 
     | 
         * _nc_wgetch(), while we want to return a wide character or KEY_xxx code.  | 
    
    
    81  | 
     | 
     | 
         */  | 
    
    
    82  | 
     | 
     | 
        _nc_lock_global(curses);  | 
    
    
    83  | 
     | 
     | 
        sp = _nc_screen_of(win);  | 
    
    
    84  | 
     | 
     | 
        if (sp != 0) { | 
    
    
    85  | 
     | 
     | 
    	for (;;) { | 
    
    
    86  | 
     | 
     | 
    	    T(("reading %d of %d", (int) count + 1, (int) sizeof(buffer))); | 
    
    
    87  | 
     | 
     | 
    	    code = _nc_wgetch(win, &value, TRUE EVENTLIST_2nd((_nc_eventlist  | 
    
    
    88  | 
     | 
     | 
    							       *) 0));  | 
    
    
    89  | 
     | 
     | 
    	    if (code == ERR) { | 
    
    
    90  | 
     | 
     | 
    		break;  | 
    
    
    91  | 
     | 
     | 
    	    } else if (code == KEY_CODE_YES) { | 
    
    
    92  | 
     | 
     | 
    		/*  | 
    
    
    93  | 
     | 
     | 
    		 * If we were processing an incomplete multibyte character,  | 
    
    
    94  | 
     | 
     | 
    		 * return an error since we have a KEY_xxx code which  | 
    
    
    95  | 
     | 
     | 
    		 * interrupts it.  For some cases, we could improve this by  | 
    
    
    96  | 
     | 
     | 
    		 * writing a new version of lib_getch.c(!), but it is not clear  | 
    
    
    97  | 
     | 
     | 
    		 * whether the improvement would be worth the effort.  | 
    
    
    98  | 
     | 
     | 
    		 */  | 
    
    
    99  | 
     | 
     | 
    		if (count != 0) { | 
    
    
    100  | 
     | 
     | 
    		    _nc_ungetch(sp, (int) value);  | 
    
    
    101  | 
     | 
     | 
    		    code = ERR;  | 
    
    
    102  | 
     | 
     | 
    		}  | 
    
    
    103  | 
     | 
     | 
    		break;  | 
    
    
    104  | 
     | 
     | 
    	    } else if (count + 1 >= sizeof(buffer)) { | 
    
    
    105  | 
     | 
     | 
    		_nc_ungetch(sp, (int) value);  | 
    
    
    106  | 
     | 
     | 
    		code = ERR;  | 
    
    
    107  | 
     | 
     | 
    		break;  | 
    
    
    108  | 
     | 
     | 
    	    } else { | 
    
    
    109  | 
     | 
     | 
    		buffer[count++] = (char) UChar(value);  | 
    
    
    110  | 
     | 
     | 
    		reset_mbytes(state);  | 
    
    
    111  | 
     | 
     | 
    		status = count_mbytes(buffer, count, state);  | 
    
    
    112  | 
     | 
     | 
    		if (status >= 0) { | 
    
    
    113  | 
     | 
     | 
    		    reset_mbytes(state);  | 
    
    
    114  | 
     | 
     | 
    		    if (check_mbytes(wch, buffer, count, state) != status) { | 
    
    
    115  | 
     | 
     | 
    			code = ERR;	/* the two calls should match */  | 
    
    
    116  | 
     | 
     | 
    			_nc_ungetch(sp, (int) value);  | 
    
    
    117  | 
     | 
     | 
    		    }  | 
    
    
    118  | 
     | 
     | 
    		    value = wch;  | 
    
    
    119  | 
     | 
     | 
    		    break;  | 
    
    
    120  | 
     | 
     | 
    		}  | 
    
    
    121  | 
     | 
     | 
    	    }  | 
    
    
    122  | 
     | 
     | 
    	}  | 
    
    
    123  | 
     | 
     | 
        } else { | 
    
    
    124  | 
     | 
     | 
    	code = ERR;  | 
    
    
    125  | 
     | 
     | 
        }  | 
    
    
    126  | 
     | 
     | 
        *result = value;  | 
    
    
    127  | 
     | 
     | 
        _nc_unlock_global(curses);  | 
    
    
    128  | 
     | 
     | 
        T(("result %#lo", value)); | 
    
    
    129  | 
     | 
     | 
        returnCode(code);  | 
    
    
    130  | 
     | 
     | 
    }  |