- tor-commits - lists.torproject.org

[tor/master] Upgrade to the latest version of tinytest
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit 676608f9a69d7dd9c70a581109fa4eecc53a5d9a Author: Nick Mathewson <nickm(a)torproject.org> Date: Thu Jun 9 17:30:09 2011 -0400 Upgrade to the latest version of tinytest --- src/test/test.h | 3 ++- src/test/test_addr.c | 4 +++- src/test/test_dir.c | 2 +- src/test/tinytest.c | 17 ++++++++++------- src/test/tinytest_demo.c | 14 +++++++------- src/test/tinytest_macros.h | 39 ++++++++++++++++++++++++++++----------- 6 files changed, 51 insertions(+), 28 deletions(-) diff --git a/src/test/test.h b/src/test/test.h index f7ae46c..a053a7a 100644 --- a/src/test/test.h +++ b/src/test/test.h @@ -45,7 +45,8 @@ _print = tor_malloc(printlen); \ base16_encode(_print, printlen, _value, \ (len)); }, \ - { tor_free(_print); } \ + { tor_free(_print); }, \ + TT_EXIT_TEST_FUNCTION \ ); #define test_memeq(expr1, expr2, len) test_mem_op((expr1), ==, (expr2), len) diff --git a/src/test/test_addr.c b/src/test/test_addr.c index 6db4ee2..7629d95 100644 --- a/src/test/test_addr.c +++ b/src/test/test_addr.c @@ -74,7 +74,9 @@ test_addr_basic(void) cp += 2; \ if (i != 15) *cp++ = ':'; \ } \ - }, { tor_free(_print); } \ + }, \ + { tor_free(_print); }, \ + TT_EXIT_TEST_FUNCTION \ ); \ STMT_END diff --git a/src/test/test_dir.c b/src/test/test_dir.c index 4bbf768..04ca29a 100644 --- a/src/test/test_dir.c +++ b/src/test/test_dir.c @@ -298,7 +298,7 @@ test_dir_versions(void) #define tt_versionstatus_op(vs1, op, vs2) \ tt_assert_test_type(vs1,vs2,#vs1" "#op" "#vs2,version_status_t, \ - (_val1 op _val2),"%d") + (_val1 op _val2),"%d",TT_EXIT_TEST_FUNCTION) #define test_v_i_o(val, ver, lst) \ tt_versionstatus_op(val, ==, tor_version_is_obsolete(ver, lst)) diff --git a/src/test/tinytest.c b/src/test/tinytest.c index 11ffc2f..8caa4f5 100644 --- a/src/test/tinytest.c +++ b/src/test/tinytest.c @@ -28,7 +28,11 @@ #include <string.h> #include <assert.h> -#ifdef WIN32 +#ifdef TINYTEST_LOCAL +#include "tinytest_local.h" +#endif + +#ifdef _WIN32 #include <windows.h> #else #include <sys/types.h> @@ -40,9 +44,6 @@ #define __attribute__(x) #endif -#ifdef TINYTEST_LOCAL -#include "tinytest_local.h" -#endif #include "tinytest.h" #include "tinytest_macros.h" @@ -64,7 +65,7 @@ const char *cur_test_prefix = NULL; /**< prefix of the current test group */ /** Name of the current test, if we haven't logged is yet. Used for --quiet */ const char *cur_test_name = NULL; -#ifdef WIN32 +#ifdef _WIN32 /** Pointer to argv[0] for win32. */ static const char *commandname = NULL; #endif @@ -103,7 +104,7 @@ static enum outcome _testcase_run_forked(const struct testgroup_t *group, const struct testcase_t *testcase) { -#ifdef WIN32 +#ifdef _WIN32 /* Fork? On Win32? How primitive! We'll do what the smart kids do: we'll invoke our own exe (whose name we recall from the command line) with a command line that tells it to run just the test we @@ -174,6 +175,7 @@ _testcase_run_forked(const struct testgroup_t *group, exit(1); } exit(0); + return FAIL; /* unreachable */ } else { /* parent */ int status, r; @@ -239,6 +241,7 @@ testcase_run_one(const struct testgroup_t *group, if (opt_forked) { exit(outcome==OK ? 0 : (outcome==SKIP?MAGIC_EXITCODE : 1)); + return 1; /* unreachable */ } else { return (int)outcome; } @@ -287,7 +290,7 @@ tinytest_main(int c, const char **v, struct testgroup_t *groups) { int i, j, n=0; -#ifdef WIN32 +#ifdef _WIN32 commandname = v[0]; #endif for (i=1; i<c; ++i) { diff --git a/src/test/tinytest_demo.c b/src/test/tinytest_demo.c index bd33cc3..8838459 100644 --- a/src/test/tinytest_demo.c +++ b/src/test/tinytest_demo.c @@ -1,4 +1,4 @@ -/* tinytest_demo.c -- Copyright 2009 Nick Mathewson +/* tinytest_demo.c -- Copyright 2009-2010 Nick Mathewson * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions @@ -53,7 +53,7 @@ test_strcmp(void *data) } /* Pretty often, calling tt_abort_msg to indicate failure is more - heavy-weight than you want. Instead, just say: */ + heavy-weight than you want. Instead, just say: */ tt_assert(strcmp("testcase", "testcase") == 0); /* Occasionally, you don't want to stop the current testcase just @@ -91,7 +91,7 @@ test_strcmp(void *data) /* First you declare a type to hold the environment info, and functions to set it up and tear it down. */ struct data_buffer { - /* We're just going to have couple of character buffer. Using + /* We're just going to have couple of character buffer. Using setup/teardown functions is probably overkill for this case. You could also do file descriptors, complicated handles, temporary @@ -164,7 +164,7 @@ test_memcpy(void *ptr) /* ============================================================ */ -/* Now we need to make sure that our tests get invoked. First, you take +/* Now we need to make sure that our tests get invoked. First, you take a bunch of related tests and put them into an array of struct testcase_t. */ @@ -189,15 +189,15 @@ struct testgroup_t groups[] = { /* Every group has a 'prefix', and an array of tests. That's it. */ { "demo/", demo_tests }, - END_OF_GROUPS + END_OF_GROUPS }; int main(int c, const char **v) { - /* Finally, just call tinytest_main(). It lets you specify verbose - or quiet output with --verbose and --quiet. You can list + /* Finally, just call tinytest_main(). It lets you specify verbose + or quiet output with --verbose and --quiet. You can list specific tests: tinytest-demo demo/memcpy diff --git a/src/test/tinytest_macros.h b/src/test/tinytest_macros.h index a7fa64a..032393c 100644 --- a/src/test/tinytest_macros.h +++ b/src/test/tinytest_macros.h @@ -90,10 +90,10 @@ TT_STMT_BEGIN \ if (!(b)) { \ _tinytest_set_test_failed(); \ - TT_GRIPE((msg)); \ + TT_GRIPE(("%s",msg)); \ fail; \ } else { \ - TT_BLATHER((msg)); \ + TT_BLATHER(("%s",msg)); \ } \ TT_STMT_END @@ -111,7 +111,7 @@ #define tt_assert(b) tt_assert_msg((b), "assert("#b")") #define tt_assert_test_fmt_type(a,b,str_test,type,test,printf_type,printf_fmt, \ - setup_block,cleanup_block) \ + setup_block,cleanup_block,die_on_fail) \ TT_STMT_BEGIN \ type _val1 = (type)(a); \ type _val2 = (type)(b); \ @@ -135,33 +135,50 @@ cleanup_block; \ if (!_tt_status) { \ _tinytest_set_test_failed(); \ - TT_EXIT_TEST_FUNCTION; \ + die_on_fail ; \ } \ } \ TT_STMT_END -#define tt_assert_test_type(a,b,str_test,type,test,fmt) \ +#define tt_assert_test_type(a,b,str_test,type,test,fmt,die_on_fail) \ tt_assert_test_fmt_type(a,b,str_test,type,test,type,fmt, \ - {_print=_value;},{}) + {_print=_value;},{},die_on_fail) /* Helper: assert that a op b, when cast to type. Format the values with * printf format fmt on failure. */ #define tt_assert_op_type(a,op,b,type,fmt) \ - tt_assert_test_type(a,b,#a" "#op" "#b,type,(_val1 op _val2),fmt) + tt_assert_test_type(a,b,#a" "#op" "#b,type,(_val1 op _val2),fmt, \ + TT_EXIT_TEST_FUNCTION) #define tt_int_op(a,op,b) \ - tt_assert_test_type(a,b,#a" "#op" "#b,long,(_val1 op _val2),"%ld") + tt_assert_test_type(a,b,#a" "#op" "#b,long,(_val1 op _val2), \ + "%ld",TT_EXIT_TEST_FUNCTION) #define tt_uint_op(a,op,b) \ tt_assert_test_type(a,b,#a" "#op" "#b,unsigned long, \ - (_val1 op _val2),"%lu") + (_val1 op _val2),"%lu",TT_EXIT_TEST_FUNCTION) #define tt_ptr_op(a,op,b) \ tt_assert_test_type(a,b,#a" "#op" "#b,void*, \ - (_val1 op _val2),"%p") + (_val1 op _val2),"%p",TT_EXIT_TEST_FUNCTION) #define tt_str_op(a,op,b) \ tt_assert_test_type(a,b,#a" "#op" "#b,const char *, \ - (strcmp(_val1,_val2) op 0),"<%s>") + (strcmp(_val1,_val2) op 0),"<%s>",TT_EXIT_TEST_FUNCTION) + +#define tt_want_int_op(a,op,b) \ + tt_assert_test_type(a,b,#a" "#op" "#b,long,(_val1 op _val2),"%ld",(void)0) + +#define tt_want_uint_op(a,op,b) \ + tt_assert_test_type(a,b,#a" "#op" "#b,unsigned long, \ + (_val1 op _val2),"%lu",(void)0) + +#define tt_want_ptr_op(a,op,b) \ + tt_assert_test_type(a,b,#a" "#op" "#b,void*, \ + (_val1 op _val2),"%p",(void)0) + +#define tt_want_str_op(a,op,b) \ + tt_assert_test_type(a,b,#a" "#op" "#b,const char *, \ + (strcmp(_val1,_val2) op 0),"<%s>",(void)0) #endif

1 0

[obfsproxy/master] Added unit tests for parse_and_set_options() of obfs2.
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit b03e4e200a2f990829152bfce5684f3c5d84b935 Author: George Kadianakis <desnacked(a)gmail.com> Date: Thu Jun 9 19:36:30 2011 +0200 Added unit tests for parse_and_set_options() of obfs2. --- src/protocols/obfs2.c | 7 +-- src/protocols/obfs2.h | 3 + src/test/unittest_obfs2.c | 109 ++++++++++++++++++++++++++++++++++++++++++-- 3 files changed, 109 insertions(+), 10 deletions(-) diff --git a/src/protocols/obfs2.c b/src/protocols/obfs2.c index 2cb1089..edfc365 100644 --- a/src/protocols/obfs2.c +++ b/src/protocols/obfs2.c @@ -34,9 +34,6 @@ static int obfs2_state_set_shared_secret(void *s, size_t secretlen); static int set_up_vtable(void); static void usage(void); -static int parse_and_set_options(int n_options, char **options, - struct protocol_params_t *params); - static protocol_vtable *vtable=NULL; @@ -67,7 +64,7 @@ obfs2_init(int n_options, char **options, return 1; } -static int +int parse_and_set_options(int n_options, char **options, struct protocol_params_t *params) { @@ -94,7 +91,7 @@ parse_and_set_options(int n_options, char **options, struct sockaddr_storage ss_target; struct sockaddr *sa_target=NULL; int sl_target=0; - if (resolve_address_port(*options+7, 1, 0, + if (resolve_address_port(*options+7, 1, 0, &ss_target, &sl_target, NULL) < 0) return -1; assert(sl_target <= sizeof(struct sockaddr_storage)); diff --git a/src/protocols/obfs2.h b/src/protocols/obfs2.h index 7747330..d206bc8 100644 --- a/src/protocols/obfs2.h +++ b/src/protocols/obfs2.h @@ -21,6 +21,9 @@ struct listener_t; int obfs2_init(int n_options, char **options, struct protocol_params_t *params); void *obfs2_new(struct protocol_t *proto_struct, struct protocol_params_t *params); +int parse_and_set_options(int n_options, char **options, + struct protocol_params_t *params); + #ifdef CRYPT_PROTOCOL_PRIVATE diff --git a/src/test/unittest_obfs2.c b/src/test/unittest_obfs2.c index 45d1278..1348c7d 100644 --- a/src/test/unittest_obfs2.c +++ b/src/test/unittest_obfs2.c @@ -23,6 +23,84 @@ #include "../protocols/obfs2.h" #include "../network.h" +static void +test_proto_option_parsing(void *data) +{ + protocol_params_t *proto_params = calloc(1, sizeof(protocol_params_t)); + char *options[] = {"obfs2", "--shared-secret=a", "socks", "127.0.0.1:0"}; + int n_options = 4; + + tt_assert(set_up_protocol(n_options, options, + proto_params) == 1); + + /** two --dest. */ + char *options2[] = {"obfs2", "--dest=127.0.0.1:5555", "--dest=a", + "server", "127.0.0.1:5552"}; + n_options = 5; + + tt_assert(set_up_protocol(n_options, options2, + proto_params) == -1); + + /** unknown arg */ + char *options3[] = {"obfs2", "--gabura=a", + "server", "127.0.0.1:5552"}; + n_options = 4; + + tt_assert(set_up_protocol(n_options, options3, + proto_params) == -1); + + /** too many args */ + char *options4[] = {"obfs2", "1", "2", "3", "4", "5" }; + n_options = 6; + + tt_assert(set_up_protocol(n_options, options4, + proto_params) == -1); + + /** wrong mode */ + char *options5[] = {"obfs2", "--dest=1:1", + "gladiator", "127.0.0.1:5552"}; + n_options = 4; + + tt_assert(set_up_protocol(n_options, options5, + proto_params) == -1); + + /** stupid listen addr. */ + char *options6[] = {"obfs2", "--dest=1:1", + "server", "127.0.0.1:a"}; + n_options = 4; + + tt_assert(set_up_protocol(n_options, options6, + proto_params) == -1); + + /** stupid dest addr. */ + char *options7[] = {"obfs2", "--dest=1:b", + "server", "127.0.0.1:1"}; + n_options = 4; + + tt_assert(set_up_protocol(n_options, options7, + proto_params) == -1); + + /** socks with dest. */ + char *options8[] = {"obfs2", "--dest=1:2", + "socks", "127.0.0.1:1"}; + n_options = 4; + + tt_assert(set_up_protocol(n_options, options8, + proto_params) == -1); + + /** socks with dest. */ + char *options9[] = {"obfs2", "--shared-secret=a", + "server", "127.0.0.1:1"}; + n_options = 4; + + tt_assert(set_up_protocol(n_options, options9, + proto_params) == -1); + + end: + if (proto_params) + free(proto_params); +} + /* Make sure we can successfully set up a protocol state */ static void test_proto_setup(void *data) @@ -57,7 +135,6 @@ test_proto_setup(void *data) proto_destroy(client_proto); if (server_proto->state) proto_destroy(server_proto); - } static void @@ -126,6 +203,11 @@ test_proto_handshake(void *data) if (server_proto->state) proto_destroy(server_proto); + if (proto_params_client) + free(proto_params_client); + if (proto_params_serv) + free(proto_params_serv); + if (output_buffer) evbuffer_free(output_buffer); if (dummy_buffer) @@ -215,6 +297,11 @@ test_proto_transfer(void *data) if (server_proto->state) proto_destroy(server_proto); + if (proto_params_client) + free(proto_params_client); + if (proto_params_serv) + free(proto_params_serv); + if (output_buffer) evbuffer_free(output_buffer); if (dummy_buffer) @@ -378,6 +465,11 @@ test_proto_splitted_handshake(void *data) if (server_state) proto_destroy(server_proto); + if (proto_params_client) + free(proto_params_client); + if (proto_params_serv) + free(proto_params_serv); + if (output_buffer) evbuffer_free(output_buffer); if (dummy_buffer) @@ -459,6 +551,11 @@ test_proto_wrong_handshake_magic(void *data) if (server_state) proto_destroy(server_proto); + if (proto_params_client) + free(proto_params_client); + if (proto_params_serv) + free(proto_params_serv); + if (output_buffer) evbuffer_free(output_buffer); if (dummy_buffer) @@ -539,25 +636,27 @@ test_proto_wrong_handshake_plength(void *data) if (server_state) proto_destroy(server_proto); + if (proto_params_client) + free(proto_params_client); + if (proto_params_serv) + free(proto_params_serv); + if (output_buffer) evbuffer_free(output_buffer); if (dummy_buffer) evbuffer_free(dummy_buffer); } - #define T(name, flags) \ { #name, test_proto_##name, (flags), NULL, NULL } struct testcase_t protocol_tests[] = { + T(option_parsing,0), T(setup, 0), T(handshake, 0), T(transfer, 0), T(splitted_handshake, 0), T(wrong_handshake_magic, 0), -#if 0 - T(wrong_handshake_padding, 0), -#endif T(wrong_handshake_plength, 0), END_OF_TESTCASES };

1 0

[obfsproxy/master] Detached option parsing from {dummy, obfs2}_init().
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit fb76953a54809479a9527213623526d1fdf0b763 Author: George Kadianakis <desnacked(a)gmail.com> Date: Thu Jun 9 19:13:33 2011 +0200 Detached option parsing from {dummy,obfs2}_init(). It's cleaner and it's easier to test. --- src/protocols/dummy.c | 47 +++++++++++++++++++++++++++------------------ src/protocols/obfs2.c | 50 +++++++++++++++++++++++++++++------------------- 2 files changed, 58 insertions(+), 39 deletions(-) diff --git a/src/protocols/dummy.c b/src/protocols/dummy.c index 5f6b8eb..71b3313 100644 --- a/src/protocols/dummy.c +++ b/src/protocols/dummy.c @@ -19,9 +19,10 @@ static int dummy_send(void *nothing, static enum recv_ret dummy_recv(void *nothing, struct evbuffer *source, struct evbuffer *dest); static void usage(void); +static int parse_and_set_options(int n_options, char **options, + struct protocol_params_t *params); static protocol_vtable *vtable=NULL; - /** This function sets up the protocol and populates 'listner' according to 'options'. @@ -33,12 +34,35 @@ int dummy_init(int n_options, char **options, struct protocol_params_t *params) { + if (parse_and_set_options(n_options,options,params) < 0) { + usage(); + return -1; + } + + /* XXX memleak. */ + vtable = calloc(1, sizeof(protocol_vtable)); + if (!vtable) + return -1; + + vtable->destroy = NULL; + vtable->create = dummy_new; + vtable->handshake = NULL; + vtable->send = dummy_send; + vtable->recv = dummy_recv; + + return 1; +} + +static int +parse_and_set_options(int n_options, char **options, + struct protocol_params_t *params) +{ struct sockaddr_storage ss_listen; int sl_listen; const char* defport; if (n_options != 3) - goto err; + return -1; assert(!strcmp(options[0],"dummy")); params->proto = DUMMY_PROTOCOL; @@ -53,12 +77,12 @@ dummy_init(int n_options, char **options, defport = "11253"; /* 2bf5 */ params->mode = LSN_SIMPLE_SERVER; } else - goto err; + return -1; if (resolve_address_port(options[2], 1, 1, &ss_listen, &sl_listen, defport) < 0) { printf("addr\n"); - goto err; + return -1; } assert(sl_listen <= sizeof(struct sockaddr_storage)); struct sockaddr *sa_listen=NULL; @@ -66,22 +90,7 @@ dummy_init(int n_options, char **options, memcpy(&params->on_address, sa_listen, sl_listen); params->on_address_len = sl_listen; - /* XXX memleak. */ - vtable = calloc(1, sizeof(protocol_vtable)); - if (!vtable) - return -1; - - vtable->destroy = NULL; - vtable->create = dummy_new; - vtable->handshake = NULL; - vtable->send = dummy_send; - vtable->recv = dummy_recv; - return 1; - - err: - usage(); - return -1; } static void diff --git a/src/protocols/obfs2.c b/src/protocols/obfs2.c index d886594..2cb1089 100644 --- a/src/protocols/obfs2.c +++ b/src/protocols/obfs2.c @@ -34,6 +34,8 @@ static int obfs2_state_set_shared_secret(void *s, size_t secretlen); static int set_up_vtable(void); static void usage(void); +static int parse_and_set_options(int n_options, char **options, + struct protocol_params_t *params); static protocol_vtable *vtable=NULL; @@ -49,6 +51,26 @@ int obfs2_init(int n_options, char **options, struct protocol_params_t *params) { + if (parse_and_set_options(n_options,options,params) < 0) { + usage(); + return -1; + } + + if (set_up_vtable() < 0) + return -1; + + if (initialize_crypto() < 0) { + fprintf(stderr, "Can't initialize crypto; failing\n"); + return -1; + } + + return 1; +} + +static int +parse_and_set_options(int n_options, char **options, + struct protocol_params_t *params) +{ struct sockaddr_storage ss_listen; int sl_listen; int got_dest=0; @@ -57,7 +79,7 @@ obfs2_init(int n_options, char **options, if ((n_options < 3) || (n_options > 5)) { printf("wrong options number: %d\n", n_options); - goto err; + return -1; } assert(!strcmp(*options,"obfs2")); @@ -68,13 +90,13 @@ obfs2_init(int n_options, char **options, while (!strncmp(*options,"--",2)) { if (!strncmp(*options,"--dest=",7)) { if (got_dest) - goto err; + return -1; struct sockaddr_storage ss_target; struct sockaddr *sa_target=NULL; int sl_target=0; if (resolve_address_port(*options+7, 1, 0, &ss_target, &sl_target, NULL) < 0) - goto err; + return -1; assert(sl_target <= sizeof(struct sockaddr_storage)); sa_target = (struct sockaddr *)&ss_target; memcpy(&params->target_address, sa_target, sl_target); @@ -82,14 +104,14 @@ obfs2_init(int n_options, char **options, got_dest=1; } else if (!strncmp(*options,"--shared-secret=",16)) { if (got_ss) - goto err; + return -1; /* this is freed in protocol_params_free() */ params->shared_secret = strdup(*options+16); params->shared_secret_len = strlen(*options+16); got_ss=1; } else { printf("Unknown argument.\n"); - goto err; + return -1; } options++; } @@ -105,7 +127,7 @@ obfs2_init(int n_options, char **options, params->mode = LSN_SIMPLE_SERVER; } else { printf("only client/socks/server modes supported.\n"); - goto err; + return -1; } options++; @@ -113,7 +135,7 @@ obfs2_init(int n_options, char **options, if (resolve_address_port(*options, 1, 1, &ss_listen, &sl_listen, defport) < 0) - goto err; + return -1; assert(sl_listen <= sizeof(struct sockaddr_storage)); struct sockaddr *sa_listen=NULL; sa_listen = (struct sockaddr *)&ss_listen; @@ -123,27 +145,15 @@ obfs2_init(int n_options, char **options, /* Validate option selection. */ if (got_dest && (params->mode == LSN_SOCKS_CLIENT)) { printf("You can't be on socks mode and have --dest.\n"); - goto err; + return -1; } if (!got_dest && (params->mode != LSN_SOCKS_CLIENT)) { printf("client/server mode needs --dest.\n"); - goto err; - } - - if (set_up_vtable() < 0) - return -1; - - if (initialize_crypto() < 0) { - fprintf(stderr, "Can't initialize crypto; failing\n"); return -1; } return 1; - - err: - usage(); - return -1; } /**

1 0

[obfsproxy/master] Documentation updates.
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit 835464e0d6bd025faa74ff429610e970bec25933 Author: George Kadianakis <desnacked(a)gmail.com> Date: Thu Jun 9 17:27:56 2011 +0200 Documentation updates. * Tweaked tor-obfs-howto.txt for the new cli. * Fixed a comment in main.c --- doc/tor-obfs-howto.txt | 6 +++--- src/main.c | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/doc/tor-obfs-howto.txt b/doc/tor-obfs-howto.txt index 36ad7a3..ed8ba72 100644 --- a/doc/tor-obfs-howto.txt +++ b/doc/tor-obfs-howto.txt @@ -50,11 +50,11 @@ Our port setup will be like this: This command will setup an obfsproxy socks client listening to the obfs2 protocol on localhost:1050: -./obfsproxy socks obfs2 127.0.0.1:1050 +./obfsproxy obfs2 socks 127.0.0.1:1050 This command will setup an obfsproxy server listening to the -obfs2 protocol on localhost:5001: -./obfsproxy server obfs2 127.0.0.1:1051 127.0.0.1:5001 +obfs2 protocol on localhost:1051: +./obfsproxy obfs2 --dest 127.0.0.1:5001 server 127.0.0.1:1051 Setting up Tor: diff --git a/src/main.c b/src/main.c index e7680d5..8617bb3 100644 --- a/src/main.c +++ b/src/main.c @@ -112,8 +112,8 @@ main(int argc, const char **argv) /* This is an integer! It contains the number of protocols that we managed to recognize, by their protocol name. Of course it's not the *actual* actual_protocols since some of them could have wrong - options or arguments, but this will be resolved by - set_up_protocol() and listener_new(). */ + options or arguments, but this will be resolved per-protocol by + set_up_protocol(). */ int actual_protocols=0; int start;

1 0

[obfsproxy/master] Merge branch 'bug3204' into bug3204_actual_merge
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit 1e750f25009721ff183483fad831b79ee9de8ed1 Merge: f949fac afdc2d8 Author: George Kadianakis <desnacked(a)gmail.com> Date: Thu Jun 9 15:44:12 2011 +0200 Merge branch 'bug3204' into bug3204_actual_merge Conflicts: src/main.c src/protocol.c src/protocol.h src/protocols/dummy.c src/socks.h src/test/unittest_obfs2.c src/main.c | 300 +++++++++++++++++++++++++++++++++------------ src/network.c | 75 +++-------- src/network.h | 15 ++- src/protocol.c | 88 +++++++++---- src/protocol.h | 45 ++++--- src/protocols/dummy.c | 70 ++++++++++- src/protocols/dummy.h | 5 +- src/protocols/obfs2.c | 175 ++++++++++++++++++++++++-- src/protocols/obfs2.h | 5 +- src/test/unittest_obfs2.c | 158 ++++++++++++------------ src/test/unittest_socks.c | 2 +- 11 files changed, 654 insertions(+), 284 deletions(-) diff --cc src/protocol.c index bcd5330,57c0c23..32d8d0d --- a/src/protocol.c +++ b/src/protocol.c @@@ -72,7 -80,10 +80,10 @@@ proto_send(struct protocol_t *proto, vo return -1; } + /** + This function is responsible for receiving protocol data. + */ -int +enum recv_ret proto_recv(struct protocol_t *proto, void *source, void *dest) { assert(proto); if (proto->vtable->recv) diff --cc src/protocol.h index 4a17c73,0a7c452..69ea4aa --- a/src/protocol.h +++ b/src/protocol.h @@@ -39,6 -44,15 +44,15 @@@ struct protocol_t */ void *state; }; + int set_up_protocol(int n_options, char **options, + struct protocol_params_t *params); + struct protocol_t *proto_new(struct protocol_params_t *params); + void proto_destroy(struct protocol_t *proto); + int proto_handshake(struct protocol_t *proto, void *buf); + int proto_send(struct protocol_t *proto, void *source, void *dest); -int proto_recv(struct protocol_t *proto, void *source, void *dest); ++enum recv_ret proto_recv(struct protocol_t *proto, void *source, void *dest); + + void proto_params_free(protocol_params_t *params); typedef struct protocol_vtable { diff --cc src/protocols/dummy.c index 68bd7a1,2a24b56..5f6b8eb --- a/src/protocols/dummy.c +++ b/src/protocols/dummy.c @@@ -9,14 -9,16 +9,16 @@@ #include <event2/buffer.h> #include "dummy.h" + #include "../network.h" #include "../util.h" #include "../protocol.h" - +#include "../network.h" static int dummy_send(void *nothing, - struct evbuffer *source, struct evbuffer *dest); -static int dummy_recv(void *nothing, struct evbuffer *source, - struct evbuffer *dest); + struct evbuffer *source, struct evbuffer *dest); +static enum recv_ret dummy_recv(void *nothing, struct evbuffer *source, + struct evbuffer *dest); + static void usage(void); static protocol_vtable *vtable=NULL; diff --cc src/protocols/obfs2.c index c93db0e,5e91b5c..d886594 --- a/src/protocols/obfs2.c +++ b/src/protocols/obfs2.c @@@ -17,9 -17,9 +17,10 @@@ #include "obfs2_crypt.h" #include "obfs2.h" + #include "../network.h" #include "../util.h" #include "../protocol.h" +#include "../network.h" static void obfs2_state_free(void *state); static int obfs2_send_initial_message(void *state, struct evbuffer *buf); @@@ -441,14 -575,12 +585,17 @@@ obfs2_recv(void *s, struct evbuffer *so dbg(("Processing %d bytes data onto destination buffer\n", (int) evbuffer_get_length(source))); - return crypt_and_transmit(state->recv_crypto, source, dest); + crypt_and_transmit(state->recv_crypto, source, dest); + + if (r != RECV_SEND_PENDING) + r = RECV_GOOD; + + return r; } + /** + Frees obfs2 state 's' + */ static void obfs2_state_free(void *s) {

1 0

[obfsproxy/master] Revives the unittests.
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit 869aad578c26365a56890cc2ec5afb765a6048ca Author: George Kadianakis <desnacked(a)gmail.com> Date: Sun May 29 04:05:48 2011 +0200 Revives the unittests. --- src/test/unittest_obfs2.c | 158 ++++++++++++++++++++++---------------------- src/test/unittest_socks.c | 2 +- 2 files changed, 80 insertions(+), 80 deletions(-) diff --git a/src/test/unittest_obfs2.c b/src/test/unittest_obfs2.c index 052c909..c411cbe 100644 --- a/src/test/unittest_obfs2.c +++ b/src/test/unittest_obfs2.c @@ -29,22 +29,22 @@ test_proto_setup(void *data) struct protocol_t *client_proto = NULL; struct protocol_t *server_proto = NULL; - tt_assert(set_up_protocol(OBFS2_PROTOCOL) >= 0); + protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); + char *options_client[] = {"obfs2", "--shared-secret=hahaha", "socks", "127.0.0.1:1800"}; + int n_options_client = 4; protocol_params_t *proto_params_serv = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_serv); - proto_params_serv->is_initiator = 1; - proto_params_serv->shared_secret = "hahaha"; - proto_params_serv->shared_secret_len = 7; - - protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_client); - proto_params_client->is_initiator = 0; - proto_params_client->shared_secret = "hahaha"; - proto_params_client->shared_secret_len = 7; + char *options_server[] = {"obfs2", "--shared-secret=hahaha", "--dest=127.0.0.1:1500", + "server", "127.0.0.1:1800"}; + int n_options_server = 5; + + tt_assert(set_up_protocol(n_options_client, options_client, + proto_params_client) >= 0); + tt_assert(set_up_protocol(n_options_server, options_server, + proto_params_serv) >= 0); - client_proto = proto_new(OBFS2_PROTOCOL,proto_params_serv); - server_proto = proto_new(OBFS2_PROTOCOL,proto_params_client); + client_proto = proto_new(proto_params_serv); + server_proto = proto_new(proto_params_client); tt_assert(client_proto); tt_assert(server_proto); @@ -70,22 +70,22 @@ test_proto_handshake(void *data) struct protocol_t *client_proto = NULL; struct protocol_t *server_proto = NULL; - tt_assert(set_up_protocol(OBFS2_PROTOCOL) >= 0); + protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); + char *options_client[] = {"obfs2", "--shared-secret=hahaha", "socks", "127.0.0.1:1800"}; + int n_options_client = 4; protocol_params_t *proto_params_serv = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_serv); - proto_params_serv->is_initiator = 1; - proto_params_serv->shared_secret = "hahaha"; - proto_params_serv->shared_secret_len = 7; - - protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_client); - proto_params_client->is_initiator = 0; - proto_params_client->shared_secret = "hahaha"; - proto_params_client->shared_secret_len = 7; + char *options_server[] = {"obfs2", "--shared-secret=hahaha", "--dest=127.0.0.1:1500", + "server", "127.0.0.1:1800"}; + int n_options_server = 5; - client_proto = proto_new(OBFS2_PROTOCOL,proto_params_serv); - server_proto = proto_new(OBFS2_PROTOCOL,proto_params_client); + tt_assert(set_up_protocol(n_options_client, options_client, + proto_params_client) >= 0); + tt_assert(set_up_protocol(n_options_server, options_server, + proto_params_serv) >= 0); + + client_proto = proto_new(proto_params_client); + server_proto = proto_new(proto_params_serv); tt_assert(client_proto); tt_assert(server_proto); @@ -142,22 +142,22 @@ test_proto_transfer(void *data) struct protocol_t *client_proto = NULL; struct protocol_t *server_proto = NULL; - tt_assert(set_up_protocol(OBFS2_PROTOCOL) >= 0); + protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); + char *options_client[] = {"obfs2", "--shared-secret=hahaha", "socks", "127.0.0.1:1800"}; + int n_options_client = 4; protocol_params_t *proto_params_serv = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_serv); - proto_params_serv->is_initiator = 1; - proto_params_serv->shared_secret = "hahaha"; - proto_params_serv->shared_secret_len = 7; - - protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_client); - proto_params_client->is_initiator = 0; - proto_params_client->shared_secret = "hahaha"; - proto_params_client->shared_secret_len = 7; + char *options_server[] = {"obfs2", "--shared-secret=hahaha", "--dest=127.0.0.1:1500", + "server", "127.0.0.1:1800"}; + int n_options_server = 5; + + tt_assert(set_up_protocol(n_options_client, options_client, + proto_params_client) >= 0); + tt_assert(set_up_protocol(n_options_server, options_server, + proto_params_serv) >= 0); - client_proto = proto_new(OBFS2_PROTOCOL,proto_params_serv); - server_proto = proto_new(OBFS2_PROTOCOL,proto_params_client); + client_proto = proto_new(proto_params_client); + server_proto = proto_new(proto_params_serv); tt_assert(client_proto); tt_assert(server_proto); @@ -238,22 +238,22 @@ test_proto_splitted_handshake(void *data) struct protocol_t *client_proto = NULL; struct protocol_t *server_proto = NULL; - tt_assert(set_up_protocol(OBFS2_PROTOCOL) >= 0); + protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); + char *options_client[] = {"obfs2", "--shared-secret=hahaha", "socks", "127.0.0.1:1800"}; + int n_options_client = 4; protocol_params_t *proto_params_serv = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_serv); - proto_params_serv->is_initiator = 1; - proto_params_serv->shared_secret = "hahaha"; - proto_params_serv->shared_secret_len = 7; - - protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_client); - proto_params_client->is_initiator = 0; - proto_params_client->shared_secret = "hahaha"; - proto_params_client->shared_secret_len = 7; - - client_proto = proto_new(OBFS2_PROTOCOL,proto_params_serv); - server_proto = proto_new(OBFS2_PROTOCOL,proto_params_client); + char *options_server[] = {"obfs2", "--shared-secret=hahaha", "--dest=127.0.0.1:1500", + "server", "127.0.0.1:1800"}; + int n_options_server = 5; + + tt_assert(set_up_protocol(n_options_client, options_client, + proto_params_client) >= 0); + tt_assert(set_up_protocol(n_options_server, options_server, + proto_params_serv) >= 0); + + client_proto = proto_new(proto_params_client); + server_proto = proto_new(proto_params_serv); tt_assert(client_proto); tt_assert(server_proto); @@ -391,22 +391,22 @@ test_proto_wrong_handshake_magic(void *data) struct protocol_t *client_proto = NULL; struct protocol_t *server_proto = NULL; - tt_assert(set_up_protocol(OBFS2_PROTOCOL) >= 0); + protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); + char *options_client[] = {"obfs2", "--shared-secret=hahaha", "socks", "127.0.0.1:1800"}; + int n_options_client = 4; protocol_params_t *proto_params_serv = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_serv); - proto_params_serv->is_initiator = 1; - proto_params_serv->shared_secret = "hahaha"; - proto_params_serv->shared_secret_len = 7; - - protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_client); - proto_params_client->is_initiator = 0; - proto_params_client->shared_secret = "hahaha"; - proto_params_client->shared_secret_len = 7; + char *options_server[] = {"obfs2", "--shared-secret=hahaha", "--dest=127.0.0.1:1500", + "server", "127.0.0.1:1800"}; + int n_options_server = 5; + + tt_assert(set_up_protocol(n_options_client, options_client, + proto_params_client) >= 0); + tt_assert(set_up_protocol(n_options_server, options_server, + proto_params_serv) >= 0); - client_proto = proto_new(OBFS2_PROTOCOL,proto_params_serv); - server_proto = proto_new(OBFS2_PROTOCOL,proto_params_client); + client_proto = proto_new(proto_params_client); + server_proto = proto_new(proto_params_serv); tt_assert(client_proto); tt_assert(server_proto); @@ -470,23 +470,23 @@ test_proto_wrong_handshake_plength(void *data) struct protocol_t *client_proto = NULL; struct protocol_t *server_proto = NULL; - tt_assert(set_up_protocol(OBFS2_PROTOCOL) >= 0); - protocol_params_t *proto_params_serv = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_serv); - proto_params_serv->is_initiator = 1; - proto_params_serv->shared_secret = "hahaha"; - proto_params_serv->shared_secret_len = 7; - protocol_params_t *proto_params_client = calloc(1, sizeof(protocol_params_t)); - tt_assert(proto_params_client); - proto_params_client->is_initiator = 0; - proto_params_client->shared_secret = "hahaha"; - proto_params_client->shared_secret_len = 7; + char *options_client[] = {"obfs2", "--shared-secret=hahaha", "socks", "127.0.0.1:1800"}; + int n_options_client = 4; + + protocol_params_t *proto_params_serv = calloc(1, sizeof(protocol_params_t)); + char *options_server[] = {"obfs2", "--shared-secret=hahaha", "--dest=127.0.0.1:1500", + "server", "127.0.0.1:1800"}; + int n_options_server = 5; + tt_assert(set_up_protocol(n_options_client, options_client, + proto_params_client) >= 0); + tt_assert(set_up_protocol(n_options_server, options_server, + proto_params_serv) >= 0); - client_proto = proto_new(OBFS2_PROTOCOL,proto_params_serv); - server_proto = proto_new(OBFS2_PROTOCOL,proto_params_client); + client_proto = proto_new(proto_params_client); + server_proto = proto_new(proto_params_serv); tt_assert(client_proto); tt_assert(server_proto); diff --git a/src/test/unittest_socks.c b/src/test/unittest_socks.c index d7706ee..086d887 100644 --- a/src/test/unittest_socks.c +++ b/src/test/unittest_socks.c @@ -594,7 +594,7 @@ test_socks_socks4_request_reply(void *data) tt_assert(rep2[1] == SOCKS4_FAILED); /* check port */ - tt_int_op(0, ==, memcmp(rep1+2,"\x1c\xbd",2)); + tt_int_op(0, ==, memcmp(rep2+2,"\x1c\xbd",2)); /* check address */ /* tt_str_op(rep1+2+2, ==, "www.test.example"); */

1 0

[obfsproxy/master] Fixed some memory leaks and did some style changes.
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit 1c94e6df95fd8808bca3722fdcc9eba25d3bb0fb Author: George Kadianakis <desnacked(a)gmail.com> Date: Sun May 29 23:28:53 2011 +0200 Fixed some memory leaks and did some style changes. --- src/main.c | 29 ++++++++++++++++++----------- src/network.c | 12 +----------- src/protocol.c | 26 +++++++++++++++----------- src/protocol.h | 2 ++ src/protocols/obfs2.c | 1 + 5 files changed, 37 insertions(+), 33 deletions(-) diff --git a/src/main.c b/src/main.c index 68ea424..dffe210 100644 --- a/src/main.c +++ b/src/main.c @@ -19,11 +19,13 @@ #define __attribute__(x) #endif -static void usage(void) __attribute__((noreturn)); - +/* The character that seperates multiple listeners in the cli */ #define SEPERATOR "+" +/* Totally arbitrary. */ #define MAXPROTOCOLS 20 +static void usage(void) __attribute__((noreturn)); + /* protocol.c */ extern char *supported_protocols[]; extern int n_supported_protocols; @@ -85,6 +87,8 @@ is_supported_protocol(const char *name) { return 0; } +#define STUPID_BEAUTIFIER "===========================\n" + int main(int argc, const char **argv) { @@ -215,10 +219,11 @@ main(int argc, const char **argv) listener_t *temp_listener; int n_listeners=0; for (h=0;h<actual_protocols;h++) { + if (n_protocols > 1) { - printf("===========================\n" - "Spawning listener %d!\n" - "===========================\n", h+1); + dbg((STUPID_BEAUTIFIER + "Spawning listener %d!\n" + STUPID_BEAUTIFIER, h+1)); } temp_listener = listener_new(base, n_options_array[h], protocol_options[h]); @@ -232,18 +237,18 @@ main(int argc, const char **argv) continue; } - printf("Succesfully created listener.\n"); + dbg(("Succesfully created listener.\n")); listeners[n_listeners] = temp_listener; n_listeners++; } if (n_protocols > 1) { - printf("\n===========================\n" - "From the original %d protocols only %d were parsed from main.c. " - "In the end only %d survived.\n\nStarting up...\n" - "===========================\n", - n_protocols, actual_protocols,n_listeners); + dbg((STUPID_BEAUTIFIER + "From the original %d protocols only %d were parsed from main.c. " + "In the end only %d survived.\n\nStarting up...\n" + STUPID_BEAUTIFIER, + n_protocols, actual_protocols,n_listeners)); } /* run the event loop if at least a listener was created. */ @@ -258,3 +263,5 @@ main(int argc, const char **argv) return 0; } + +#undef STUPID_BEAUTIFIER diff --git a/src/network.c b/src/network.c index 1c4b853..3b3c2bf 100644 --- a/src/network.c +++ b/src/network.c @@ -71,23 +71,13 @@ listener_new(struct event_base *base, return lsn; } -static void -protocol_params_free(protocol_params_t *params) -{ - assert(params); - - if (params->shared_secret) - free(params->shared_secret); - free(params); -} - void listener_free(listener_t *lsn) { if (lsn->listener) evconnlistener_free(lsn->listener); if (lsn->proto_params) - protocol_params_free(lsn->proto_params); + proto_params_free(lsn->proto_params); memset(lsn, 0xb0, sizeof(listener_t)); free(lsn); } diff --git a/src/protocol.c b/src/protocol.c index b86bb5f..1c93150 100644 --- a/src/protocol.c +++ b/src/protocol.c @@ -43,20 +43,12 @@ proto_new(protocol_params_t *params) { if (!proto) return NULL; - proto->vtable = calloc(1, sizeof(struct protocol_vtable)); - if (!proto->vtable) - return NULL; - - if (params->proto == OBFS2_PROTOCOL) { + if (params->proto == OBFS2_PROTOCOL) proto->state = obfs2_new(proto, params); - } else if (params->proto == DUMMY_PROTOCOL) { + else if (params->proto == DUMMY_PROTOCOL) proto->state = dummy_new(proto, NULL); - } - if (proto->state) - return proto; - else - return NULL; + return proto->state ? proto : NULL; } int @@ -93,4 +85,16 @@ proto_destroy(struct protocol_t *proto) { if (proto->vtable->destroy) proto->vtable->destroy(proto->state); + + free(proto); +} + +void +proto_params_free(protocol_params_t *params) +{ + assert(params); + + if (params->shared_secret) + free(params->shared_secret); + free(params); } diff --git a/src/protocol.h b/src/protocol.h index 0f45f22..0a7c452 100644 --- a/src/protocol.h +++ b/src/protocol.h @@ -52,6 +52,8 @@ int proto_handshake(struct protocol_t *proto, void *buf); int proto_send(struct protocol_t *proto, void *source, void *dest); int proto_recv(struct protocol_t *proto, void *source, void *dest); +void proto_params_free(protocol_params_t *params); + typedef struct protocol_vtable { /* Initialization function: Fills in the protocol vtable. */ diff --git a/src/protocols/obfs2.c b/src/protocols/obfs2.c index 41a97aa..115337e 100644 --- a/src/protocols/obfs2.c +++ b/src/protocols/obfs2.c @@ -136,6 +136,7 @@ obfs2_init(int n_options, char **options, } return 1; + err: usage(); return -1;

1 0

[obfsproxy/master] Updated obfs2.c function documentation.
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit 27e11a547181401016bdbf5fbda062e541a941f3 Author: George Kadianakis <desnacked(a)gmail.com> Date: Tue Jun 7 01:55:05 2011 +0200 Updated obfs2.c function documentation. --- src/protocols/obfs2.c | 30 +++++++++++++++++++++++------- 1 files changed, 23 insertions(+), 7 deletions(-) diff --git a/src/protocols/obfs2.c b/src/protocols/obfs2.c index 6f6f84b..5e91b5c 100644 --- a/src/protocols/obfs2.c +++ b/src/protocols/obfs2.c @@ -38,8 +38,8 @@ static void usage(void); static protocol_vtable *vtable=NULL; /* - This function parses 'options' and fills the protocol_params_t - 'params'. + This function parses 'options' and fills the protocol parameters + structure 'params'. It then fills the obfs2 vtable and initializes the crypto subsystem. Returns 0 on success, -1 on fail. @@ -233,6 +233,10 @@ derive_key(void *s, const char *keytype) return cryptstate; } +/** + Derive and return padding key of type 'keytype' from the seeds + currently set in state 's'. Returns NULL on failure. +*/ static crypt_t * derive_padding_key(void *s, const uchar *seed, const char *keytype) @@ -268,6 +272,14 @@ derive_padding_key(void *s, const uchar *seed, return cryptstate; } +/** + This is called everytime we get a connection for the obfs2 + protocol. + + It sets up the protocol vtable in 'proto_struct' and then attempts + to create and return a protocol state according to the protocol + parameters 'params'. +*/ void * obfs2_new(struct protocol_t *proto_struct, protocol_params_t *params) @@ -279,9 +291,8 @@ obfs2_new(struct protocol_t *proto_struct, } /** - Return a new object to handle protocol state. If 'initiator' is true, - we're the handshake initiator. Otherwise, we're the responder. Return - NULL on failure. + Returns an obfs2 state according to the protocol parameters + 'params'. If something goes wrong it returns NULL. */ static void * obfs2_state_new(protocol_params_t *params) @@ -324,7 +335,9 @@ obfs2_state_new(protocol_params_t *params) return state; } -/** Set the shared secret to be used with this protocol state. */ +/** + Sets the shared 'secret' to be used, on the protocol state 's'. +*/ static int obfs2_state_set_shared_secret(void *s, const char *secret, size_t secretlen) @@ -349,7 +362,7 @@ obfs2_state_set_shared_secret(void *s, const char *secret, } /** - Write the initial protocol setup and padding message for 'state' to + Write the initial protocol setup and padding message for state 's' to the evbuffer 'buf'. Return 0 on success, -1 on failure. */ static int @@ -565,6 +578,9 @@ obfs2_recv(void *s, struct evbuffer *source, return crypt_and_transmit(state->recv_crypto, source, dest); } +/** + Frees obfs2 state 's' +*/ static void obfs2_state_free(void *s) {

1 0

[obfsproxy/master] Kept argument parsing (set_up_protocol()) to main.c. We now just pass the protocol_params_t to listener_new().
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit afa43eb36053c3f0a292d43cd282dba104fc9648 Author: George Kadianakis <desnacked(a)gmail.com> Date: Tue Jun 7 01:47:11 2011 +0200 Kept argument parsing (set_up_protocol()) to main.c. We now just pass the protocol_params_t to listener_new(). --- src/main.c | 10 ++++++++-- src/network.c | 13 +++++-------- src/network.h | 3 ++- 3 files changed, 15 insertions(+), 11 deletions(-) diff --git a/src/main.c b/src/main.c index f7d3d35..065e011 100644 --- a/src/main.c +++ b/src/main.c @@ -238,6 +238,7 @@ main(int argc, const char **argv) listener_t *listeners[actual_protocols]; listener_t *temp_listener; int n_listeners=0; + protocol_params_t *proto_params; for (h=0;h<actual_protocols;h++) { if (n_protocols > 1) { @@ -246,8 +247,13 @@ main(int argc, const char **argv) STUPID_BEAUTIFIER, h+1)); } - temp_listener = listener_new(base, n_options_array[h], - protocol_options[h]); + /** free'd in listener_free() */ + proto_params = calloc(1, sizeof(protocol_params_t)); + if (set_up_protocol(n_options_array[h],protocol_options[h], + proto_params)<0) + continue; + + temp_listener = listener_new(base, proto_params); /** Free the space allocated for this protocol's options. */ free(protocol_options[h]); diff --git a/src/network.c b/src/network.c index c7dab85..3d64037 100644 --- a/src/network.c +++ b/src/network.c @@ -46,23 +46,20 @@ static void output_event_cb(struct bufferevent *bev, short what, void *arg); */ listener_t * listener_new(struct event_base *base, - int n_options, char **options) + protocol_params_t *proto_params) { const unsigned flags = LEV_OPT_CLOSE_ON_FREE|LEV_OPT_CLOSE_ON_EXEC|LEV_OPT_REUSEABLE; listener_t *lsn = calloc(1, sizeof(listener_t)); - if (!lsn) + if (!lsn) { + if (proto_params) + free(proto_params); return NULL; + } - protocol_params_t *proto_params = calloc(1, sizeof(protocol_params_t)); lsn->proto_params = proto_params; - if (set_up_protocol(n_options,options,lsn->proto_params)<0) { - listener_free(lsn); - return NULL; - } - lsn->listener = evconnlistener_new_bind(base, simple_listener_cb, lsn, flags, -1, diff --git a/src/network.h b/src/network.h index 5537680..120af80 100644 --- a/src/network.h +++ b/src/network.h @@ -21,6 +21,7 @@ typedef struct listener_t *listener; struct sockaddr; struct event_base; struct socks_state_t; +struct protocol_params_t; #define LSN_SIMPLE_CLIENT 1 #define LSN_SIMPLE_SERVER 2 @@ -30,7 +31,7 @@ typedef struct listener_t listener_t; struct addrinfo; listener_t *listener_new(struct event_base *base, - int n_options, char **options); + struct protocol_params_t *params); void listener_free(listener_t *listener); #ifdef NETWORK_PRIVATE

1 0

[obfsproxy/master] minor fix
by nickm＠torproject.org 09 Jun '11

09 Jun '11

commit afdc2d87cd073f3b9521cc4764f9bf4230a8a6dd Author: George Kadianakis <desnacked(a)gmail.com> Date: Tue Jun 7 04:00:13 2011 +0200 minor fix --- src/main.c | 11 ++++++----- 1 files changed, 6 insertions(+), 5 deletions(-) diff --git a/src/main.c b/src/main.c index 065e011..e7680d5 100644 --- a/src/main.c +++ b/src/main.c @@ -238,7 +238,7 @@ main(int argc, const char **argv) listener_t *listeners[actual_protocols]; listener_t *temp_listener; int n_listeners=0; - protocol_params_t *proto_params; + protocol_params_t *proto_params=NULL; for (h=0;h<actual_protocols;h++) { if (n_protocols > 1) { @@ -247,20 +247,21 @@ main(int argc, const char **argv) STUPID_BEAUTIFIER, h+1)); } - /** free'd in listener_free() */ + /** normally free'd in listener_free() */ proto_params = calloc(1, sizeof(protocol_params_t)); if (set_up_protocol(n_options_array[h],protocol_options[h], - proto_params)<0) + proto_params)<0) { + free(proto_params); continue; + } temp_listener = listener_new(base, proto_params); /** Free the space allocated for this protocol's options. */ free(protocol_options[h]); - if (!temp_listener) { + if (!temp_listener) continue; - } dbg(("Succesfully created listener.\n")); listeners[n_listeners] = temp_listener;

1 0