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