1 files changed, 3 insertions, 1 deletions

diff --git a/net/vmw_vsock/virtio_transport_common.c b/net/vmw_vsock/virtio_transport_common.c
index f3f3d06cb6d8..e30f53728725 100644
--- a/net/vmw_vsock/virtio_transport_common.c
+++ b/net/vmw_vsock/virtio_transport_common.c
@@ -871,8 +871,10 @@ virtio_transport_recv_connected(struct sock *sk,
                 if (le32_to_cpu(pkt->hdr.flags) & VIRTIO_VSOCK_SHUTDOWN_SEND)
                         vsk->peer_shutdown |= SEND_SHUTDOWN;
                 if (vsk->peer_shutdown == SHUTDOWN_MASK &&
-                    vsock_stream_has_data(vsk) <= 0)
+                    vsock_stream_has_data(vsk) <= 0) {
+                        sock_set_flag(sk, SOCK_DONE);
                         sk->sk_state = TCP_CLOSING;
+                }
                 if (le32_to_cpu(pkt->hdr.flags))
                         sk->sk_state_change(sk);
                 break;