Commit 22f21358 authored by Benjamin Peterson's avatar Benjamin Peterson

fix load_verify_locations on unicode paths (closes #22244)

parent c2d86044
......@@ -850,11 +850,14 @@ class ContextTests(unittest.TestCase):
ctx.load_verify_locations(cafile=CERTFILE, capath=None)
ctx.load_verify_locations(BYTES_CERTFILE)
ctx.load_verify_locations(cafile=BYTES_CERTFILE, capath=None)
ctx.load_verify_locations(cafile=BYTES_CERTFILE.decode('utf-8'))
self.assertRaises(TypeError, ctx.load_verify_locations)
self.assertRaises(TypeError, ctx.load_verify_locations, None, None, None)
with self.assertRaises(IOError) as cm:
ctx.load_verify_locations(WRONGCERT)
self.assertEqual(cm.exception.errno, errno.ENOENT)
with self.assertRaises(IOError):
ctx.load_verify_locations(u'')
with self.assertRaisesRegexp(ssl.SSLError, "PEM lib"):
ctx.load_verify_locations(BADCERT)
ctx.load_verify_locations(CERTFILE, CAPATH)
......
......@@ -2628,16 +2628,32 @@ load_verify_locations(PySSLContext *self, PyObject *args, PyObject *kwds)
}
if (cafile) {
cafile_bytes = PyString_AsEncodedObject(
cafile, Py_FileSystemDefaultEncoding, "strict");
if (!cafile_bytes) {
if (PyString_Check(cafile)) {
Py_INCREF(cafile);
cafile_bytes = cafile;
} else {
PyObject *u = PyUnicode_FromObject(cafile);
if (!u)
goto error;
cafile_bytes = PyUnicode_AsEncodedString(
u, Py_FileSystemDefaultEncoding, NULL);
Py_DECREF(u);
if (!cafile_bytes)
goto error;
}
}
if (capath) {
capath_bytes = PyString_AsEncodedObject(
capath, Py_FileSystemDefaultEncoding, "strict");
if (!capath_bytes) {
if (PyString_Check(capath)) {
Py_INCREF(capath);
capath_bytes = capath;
} else {
PyObject *u = PyUnicode_FromObject(capath);
if (!u)
goto error;
capath_bytes = PyUnicode_AsEncodedString(
u, Py_FileSystemDefaultEncoding, NULL);
Py_DECREF(u);
if (!capath_bytes)
goto error;
}
}
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment