overlapped.c: Fix usage of the union

* read_buffer can only be used for TYPE_READ and TYPE_ACCEPT types
* write_buffer can only be used for TYPE_WRITE type
This commit is contained in:
Victor Stinner 2014-01-30 19:06:44 +01:00
parent 63a615cfa1
commit 91445fbeb0
1 changed files with 11 additions and 9 deletions

View File

@ -45,9 +45,9 @@ typedef struct {
/* Type of operation */
DWORD type;
union {
/* Buffer used for reading (optional) */
/* Buffer used for reading: TYPE_READ and TYPE_ACCEPT */
PyObject *read_buffer;
/* Buffer used for writing (optional) */
/* Buffer used for writing: TYPE_WRITE */
Py_buffer write_buffer;
};
} OverlappedObject;
@ -568,13 +568,15 @@ Overlapped_dealloc(OverlappedObject *self)
if (self->overlapped.hEvent != NULL)
CloseHandle(self->overlapped.hEvent);
if (self->write_buffer.obj)
PyBuffer_Release(&self->write_buffer);
switch (self->type) {
case TYPE_READ:
case TYPE_ACCEPT:
Py_CLEAR(self->read_buffer);
case TYPE_READ:
case TYPE_ACCEPT:
Py_CLEAR(self->read_buffer);
break;
case TYPE_WRITE:
if (self->write_buffer.obj)
PyBuffer_Release(&self->write_buffer);
break;
}
PyObject_Del(self);
SetLastError(olderr);
@ -648,7 +650,7 @@ Overlapped_getresult(OverlappedObject *self, PyObject *args)
case ERROR_MORE_DATA:
break;
case ERROR_BROKEN_PIPE:
if (self->read_buffer != NULL)
if ((self->type == TYPE_READ || self->type == TYPE_ACCEPT) && self->read_buffer != NULL)
break;
/* fall through */
default: