code.delx.au - pulseaudio/blob - src/tests/ipacl-test.c

   1 #ifdef HAVE_CONFIG_H
   2 #include <config.h>
   3 #endif
   4
   5 #include <sys/types.h>
   6 #include <stdio.h>
   7 #include <unistd.h>
   8 #include <assert.h>
   9 #include <string.h>
  10
  11 #ifdef HAVE_NETINET_IN_H
  12 #include <netinet/in.h>
  13 #endif
  14 #ifdef HAVE_NETINET_IN_SYSTM_H
  15 #include <netinet/in_systm.h>
  16 #endif
  17 #ifdef HAVE_NETINET_IP_H
  18 #include <netinet/ip.h>
  19 #endif
  20 #ifdef HAVE_ARPA_INET_H
  21 #include <arpa/inet.h>
  22 #endif
  23
  24 #include <pulsecore/socket.h>
  25 #include <pulsecore/macro.h>
  26 #include <pulsecore/ipacl.h>
  27 #include <pulsecore/inet_pton.h>
  28
  29 int main(int argc, char *argv[]) {
  30     struct sockaddr_in sa;
  31 #ifdef HAVE_IPV6
  32     struct sockaddr_in6 sa6;
  33 #endif
  34     int fd;
  35     int r;
  36     pa_ip_acl *acl;
  37
  38     fd = socket(PF_INET, SOCK_STREAM, 0);
  39     assert(fd >= 0);
  40
  41     sa.sin_family = AF_INET;
  42     sa.sin_port = htons(22);
  43     sa.sin_addr.s_addr = inet_addr("127.0.0.1");
  44
  45     r = connect(fd, (struct sockaddr*) &sa, sizeof(sa));
  46     assert(r >= 0);
  47
  48     acl = pa_ip_acl_new("127.0.0.1");
  49     assert(acl);
  50     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
  51     pa_ip_acl_free(acl);
  52
  53     acl = pa_ip_acl_new("127.0.0.2/0");
  54     assert(acl);
  55     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
  56     pa_ip_acl_free(acl);
  57
  58     acl = pa_ip_acl_new("127.0.0.1/32");
  59     assert(acl);
  60     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
  61     pa_ip_acl_free(acl);
  62
  63     acl = pa_ip_acl_new("127.0.0.1/7");
  64     assert(acl);
  65     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
  66     pa_ip_acl_free(acl);
  67
  68     acl = pa_ip_acl_new("127.0.0.2");
  69     assert(acl);
  70     printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd));
  71     pa_ip_acl_free(acl);
  72
  73     acl = pa_ip_acl_new("127.0.0.0/8;0.0.0.0/32");
  74     assert(acl);
  75     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
  76     pa_ip_acl_free(acl);
  77
  78     acl = pa_ip_acl_new("128.0.0.2/9");
  79     assert(acl);
  80     printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd));
  81     pa_ip_acl_free(acl);
  82
  83     acl = pa_ip_acl_new("::1/9");
  84     assert(acl);
  85     printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd));
  86     pa_ip_acl_free(acl);
  87
  88     close(fd);
  89
  90 #ifdef HAVE_IPV6
  91     if ( (fd = socket(PF_INET6, SOCK_STREAM, 0)) < 0 ) {
  92       printf("Unable to open IPv6 socket, IPv6 tests ignored");
  93       return 0;
  94     }
  95
  96     memset(&sa6, 0, sizeof(sa6));
  97     sa6.sin6_family = AF_INET6;
  98     sa6.sin6_port = htons(22);
  99     pa_assert_se(inet_pton(AF_INET6, "::1", &sa6.sin6_addr) == 1);
 100
 101     r = connect(fd, (struct sockaddr*) &sa6, sizeof(sa6));
 102     assert(r >= 0);
 103
 104     acl = pa_ip_acl_new("::1");
 105     assert(acl);
 106     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
 107     pa_ip_acl_free(acl);
 108
 109     acl = pa_ip_acl_new("::1/9");
 110     assert(acl);
 111     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
 112     pa_ip_acl_free(acl);
 113
 114     acl = pa_ip_acl_new("::/0");
 115     assert(acl);
 116     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
 117     pa_ip_acl_free(acl);
 118
 119     acl = pa_ip_acl_new("::2/128");
 120     assert(acl);
 121     printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd));
 122     pa_ip_acl_free(acl);
 123
 124     acl = pa_ip_acl_new("::2/127");
 125     assert(acl);
 126     printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd));
 127     pa_ip_acl_free(acl);
 128
 129     acl = pa_ip_acl_new("::2/126");
 130     assert(acl);
 131     printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd));
 132     pa_ip_acl_free(acl);
 133
 134     close(fd);
 135 #endif
 136
 137     return 0;
 138 }