File: test_connect.c

package info (click to toggle)
torsocks 2.5.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 916 kB
  • sloc: ansic: 7,437; sh: 1,038; makefile: 105
file content (112 lines) | stat: -rw-r--r-- 3,689 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*
 * 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 8

/* Suppress output messages. */
int tsocks_loglevel = MSGNONE;
//int tsocks_loglevel = MSGDEBUG;

static void test_connect_deny(void)
{
	int fd, ret;
	struct sockaddr_in sin = {0};
	struct sockaddr_in6 sin6 = {0};

	fd = tsocks_libc_socket(AF_INET, SOCK_RAW, IPPROTO_TCP);
	skip_start(fd == -1, 1, "Couldn't create raw socket");
	sin.sin_family = AF_INET;
	ret = connect(fd, (struct sockaddr *) &sin, sizeof(sin));
	ok (ret == -1 && errno == EPERM,
	    "Connect with RAW socket NOT valid. ret=%d, errno=%d: %s",
	    ret, errno, strerror(errno));
	close(fd);
	skip_end();

	sin.sin_family = AF_INET;
	fd = tsocks_libc_socket(sin.sin_family, SOCK_DGRAM, 0);
	ret = connect(fd, (struct sockaddr *) &sin, sizeof(sin));
	ok (ret == -1 && errno == EPERM,
	    "Connect with UDP socket NOT valid. ret=%d, errno=%d: %s",
	    ret, errno, strerror(errno));
	close(fd);

	inet_pton(sin.sin_family, "0.0.0.0", &sin.sin_addr);
	fd = tsocks_libc_socket(sin.sin_family, SOCK_STREAM, 0);
	ret = connect(fd, (struct sockaddr *) &sin, sizeof(sin));
	ok (ret == -1 && errno == EPERM,
	    "Connect with ANY address is NOT valid. ret=%d, errno=%d: %s",
	    ret, errno, strerror(errno));
	close(fd);

	inet_pton(sin.sin_family, "127.0.0.1", &sin.sin_addr);
	fd = tsocks_libc_socket(sin.sin_family, SOCK_STREAM, 0);
	ret = connect(fd, (struct sockaddr *) &sin, sizeof(sin));
	ok (ret == -1 && errno == EPERM,
	    "Connect with local address is NOT valid. ret=%d, errno=%d: %s",
	    ret, errno, strerror(errno));
	close(fd);

	sin6.sin6_family = AF_INET6;
	fd = tsocks_libc_socket(sin6.sin6_family, SOCK_DGRAM, 0);
	ret = connect(fd, (struct sockaddr *) &sin6, sizeof(sin6));
	ok (ret == -1 && errno == EPERM,
	    "Connect with UDPv6 socket NOT valid. ret=%d, errno=%d: %s",
	    ret, errno, strerror(errno));
	close(fd);

	inet_pton(sin6.sin6_family, "::", &sin6.sin6_addr);
	fd = tsocks_libc_socket(sin6.sin6_family, SOCK_STREAM, 0);
	ret = connect(fd, (struct sockaddr *) &sin6, sizeof(sin6));
	ok (ret == -1 && errno == EPERM,
	    "Connect with ANYv6 address is NOT valid. ret=%d, errno=%d: %s",
	    ret, errno, strerror(errno));
	close(fd);

	inet_pton(sin6.sin6_family, "::1", &sin6.sin6_addr);
	fd = tsocks_libc_socket(sin6.sin6_family, SOCK_STREAM, 0);
	ret = connect(fd, (struct sockaddr *) &sin6, sizeof(sin6));
	ok (ret == -1 && errno == EPERM,
	    "Connect with local v6 address is NOT valid. ret=%d, errno=%d: %s",
	    ret, errno, strerror(errno));
	close(fd);

	/* Bad fd. */
	ret = connect(42, (struct sockaddr *) &sin, 42);
	ok (ret == -1 && errno == EBADF,
	    "Bad socket FD. ret=%d errno=%d: %s",
	    ret, errno, strerror(errno));
}

int main(int argc, char **argv)
{
	/* Libtap call for the number of tests planned. */
	plan_tests(NUM_TESTS);

	test_connect_deny();

	return exit_status();
}