// 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); }