X-Git-Url: https://asedeno.scripts.mit.edu/gitweb/?a=blobdiff_plain;f=winnet.c;h=cc0b080c63c76148b31ddf91fc4f7e4a22efc624;hb=9b15a8010102fc83dcd7eef78014a65f11619d63;hp=8602802103aacb5bc169974787f12988ef76e266;hpb=930fcdadebdb87f7dd2f8b629fda6107becd5c49;p=PuTTY.git diff --git a/winnet.c b/winnet.c index 86028021..cc0b080c 100644 --- a/winnet.c +++ b/winnet.c @@ -560,7 +560,7 @@ void sk_write_oob(Socket s, char *buf, int len) { } int select_result(WPARAM wParam, LPARAM lParam) { - int ret; + int ret, open; DWORD err; char buf[BUFFER_GRANULE]; Socket s; @@ -626,9 +626,19 @@ int select_result(WPARAM wParam, LPARAM lParam) { try_send(s); break; case FD_CLOSE: - /* Signal a close on the socket. */ - return s->receiver(s, 0, NULL, 0); - break; + /* Signal a close on the socket. First read any outstanding data. */ + open = 1; + do { + ret = recv(s->s, buf, sizeof(buf), 0); + if (ret < 0) { + err = WSAGetLastError(); + if (err == WSAEWOULDBLOCK) + break; + return s->receiver(s, 3, winsock_error_string(err), err); + } else + open &= s->receiver(s, 0, buf, ret); + } while (ret > 0); + return open; } return 1;