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