Received: by 2002:a05:6a10:5bc5:0:0:0:0 with SMTP id os5csp432331pxb; Tue, 19 Oct 2021 06:00:46 -0700 (PDT) X-Google-Smtp-Source: ABdhPJxUul4LaotfC6v+cHks3pipyzLRVfR1YUR9Dvh5V7r6xtjAhMbfI4LDFDamYSeRKHJgPJ6M X-Received: by 2002:a17:90a:19e:: with SMTP id 30mr6531439pjc.140.1634648446665; Tue, 19 Oct 2021 06:00:46 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1634648446; cv=none; d=google.com; s=arc-20160816; b=ktZmBpc8xcOaBbt+j6DYu+QAEBwAlyPui0UohUb4OBpxSkMskPyi8XFaTfqwkgKPav TBrwCE6mLx6QDN7i0/4yU1y8pOy9SwNxE4qVXtuq6s+rJDiXmT5B9eVCaX/zQs8d1Eva UUcs4GeaNuNMdkc05mFBvumFzazAoFekfiKF/21kNtuUU7HfaiL4NiWdlKpoUOsiV1/f qDIPmOtXTE8oXh0xpbEQ5IX+gXZlu7VBGeGQLNwUswZHwyBaoTIfwDDHYo+Cb6lbTPDX uNPWUNupXIdZTtIZNUjSCx0GnKn9pnw1yOoIL3gCwNU0ZPJQfJII2jZrinptOeQUnnW2 WdoA== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=list-id:precedence:content-transfer-encoding:mime-version :message-id:date:subject:cc:to:from; bh=zmlD3BO43+2fiIDdi9pvbi/7y5DbhFPUrtNDDOxdQS8=; b=Dvp8qOFs0gIy/6TQloY5d6l5wnzi3uV7TLEpxp5wwKGcHtWyl+nARtwoKtsOf/QVuZ sknhhye5axK25o70VW0OaYDfpKwJb6Ks2u4NuEb1kvufZIEVU5saqBtWp3XslkUYMpHf +CXWe1TYO5YYSy/HLQ1+Dvibj7t8z6/BDfHcEIungR3cY/aI558vOmfDS97No2v8MSd4 r/s89nv5HHS70xromr3Kx6z4NU3E+ePKgqGDwQqr8FHngzHj8Vgu2DVXScm+Lmshie2C GG8f3a0sb1gTeKjB6CLSjWF4lXritHWzTwIMp7P6c+jXmLbYL9Y8XN7As7rs6rvv6JAD 4SUw== ARC-Authentication-Results: i=1; mx.google.com; spf=pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 23.128.96.18 as permitted sender) smtp.mailfrom=linux-kernel-owner@vger.kernel.org; dmarc=fail (p=NONE sp=NONE dis=NONE) header.from=huawei.com Return-Path: Received: from vger.kernel.org (vger.kernel.org. [23.128.96.18]) by mx.google.com with ESMTP id 193si18928038pgf.582.2021.10.19.06.00.32; Tue, 19 Oct 2021 06:00:46 -0700 (PDT) Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 23.128.96.18 as permitted sender) client-ip=23.128.96.18; Authentication-Results: mx.google.com; spf=pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 23.128.96.18 as permitted sender) smtp.mailfrom=linux-kernel-owner@vger.kernel.org; dmarc=fail (p=NONE sp=NONE dis=NONE) header.from=huawei.com Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S235764AbhJSNB3 (ORCPT + 99 others); Tue, 19 Oct 2021 09:01:29 -0400 Received: from szxga01-in.huawei.com ([45.249.212.187]:29909 "EHLO szxga01-in.huawei.com" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S235763AbhJSNB1 (ORCPT ); Tue, 19 Oct 2021 09:01:27 -0400 Received: from dggemv704-chm.china.huawei.com (unknown [172.30.72.55]) by szxga01-in.huawei.com (SkyGuard) with ESMTP id 4HYYbz40RQzbnFn; Tue, 19 Oct 2021 20:54:39 +0800 (CST) Received: from dggpeml500011.china.huawei.com (7.185.36.84) by dggemv704-chm.china.huawei.com (10.3.19.47) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256) id 15.1.2308.15; Tue, 19 Oct 2021 20:59:12 +0800 Received: from localhost.localdomain (10.175.101.6) by dggpeml500011.china.huawei.com (7.185.36.84) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256) id 15.1.2308.15; Tue, 19 Oct 2021 20:59:11 +0800 From: Di Zhu To: , , , , CC: , , Subject: [PATCH] bpf: support BPF_PROG_QUERY for progs attached to sockmap Date: Tue, 19 Oct 2021 20:58:56 +0800 Message-ID: <20211019125856.2566882-1-zhudi2@huawei.com> X-Mailer: git-send-email 2.27.0 MIME-Version: 1.0 Content-Transfer-Encoding: 7BIT Content-Type: text/plain; charset=US-ASCII X-Originating-IP: [10.175.101.6] X-ClientProxiedBy: dggems701-chm.china.huawei.com (10.3.19.178) To dggpeml500011.china.huawei.com (7.185.36.84) X-CFilter-Loop: Reflected Precedence: bulk List-ID: X-Mailing-List: linux-kernel@vger.kernel.org Right now there is no way to query whether BPF programs are attached to a sockmap or not. we can use the standard interface in libbpf to query, such as: bpf_prog_query(mapFd, BPF_SK_SKB_STREAM_PARSER, 0, NULL, ...); the mapFd is the fd of sockmap. Signed-off-by: Di Zhu --- include/linux/bpf.h | 8 +++++ kernel/bpf/syscall.c | 4 +++ net/core/sock_map.c | 81 ++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 93 insertions(+) diff --git a/include/linux/bpf.h b/include/linux/bpf.h index 1c7fd7c4c6d3..69cf70b077d5 100644 --- a/include/linux/bpf.h +++ b/include/linux/bpf.h @@ -1959,6 +1959,8 @@ int bpf_prog_test_run_syscall(struct bpf_prog *prog, int sock_map_get_from_fd(const union bpf_attr *attr, struct bpf_prog *prog); int sock_map_prog_detach(const union bpf_attr *attr, enum bpf_prog_type ptype); int sock_map_update_elem_sys(struct bpf_map *map, void *key, void *value, u64 flags); +int sockmap_bpf_prog_query(const union bpf_attr *attr, + union bpf_attr __user *uattr); void sock_map_unhash(struct sock *sk); void sock_map_close(struct sock *sk, long timeout); #else @@ -2012,6 +2014,12 @@ static inline int sock_map_update_elem_sys(struct bpf_map *map, void *key, void { return -EOPNOTSUPP; } + +static inline int sockmap_bpf_prog_query(const union bpf_attr *attr, + union bpf_attr __user *uattr) +{ + return -EINVAL; +} #endif /* CONFIG_BPF_SYSCALL */ #endif /* CONFIG_NET && CONFIG_BPF_SYSCALL */ diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c index 4e50c0bfdb7d..3049b9506583 100644 --- a/kernel/bpf/syscall.c +++ b/kernel/bpf/syscall.c @@ -3275,6 +3275,10 @@ static int bpf_prog_query(const union bpf_attr *attr, case BPF_FLOW_DISSECTOR: case BPF_SK_LOOKUP: return netns_bpf_prog_query(attr, uattr); + case BPF_SK_SKB_STREAM_PARSER: + case BPF_SK_SKB_STREAM_VERDICT: + case BPF_SK_MSG_VERDICT: + return sockmap_bpf_prog_query(attr, uattr); default: return -EINVAL; } diff --git a/net/core/sock_map.c b/net/core/sock_map.c index e252b8ec2b85..32688a1b78c6 100644 --- a/net/core/sock_map.c +++ b/net/core/sock_map.c @@ -1451,6 +1451,87 @@ static int sock_map_prog_update(struct bpf_map *map, struct bpf_prog *prog, return 0; } +static int sock_map_prog_lookup(struct bpf_map *map, struct bpf_prog **prog, + u32 which) +{ + struct sk_psock_progs *progs = sock_map_progs(map); + + if (!progs) + return -EOPNOTSUPP; + + switch (which) { + case BPF_SK_MSG_VERDICT: + *prog = READ_ONCE(progs->msg_parser); + break; +#if IS_ENABLED(CONFIG_BPF_STREAM_PARSER) + case BPF_SK_SKB_STREAM_PARSER: + *prog = READ_ONCE(progs->skb_parser); + break; +#endif + case BPF_SK_SKB_STREAM_VERDICT: + *prog = READ_ONCE(progs->skb_verdict); + break; + default: + return -EOPNOTSUPP; + } + + return 0; +} + +int sockmap_bpf_prog_query(const union bpf_attr *attr, + union bpf_attr __user *uattr) +{ + __u32 __user *prog_ids = u64_to_user_ptr(attr->query.prog_ids); + u32 prog_cnt = 0, flags = 0; + u32 ufd = attr->target_fd; + struct bpf_map *map; + struct bpf_prog *prog; + struct fd f; + int ret; + + if (attr->query.query_flags) + return -EINVAL; + + if (copy_to_user(&uattr->query.attach_flags, &flags, sizeof(flags))) + return -EFAULT; + + f = fdget(ufd); + map = __bpf_map_get(f); + if (IS_ERR(map)) + return PTR_ERR(map); + + rcu_read_lock(); + + ret = sock_map_prog_lookup(map, &prog, attr->query.attach_type); + if (ret) + goto end; + + prog_cnt = (!prog) ? 0 : 1; + if (copy_to_user(&uattr->query.prog_cnt, &prog_cnt, sizeof(prog_cnt))) { + ret = -EFAULT; + goto end; + } + + if (!attr->query.prog_cnt || !prog_ids || !prog_cnt) + goto end; + + prog = bpf_prog_inc_not_zero(prog); + if (IS_ERR(prog)) { + ret = PTR_ERR(prog); + goto end; + } + + if (copy_to_user(prog_ids, &prog->aux->id, sizeof(u32))) + ret = -EFAULT; + + bpf_prog_put(prog); + +end: + rcu_read_unlock(); + fdput(f); + return ret; +} + static void sock_map_unlink(struct sock *sk, struct sk_psock_link *link) { switch (link->map->map_type) { -- 2.23.0