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