mirror of
https://gitlab.torproject.org/tpo/core/tor.git
synced 2025-02-25 23:21:38 +01:00
Add some unit tests.
This commit is contained in:
parent
5492de76dd
commit
ce419a78c5
3 changed files with 152 additions and 0 deletions
|
@ -19,6 +19,7 @@ test_SOURCES = \
|
||||||
test_data.c \
|
test_data.c \
|
||||||
test_dir.c \
|
test_dir.c \
|
||||||
test_microdesc.c \
|
test_microdesc.c \
|
||||||
|
test_pt.c \
|
||||||
test_util.c \
|
test_util.c \
|
||||||
tinytest.c
|
tinytest.c
|
||||||
|
|
||||||
|
|
|
@ -1261,6 +1261,7 @@ extern struct testcase_t container_tests[];
|
||||||
extern struct testcase_t util_tests[];
|
extern struct testcase_t util_tests[];
|
||||||
extern struct testcase_t dir_tests[];
|
extern struct testcase_t dir_tests[];
|
||||||
extern struct testcase_t microdesc_tests[];
|
extern struct testcase_t microdesc_tests[];
|
||||||
|
extern struct testcase_t pt_tests[];
|
||||||
|
|
||||||
static struct testgroup_t testgroups[] = {
|
static struct testgroup_t testgroups[] = {
|
||||||
{ "", test_array },
|
{ "", test_array },
|
||||||
|
@ -1270,6 +1271,7 @@ static struct testgroup_t testgroups[] = {
|
||||||
{ "util/", util_tests },
|
{ "util/", util_tests },
|
||||||
{ "dir/", dir_tests },
|
{ "dir/", dir_tests },
|
||||||
{ "dir/md/", microdesc_tests },
|
{ "dir/md/", microdesc_tests },
|
||||||
|
{ "pt/", pt_tests },
|
||||||
END_OF_GROUPS
|
END_OF_GROUPS
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
149
src/test/test_pt.c
Normal file
149
src/test/test_pt.c
Normal file
|
@ -0,0 +1,149 @@
|
||||||
|
/* Copyright (c) 2001-2004, Roger Dingledine.
|
||||||
|
* Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
|
||||||
|
* Copyright (c) 2007-2011, The Tor Project, Inc. */
|
||||||
|
/* See LICENSE for licensing information */
|
||||||
|
|
||||||
|
#include "orconfig.h"
|
||||||
|
#define PT_PRIVATE
|
||||||
|
#include "or.h"
|
||||||
|
#include "pluggable_transports.h"
|
||||||
|
#include "circuitbuild.h"
|
||||||
|
#include "test.h"
|
||||||
|
|
||||||
|
static void
|
||||||
|
reset_mp(managed_proxy_t *mp)
|
||||||
|
{
|
||||||
|
mp->conf_state = PT_PROTO_INFANT;
|
||||||
|
SMARTLIST_FOREACH(mp->transports, transport_t *, t, transport_free(t));
|
||||||
|
smartlist_clear(mp->transports);
|
||||||
|
smartlist_free(mp->transports);
|
||||||
|
mp->transports = smartlist_create();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
test_pt_parsing(void)
|
||||||
|
{
|
||||||
|
char line[200];
|
||||||
|
|
||||||
|
managed_proxy_t *mp = tor_malloc(sizeof(managed_proxy_t));
|
||||||
|
mp->conf_state = PT_PROTO_INFANT;
|
||||||
|
mp->transports = smartlist_create();
|
||||||
|
|
||||||
|
/* incomplete cmethod */
|
||||||
|
strcpy(line,"CMETHOD trebuchet");
|
||||||
|
test_assert(parse_cmethod_line(line, mp) < 0);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
/* wrong proxy type */
|
||||||
|
strcpy(line,"CMETHOD trebuchet dog 127.0.0.1:1999");
|
||||||
|
test_assert(parse_cmethod_line(line, mp) < 0);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
/* wrong addrport */
|
||||||
|
strcpy(line,"CMETHOD trebuchet socks4 abcd");
|
||||||
|
test_assert(parse_cmethod_line(line, mp) < 0);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
/* correct line */
|
||||||
|
strcpy(line,"CMETHOD trebuchet socks5 127.0.0.1:1999");
|
||||||
|
test_assert(parse_cmethod_line(line, mp) == 0);
|
||||||
|
test_assert(smartlist_len(mp->transports));
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
/* incomplete smethod */
|
||||||
|
strcpy(line,"SMETHOD trebuchet");
|
||||||
|
test_assert(parse_smethod_line(line, mp) < 0);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
/* wrong addr type */
|
||||||
|
strcpy(line,"SMETHOD trebuchet abcd");
|
||||||
|
test_assert(parse_smethod_line(line, mp) < 0);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
/* cowwect */
|
||||||
|
strcpy(line,"SMETHOD trebuchy 127.0.0.1:1999");
|
||||||
|
test_assert(parse_smethod_line(line, mp) == 0);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
/* unsupported version */
|
||||||
|
strcpy(line,"VERSION 666");
|
||||||
|
test_assert(parse_version(line, mp) < 0);
|
||||||
|
|
||||||
|
/* incomplete VERSION */
|
||||||
|
strcpy(line,"VERSION ");
|
||||||
|
test_assert(parse_version(line, mp) < 0);
|
||||||
|
|
||||||
|
/* correct VERSION */
|
||||||
|
strcpy(line,"VERSION 1");
|
||||||
|
test_assert(parse_version(line, mp) == 0);
|
||||||
|
|
||||||
|
done:
|
||||||
|
tor_free(mp);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
test_pt_protocol(void)
|
||||||
|
{
|
||||||
|
char line[200];
|
||||||
|
|
||||||
|
managed_proxy_t *mp = tor_malloc(sizeof(managed_proxy_t));
|
||||||
|
mp->conf_state = PT_PROTO_INFANT;
|
||||||
|
mp->transports = smartlist_create();
|
||||||
|
|
||||||
|
/* various wrong protocol runs: */
|
||||||
|
|
||||||
|
strcpy(line, "TEST TEST");
|
||||||
|
handle_proxy_line(line, mp);
|
||||||
|
test_assert(mp->conf_state == PT_PROTO_BROKEN);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
strcpy(line,"VERSION 1");
|
||||||
|
handle_proxy_line(line, mp);
|
||||||
|
test_assert(mp->conf_state == PT_PROTO_ACCEPTING_METHODS);
|
||||||
|
|
||||||
|
strcpy(line,"VERSION 1");
|
||||||
|
handle_proxy_line(line, mp);
|
||||||
|
test_assert(mp->conf_state == PT_PROTO_BROKEN);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
strcpy(line,"CMETHOD trebuchet socks5 127.0.0.1:1999");
|
||||||
|
handle_proxy_line(line, mp);
|
||||||
|
test_assert(mp->conf_state == PT_PROTO_BROKEN);
|
||||||
|
|
||||||
|
reset_mp(mp);
|
||||||
|
|
||||||
|
/* correct protocol run: */
|
||||||
|
strcpy(line,"VERSION 1");
|
||||||
|
handle_proxy_line(line, mp);
|
||||||
|
test_assert(mp->conf_state == PT_PROTO_ACCEPTING_METHODS);
|
||||||
|
|
||||||
|
strcpy(line,"CMETHOD trebuchet socks5 127.0.0.1:1999");
|
||||||
|
handle_proxy_line(line, mp);
|
||||||
|
test_assert(mp->conf_state == PT_PROTO_ACCEPTING_METHODS);
|
||||||
|
|
||||||
|
strcpy(line,"CMETHODS DONE");
|
||||||
|
handle_proxy_line(line, mp);
|
||||||
|
test_assert(mp->conf_state == PT_PROTO_CONFIGURED);
|
||||||
|
|
||||||
|
done:
|
||||||
|
tor_free(mp);
|
||||||
|
}
|
||||||
|
|
||||||
|
#define PT_LEGACY(name) \
|
||||||
|
{ #name, legacy_test_helper, 0, &legacy_setup, test_pt_ ## name }
|
||||||
|
|
||||||
|
struct testcase_t pt_tests[] = {
|
||||||
|
PT_LEGACY(parsing),
|
||||||
|
PT_LEGACY(protocol),
|
||||||
|
END_OF_TESTCASES
|
||||||
|
};
|
||||||
|
|
Loading…
Add table
Reference in a new issue