@@ -57,8 +57,22 @@ struct vhost_vsock_ref {
static bool vhost_transport_contain_cid(u32 cid)
{
+ unsigned int index;
+ struct vhost_vsock_ref *ref;
+
if (cid == VHOST_VSOCK_DEFAULT_HOST_CID)
return true;
+
+ mutex_lock(&valid_host_mutex);
+ hash_for_each(valid_host_hash, index, ref, ref_hash) {
+ u32 other_cid = ref->cid;
+
+ if (other_cid == cid) {
+ mutex_unlock(&valid_host_mutex);
+ return true;
+ }
+ }
+ mutex_unlock(&valid_host_mutex);
return false;
}
@@ -101,6 +115,21 @@ vhost_vsock_contain_cid(struct vhost_vsock *vsock, u32 cid)
return false;
}
+/* Check if a cid is valid for the pkt to be received. */
+static bool
+vhost_vsock_contain_host_cid(struct vhost_vsock *vsock, u32 dst_cid)
+{
+ uint32_t index;
+
+ if (dst_cid == VHOST_VSOCK_DEFAULT_HOST_CID)
+ return true;
+ for (index = 0; index < vsock->num_host_cid; index++) {
+ if (vsock->hostcids[index] == dst_cid)
+ return true;
+ }
+ return false;
+}
+
static u32 vhost_transport_get_local_cid(void)
{
return VHOST_VSOCK_DEFAULT_HOST_CID;
@@ -128,6 +157,13 @@ static struct vhost_vsock *vhost_vsock_get(u32 guest_cid)
return NULL;
}
+/* This function checks if the cid is used by one of the guests. */
+static bool
+vhost_transport_contain_opposite_cid(u32 cid)
+{
+ return vhost_vsock_get(cid) != NULL;
+}
+
/* Callers that dereference the return value must hold vhost_vsock_mutex or the
* RCU read lock.
*/
@@ -512,6 +548,7 @@ static struct virtio_transport vhost_transport = {
.get_local_cid = vhost_transport_get_local_cid,
.contain_cid = vhost_transport_contain_cid,
+ .contain_opposite_cid = vhost_transport_contain_opposite_cid,
.init = virtio_transport_do_socket_init,
.destruct = virtio_transport_destruct,
@@ -629,7 +666,7 @@ static void vhost_vsock_handle_tx_kick(struct vhost_work *work)
/* Only accept correctly addressed packets */
if (vsock->num_cid > 0 &&
vhost_vsock_contain_cid(vsock, pkt->hdr.src_cid) &&
- le64_to_cpu(pkt->hdr.dst_cid) == vhost_transport_get_local_cid())
+ vhost_vsock_contain_host_cid(vsock, le64_to_cpu(pkt->hdr.dst_cid)))
virtio_transport_recv_pkt(&vhost_transport, pkt);
else
virtio_transport_free_pkt(pkt);
@@ -171,6 +171,10 @@ struct vsock_transport {
/* Addressing. */
u32 (*get_local_cid)(void);
bool (*contain_cid)(u32 cid);
+ /* For transport_g2h, this checks if the cid is used by its host. */
+ /* For transport_h2g, this checks if the cid is used by one of its guests. */
+ /* This function is set to NULL for loopback_transport. */
+ bool (*contain_opposite_cid)(u32 cid);
};
/**** CORE ****/
@@ -397,9 +397,9 @@ static bool vsock_use_local_transport(unsigned int remote_cid)
return true;
if (transport_g2h) {
- return remote_cid == transport_g2h->get_local_cid();
+ return transport_g2h->contain_cid(remote_cid);
} else {
- return remote_cid == VMADDR_CID_HOST;
+ return transport_h2g->contain_cid(remote_cid);
}
}
@@ -423,7 +423,9 @@ static void vsock_deassign_transport(struct vsock_sock *vsk)
* g2h is not loaded, will use local transport;
* - remote CID <= VMADDR_CID_HOST or h2g is not loaded or remote flags field
* includes VMADDR_FLAG_TO_HOST flag value, will use guest->host transport;
- * - remote CID > VMADDR_CID_HOST will use host->guest transport;
+ * - remote CID > VMADDR_CID_HOST will use host->guest transport if
+ * guest->host transport is not loaded. Otherwise, if guest->host transport
+ * contains the remote_cid, then use the guest->host transport.
*/
int vsock_assign_transport(struct vsock_sock *vsk, struct vsock_sock *psk)
{
@@ -434,15 +436,18 @@ int vsock_assign_transport(struct vsock_sock *vsk, struct vsock_sock *psk)
int ret;
/* If the packet is coming with the source and destination CIDs higher
- * than VMADDR_CID_HOST, then a vsock channel where all the packets are
+ * than VMADDR_CID_HOST, and the source and destination CIDs are not
+ * used by the host, then a vsock channel where all the packets are
* forwarded to the host should be established. Then the host will
* need to forward the packets to the guest.
*
* The flag is set on the (listen) receive path (psk is not NULL). On
* the connect path the flag can be set by the user space application.
*/
- if (psk && vsk->local_addr.svm_cid > VMADDR_CID_HOST &&
- vsk->remote_addr.svm_cid > VMADDR_CID_HOST)
+ if (psk && transport_h2g && vsk->local_addr.svm_cid > VMADDR_CID_HOST &&
+ !transport_h2g->contain_cid(vsk->local_addr.svm_cid) &&
+ vsk->remote_addr.svm_cid > VMADDR_CID_HOST &&
+ !transport_h2g->contain_cid(vsk->remote_addr.svm_cid))
vsk->remote_addr.svm_flags |= VMADDR_FLAG_TO_HOST;
remote_flags = vsk->remote_addr.svm_flags;
@@ -458,6 +463,9 @@ int vsock_assign_transport(struct vsock_sock *vsk, struct vsock_sock *psk)
else if (remote_cid <= VMADDR_CID_HOST || !transport_h2g ||
(remote_flags & VMADDR_FLAG_TO_HOST))
new_transport = transport_g2h;
+ else if (remote_cid > VMADDR_CID_HOST && transport_g2h &&
+ transport_g2h->contain_opposite_cid(remote_cid))
+ new_transport = transport_g2h;
else
new_transport = transport_h2g;
break;
@@ -99,6 +99,35 @@ static bool virtio_transport_contain_cid(u32 cid)
return ret;
}
+/* This function checks if the transport_g2h is using the cid. */
+static bool virtio_transport_contain_opposite_cid(u32 cid)
+{
+ struct virtio_vsock *vsock;
+ bool ret;
+ u32 num_host_cid;
+
+ if (cid == VMADDR_CID_HOST)
+ return true;
+ num_host_cid = 0;
+ rcu_read_lock();
+ vsock = rcu_dereference(the_virtio_vsock);
+ if (!vsock || vsock->number_host_cid == 0) {
+ ret = false;
+ goto out_rcu;
+ }
+
+ for (num_host_cid = 0; num_host_cid < vsock->number_host_cid; num_host_cid++) {
+ if (vsock->host_cids[num_host_cid] == cid) {
+ ret = true;
+ goto out_rcu;
+ }
+ }
+ ret = false;
+out_rcu:
+ rcu_read_unlock();
+ return ret;
+}
+
static u32 virtio_transport_get_local_cid(void)
{
struct virtio_vsock *vsock;
@@ -532,6 +561,7 @@ static struct virtio_transport virtio_transport = {
.get_local_cid = virtio_transport_get_local_cid,
.contain_cid = virtio_transport_contain_cid,
+ .contain_opposite_cid = virtio_transport_contain_opposite_cid,
.init = virtio_transport_do_socket_init,
.destruct = virtio_transport_destruct,
This patch allows the user to use the additional host CIDS to communicate with the guest. As server, the host can bind to any CIDS as long as the cid can be mapped to one guest. The VHOST_DEFAULT_CID can be used as normal. As client, when connect to a remote server, if no address is specified to be used, then it will use the first cid in the array. If the user wants to use a specific cid, then the user can perfrom bind before the connect operation, so that vsock_auto_bind will not be performed. The patch depends on the previous patch which enables hypervisors such as qemu to specify multiple cids for host and guest. Signed-off-by: fuguancheng <fuguancheng@bytedance.com> --- drivers/vhost/vsock.c | 39 ++++++++++++++++++++++++++++++++++++++- include/net/af_vsock.h | 4 ++++ net/vmw_vsock/af_vsock.c | 20 ++++++++++++++------ net/vmw_vsock/virtio_transport.c | 30 ++++++++++++++++++++++++++++++ 4 files changed, 86 insertions(+), 7 deletions(-)