diff -r 4d1236b180be Include/unicodeobject.h --- a/Include/unicodeobject.h Mon Apr 13 23:12:42 2015 -0700 +++ b/Include/unicodeobject.h Tue Apr 14 14:33:48 2015 +0300 @@ -2060,12 +2060,6 @@ PyAPI_FUNC(int) PyUnicode_Contains( PyObject *element /* Element string */ ); -/* Checks whether the string contains any NUL characters. */ - -#ifndef Py_LIMITED_API -PyAPI_FUNC(int) _PyUnicode_HasNULChars(PyObject *); -#endif - /* Checks whether argument is a valid identifier. */ PyAPI_FUNC(int) PyUnicode_IsIdentifier(PyObject *s); diff -r 4d1236b180be Lib/test/test_posix.py --- a/Lib/test/test_posix.py Mon Apr 13 23:12:42 2015 -0700 +++ b/Lib/test/test_posix.py Tue Apr 14 14:33:48 2015 +0300 @@ -1138,6 +1138,42 @@ class PosixTester(unittest.TestCase): else: self.fail("No valid path_error2() test for os." + name) + def test_path_with_null_character(self): + fn = support.TESTFN + fn_with_NUL = fn + '\0' + self.addCleanup(support.unlink, fn) + support.unlink(fn) + fd = None + try: + with self.assertRaises(ValueError): + fd = os.open(fn_with_NUL, os.O_WRONLY | os.O_CREAT) # raises + finally: + if fd is not None: + os.close(fd) + self.assertFalse(os.path.exists(fn)) + self.assertRaises(ValueError, os.mkdir, fn_with_NUL) + self.assertFalse(os.path.exists(fn)) + open(fn, 'wb').close() + self.assertRaises(ValueError, os.stat, fn_with_NUL) + + def test_path_with_null_byte(self): + fn = os.fsencode(support.TESTFN) + fn_with_NUL = fn + b'\0' + self.addCleanup(support.unlink, fn) + support.unlink(fn) + fd = None + try: + with self.assertRaises(ValueError): + fd = os.open(fn_with_NUL, os.O_WRONLY | os.O_CREAT) # raises + finally: + if fd is not None: + os.close(fd) + self.assertFalse(os.path.exists(fn)) + self.assertRaises(ValueError, os.mkdir, fn_with_NUL) + self.assertFalse(os.path.exists(fn)) + open(fn, 'wb').close() + self.assertRaises(ValueError, os.stat, fn_with_NUL) + class PosixGroupsTester(unittest.TestCase): def setUp(self): diff -r 4d1236b180be Modules/_io/fileio.c --- a/Modules/_io/fileio.c Mon Apr 13 23:12:42 2015 -0700 +++ b/Modules/_io/fileio.c Tue Apr 14 14:33:48 2015 +0300 @@ -243,15 +243,14 @@ fileio_init(PyObject *oself, PyObject *a #ifdef MS_WINDOWS if (PyUnicode_Check(nameobj)) { - int rv = _PyUnicode_HasNULChars(nameobj); - if (rv) { - if (rv != -1) - PyErr_SetString(PyExc_ValueError, "embedded null character"); + Py_ssize_t length; + widename = PyUnicode_AsUnicodeAndSize(nameobj, &length); + if (widename == NULL) + return -1; + if (wcslen(widename) != length) { + PyErr_SetString(PyExc_ValueError, "embedded null character"); return -1; } - widename = PyUnicode_AsUnicode(nameobj); - if (widename == NULL) - return -1; } else #endif if (fd < 0) diff -r 4d1236b180be Modules/posixmodule.c --- a/Modules/posixmodule.c Mon Apr 13 23:12:42 2015 -0700 +++ b/Modules/posixmodule.c Tue Apr 14 14:33:48 2015 +0300 @@ -866,6 +866,11 @@ path_converter(PyObject *o, void *p) { Py_DECREF(unicode); return 0; } + if (wcslen(wide) != length) { + FORMAT_EXCEPTION(PyExc_ValueError, "embedded null character"); + Py_DECREF(unicode); + return 0; + } path->wide = wide; path->narrow = NULL; diff -r 4d1236b180be Objects/unicodeobject.c --- a/Objects/unicodeobject.c Mon Apr 13 23:12:42 2015 -0700 +++ b/Objects/unicodeobject.c Tue Apr 14 14:33:48 2015 +0300 @@ -3607,21 +3607,6 @@ PyUnicode_DecodeFSDefaultAndSize(const c int -_PyUnicode_HasNULChars(PyObject* str) -{ - Py_ssize_t pos; - - if (PyUnicode_READY(str) == -1) - return -1; - pos = findchar(PyUnicode_DATA(str), PyUnicode_KIND(str), - PyUnicode_GET_LENGTH(str), '\0', 1); - if (pos == -1) - return 0; - else - return 1; -} - -int PyUnicode_FSConverter(PyObject* arg, void* addr) { PyObject *output = NULL;