torproject/tor
1
0
Fork 0
mirror of https://gitlab.torproject.org/tpo/core/tor.git synced 2024-12-06 19:41:15 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add GETINFO "rephist/{tap,ntor}/onion_handshakes_{assigned/requested}" test

Browse Source
This commit is contained in:
Neel Chauhan
2020-05-27 21:49:22 -07:00
committed by David Goulet
parent 4477317cbe
commit 4173876bc4
1 changed files with 87 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/test_controller.c
+87
View File
@@ -19,6 +19,7 @@
#include "feature/rend/rendservice.h"
#include "feature/nodelist/authcert.h"
#include "feature/nodelist/nodelist.h"
#include "feature/stats/rephist.h"
#include "test/test.h"
#include "test/test_helpers.h"
#include "lib/net/resolve.h"
@@ -2112,6 +2113,91 @@ test_control_getconf(void *arg)
smartlist_free(reply_strs);
}
static int
mock_rep_hist_get_circuit_handshake(uint16_t type)
{
int ret;
switch (type) {
case ONION_HANDSHAKE_TYPE_NTOR:
ret = 80;
break;
case ONION_HANDSHAKE_TYPE_TAP:
ret = 86;
break;
default:
ret = 0;
break;
}
return ret;
}
static void
test_rep_hist(void *arg)
{
/* We just need one of these to pass, it doesn't matter what's in it */
control_connection_t dummy;
/* Get results out */
char *answer = NULL;
const char *errmsg = NULL;
(void) arg;
/* We need these for returning the (mock) rephist. */
MOCK(rep_hist_get_circuit_handshake_requested,
mock_rep_hist_get_circuit_handshake);
MOCK(rep_hist_get_circuit_handshake_assigned,
mock_rep_hist_get_circuit_handshake);
/* NTor tests */
getinfo_helper_rephist(&dummy, "stats/ntor/requested",
&answer, &errmsg);
tt_ptr_op(answer, OP_NE, NULL);
tt_ptr_op(errmsg, OP_EQ, NULL);
tt_str_op(answer, OP_EQ, "80");
tor_free(answer);
errmsg = NULL;
getinfo_helper_rephist(&dummy, "stats/ntor/assigned",
&answer, &errmsg);
tt_ptr_op(answer, OP_NE, NULL);
tt_ptr_op(errmsg, OP_EQ, NULL);
tt_str_op(answer, OP_EQ, "80");
tor_free(answer);
errmsg = NULL;
/* TAP tests */
getinfo_helper_rephist(&dummy, "stats/tap/requested",
&answer, &errmsg);
tt_ptr_op(answer, OP_NE, NULL);
tt_ptr_op(errmsg, OP_EQ, NULL);
tt_str_op(answer, OP_EQ, "86");
tor_free(answer);
errmsg = NULL;
getinfo_helper_rephist(&dummy, "stats/tap/assigned",
&answer, &errmsg);
tt_ptr_op(answer, OP_NE, NULL);
tt_ptr_op(errmsg, OP_EQ, NULL);
tt_str_op(answer, OP_EQ, "86");
tor_free(answer);
errmsg = NULL;
getinfo_helper_rephist(&dummy, "stats/tap/onion_circuits_ddosed",
&answer, &errmsg);
tt_ptr_op(answer, OP_EQ, NULL);
tt_str_op(errmsg, OP_EQ, "Unrecognized handshake type");
errmsg = NULL;
done:
UNMOCK(rep_hist_get_circuit_handshake_requested);
UNMOCK(rep_hist_get_circuit_handshake_assigned);
tor_free(answer);
return;
}
#ifndef COCCI
#define PARSER_TEST(type) \
{ "parse/" #type, test_controller_parse_cmd, 0, &passthrough_setup, \
@@ -2146,5 +2232,6 @@ struct testcase_t controller_tests[] = {
{ "getinfo_md_all", test_getinfo_md_all, 0, NULL, NULL },
{ "control_reply", test_control_reply, 0, NULL, NULL },
{ "control_getconf", test_control_getconf, 0, NULL, NULL },
{ "rep_hist", test_rep_hist, 0, NULL, NULL },
END_OF_TESTCASES
};