LXR linux/tools/testing/selftests/net/socket.c

   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}
  93