| // SPDX-License-Identifier: GPL-2.0 |
| // Copyright (c) 2020 Cloudflare |
| |
| #include "test_progs.h" |
| |
| static int connected_socket_v4(void) |
| { |
| struct sockaddr_in addr = { |
| .sin_family = AF_INET, |
| .sin_port = htons(80), |
| .sin_addr = { inet_addr("127.0.0.1") }, |
| }; |
| socklen_t len = sizeof(addr); |
| int s, repair, err; |
| |
| s = socket(AF_INET, SOCK_STREAM, 0); |
| if (CHECK_FAIL(s == -1)) |
| goto error; |
| |
| repair = TCP_REPAIR_ON; |
| err = setsockopt(s, SOL_TCP, TCP_REPAIR, &repair, sizeof(repair)); |
| if (CHECK_FAIL(err)) |
| goto error; |
| |
| err = connect(s, (struct sockaddr *)&addr, len); |
| if (CHECK_FAIL(err)) |
| goto error; |
| |
| repair = TCP_REPAIR_OFF_NO_WP; |
| err = setsockopt(s, SOL_TCP, TCP_REPAIR, &repair, sizeof(repair)); |
| if (CHECK_FAIL(err)) |
| goto error; |
| |
| return s; |
| error: |
| perror(__func__); |
| close(s); |
| return -1; |
| } |
| |
| /* Create a map, populate it with one socket, and free the map. */ |
| static void test_sockmap_create_update_free(enum bpf_map_type map_type) |
| { |
| const int zero = 0; |
| int s, map, err; |
| |
| s = connected_socket_v4(); |
| if (CHECK_FAIL(s == -1)) |
| return; |
| |
| map = bpf_create_map(map_type, sizeof(int), sizeof(int), 1, 0); |
| if (CHECK_FAIL(map == -1)) { |
| perror("bpf_create_map"); |
| goto out; |
| } |
| |
| err = bpf_map_update_elem(map, &zero, &s, BPF_NOEXIST); |
| if (CHECK_FAIL(err)) { |
| perror("bpf_map_update"); |
| goto out; |
| } |
| |
| out: |
| close(map); |
| close(s); |
| } |
| |
| void test_sockmap_basic(void) |
| { |
| if (test__start_subtest("sockmap create_update_free")) |
| test_sockmap_create_update_free(BPF_MAP_TYPE_SOCKMAP); |
| if (test__start_subtest("sockhash create_update_free")) |
| test_sockmap_create_update_free(BPF_MAP_TYPE_SOCKHASH); |
| } |