8000 bpo-47003: Cleanup _overlapped module by asvetlov · Pull Request #31848 · python/cpython · GitHub
[go: up one dir, main page]
Skip to content

bpo-47003: Cleanup _overlapped module #31848

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Mar 13, 2022
Merged

bpo-47003: Cleanup _overlapped module #31848

Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
8f635f4
Cleanup _overlapped module
asvetlov Mar 13, 2022
2243f0b
Run CancelIOEx with GIL released
asvetlov Mar 13, 2022
357df44
Fix compiler errors
asvetlov Mar 13, 2022
720fd84
Fix memory management
asvetlov Mar 13, 2022
50b1380
Fix compile error
asvetlov Mar 13, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix compiler errors
  • Loading branch information
@asvetlov
asvetlov committed Mar 13, 2022
commit 357df44fcef6df98497eb7178183e01775eb3209
33 changes: 16 additions & 17 deletions Modules/overlapped.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ typedef struct {
/* Buffer allocated by us: TYPE_READ and TYPE_ACCEPT */
PyObject *allocated_buffer;
/* Buffer passed by the user: TYPE_WRITE, TYPE_WRITE_TO, and TYPE_READINTO */
Py_buffer user_buffer;
Py_buffer *user_buffer;

/* Data used for reading from a connectionless socket:
TYPE_READ_FROM */
Expand Down Expand Up @@ -163,7 +163,6 @@ initialize_function_pointers(void)
GUID GuidConnectEx = WSAID_CONNECTEX;
GUID GuidDisconnectEx = WSAID_DISCONNECTEX;
GUID GuidTransmitFile = WSAID_TRANSMITFILE;
HINSTANCE hKernel32;
SOCKET s;
DWORD dwBytes;

Expand Down Expand Up @@ -645,7 +644,7 @@ _overlapped_Overlapped_impl(PyTypeObject *type, HANDLE event)
self->type = TYPE_NONE;
self->allocated_buffer = NULL;
memset(&self->overlapped, 0, sizeof(OVERLAPPED));
memset(&self->user_buffer, 0, sizeof(Py_buffer));
memset(self->user_buffer, 0, sizeof(Py_buffer));
if (event)
self->overlapped.hEvent = event;
return (PyObject *)self;
Expand Down Expand Up @@ -686,8 +685,8 @@ Overlapped_clear(OverlappedObject *self)
case TYPE_WRITE:
case TYPE_WRITE_TO:
case TYPE_READINTO: {
if (self->user_buffer.obj) {
PyBuffer_Release(&self->user_buffer);
if (self->user_buffer->obj) {
PyBuffer_Release(self->user_buffer);
}
break;
}
Expand Down Expand Up @@ -1061,8 +1060,8 @@ _overlapped_Overlapped_ReadFileInto_impl(OverlappedObject *self,
self->type = TYPE_READINTO;
self->handle = handle;

return do_ReadFile(self, handle, self->user_buffer.buf,
(DWORD)self->user_buffer.len);
return do_ReadFile(self, handle, self->user_buffer->buf,
(DWORD)self->user_buffer->len);
}

static PyObject *
Expand Down Expand Up @@ -1168,8 +1167,8 @@ _overlapped_Overlapped_WSARecvInto_impl(OverlappedObject *self,
self->type = TYPE_READINTO;
self->handle = handle;

return do_WSARecv(self, handle, self->user_buffer.buf,
(DWORD)self->user_buffer.len, flags);
return do_WSARecv(self, handle, self->user_buffer->buf,
(DWORD)self->user_buffer->len, flags);
}

/*[clinic input]
Expand Down Expand Up @@ -1208,8 +1207,8 @@ _overlapped_Overlapped_WriteFile_impl(OverlappedObject *self, HANDLE handle,
self->handle = handle;

Py_BEGIN_ALLOW_THREADS
ret = WriteFile(handle, self->user_buffer.buf,
(DWORD)self->user_buffer.len,
ret = WriteFile(handle, self->user_buffer->buf,
(DWORD)self->user_buffer->len,
&written, &self->overlapped);
Py_END_ALLOW_THREADS

Expand Down Expand Up @@ -1260,8 +1259,8 @@ _overlapped_Overlapped_WSASend_impl(OverlappedObject *self, HANDLE handle,

self->type = TYPE_WRITE;
self->handle = handle;
wsabuf.len = (DWORD)self->user_buffer.len;
wsabuf.buf = self->user_buffer.buf;
wsabuf.len = (DWORD)self->user_buffer->len;
wsabuf.buf = self->user_buffer->buf;

Py_BEGIN_ALLOW_THREADS
ret = WSASend((SOCKET)handle, &wsabuf, 1, &written, flags,
Expand Down Expand Up @@ -1646,8 +1645,8 @@ Overlapped_traverse(OverlappedObject *self, visitproc visit, void *arg)
case TYPE_WRITE:
case TYPE_WRITE_TO:
case TYPE_READINTO:
if (self->user_buffer.obj) {
Py_VISIT(&self->user_buffer.obj);
if (self->user_buffer->obj) {
Py_VISIT(&self->user_buffer->obj);
}
break;
case TYPE_READ_FROM:
Expand Down Expand Up @@ -1760,8 +1759,8 @@ _overlapped_Overlapped_WSASendTo_impl(OverlappedObject *self, HANDLE handle,

self->type = TYPE_WRITE_TO;
self->handle = handle;
wsabuf.len = (DWORD)self->user_buffer.len;
wsabuf.buf = self->user_buffer.buf;
wsabuf.len = (DWORD)self->user_buffer->len;
wsabuf.buf = self->user_buffer->buf;

Py_BEGIN_ALLOW_THREADS
ret = WSASendTo((SOCKET)handle, &wsabuf, 1, &written, flags,
Expand Down
0