X-Git-Url: https://code.delx.au/pulseaudio/blobdiff_plain/126e4cf89f5ccfe393e60a930c88ab513eeee085..a3a795ef3d740b9b41d188f0a00296e0561831f0:/src/tests/ipacl-test.c?ds=sidebyside diff --git a/src/tests/ipacl-test.c b/src/tests/ipacl-test.c index bcdd469a..4ce3490c 100644 --- a/src/tests/ipacl-test.c +++ b/src/tests/ipacl-test.c @@ -5,12 +5,10 @@ #include #include #include -#include #include -#ifdef HAVE_SYS_SOCKET_H -#include -#endif +#include + #ifdef HAVE_NETINET_IN_H #include #endif @@ -20,115 +18,99 @@ #ifdef HAVE_NETINET_IP_H #include #endif -#ifdef HAVE_ARPA_INET_H -#include -#endif - -#include "../pulsecore/winsock.h" +#include +#include +#include #include +#include -int main(int argc, char *argv[]) { +static void do_ip_acl_check(const char *s, int fd, int expected) { + pa_ip_acl *acl; + int result; + + acl = pa_ip_acl_new(s); + fail_unless(acl != NULL); + result = pa_ip_acl_check(acl, fd); + pa_ip_acl_free(acl); + + pa_log_info("%-20s result=%u (should be %u)", s, result, expected); + fail_unless(result == expected); +} + +START_TEST (ipacl_test) { struct sockaddr_in sa; +#ifdef HAVE_IPV6 struct sockaddr_in6 sa6; +#endif int fd; int r; - pa_ip_acl *acl; fd = socket(PF_INET, SOCK_STREAM, 0); - assert(fd >= 0); + fail_unless(fd >= 0); sa.sin_family = AF_INET; sa.sin_port = htons(22); sa.sin_addr.s_addr = inet_addr("127.0.0.1"); r = connect(fd, (struct sockaddr*) &sa, sizeof(sa)); - assert(r >= 0); - - acl = pa_ip_acl_new("127.0.0.1"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); - - acl = pa_ip_acl_new("127.0.0.2/0"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); - - acl = pa_ip_acl_new("127.0.0.1/32"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); - - acl = pa_ip_acl_new("127.0.0.1/7"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); - - acl = pa_ip_acl_new("127.0.0.2"); - assert(acl); - printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); - - acl = pa_ip_acl_new("127.0.0.0/8;0.0.0.0/32"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); + fail_unless(r >= 0); - acl = pa_ip_acl_new("128.0.0.2/9"); - assert(acl); - printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); - - acl = pa_ip_acl_new("::1/9"); - assert(acl); - printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); + do_ip_acl_check("127.0.0.1", fd, 1); + do_ip_acl_check("127.0.0.2/0", fd, 1); + do_ip_acl_check("127.0.0.1/32", fd, 1); + do_ip_acl_check("127.0.0.1/7", fd, 1); + do_ip_acl_check("127.0.0.2", fd, 0); + do_ip_acl_check("127.0.0.0/8;0.0.0.0/32", fd, 1); + do_ip_acl_check("128.0.0.2/9", fd, 0); + do_ip_acl_check("::1/9", fd, 0); close(fd); - fd = socket(PF_INET6, SOCK_STREAM, 0); - assert(fd >= 0); +#ifdef HAVE_IPV6 + if ( (fd = socket(PF_INET6, SOCK_STREAM, 0)) < 0 ) { + pa_log_error("Unable to open IPv6 socket, IPv6 tests ignored"); + return; + } memset(&sa6, 0, sizeof(sa6)); sa6.sin6_family = AF_INET6; sa6.sin6_port = htons(22); - inet_pton(AF_INET6, "::1", &sa6.sin6_addr); + fail_unless(inet_pton(AF_INET6, "::1", &sa6.sin6_addr) == 1); r = connect(fd, (struct sockaddr*) &sa6, sizeof(sa6)); - assert(r >= 0); - - acl = pa_ip_acl_new("::1"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); + fail_unless(r >= 0); - acl = pa_ip_acl_new("::1/9"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); + do_ip_acl_check("::1", fd, 1); + do_ip_acl_check("::1/9", fd, 1); + do_ip_acl_check("::/0", fd, 1); + do_ip_acl_check("::2/128", fd, 0); + do_ip_acl_check("::2/127", fd, 0); + do_ip_acl_check("::2/126", fd, 1); - acl = pa_ip_acl_new("::/0"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); + close(fd); +#endif +} +END_TEST - acl = pa_ip_acl_new("::2/128"); - assert(acl); - printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); +int main(int argc, char *argv[]) { + int failed = 0; + Suite *s; + TCase *tc; + SRunner *sr; - acl = pa_ip_acl_new("::2/127"); - assert(acl); - printf("result=%u (should be 0)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); + if (!getenv("MAKE_CHECK")) + pa_log_set_level(PA_LOG_DEBUG); - acl = pa_ip_acl_new("::2/126"); - assert(acl); - printf("result=%u (should be 1)\n", pa_ip_acl_check(acl, fd)); - pa_ip_acl_free(acl); + s = suite_create("IP ACL"); + tc = tcase_create("ipacl"); + tcase_add_test(tc, ipacl_test); + suite_add_tcase(s, tc); - close(fd); + sr = srunner_create(s); + srunner_run_all(sr, CK_NORMAL); + failed = srunner_ntests_failed(sr); + srunner_free(sr); - return 0; + return (failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE; }