commit df40ceaaa17dbe4fb5431f6cb41bc1560192f9cb Author: David Goulet <dgoulet@ev0ke.net> Date: Mon Mar 31 13:49:43 2014 -0400 Tests: add socket() test Add unit test for socket() and socketpair(). Signed-off-by: David Goulet <dgoulet@ev0ke.net> --- .gitignore | 1 + tests/Makefile.am | 5 ++- tests/test_list | 1 + tests/test_socket.c | 123 +++++++++++++++++++++++++++++++++++++++++++++++++++ 4 files changed, 129 insertions(+), 1 deletion(-) diff --git a/.gitignore b/.gitignore index 59fcadb..c474007 100644 --- a/.gitignore +++ b/.gitignore @@ -44,6 +44,7 @@ doc/usewithtor.1 src/bin/torsocks tests/test_dns +tests/test_socket tests/unit/test_onion tests/unit/test_connection tests/unit/test_utils diff --git a/tests/Makefile.am b/tests/Makefile.am index 18eb5a9..ae3c2d1 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -6,11 +6,14 @@ LIBTAP=$(top_builddir)/tests/utils/tap/libtap.la LIBTORSOCKS=$(top_builddir)/src/lib/libtorsocks.la -noinst_PROGRAMS = test_dns +noinst_PROGRAMS = test_dns test_socket test_dns_SOURCES = test_dns.c test_dns_LDADD = $(LIBTAP) $(LIBTORSOCKS) +test_socket_SOURCES = test_socket.c +test_socket_LDADD = $(LIBTAP) $(LIBTORSOCKS) + check-am: ./run.sh test_list diff --git a/tests/test_list b/tests/test_list index d5f09ba..542b93f 100644 --- a/tests/test_list +++ b/tests/test_list @@ -1,4 +1,5 @@ ./test_dns +./test_socket ./unit/test_onion ./unit/test_connection ./unit/test_utils diff --git a/tests/test_socket.c b/tests/test_socket.c new file mode 100644 index 0000000..55f1586 --- /dev/null +++ b/tests/test_socket.c @@ -0,0 +1,123 @@ +/* + * Copyright (C) 2014 - David Goulet <dgoulet@ev0ke.net> + * + * This program is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License, version 2 only, as + * published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for + * more details. + * + * You should have received a copy of the GNU General Public License along with + * this program; if not, write to the Free Software Foundation, Inc., 51 + * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#include <arpa/inet.h> +#include <netinet/in.h> +#include <stdio.h> +#include <sys/socket.h> + +#include <lib/torsocks.h> + +#include <tap/tap.h> + +#define NUM_TESTS 18 + +static void test_socketpair_types(void) +{ + int fd[2], ret1, ret2, err; + + err = socketpair(AF_UNIX, SOCK_STREAM, 0, fd); + ret1 = close(fd[0]); + ret2 = close(fd[1]); + ok (fd[0] != -1 && fd[1] != -1 && !err && ret1 == 0 && ret2 == 0, + "Unix socket valid for socketpair"); + + err = socketpair(AF_INET, SOCK_STREAM, 0, fd); + ok (err == -1 && errno == EPERM, "INET socket NOT valid for socketpair"); + + err = socketpair(AF_INET6, SOCK_STREAM, 0, fd); + ok (err == -1 && errno == EPERM, "INET6 socket NOT valid for socketpair"); +} + +static void test_socket_types(void) +{ + int fd, ret; + + fd = socket(AF_UNIX, SOCK_STREAM, 0); + ret = close(fd); + ok (fd != -1 && ret == 0, "Unix socket is valid"); + + fd = socket(AF_LOCAL, SOCK_STREAM, 0); + ret = close(fd); + ok (fd != -1 && ret == 0, "AF local socket is valid"); + + fd = socket(AF_INET, SOCK_STREAM, IPPROTO_IP); + ret = close(fd); + ok (fd != -1 && ret == 0, "IPv4 TCP socket is valid"); + + fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, IPPROTO_IP); + ret = close(fd); + ok (fd != -1 && ret == 0, "IPv4 TCP non block socket is valid"); + + fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, + IPPROTO_IP); + ret = close(fd); + ok (fd != -1 && ret == 0, "IPv4 TCP non block/cloexec socket is valid"); + + fd = socket(AF_INET6, SOCK_STREAM, IPPROTO_IP); + ret = close(fd); + ok (fd != -1 && ret == 0, "IPv6 TCP socket is valid"); + + fd = socket(AF_INET6, SOCK_STREAM | SOCK_NONBLOCK, IPPROTO_IP); + ret = close(fd); + ok (fd != -1 && ret == 0, "IPv6 TCP non block socket is valid"); + + fd = socket(AF_INET6, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, + IPPROTO_IP); + ret = close(fd); + ok (fd != -1 && ret == 0, "IPv6 TCP non block/cloexec socket is valid"); + + fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); + ok (fd == -1 && errno == EPERM, + "IPv4 UDP socket is NOT valid"); + + fd = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK, IPPROTO_UDP); + ok (fd == -1 && errno == EPERM, + "IPv4 UDP non block socket is NOT valid"); + + fd = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK | SOCK_CLOEXEC, + IPPROTO_UDP); + ok (fd == -1 && errno == EPERM, + "IPv4 UDP non block/cloexec socket is NOT valid"); + + fd = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP); + ok (fd == -1 && errno == EPERM, + "IPv6 UDP socket is NOT valid"); + + fd = socket(AF_INET, SOCK_RAW, IPPROTO_RAW); + ok (fd == -1 && errno == EPERM, + "IPv4 RAW socket is NOT valid"); + + fd = socket(AF_INET, SOCK_RAW, IPPROTO_ICMP); + ok (fd == -1 && errno == EPERM, + "IPv4 RAW ICMP socket is NOT valid"); + + fd = socket(AF_INET6, SOCK_RAW, IPPROTO_RAW); + ok (fd == -1 && errno == EPERM, + "IPv6 RAW socket is NOT valid"); +} + +int main(int argc, char **argv) +{ + /* Libtap call for the number of tests planned. */ + plan_tests(NUM_TESTS); + + test_socket_types(); + test_socketpair_types(); + + return 0; +}