qiurui
/
Centos-kernel-stream-9
mirror of https://gitlab.com/redhat/centos-stream/src/kernel/centos-stream-9.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

selftests/bpf: Verify optval=NULL case 

Bugzilla: https://bugzilla.redhat.com/2221599

commit 833d67ecdc5f35f1ebf59d0fccc1ce771434be9c
Author: Stanislav Fomichev <sdf@google.com>
Date:   Tue Apr 18 15:53:39 2023 -0700

    selftests/bpf: Verify optval=NULL case
    
    Make sure we get optlen exported instead of getting EFAULT.
    
    Signed-off-by: Stanislav Fomichev <sdf@google.com>
    Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>
    Link: https://lore.kernel.org/bpf/20230418225343.553806-3-sdf@google.com

Signed-off-by: Artem Savkov <asavkov@redhat.com>
This commit is contained in:
Artem Savkov 2023-07-11 10:07:21 +02:00
parent 7cbf3a2981
commit a64a639f30
2 changed files with 40 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tools/testing/selftests/bpf/prog_tests/sockopt_sk.c
View File

@ -3,6 +3,7 @@
#include "cgroup_helpers.h" #include "cgroup_helpers.h"
#include <linux/tcp.h> #include <linux/tcp.h>
#include <linux/netlink.h>
#include "sockopt_sk.skel.h" #include "sockopt_sk.skel.h"
#ifndef SOL_TCP #ifndef SOL_TCP
@ -183,6 +184,33 @@ static int getsetsockopt(void)
goto err; goto err;
} }
/* optval=NULL case is handled correctly */
close(fd);
fd = socket(AF_NETLINK, SOCK_RAW, 0);
if (fd < 0) {
log_err("Failed to create AF_NETLINK socket");
return -1;
}
buf.u32 = 1;
optlen = sizeof(__u32);
err = setsockopt(fd, SOL_NETLINK, NETLINK_ADD_MEMBERSHIP, &buf, optlen);
if (err) {
log_err("Unexpected getsockopt(NETLINK_ADD_MEMBERSHIP) err=%d errno=%d",
err, errno);
goto err;
}
optlen = 0;
err = getsockopt(fd, SOL_NETLINK, NETLINK_LIST_MEMBERSHIPS, NULL, &optlen);
if (err) {
log_err("Unexpected getsockopt(NETLINK_LIST_MEMBERSHIPS) err=%d errno=%d",
err, errno);
goto err;
}
ASSERT_EQ(optlen, 4, "Unexpected NETLINK_LIST_MEMBERSHIPS value");
free(big_buf); free(big_buf);
close(fd); close(fd);
return 0; return 0;

12
tools/testing/selftests/bpf/progs/sockopt_sk.c
View File

@ -32,6 +32,12 @@ int _getsockopt(struct bpf_sockopt *ctx)
__u8 *optval_end = ctx->optval_end; __u8 *optval_end = ctx->optval_end;
__u8 *optval = ctx->optval; __u8 *optval = ctx->optval;
struct sockopt_sk *storage; struct sockopt_sk *storage;
struct bpf_sock *sk;
/* Bypass AF_NETLINK. */
sk = ctx->sk;
if (sk && sk->family == AF_NETLINK)
return 1;
/* Make sure bpf_get_netns_cookie is callable. /* Make sure bpf_get_netns_cookie is callable.
*/ */
@ -131,6 +137,12 @@ int _setsockopt(struct bpf_sockopt *ctx)
__u8 *optval_end = ctx->optval_end; __u8 *optval_end = ctx->optval_end;
__u8 *optval = ctx->optval; __u8 *optval = ctx->optval;
struct sockopt_sk *storage; struct sockopt_sk *storage;
struct bpf_sock *sk;
/* Bypass AF_NETLINK. */
sk = ctx->sk;
if (sk && sk->family == AF_NETLINK)
return 1;
/* Make sure bpf_get_netns_cookie is callable. /* Make sure bpf_get_netns_cookie is callable.
*/ */