djsundog/circuitpython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

extmod/modlwip: Add support for polling UDP sockets for writability.

Browse Source
This commit is contained in:
Damien George 2019-01-31 11:22:03 +11:00
parent 4bed17e786
commit 2a7a307baa
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
extmod/modlwip.c
View File

@ -1231,9 +1231,15 @@ STATIC mp_uint_t lwip_socket_ioctl(mp_obj_t self_in, mp_uint_t request, uintptr_
} }
} }
// Note: pcb.tcp==NULL if state<0, and in this case we can't call tcp_sndbuf if (flags & MP_STREAM_POLL_WR) {
if (flags & MP_STREAM_POLL_WR && socket->pcb.tcp != NULL && tcp_sndbuf(socket->pcb.tcp) > 0) { if (socket->type == MOD_NETWORK_SOCK_DGRAM && socket->pcb.udp != NULL) {
ret |= MP_STREAM_POLL_WR; // UDP socket is writable
ret |= MP_STREAM_POLL_WR;
} else if (socket->pcb.tcp != NULL && tcp_sndbuf(socket->pcb.tcp) > 0) {
// TCP socket is writable
// Note: pcb.tcp==NULL if state<0, and in this case we can't call tcp_sndbuf
ret |= MP_STREAM_POLL_WR;
}
} }
if (socket->state == STATE_NEW) { if (socket->state == STATE_NEW) {