mirror of
https://gitlab.torproject.org/tpo/core/tor.git
synced 2024-11-11 05:33:47 +01:00
Basic unit tests for relay_crypto.c
These tests handle incoming and outgoing cells on a three-hop circuit, and make sure that the crypto works end-to-end. They don't yet test spec conformance, leaky-pipe, or various error cases.
This commit is contained in:
parent
5ecad6c95d
commit
7db4d0c55f
@ -150,6 +150,7 @@ src_test_test_SOURCES = \
|
||||
src/test/test_pubsub.c \
|
||||
src/test/test_relay.c \
|
||||
src/test/test_relaycell.c \
|
||||
src/test/test_relaycrypt.c \
|
||||
src/test/test_rendcache.c \
|
||||
src/test/test_replay.c \
|
||||
src/test/test_router.c \
|
||||
|
@ -1242,6 +1242,7 @@ struct testgroup_t testgroups[] = {
|
||||
{ "pt/", pt_tests },
|
||||
{ "relay/" , relay_tests },
|
||||
{ "relaycell/", relaycell_tests },
|
||||
{ "relaycrypt/", relaycrypt_tests },
|
||||
{ "rend_cache/", rend_cache_tests },
|
||||
{ "replaycache/", replaycache_tests },
|
||||
{ "router/", router_tests },
|
||||
|
@ -237,6 +237,7 @@ extern struct testcase_t pubsub_tests[];
|
||||
extern struct testcase_t pt_tests[];
|
||||
extern struct testcase_t relay_tests[];
|
||||
extern struct testcase_t relaycell_tests[];
|
||||
extern struct testcase_t relaycrypt_tests[];
|
||||
extern struct testcase_t rend_cache_tests[];
|
||||
extern struct testcase_t replaycache_tests[];
|
||||
extern struct testcase_t router_tests[];
|
||||
|
184
src/test/test_relaycrypt.c
Normal file
184
src/test/test_relaycrypt.c
Normal file
@ -0,0 +1,184 @@
|
||||
/* Copyright 2001-2004 Roger Dingledine.
|
||||
* Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
|
||||
* Copyright (c) 2007-2018, The Tor Project, Inc. */
|
||||
/* See LICENSE for licensing information */
|
||||
|
||||
#include "or.h"
|
||||
#include "circuitbuild.h"
|
||||
#define CIRCUITLIST_PRIVATE
|
||||
#include "circuitlist.h"
|
||||
#include "relay.h"
|
||||
#include "relay_crypto.h"
|
||||
#include "test.h"
|
||||
|
||||
static const char KEY_MATERIAL[3][CPATH_KEY_MATERIAL_LEN] = {
|
||||
" 'My public key is in this signed x509 object', said Tom assertively.",
|
||||
"'Let's chart the pedal phlanges in the tomb', said Tom cryptographically",
|
||||
" 'Segmentation fault bugs don't _just happen_', said Tom seethingly.",
|
||||
};
|
||||
|
||||
typedef struct testing_circuitset_t {
|
||||
or_circuit_t *or_circ[3];
|
||||
origin_circuit_t *origin_circ;
|
||||
} testing_circuitset_t;
|
||||
|
||||
static int testing_circuitset_teardown(const struct testcase_t *testcase,
|
||||
void *ptr);
|
||||
|
||||
static void *
|
||||
testing_circuitset_setup(const struct testcase_t *testcase)
|
||||
{
|
||||
testing_circuitset_t *cs = tor_malloc_zero(sizeof(testing_circuitset_t));
|
||||
int i;
|
||||
|
||||
for (i=0; i<3; ++i) {
|
||||
cs->or_circ[i] = or_circuit_new(0, NULL);
|
||||
tt_int_op(0, OP_EQ,
|
||||
relay_crypto_init(&cs->or_circ[i]->crypto,
|
||||
KEY_MATERIAL[i], sizeof(KEY_MATERIAL[i]),
|
||||
0, 0));
|
||||
}
|
||||
|
||||
cs->origin_circ = origin_circuit_new();
|
||||
cs->origin_circ->base_.purpose = CIRCUIT_PURPOSE_C_GENERAL;
|
||||
for (i=0; i<3; ++i) {
|
||||
crypt_path_t *hop = tor_malloc_zero(sizeof(*hop));
|
||||
relay_crypto_init(&hop->crypto, KEY_MATERIAL[i], sizeof(KEY_MATERIAL[i]),
|
||||
0, 0);
|
||||
hop->state = CPATH_STATE_OPEN;
|
||||
onion_append_to_cpath(&cs->origin_circ->cpath, hop);
|
||||
tt_ptr_op(hop, OP_EQ, cs->origin_circ->cpath->prev);
|
||||
}
|
||||
|
||||
return cs;
|
||||
done:
|
||||
testing_circuitset_teardown(testcase, cs);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static int
|
||||
testing_circuitset_teardown(const struct testcase_t *testcase, void *ptr)
|
||||
{
|
||||
(void)testcase;
|
||||
testing_circuitset_t *cs = ptr;
|
||||
int i;
|
||||
for (i=0; i<3; ++i) {
|
||||
circuit_free_(TO_CIRCUIT(cs->or_circ[i]));
|
||||
}
|
||||
circuit_free_(TO_CIRCUIT(cs->origin_circ));
|
||||
tor_free(cs);
|
||||
return 1;
|
||||
}
|
||||
|
||||
static const struct testcase_setup_t relaycrypt_setup = {
|
||||
testing_circuitset_setup, testing_circuitset_teardown
|
||||
};
|
||||
|
||||
/* Test encrypting a cell to the final hop on a circuit, decrypting it
|
||||
* at each hop, and recognizing it at the other end. Then do it again
|
||||
* and again as the state evolves. */
|
||||
static void
|
||||
test_relaycrypt_outbound(void *arg)
|
||||
{
|
||||
testing_circuitset_t *cs = arg;
|
||||
tt_assert(cs);
|
||||
|
||||
relay_header_t rh;
|
||||
cell_t orig;
|
||||
cell_t encrypted;
|
||||
int i, j;
|
||||
|
||||
for (i = 0; i < 50; ++i) {
|
||||
crypto_rand((char *)&orig, sizeof(orig));
|
||||
|
||||
relay_header_unpack(&rh, orig.payload);
|
||||
rh.recognized = 0;
|
||||
memset(rh.integrity, 0, sizeof(rh.integrity));
|
||||
relay_header_pack(orig.payload, &rh);
|
||||
|
||||
memcpy(&encrypted, &orig, sizeof(orig));
|
||||
|
||||
/* Encrypt the cell to the last hop */
|
||||
relay_encrypt_cell_outbound(&encrypted, cs->origin_circ,
|
||||
cs->origin_circ->cpath->prev);
|
||||
|
||||
for (j = 0; j < 3; ++j) {
|
||||
crypt_path_t *layer_hint = NULL;
|
||||
char recognized = 0;
|
||||
int r = relay_decrypt_cell(TO_CIRCUIT(cs->or_circ[j]),
|
||||
&encrypted,
|
||||
CELL_DIRECTION_OUT,
|
||||
&layer_hint, &recognized);
|
||||
tt_int_op(r, OP_EQ, 0);
|
||||
tt_ptr_op(layer_hint, OP_EQ, NULL);
|
||||
tt_int_op(recognized != 0, OP_EQ, j == 2);
|
||||
}
|
||||
|
||||
tt_mem_op(orig.payload, OP_EQ, encrypted.payload, CELL_PAYLOAD_SIZE);
|
||||
}
|
||||
|
||||
done:
|
||||
;
|
||||
}
|
||||
|
||||
/* As above, but simulate inbound cells from the last hop. */
|
||||
static void
|
||||
test_relaycrypt_inbound(void *arg)
|
||||
{
|
||||
testing_circuitset_t *cs = arg;
|
||||
tt_assert(cs);
|
||||
|
||||
relay_header_t rh;
|
||||
cell_t orig;
|
||||
cell_t encrypted;
|
||||
int i, j;
|
||||
|
||||
for (i = 0; i < 50; ++i) {
|
||||
crypto_rand((char *)&orig, sizeof(orig));
|
||||
|
||||
relay_header_unpack(&rh, orig.payload);
|
||||
rh.recognized = 0;
|
||||
memset(rh.integrity, 0, sizeof(rh.integrity));
|
||||
relay_header_pack(orig.payload, &rh);
|
||||
|
||||
memcpy(&encrypted, &orig, sizeof(orig));
|
||||
|
||||
/* Encrypt the cell to the last hop */
|
||||
relay_encrypt_cell_inbound(&encrypted, cs->or_circ[2]);
|
||||
|
||||
crypt_path_t *layer_hint = NULL;
|
||||
char recognized = 0;
|
||||
int r;
|
||||
for (j = 1; j >= 0; --j) {
|
||||
r = relay_decrypt_cell(TO_CIRCUIT(cs->or_circ[j]),
|
||||
&encrypted,
|
||||
CELL_DIRECTION_IN,
|
||||
&layer_hint, &recognized);
|
||||
tt_int_op(r, OP_EQ, 0);
|
||||
tt_ptr_op(layer_hint, OP_EQ, NULL);
|
||||
tt_int_op(recognized, OP_EQ, 0);
|
||||
}
|
||||
|
||||
relay_decrypt_cell(TO_CIRCUIT(cs->origin_circ),
|
||||
&encrypted,
|
||||
CELL_DIRECTION_IN,
|
||||
&layer_hint, &recognized);
|
||||
tt_int_op(r, OP_EQ, 0);
|
||||
tt_int_op(recognized, OP_EQ, 1);
|
||||
tt_ptr_op(layer_hint, OP_EQ, cs->origin_circ->cpath->prev);
|
||||
|
||||
tt_mem_op(orig.payload, OP_EQ, encrypted.payload, CELL_PAYLOAD_SIZE);
|
||||
}
|
||||
done:
|
||||
;
|
||||
}
|
||||
|
||||
#define TEST(name) \
|
||||
{ # name, test_relaycrypt_ ## name, 0, &relaycrypt_setup, NULL }
|
||||
|
||||
struct testcase_t relaycrypt_tests[] = {
|
||||
TEST(outbound),
|
||||
TEST(inbound),
|
||||
END_OF_TESTCASES
|
||||
};
|
||||
|
Loading…
Reference in New Issue
Block a user