Kaydet (Commit) 90374f51 authored tarafından Tim Golden's avatar Tim Golden

Issue #3210: Ensure stdio handles are closed if CreateProcess fails

üst 3ad323ec
......@@ -533,6 +533,26 @@ class ProcessTestCase(BaseTestCase):
if c.exception.errno != 2: # ignore "no such file"
raise c.exception
def test_handles_closed_on_exception(self):
# If CreateProcess exits with an error, ensure the
# duplicate output handles are released
ifhandle, ifname = mkstemp()
ofhandle, ofname = mkstemp()
efhandle, efname = mkstemp()
try:
subprocess.Popen (["*"], stdin=ifhandle, stdout=ofhandle,
stderr=efhandle)
except OSError:
os.close(ifhandle)
os.remove(ifname)
os.close(ofhandle)
os.remove(ofname)
os.close(efhandle)
os.remove(efname)
self.assertFalse(os.path.exists(ifname))
self.assertFalse(os.path.exists(ofname))
self.assertFalse(os.path.exists(efname))
# context manager
class _SuppressCoreFiles(object):
......
......@@ -425,6 +425,7 @@ sp_CreateProcess(PyObject* self, PyObject* args)
PyObject* env_mapping;
char* current_directory;
PyObject* startup_info;
DWORD error;
if (! PyArg_ParseTuple(args, "zzOOiiOzO:CreateProcess",
&application_name,
......@@ -474,8 +475,22 @@ sp_CreateProcess(PyObject* self, PyObject* args)
Py_XDECREF(environment);
if (! result)
return PyErr_SetFromWindowsErr(GetLastError());
if (! result) {
error = GetLastError();
if(si.hStdInput != INVALID_HANDLE_VALUE) {
CloseHandle(si.hStdInput);
si.hStdInput = INVALID_HANDLE_VALUE;
}
if(si.hStdOutput != INVALID_HANDLE_VALUE) {
CloseHandle(si.hStdOutput);
si.hStdOutput = INVALID_HANDLE_VALUE;
}
if(si.hStdError != INVALID_HANDLE_VALUE) {
CloseHandle(si.hStdError);
si.hStdError = INVALID_HANDLE_VALUE;
}
return PyErr_SetFromWindowsErr(error);
}
return Py_BuildValue("NNii",
sp_handle_new(pi.hProcess),
......
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