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