Skip to content

Commit b08f205

Browse files
mmhalMartin KaFai Lau
authored andcommitted
selftests/bpf: Socket pair creation, cleanups
Following create_pair() changes, remove unused function argument in create_socket_pairs() and adapt its callers, i.e. drop the open-coded loopback socket creation. Reviewed-by: Jakub Sitnicki <[email protected]> Tested-by: Jakub Sitnicki <[email protected]> Suggested-by: Jakub Sitnicki <[email protected]> Signed-off-by: Michal Luczaj <[email protected]> Link: https://lore.kernel.org/r/[email protected] Signed-off-by: Martin KaFai Lau <[email protected]>
1 parent 190de54 commit b08f205

File tree

3 files changed

+13
-26
lines changed

3 files changed

+13
-26
lines changed

tools/testing/selftests/bpf/prog_tests/sockmap_basic.c

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -503,8 +503,8 @@ static void test_sockmap_skb_verdict_shutdown(void)
503503

504504
static void test_sockmap_skb_verdict_fionread(bool pass_prog)
505505
{
506+
int err, map, verdict, c0 = -1, c1 = -1, p0 = -1, p1 = -1;
506507
int expected, zero = 0, sent, recvd, avail;
507-
int err, map, verdict, s, c0 = -1, c1 = -1, p0 = -1, p1 = -1;
508508
struct test_sockmap_pass_prog *pass = NULL;
509509
struct test_sockmap_drop_prog *drop = NULL;
510510
char buf[256] = "0123456789";
@@ -531,11 +531,8 @@ static void test_sockmap_skb_verdict_fionread(bool pass_prog)
531531
if (!ASSERT_OK(err, "bpf_prog_attach"))
532532
goto out;
533533

534-
s = socket_loopback(AF_INET, SOCK_STREAM);
535-
if (!ASSERT_GT(s, -1, "socket_loopback(s)"))
536-
goto out;
537-
err = create_socket_pairs(s, AF_INET, SOCK_STREAM, &c0, &c1, &p0, &p1);
538-
if (!ASSERT_OK(err, "create_socket_pairs(s)"))
534+
err = create_socket_pairs(AF_INET, SOCK_STREAM, &c0, &c1, &p0, &p1);
535+
if (!ASSERT_OK(err, "create_socket_pairs()"))
539536
goto out;
540537

541538
err = bpf_map_update_elem(map, &zero, &c1, BPF_NOEXIST);

tools/testing/selftests/bpf/prog_tests/sockmap_helpers.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -437,8 +437,8 @@ static inline int create_pair(int family, int sotype, int *p0, int *p1)
437437
return err;
438438
}
439439

440-
static inline int create_socket_pairs(int s, int family, int sotype,
441-
int *c0, int *c1, int *p0, int *p1)
440+
static inline int create_socket_pairs(int family, int sotype, int *c0, int *c1,
441+
int *p0, int *p1)
442442
{
443443
int err;
444444

tools/testing/selftests/bpf/prog_tests/sockmap_listen.c

Lines changed: 8 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -677,21 +677,18 @@ static void redir_to_connected(int family, int sotype, int sock_mapfd,
677677
int verd_mapfd, enum redir_mode mode)
678678
{
679679
const char *log_prefix = redir_mode_str(mode);
680-
int s, c0, c1, p0, p1;
680+
int c0, c1, p0, p1;
681681
unsigned int pass;
682682
int err, n;
683683
u32 key;
684684
char b;
685685

686686
zero_verdict_count(verd_mapfd);
687687

688-
s = socket_loopback(family, sotype | SOCK_NONBLOCK);
689-
if (s < 0)
690-
return;
691-
692-
err = create_socket_pairs(s, family, sotype, &c0, &c1, &p0, &p1);
688+
err = create_socket_pairs(family, sotype | SOCK_NONBLOCK, &c0, &c1,
689+
&p0, &p1);
693690
if (err)
694-
goto close_srv;
691+
return;
695692

696693
err = add_to_sockmap(sock_mapfd, p0, p1);
697694
if (err)
@@ -722,8 +719,6 @@ static void redir_to_connected(int family, int sotype, int sock_mapfd,
722719
xclose(c1);
723720
xclose(p0);
724721
xclose(c0);
725-
close_srv:
726-
xclose(s);
727722
}
728723

729724
static void test_skb_redir_to_connected(struct test_sockmap_listen *skel,
@@ -909,7 +904,7 @@ static void test_msg_redir_to_listening_with_link(struct test_sockmap_listen *sk
909904

910905
static void redir_partial(int family, int sotype, int sock_map, int parser_map)
911906
{
912-
int s, c0 = -1, c1 = -1, p0 = -1, p1 = -1;
907+
int c0 = -1, c1 = -1, p0 = -1, p1 = -1;
913908
int err, n, key, value;
914909
char buf[] = "abc";
915910

@@ -919,13 +914,10 @@ static void redir_partial(int family, int sotype, int sock_map, int parser_map)
919914
if (err)
920915
return;
921916

922-
s = socket_loopback(family, sotype | SOCK_NONBLOCK);
923-
if (s < 0)
924-
goto clean_parser_map;
925-
926-
err = create_socket_pairs(s, family, sotype, &c0, &c1, &p0, &p1);
917+
err = create_socket_pairs(family, sotype | SOCK_NONBLOCK, &c0, &c1,
918+
&p0, &p1);
927919
if (err)
928-
goto close_srv;
920+
goto clean_parser_map;
929921

930922
err = add_to_sockmap(sock_map, p0, p1);
931923
if (err)
@@ -944,8 +936,6 @@ static void redir_partial(int family, int sotype, int sock_map, int parser_map)
944936
xclose(p0);
945937
xclose(c1);
946938
xclose(p1);
947-
close_srv:
948-
xclose(s);
949939

950940
clean_parser_map:
951941
key = 0;

0 commit comments

Comments
 (0)