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