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