diff options
author | Jesse Noller <jnoller@gmail.com> | 2009-04-02 04:22:09 (GMT) |
---|---|---|
committer | Jesse Noller <jnoller@gmail.com> | 2009-04-02 04:22:09 (GMT) |
commit | 5053fbbb38b7b8bc8207e2d27c32c7b82ce3172f (patch) | |
tree | 530d74ee5a347857bd40d4f3fb559cc716bdb913 /Modules/_multiprocessing | |
parent | 238cedcd6339e747daa6e2a383678d893482ef6d (diff) | |
download | cpython-5053fbbb38b7b8bc8207e2d27c32c7b82ce3172f.zip cpython-5053fbbb38b7b8bc8207e2d27c32c7b82ce3172f.tar.gz cpython-5053fbbb38b7b8bc8207e2d27c32c7b82ce3172f.tar.bz2 |
Issue 3551: Raise ValueError if the size causes ERROR_NO_SYSTEM_RESOURCES
Diffstat (limited to 'Modules/_multiprocessing')
-rw-r--r-- | Modules/_multiprocessing/connection.h | 8 | ||||
-rw-r--r-- | Modules/_multiprocessing/pipe_connection.c | 6 |
2 files changed, 12 insertions, 2 deletions
diff --git a/Modules/_multiprocessing/connection.h b/Modules/_multiprocessing/connection.h index 458d1d3..c2f8563 100644 --- a/Modules/_multiprocessing/connection.h +++ b/Modules/_multiprocessing/connection.h @@ -131,8 +131,12 @@ connection_sendbytes(ConnectionObject *self, PyObject *args) res = conn_send_string(self, buffer + offset, size); - if (res < 0) - return mp_SetError(PyExc_IOError, res); + if (res < 0) { + if (PyErr_Occurred()) + return NULL; + else + return mp_SetError(PyExc_IOError, res); + } Py_RETURN_NONE; } diff --git a/Modules/_multiprocessing/pipe_connection.c b/Modules/_multiprocessing/pipe_connection.c index 27e79dd..66947c8 100644 --- a/Modules/_multiprocessing/pipe_connection.c +++ b/Modules/_multiprocessing/pipe_connection.c @@ -23,6 +23,12 @@ conn_send_string(ConnectionObject *conn, char *string, size_t length) Py_BEGIN_ALLOW_THREADS ret = WriteFile(conn->handle, string, length, &amount_written, NULL); Py_END_ALLOW_THREADS + + if (ret == 0 && GetLastError() == ERROR_NO_SYSTEM_RESOURCES) { + PyErr_Format(PyExc_ValueError, "Cannnot send %" PY_FORMAT_SIZE_T "d bytes over connection", length); + return MP_STANDARD_ERROR; + } + return ret ? MP_SUCCESS : MP_STANDARD_ERROR; } |