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
 9struct 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
25static 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
36static 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
87int main(void)
88{
89	int err = run_tests();
90
91	return err;
92}