ncurses 6.2 - patch 20210418

[ncurses.git] / ncurses / widechar / lib_get_wch.c

diff --git a/ncurses/widechar/lib_get_wch.c b/ncurses/widechar/lib_get_wch.c
index fa024d2ccb047903ce7a4c0c0e838bef1735f269..9f10d3bcf121063a791e411602c3801d699e5650 100644 (file)
--- a/ncurses/widechar/lib_get_wch.c
+++ b/ncurses/widechar/lib_get_wch.c
@@ -1,5 +1,6 @@
 /****************************************************************************
- * Copyright (c) 2002-2008,2009 Free Software Foundation, Inc.              *
+ * Copyright 2020,2021 Thomas E. Dickey                                     *
+ * Copyright 2002-2011,2016 Free Software Foundation, Inc.                  *
  *                                                                          *
  * Permission is hereby granted, free of charge, to any person obtaining a  *
  * copy of this software and associated documentation files (the            *
@@ -40,18 +41,14 @@
 #include <curses.priv.h>
 #include <ctype.h>
 
-MODULE_ID("$Id: lib_get_wch.c,v 1.20 2009/10/24 22:42:25 tom Exp $")
+MODULE_ID("$Id: lib_get_wch.c,v 1.26 2021/04/17 16:12:54 tom Exp $")
 
 NCURSES_EXPORT(int)
 wget_wch(WINDOW *win, wint_t *result)
 {
     SCREEN *sp;
     int code;
-    char buffer[(MB_LEN_MAX * 9) + 1]; /* allow some redundant shifts */
-    int status;
-    size_t count = 0;
-    unsigned long value;
-    wchar_t wch;
+    int value = 0;
 #ifndef state_unused
     mbstate_t state;
 #endif
@@ -64,8 +61,13 @@ wget_wch(WINDOW *win, wint_t *result)
      */
     _nc_lock_global(curses);
     sp = _nc_screen_of(win);
+
     if (sp != 0) {
+       size_t count = 0;
+
        for (;;) {
+           char buffer[(MB_LEN_MAX * 9) + 1];  /* allow some redundant shifts */
+
            T(("reading %d of %d", (int) count + 1, (int) sizeof(buffer)));
            code = _nc_wgetch(win, &value, TRUE EVENTLIST_2nd((_nc_eventlist
                                                               *) 0));
@@ -80,23 +82,26 @@ wget_wch(WINDOW *win, wint_t *result)
                 * whether the improvement would be worth the effort.
                 */
                if (count != 0) {
-                   safe_ungetch(SP_PARM, (int) value);
+                   safe_ungetch(SP_PARM, value);
                    code = ERR;
                }
                break;
            } else if (count + 1 >= sizeof(buffer)) {
-               safe_ungetch(SP_PARM, (int) value);
+               safe_ungetch(SP_PARM, value);
                code = ERR;
                break;
            } else {
+               int status;
+
                buffer[count++] = (char) UChar(value);
                reset_mbytes(state);
                status = count_mbytes(buffer, count, state);
                if (status >= 0) {
+                   wchar_t wch;
                    reset_mbytes(state);
                    if (check_mbytes(wch, buffer, count, state) != status) {
                        code = ERR;     /* the two calls should match */
-                       safe_ungetch(SP_PARM, (int) value);
+                       safe_ungetch(SP_PARM, value);
                    }
                    value = wch;
                    break;
@@ -106,8 +111,11 @@ wget_wch(WINDOW *win, wint_t *result)
     } else {
        code = ERR;
     }
-    *result = value;
+
+    if (result != 0)
+       *result = (wint_t) value;
+
     _nc_unlock_global(curses);
-    T(("result %#lo", value));
+    T(("result %#o", value));
     returnCode(code);
 }