diff mbox

[v2,1/1] Fix unbound rx buffer

Message ID 20160804140957.23693-2-ggarcia@deic.uab.cat
State Changes Requested, archived
Delegated to: David Miller
Headers show

Commit Message

ggarcia@abra.uab.cat Aug. 4, 2016, 2:09 p.m. UTC
From: Gerard Garcia <ggarcia@deic.uab.cat>

Reset connection and close rx socket when the sender is ignoring our announced available buffer.

Signed-off-by: Gerard Garcia <ggarcia@deic.uab.cat>

---
v2:
 * Get vvs->rx_lock lock before checking if next packet is going to
    overflow the rx buffer.

 net/vmw_vsock/virtio_transport_common.c | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

Comments

Stefan Hajnoczi Aug. 10, 2016, 12:25 p.m. UTC | #1
On Thu, Aug 04, 2016 at 04:09:57PM +0200, ggarcia@abra.uab.cat wrote:
> From: Gerard Garcia <ggarcia@deic.uab.cat>
> 
> Reset connection and close rx socket when the sender is ignoring our announced available buffer.
> 
> Signed-off-by: Gerard Garcia <ggarcia@deic.uab.cat>
> 
> ---
> v2:
>  * Get vvs->rx_lock lock before checking if next packet is going to
>     overflow the rx buffer.
> 
>  net/vmw_vsock/virtio_transport_common.c | 20 +++++++++++++++-----
>  1 file changed, 15 insertions(+), 5 deletions(-)

Acked-by: Stefan Hajnoczi <stefanha@redhat.com>
David Miller Aug. 10, 2016, 10:12 p.m. UTC | #2
All of these patches need to be resubmitted with a proper subsystem
prefix in the Subject line.

Thank you.
diff mbox

Patch

diff --git a/net/vmw_vsock/virtio_transport_common.c b/net/vmw_vsock/virtio_transport_common.c
index a53b3a1..8d4abb6 100644
--- a/net/vmw_vsock/virtio_transport_common.c
+++ b/net/vmw_vsock/virtio_transport_common.c
@@ -774,12 +774,22 @@  virtio_transport_recv_connected(struct sock *sk,
 		pkt->off = 0;
 
 		spin_lock_bh(&vvs->rx_lock);
-		virtio_transport_inc_rx_pkt(vvs, pkt);
-		list_add_tail(&pkt->list, &vvs->rx_queue);
-		spin_unlock_bh(&vvs->rx_lock);
+		if (vvs->rx_bytes + pkt->len <= vvs->buf_size) {
+			virtio_transport_inc_rx_pkt(vvs, pkt);
+			list_add_tail(&pkt->list, &vvs->rx_queue);
+			spin_unlock_bh(&vvs->rx_lock);
+
+			sk->sk_data_ready(sk);
+			return err;
+		} else {
+			/* Sender is ignoring our buf_alloc */
+			spin_unlock_bh(&vvs->rx_lock);
+
+			virtio_transport_reset(vsk, pkt);
+			virtio_transport_do_close(vsk, true);
+			break;
+		}
 
-		sk->sk_data_ready(sk);
-		return err;
 	case VIRTIO_VSOCK_OP_CREDIT_UPDATE:
 		sk->sk_write_space(sk);
 		break;