diff options
author | Andrew M. Kuchling <amk@amk.ca> | 2009-09-25 22:23:54 +0000 |
---|---|---|
committer | Andrew M. Kuchling <amk@amk.ca> | 2009-09-25 22:23:54 +0000 |
commit | b49e53ee16eba81fb9548843d24d1cc328f2095a (patch) | |
tree | a507805bd930a95bf0798bc6de3d9441f3414179 | |
parent | a9a7611fb6b1a69d0ab9474da2b9cf9e8b294b54 (diff) | |
download | cpython-git-b49e53ee16eba81fb9548843d24d1cc328f2095a.tar.gz |
#6243: fix segfault when keyname() returns a NULL pointer.
Bug noted by Trundle, patched by Trundle and Jerry Chen.
-rw-r--r-- | Lib/test/test_curses.py | 5 | ||||
-rw-r--r-- | Modules/_cursesmodule.c | 11 |
2 files changed, 12 insertions, 4 deletions
diff --git a/Lib/test/test_curses.py b/Lib/test/test_curses.py index 1b3489c49b..2f7c04c3c7 100644 --- a/Lib/test/test_curses.py +++ b/Lib/test/test_curses.py @@ -257,6 +257,10 @@ def test_resize_term(stdscr): if curses.LINES != lines - 1 or curses.COLS != cols + 1: raise RuntimeError, "Expected resizeterm to update LINES and COLS" +def test_issue6243(stdscr): + curses.ungetch(1025) + stdscr.getkey() + def main(stdscr): curses.savetty() try: @@ -264,6 +268,7 @@ def main(stdscr): window_funcs(stdscr) test_userptr_without_set(stdscr) test_resize_term(stdscr) + test_issue6243(stdscr) finally: curses.resetty() diff --git a/Modules/_cursesmodule.c b/Modules/_cursesmodule.c index cd143a4ae1..53fa432dc0 100644 --- a/Modules/_cursesmodule.c +++ b/Modules/_cursesmodule.c @@ -882,14 +882,17 @@ PyCursesWindow_GetKey(PyCursesWindowObject *self, PyObject *args) /* getch() returns ERR in nodelay mode */ PyErr_SetString(PyCursesError, "no input"); return NULL; - } else if (rtn<=255) + } else if (rtn<=255) { return Py_BuildValue("c", rtn); - else + } else { + const char *knp; #if defined(__NetBSD__) - return PyString_FromString(unctrl(rtn)); + knp = unctrl(rtn); #else - return PyString_FromString((char *)keyname(rtn)); + knp = keyname(rtn); #endif + return PyString_FromString((knp == NULL) ? "" : knp); + } } static PyObject * |