Stanislav Fomichev <s...@google.com> [Thu, 2019-01-17 16:41 -0800]:
> Count the number of times hooks were executed in the cgroup
> local storage to make sure they trigger.
>
> Signed-off-by: Stanislav Fomichev <s...@google.com>
> ---
> tools/testing/selftests/bpf/test_sock_addr.c | 131 ++++++++++++++++++-
> 1 file changed, 130 insertions(+), 1 deletion(-)
>
> diff --git a/tools/testing/selftests/bpf/test_sock_addr.c
> b/tools/testing/selftests/bpf/test_sock_addr.c
> index 3f110eaaf29c..92845bf6fac9 100644
> --- a/tools/testing/selftests/bpf/test_sock_addr.c
> +++ b/tools/testing/selftests/bpf/test_sock_addr.c
> @@ -82,6 +82,9 @@ struct sock_addr_test {
> } expected_result;
> };
>
> +/* Per-test storage for additional cgroup storage map */
> +static int cg_storage_fd;
> +
> static int bind4_prog_load(const struct sock_addr_test *test);
> static int bind6_prog_load(const struct sock_addr_test *test);
> static int connect4_prog_load(const struct sock_addr_test *test);
> @@ -94,6 +97,7 @@ static int sendmsg6_rw_asm_prog_load(const struct
> sock_addr_test *test);
> static int sendmsg6_rw_c_prog_load(const struct sock_addr_test *test);
> static int sendmsg6_rw_v4mapped_prog_load(const struct sock_addr_test *test);
> static int sendmsg6_rw_wildcard_prog_load(const struct sock_addr_test *test);
> +static int release46_prog_load(const struct sock_addr_test *test);
>
> static struct sock_addr_test tests[] = {
> /* bind */
> @@ -507,6 +511,35 @@ static struct sock_addr_test tests[] = {
> SRC6_REWRITE_IP,
> SYSCALL_EPERM,
> },
> + /* release */
> + {
> + "release4: make sure the hook triggers",
> + release46_prog_load,
> + BPF_CGROUP_INET4_SOCK_RELEASE,
> + BPF_CGROUP_INET4_SOCK_RELEASE,
> + AF_INET,
> + SOCK_STREAM,
> + SERV4_REWRITE_IP,
> + 0,
> + SERV4_REWRITE_IP,
> + 0,
> + NULL,
> + SUCCESS,
> + },
> + {
> + "release6: make sure the hook triggers",
> + release46_prog_load,
> + BPF_CGROUP_INET6_SOCK_RELEASE,
> + BPF_CGROUP_INET6_SOCK_RELEASE,
> + AF_INET6,
> + SOCK_DGRAM,
> + SERV6_REWRITE_IP,
> + 0,
> + SERV6_REWRITE_IP,
> + 0,
> + NULL,
> + SUCCESS,
> + },
> };
>
> static int mk_sockaddr(int domain, const char *ip, unsigned short port,
> @@ -554,7 +587,15 @@ static int load_insns(const struct sock_addr_test *test,
> int ret;
>
> memset(&load_attr, 0, sizeof(struct bpf_load_program_attr));
> - load_attr.prog_type = BPF_PROG_TYPE_CGROUP_SOCK_ADDR;
> + switch (test->expected_attach_type) {
> + case BPF_CGROUP_INET4_SOCK_RELEASE:
> + case BPF_CGROUP_INET6_SOCK_RELEASE:
> + load_attr.prog_type = BPF_PROG_TYPE_CGROUP_SOCK;
> + break;
> + default:
> + load_attr.prog_type = BPF_PROG_TYPE_CGROUP_SOCK_ADDR;
> + break;
> + }
> load_attr.expected_attach_type = test->expected_attach_type;
> load_attr.insns = insns;
> load_attr.insns_cnt = insns_cnt;
> @@ -916,6 +957,35 @@ static int sendmsg6_rw_c_prog_load(const struct
> sock_addr_test *test)
> return load_path(test, SENDMSG6_PROG_PATH);
> }
>
> +static int release46_prog_load(const struct sock_addr_test *test)
> +{
> + struct bpf_cgroup_storage_key key;
> + unsigned long long value;
> + struct bpf_insn insns[] = {
> + BPF_LD_MAP_FD(BPF_REG_1, 0), /* map fd */
> + BPF_MOV64_IMM(BPF_REG_2, 0), /* flags, not used */
> + BPF_RAW_INSN(BPF_JMP | BPF_CALL, 0, 0, 0,
> + BPF_FUNC_get_local_storage),
> + BPF_MOV64_IMM(BPF_REG_1, 1),
> + BPF_STX_XADD(BPF_DW, BPF_REG_0, BPF_REG_1, 0),
> + BPF_LDX_MEM(BPF_W, BPF_REG_1, BPF_REG_0, 0),
> + BPF_ALU64_IMM(BPF_AND, BPF_REG_1, 0x1),
> + BPF_MOV64_REG(BPF_REG_0, BPF_REG_1),
> + BPF_EXIT_INSN(),
> + };
> +
> + cg_storage_fd = bpf_create_map(BPF_MAP_TYPE_CGROUP_STORAGE, sizeof(key),
> + sizeof(value), 0, 0);
> + if (cg_storage_fd < 0) {
> + printf("Failed to create map: %s\n", strerror(errno));
> + return -1;
> + }
> +
> + insns[0].imm = cg_storage_fd;
> +
> + return load_insns(test, insns, sizeof(insns) / sizeof(struct bpf_insn));
> +}
> +
> static int cmp_addr(const struct sockaddr_storage *addr1,
> const struct sockaddr_storage *addr2, int cmp_port)
> {
> @@ -1346,6 +1416,57 @@ static int run_sendmsg_test_case(const struct
> sock_addr_test *test)
> return err;
> }
>
> +static int run_release_test_case(const struct sock_addr_test *test)
> +{
> + socklen_t addr_len = sizeof(struct sockaddr_storage);
> + struct sockaddr_storage requested_addr;
> + struct sockaddr_storage expected_addr;
> + struct bpf_cgroup_storage_key key;
> + unsigned long long value;
> + int servfd = -1;
> +
> + if (bpf_map_get_next_key(cg_storage_fd, NULL, &key)) {
> + printf("Failed to get the first key in cgroup storage\n");
> + return -1;
> + }
> +
> + if (bpf_map_lookup_elem(cg_storage_fd, &key, &value)) {
> + printf("Failed to lookup cgroup storage 0\n");
> + return -1;
> + }
> +
> + if (value != 0) {
> + printf("Unexpected data in the cgroup storage: %llu\n", value);
> + return -1;
> + }
> +
> + if (init_addrs(test, &requested_addr, &expected_addr, NULL))
> + return -1;
> +
> + servfd = start_server(test->type, &requested_addr, addr_len);
> + if (servfd < 0)
> + return -1;
> +
> + close(servfd);
> +
> + if (bpf_map_get_next_key(cg_storage_fd, NULL, &key)) {
> + printf("Failed to get the first key in cgroup storage\n");
> + return -1;
> + }
> +
> + if (bpf_map_lookup_elem(cg_storage_fd, &key, &value)) {
> + printf("Failed to lookup cgroup storage 0\n");
> + return -1;
> + }
> +
> + if (value != 1) {
> + printf("Unexpected data in the cgroup storage: %llu\n", value);
> + return -1;
> + }
> +
> + return 0;
> +}
> +
> static int run_test_case(int cgfd, const struct sock_addr_test *test)
> {
> int progfd = -1;
> @@ -1381,6 +1502,10 @@ static int run_test_case(int cgfd, const struct
> sock_addr_test *test)
> case BPF_CGROUP_UDP6_SENDMSG:
> err = run_sendmsg_test_case(test);
> break;
> + case BPF_CGROUP_INET4_SOCK_RELEASE:
> + case BPF_CGROUP_INET6_SOCK_RELEASE:
> + err = run_release_test_case(test);
> + break;
> default:
> goto err;
> }
> @@ -1405,6 +1530,10 @@ static int run_test_case(int cgfd, const struct
> sock_addr_test *test)
> /* Detaching w/o checking return code: best effort attempt. */
> if (progfd != -1)
> bpf_prog_detach(cgfd, test->attach_type);
> + if (cg_storage_fd > 0) {
> + close(cg_storage_fd);
> + cg_storage_fd = 0;
Nit: zero is valid fd (stdin) and should not be used as "undefined" value.
That's why -1 is used with progfd and other descriptors in the test.
> + }
> close(progfd);
> printf("[%s]\n", err ? "FAIL" : "PASS");
> return err;
> --
> 2.20.1.321.g9e740568ce-goog
>
--
Andrey Ignatov
