xref: /openbmc/linux/tools/testing/selftests/net/socket.c (revision b24413180f5600bcb3bb70fbed5cf186b60864bd)
1*b2441318SGreg Kroah-Hartman // SPDX-License-Identifier: GPL-2.0
2a6f68034SDavid S. Miller #include <stdio.h>
3a6f68034SDavid S. Miller #include <errno.h>
4a6f68034SDavid S. Miller #include <unistd.h>
5a6f68034SDavid S. Miller #include <string.h>
6a6f68034SDavid S. Miller #include <sys/types.h>
7a6f68034SDavid S. Miller #include <sys/socket.h>
8a6f68034SDavid S. Miller #include <netinet/in.h>
9a6f68034SDavid S. Miller 
10a6f68034SDavid S. Miller struct socket_testcase {
11a6f68034SDavid S. Miller 	int	domain;
12a6f68034SDavid S. Miller 	int	type;
13a6f68034SDavid S. Miller 	int	protocol;
14a6f68034SDavid S. Miller 
15a6f68034SDavid S. Miller 	/* 0    = valid file descriptor
16a6f68034SDavid S. Miller 	 * -foo = error foo
17a6f68034SDavid S. Miller 	 */
18a6f68034SDavid S. Miller 	int	expect;
19a6f68034SDavid S. Miller 
20a6f68034SDavid S. Miller 	/* If non-zero, accept EAFNOSUPPORT to handle the case
21a6f68034SDavid S. Miller 	 * of the protocol not being configured into the kernel.
22a6f68034SDavid S. Miller 	 */
23a6f68034SDavid S. Miller 	int	nosupport_ok;
24a6f68034SDavid S. Miller };
25a6f68034SDavid S. Miller 
26a6f68034SDavid S. Miller static struct socket_testcase tests[] = {
27a6f68034SDavid S. Miller 	{ AF_MAX,  0,           0,           -EAFNOSUPPORT,    0 },
28a6f68034SDavid S. Miller 	{ AF_INET, SOCK_STREAM, IPPROTO_TCP, 0,                1  },
29a6f68034SDavid S. Miller 	{ AF_INET, SOCK_DGRAM,  IPPROTO_TCP, -EPROTONOSUPPORT, 1  },
30a6f68034SDavid S. Miller 	{ AF_INET, SOCK_DGRAM,  IPPROTO_UDP, 0,                1  },
31a6f68034SDavid S. Miller 	{ AF_INET, SOCK_STREAM, IPPROTO_UDP, -EPROTONOSUPPORT, 1  },
32a6f68034SDavid S. Miller };
33a6f68034SDavid S. Miller 
34a6f68034SDavid S. Miller #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
35a6f68034SDavid S. Miller #define ERR_STRING_SZ	64
36a6f68034SDavid S. Miller 
37a6f68034SDavid S. Miller static int run_tests(void)
38a6f68034SDavid S. Miller {
39a6f68034SDavid S. Miller 	char err_string1[ERR_STRING_SZ];
40a6f68034SDavid S. Miller 	char err_string2[ERR_STRING_SZ];
41a6f68034SDavid S. Miller 	int i, err;
42a6f68034SDavid S. Miller 
43a6f68034SDavid S. Miller 	err = 0;
44a6f68034SDavid S. Miller 	for (i = 0; i < ARRAY_SIZE(tests); i++) {
45a6f68034SDavid S. Miller 		struct socket_testcase *s = &tests[i];
46a6f68034SDavid S. Miller 		int fd;
47a6f68034SDavid S. Miller 
48a6f68034SDavid S. Miller 		fd = socket(s->domain, s->type, s->protocol);
49a6f68034SDavid S. Miller 		if (fd < 0) {
50a6f68034SDavid S. Miller 			if (s->nosupport_ok &&
51a6f68034SDavid S. Miller 			    errno == EAFNOSUPPORT)
52a6f68034SDavid S. Miller 				continue;
53a6f68034SDavid S. Miller 
54a6f68034SDavid S. Miller 			if (s->expect < 0 &&
55a6f68034SDavid S. Miller 			    errno == -s->expect)
56a6f68034SDavid S. Miller 				continue;
57a6f68034SDavid S. Miller 
58a6f68034SDavid S. Miller 			strerror_r(-s->expect, err_string1, ERR_STRING_SZ);
59a6f68034SDavid S. Miller 			strerror_r(errno, err_string2, ERR_STRING_SZ);
60a6f68034SDavid S. Miller 
61a6f68034SDavid S. Miller 			fprintf(stderr, "socket(%d, %d, %d) expected "
62a6f68034SDavid S. Miller 				"err (%s) got (%s)\n",
63a6f68034SDavid S. Miller 				s->domain, s->type, s->protocol,
64a6f68034SDavid S. Miller 				err_string1, err_string2);
65a6f68034SDavid S. Miller 
66a6f68034SDavid S. Miller 			err = -1;
67a6f68034SDavid S. Miller 			break;
68a6f68034SDavid S. Miller 		} else {
69a6f68034SDavid S. Miller 			close(fd);
70a6f68034SDavid S. Miller 
71a6f68034SDavid S. Miller 			if (s->expect < 0) {
72a6f68034SDavid S. Miller 				strerror_r(errno, err_string1, ERR_STRING_SZ);
73a6f68034SDavid S. Miller 
74a6f68034SDavid S. Miller 				fprintf(stderr, "socket(%d, %d, %d) expected "
75a6f68034SDavid S. Miller 					"success got err (%s)\n",
76a6f68034SDavid S. Miller 					s->domain, s->type, s->protocol,
77a6f68034SDavid S. Miller 					err_string1);
78a6f68034SDavid S. Miller 
79a6f68034SDavid S. Miller 				err = -1;
80a6f68034SDavid S. Miller 				break;
81a6f68034SDavid S. Miller 			}
82a6f68034SDavid S. Miller 		}
83a6f68034SDavid S. Miller 	}
84a6f68034SDavid S. Miller 
85a6f68034SDavid S. Miller 	return err;
86a6f68034SDavid S. Miller }
87a6f68034SDavid S. Miller 
88a6f68034SDavid S. Miller int main(void)
89a6f68034SDavid S. Miller {
90a6f68034SDavid S. Miller 	int err = run_tests();
91a6f68034SDavid S. Miller 
92a6f68034SDavid S. Miller 	return err;
93a6f68034SDavid S. Miller }
94