123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199 |
- /* Copyright (c) 2001-2004, Roger Dingledine.
- * Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
- * Copyright (c) 2007-2013, The Tor Project, Inc. */
- /* See LICENSE for licensing information */
- #include "orconfig.h"
- #define PT_PRIVATE
- #include "or.h"
- #include "config.h"
- #include "confparse.h"
- #include "transports.h"
- #include "circuitbuild.h"
- #include "test.h"
- static void
- reset_mp(managed_proxy_t *mp)
- {
- mp->conf_state = PT_PROTO_LAUNCHED;
- SMARTLIST_FOREACH(mp->transports, transport_t *, t, transport_free(t));
- smartlist_clear(mp->transports);
- }
- 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_new();
- /* incomplete cmethod */
- strlcpy(line,"CMETHOD trebuchet",sizeof(line));
- test_assert(parse_cmethod_line(line, mp) < 0);
- reset_mp(mp);
- /* wrong proxy type */
- strlcpy(line,"CMETHOD trebuchet dog 127.0.0.1:1999",sizeof(line));
- test_assert(parse_cmethod_line(line, mp) < 0);
- reset_mp(mp);
- /* wrong addrport */
- strlcpy(line,"CMETHOD trebuchet socks4 abcd",sizeof(line));
- test_assert(parse_cmethod_line(line, mp) < 0);
- reset_mp(mp);
- /* correct line */
- strlcpy(line,"CMETHOD trebuchet socks5 127.0.0.1:1999",sizeof(line));
- test_assert(parse_cmethod_line(line, mp) == 0);
- test_assert(smartlist_len(mp->transports));
- reset_mp(mp);
- /* incomplete smethod */
- strlcpy(line,"SMETHOD trebuchet",sizeof(line));
- test_assert(parse_smethod_line(line, mp) < 0);
- reset_mp(mp);
- /* wrong addr type */
- strlcpy(line,"SMETHOD trebuchet abcd",sizeof(line));
- test_assert(parse_smethod_line(line, mp) < 0);
- reset_mp(mp);
- /* cowwect */
- strlcpy(line,"SMETHOD trebuchy 127.0.0.1:1999",sizeof(line));
- test_assert(parse_smethod_line(line, mp) == 0);
- reset_mp(mp);
- /* unsupported version */
- strlcpy(line,"VERSION 666",sizeof(line));
- test_assert(parse_version(line, mp) < 0);
- /* incomplete VERSION */
- strlcpy(line,"VERSION ",sizeof(line));
- test_assert(parse_version(line, mp) < 0);
- /* correct VERSION */
- strlcpy(line,"VERSION 1",sizeof(line));
- test_assert(parse_version(line, mp) == 0);
- done:
- tor_free(mp);
- }
- static void
- test_pt_get_transport_options(void *arg)
- {
- char **execve_args;
- smartlist_t *transport_list = smartlist_new();
- managed_proxy_t *mp;
- or_options_t *options = get_options_mutable();
- char *opt_str = NULL;
- config_line_t *cl = NULL;
- (void)arg;
- execve_args = tor_malloc(sizeof(char*)*2);
- execve_args[0] = tor_strdup("cheeseshop");
- execve_args[1] = NULL;
- mp = managed_proxy_create(transport_list, execve_args, 1);
- tt_ptr_op(mp, !=, NULL);
- opt_str = get_transport_options_for_server_proxy(mp);
- tt_ptr_op(opt_str, ==, NULL);
- smartlist_add(mp->transports_to_launch, tor_strdup("gruyere"));
- smartlist_add(mp->transports_to_launch, tor_strdup("roquefort"));
- smartlist_add(mp->transports_to_launch, tor_strdup("stnectaire"));
- tt_assert(options);
- cl = tor_malloc_zero(sizeof(config_line_t));
- cl->value = tor_strdup("gruyere melty=10 hardness=se;ven");
- options->ServerTransportOptions = cl;
- cl = tor_malloc_zero(sizeof(config_line_t));
- cl->value = tor_strdup("stnectaire melty=4 hardness=three");
- cl->next = options->ServerTransportOptions;
- options->ServerTransportOptions = cl;
- cl = tor_malloc_zero(sizeof(config_line_t));
- cl->value = tor_strdup("pepperjack melty=12 hardness=five");
- cl->next = options->ServerTransportOptions;
- options->ServerTransportOptions = cl;
- opt_str = get_transport_options_for_server_proxy(mp);
- tt_str_op(opt_str, ==,
- "gruyere:melty=10;gruyere:hardness=se\\;ven;"
- "stnectaire:melty=4;stnectaire:hardness=three");
- done:
- tor_free(opt_str);
- config_free_lines(cl);
- managed_proxy_destroy(mp, 0);
- smartlist_free(transport_list);
- }
- static void
- test_pt_protocol(void)
- {
- char line[200];
- managed_proxy_t *mp = tor_malloc_zero(sizeof(managed_proxy_t));
- mp->conf_state = PT_PROTO_LAUNCHED;
- mp->transports = smartlist_new();
- mp->argv = tor_malloc_zero(sizeof(char*)*2);
- mp->argv[0] = tor_strdup("<testcase>");
- /* various wrong protocol runs: */
- strlcpy(line,"VERSION 1",sizeof(line));
- handle_proxy_line(line, mp);
- test_assert(mp->conf_state == PT_PROTO_ACCEPTING_METHODS);
- strlcpy(line,"VERSION 1",sizeof(line));
- handle_proxy_line(line, mp);
- test_assert(mp->conf_state == PT_PROTO_BROKEN);
- reset_mp(mp);
- strlcpy(line,"CMETHOD trebuchet socks5 127.0.0.1:1999",sizeof(line));
- handle_proxy_line(line, mp);
- test_assert(mp->conf_state == PT_PROTO_BROKEN);
- reset_mp(mp);
- /* correct protocol run: */
- strlcpy(line,"VERSION 1",sizeof(line));
- handle_proxy_line(line, mp);
- test_assert(mp->conf_state == PT_PROTO_ACCEPTING_METHODS);
- strlcpy(line,"CMETHOD trebuchet socks5 127.0.0.1:1999",sizeof(line));
- handle_proxy_line(line, mp);
- test_assert(mp->conf_state == PT_PROTO_ACCEPTING_METHODS);
- strlcpy(line,"CMETHODS DONE",sizeof(line));
- 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),
- { "get_transport_options", test_pt_get_transport_options, TT_FORK,
- NULL, NULL },
- END_OF_TESTCASES
- };
|