From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from vger.kernel.org (vger.kernel.org [23.128.96.18]) by smtp.lore.kernel.org (Postfix) with ESMTP id 93115C433EF for ; Thu, 10 Mar 2022 12:50:10 +0000 (UTC) Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S242122AbiCJMvJ (ORCPT ); Thu, 10 Mar 2022 07:51:09 -0500 Received: from lindbergh.monkeyblade.net ([23.128.96.19]:46036 "EHLO lindbergh.monkeyblade.net" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S242119AbiCJMvG (ORCPT ); Thu, 10 Mar 2022 07:51:06 -0500 Received: from mail-yw1-x1149.google.com (mail-yw1-x1149.google.com [IPv6:2607:f8b0:4864:20::1149]) by lindbergh.monkeyblade.net (Postfix) with ESMTPS id ECBFC14995D for ; Thu, 10 Mar 2022 04:50:03 -0800 (PST) Received: by mail-yw1-x1149.google.com with SMTP id 00721157ae682-2dbf52cc4b9so39466187b3.18 for ; Thu, 10 Mar 2022 04:50:03 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20210112; h=date:in-reply-to:message-id:mime-version:references:subject:from:to :cc; bh=4/lo4O5bvYf7GLQ1V3jQOnUyhynYGZYN44Y/DiquDY0=; b=GRWZpOfRRaVzPayOOzdrRZNOKRBkM+GEqZY7EKurLwc1B/K8D0jMk7pVX3ANTsohHj UNNqNLswmUTzkqbYlTtk+xT7geLaU3n2yQrKuYIv3iU2xZkuf7Mwft9ZmbNEFJpMXUGK Yj9ninU2FdEkHjsdLbzw1DGx1Vpcr/1FjLat0AZBEmYu5zR2YNVEw+6gWICFDVmbiGSM u8Bhmleq/2w/k9dF9BgOOYi5cinjdXUoU8JuogHx6oHHjgA1c54GEgkqPHadW1Hchm88 4gsM2PNyiBd7o+ZYva8wsOaNmJiWM+iVtGGQ8FcWaGSDRbeBGb5WtL73DHWtY0Ok73pV 8rhQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20210112; h=x-gm-message-state:date:in-reply-to:message-id:mime-version :references:subject:from:to:cc; bh=4/lo4O5bvYf7GLQ1V3jQOnUyhynYGZYN44Y/DiquDY0=; b=n3U0cPPWEjr7JA9RwcxYliS4aeG5LbhnTJLKnIXeSoSlWVA09tsX7HLa6GlqYMdQbb m1zwn9QchnaCcyuwx1HmNAsRCV1qhPUErCJ+QfpUutoNMwPV0s0dhtVMSDRzBUq3xApT joSPKy7naAd9LzBnH7y2niv1JPCALUy9s5W0JIQd1TS78uiIprb7fuUtmJ2uK6Gyoj0r b0Z3bYlhCcytAWAjnbtclHhG8k0t8FuhpHgyi3ygPLligUfopjPY+aOn/Nqdc+reTbG7 0YiDISbgoopWcaWsILdQ2eUPAQMAIsJ8zNcLpvC97Zxj5yP60Ul0KI9uee7YuPYj1Hco 5TtA== X-Gm-Message-State: AOAM530JEig6B58+826byGbt3O6ViWRyuMANkcxAbhsx5qiRscRmmD7t yqM6AwtVKEyXCa1PyZkrq7WZOntCli4= X-Google-Smtp-Source: ABdhPJy6F7MW4VtL0aXxgXM3QIkhcr6AWkkBeJdpgJm826p8XWN9AkS7ytuIQkNvXcYDAVxBL2VHblQalUI= X-Received: from jiyong.seo.corp.google.com ([2401:fa00:d:11:f59e:134:eb7:e1d2]) (user=jiyong job=sendgmr) by 2002:a0d:ed82:0:b0:2dc:3c9e:430b with SMTP id w124-20020a0ded82000000b002dc3c9e430bmr3656657ywe.115.1646916603123; Thu, 10 Mar 2022 04:50:03 -0800 (PST) Date: Thu, 10 Mar 2022 21:49:36 +0900 In-Reply-To: <20220310124936.4179591-1-jiyong@google.com> Message-Id: <20220310124936.4179591-3-jiyong@google.com> Mime-Version: 1.0 References: <20220310124936.4179591-1-jiyong@google.com> X-Mailer: git-send-email 2.35.1.723.g4982287a31-goog Subject: [PATCH 2/2] vsock: refactor vsock_for_each_connected_socket From: Jiyong Park To: Stefan Hajnoczi , Stefano Garzarella , "Michael S. Tsirkin" , Jason Wang , "David S. Miller" , Jakub Kicinski Cc: adelva@google.com, Jiyong Park , kvm@vger.kernel.org, virtualization@lists.linux-foundation.org, netdev@vger.kernel.org, linux-kernel@vger.kernel.org Content-Type: text/plain; charset="UTF-8" Precedence: bulk List-ID: X-Mailing-List: linux-kernel@vger.kernel.org vsock_for_each_connected_socket now cycles over sockets of a specific transport only, rather than asking callers to do the filtering manually, which is error-prone. Signed-off-by: Jiyong Park --- drivers/vhost/vsock.c | 7 ++----- include/net/af_vsock.h | 3 ++- net/vmw_vsock/af_vsock.c | 9 +++++++-- net/vmw_vsock/virtio_transport.c | 12 ++++-------- net/vmw_vsock/vmci_transport.c | 8 ++------ 5 files changed, 17 insertions(+), 22 deletions(-) diff --git a/drivers/vhost/vsock.c b/drivers/vhost/vsock.c index 853ddac00d5b..e6c9d41db1de 100644 --- a/drivers/vhost/vsock.c +++ b/drivers/vhost/vsock.c @@ -722,10 +722,6 @@ static void vhost_vsock_reset_orphans(struct sock *sk) * executing. */ - /* Only handle our own sockets */ - if (vsk->transport != &vhost_transport.transport) - return; - /* If the peer is still valid, no need to reset connection */ if (vhost_vsock_get(vsk->remote_addr.svm_cid)) return; @@ -757,7 +753,8 @@ static int vhost_vsock_dev_release(struct inode *inode, struct file *file) /* Iterating over all connections for all CIDs to find orphans is * inefficient. Room for improvement here. */ - vsock_for_each_connected_socket(vhost_vsock_reset_orphans); + vsock_for_each_connected_socket(&vhost_transport.transport, + vhost_vsock_reset_orphans); /* Don't check the owner, because we are in the release path, so we * need to stop the vsock device in any case. diff --git a/include/net/af_vsock.h b/include/net/af_vsock.h index ab207677e0a8..f742e50207fb 100644 --- a/include/net/af_vsock.h +++ b/include/net/af_vsock.h @@ -205,7 +205,8 @@ struct sock *vsock_find_bound_socket(struct sockaddr_vm *addr); struct sock *vsock_find_connected_socket(struct sockaddr_vm *src, struct sockaddr_vm *dst); void vsock_remove_sock(struct vsock_sock *vsk); -void vsock_for_each_connected_socket(void (*fn)(struct sock *sk)); +void vsock_for_each_connected_socket(struct vsock_transport *transport, + void (*fn)(struct sock *sk)); int vsock_assign_transport(struct vsock_sock *vsk, struct vsock_sock *psk); bool vsock_find_cid(unsigned int cid); diff --git a/net/vmw_vsock/af_vsock.c b/net/vmw_vsock/af_vsock.c index 38baeb189d4e..f04abf662ec6 100644 --- a/net/vmw_vsock/af_vsock.c +++ b/net/vmw_vsock/af_vsock.c @@ -334,7 +334,8 @@ void vsock_remove_sock(struct vsock_sock *vsk) } EXPORT_SYMBOL_GPL(vsock_remove_sock); -void vsock_for_each_connected_socket(void (*fn)(struct sock *sk)) +void vsock_for_each_connected_socket(struct vsock_transport *transport, + void (*fn)(struct sock *sk)) { int i; @@ -343,8 +344,12 @@ void vsock_for_each_connected_socket(void (*fn)(struct sock *sk)) for (i = 0; i < ARRAY_SIZE(vsock_connected_table); i++) { struct vsock_sock *vsk; list_for_each_entry(vsk, &vsock_connected_table[i], - connected_table) + connected_table) { + if (vsk->transport != transport) + continue; + fn(sk_vsock(vsk)); + } } spin_unlock_bh(&vsock_table_lock); diff --git a/net/vmw_vsock/virtio_transport.c b/net/vmw_vsock/virtio_transport.c index 61b24eb31d4b..5afc194a58bb 100644 --- a/net/vmw_vsock/virtio_transport.c +++ b/net/vmw_vsock/virtio_transport.c @@ -358,17 +358,11 @@ static void virtio_vsock_event_fill(struct virtio_vsock *vsock) static void virtio_vsock_reset_sock(struct sock *sk) { - struct vsock_sock *vsk = vsock_sk(sk); - /* vmci_transport.c doesn't take sk_lock here either. At least we're * under vsock_table_lock so the sock cannot disappear while we're * executing. */ - /* Only handle our own sockets */ - if (vsk->transport != &virtio_transport.transport) - return; - sk->sk_state = TCP_CLOSE; sk->sk_err = ECONNRESET; sk_error_report(sk); @@ -391,7 +385,8 @@ static void virtio_vsock_event_handle(struct virtio_vsock *vsock, switch (le32_to_cpu(event->id)) { case VIRTIO_VSOCK_EVENT_TRANSPORT_RESET: virtio_vsock_update_guest_cid(vsock); - vsock_for_each_connected_socket(virtio_vsock_reset_sock); + vsock_for_each_connected_socket(&virtio_transport.transport, + virtio_vsock_reset_sock); break; } } @@ -669,7 +664,8 @@ static void virtio_vsock_remove(struct virtio_device *vdev) synchronize_rcu(); /* Reset all connected sockets when the device disappear */ - vsock_for_each_connected_socket(virtio_vsock_reset_sock); + vsock_for_each_connected_socket(&virtio_transport.transport, + virtio_vsock_reset_sock); /* Stop all work handlers to make sure no one is accessing the device, * so we can safely call virtio_reset_device(). diff --git a/net/vmw_vsock/vmci_transport.c b/net/vmw_vsock/vmci_transport.c index cd2f01513fae..735d5e14608a 100644 --- a/net/vmw_vsock/vmci_transport.c +++ b/net/vmw_vsock/vmci_transport.c @@ -803,11 +803,6 @@ static void vmci_transport_handle_detach(struct sock *sk) struct vsock_sock *vsk; vsk = vsock_sk(sk); - - /* Only handle our own sockets */ - if (vsk->transport != &vmci_transport) - return; - if (!vmci_handle_is_invalid(vmci_trans(vsk)->qp_handle)) { sock_set_flag(sk, SOCK_DONE); @@ -887,7 +882,8 @@ static void vmci_transport_qp_resumed_cb(u32 sub_id, const struct vmci_event_data *e_data, void *client_data) { - vsock_for_each_connected_socket(vmci_transport_handle_detach); + vsock_for_each_connected_socket(&vmci_transport, + vmci_transport_handle_detach); } static void vmci_transport_recv_pkt_work(struct work_struct *work) -- 2.35.1.723.g4982287a31-goog