Commit e4b1ef15 authored by Arseny Krasnov's avatar Arseny Krasnov Committed by David S. Miller
Browse files

virtio/vsock: add SEQPACKET receive logic 



Update current receive logic for SEQPACKET support: performs
check for packet and socket types on receive(if mismatch, then
reset connection). Increment EOR counter on receive. Also if
buffer of new packet was appended to buffer of last packet in
rx queue, update flags of last packet with flags of new packet.

Signed-off-by: default avatarArseny Krasnov <arseny.krasnov@kaspersky.com>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent 44931195

net/vmw_vsock/virtio_transport_common.c

+31 −3
Original line number Diff line number Diff line
@@ -165,6 +165,14 @@ void virtio_transport_deliver_tap_pkt(struct virtio_vsock_pkt *pkt) 
}

EXPORT_SYMBOL_GPL(virtio_transport_deliver_tap_pkt);



static u16 virtio_transport_get_type(struct sock *sk)

{

	if (sk->sk_type == SOCK_STREAM)

		return VIRTIO_VSOCK_TYPE_STREAM;

	else

		return VIRTIO_VSOCK_TYPE_SEQPACKET;

}



/* This function can only be used on connecting/connected sockets,

 * since a socket assigned to a transport is required.

 *
@@ -987,6 +995,9 @@ virtio_transport_recv_enqueue(struct vsock_sock *vsk, 
		goto out;

	}



	if (le32_to_cpu(pkt->hdr.flags) & VIRTIO_VSOCK_SEQ_EOR)

		vvs->msg_count++;



	/* Try to copy small packets into the buffer of last packet queued,

	 * to avoid wasting memory queueing the entire buffer with a small

	 * payload.
@@ -998,13 +1009,18 @@ virtio_transport_recv_enqueue(struct vsock_sock *vsk, 
					   struct virtio_vsock_pkt, list);



		/* If there is space in the last packet queued, we copy the

		 * new packet in its buffer.

		 * new packet in its buffer. We avoid this if the last packet

		 * queued has VIRTIO_VSOCK_SEQ_EOR set, because this is

		 * delimiter of SEQPACKET record, so 'pkt' is the first packet

		 * of a new record.

		 */

		if (pkt->len <= last_pkt->buf_len - last_pkt->len) {

		if ((pkt->len <= last_pkt->buf_len - last_pkt->len) &&

		    !(le32_to_cpu(last_pkt->hdr.flags) & VIRTIO_VSOCK_SEQ_EOR)) {

			memcpy(last_pkt->buf + last_pkt->len, pkt->buf,

			       pkt->len);

			last_pkt->len += pkt->len;

			free_pkt = true;

			last_pkt->hdr.flags |= pkt->hdr.flags;

			goto out;

		}

	}
@@ -1170,6 +1186,12 @@ virtio_transport_recv_listen(struct sock *sk, struct virtio_vsock_pkt *pkt, 
	return 0;

}



static bool virtio_transport_valid_type(u16 type)

{

	return (type == VIRTIO_VSOCK_TYPE_STREAM) ||

	       (type == VIRTIO_VSOCK_TYPE_SEQPACKET);

}



/* We are under the virtio-vsock's vsock->rx_lock or vhost-vsock's vq->mutex

 * lock.

 */
@@ -1195,7 +1217,7 @@ void virtio_transport_recv_pkt(struct virtio_transport *t, 
					le32_to_cpu(pkt->hdr.buf_alloc),

					le32_to_cpu(pkt->hdr.fwd_cnt));



	if (le16_to_cpu(pkt->hdr.type) != VIRTIO_VSOCK_TYPE_STREAM) {

	if (!virtio_transport_valid_type(le16_to_cpu(pkt->hdr.type))) {

		(void)virtio_transport_reset_no_sock(t, pkt);

		goto free_pkt;

	}
@@ -1212,6 +1234,12 @@ void virtio_transport_recv_pkt(struct virtio_transport *t, 
		}

	}



	if (virtio_transport_get_type(sk) != le16_to_cpu(pkt->hdr.type)) {

		(void)virtio_transport_reset_no_sock(t, pkt);

		sock_put(sk);

		goto free_pkt;

	}



	vsk = vsock_sk(sk);



	lock_sock(sk);