i2p_tests.cpp raw
1 // Copyright (c) 2021-2022 The Limenka developers
2 // Distributed under the MIT software license, see the accompanying
3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5 #include <common/args.h>
6 #include <i2p.h>
7 #include <logging.h>
8 #include <netaddress.h>
9 #include <netbase.h>
10 #include <test/util/logging.h>
11 #include <test/util/net.h>
12 #include <test/util/setup_common.h>
13 #include <util/readwritefile.h>
14 #include <util/threadinterrupt.h>
15
16 #include <boost/test/unit_test.hpp>
17
18 #include <memory>
19 #include <string>
20
21 /// Save the log level and the value of CreateSock and restore them when the test ends.
22 class EnvTestingSetup : public BasicTestingSetup
23 {
24 public:
25 explicit EnvTestingSetup(const ChainType chainType = ChainType::MAIN,
26 TestOpts opts = {})
27 : BasicTestingSetup{chainType, opts},
28 m_prev_log_level{LogInstance().LogLevel()},
29 m_create_sock_orig{CreateSock}
30 {
31 LogInstance().SetLogLevel(BCLog::Level::Trace);
32 }
33
34 ~EnvTestingSetup()
35 {
36 CreateSock = m_create_sock_orig;
37 LogInstance().SetLogLevel(m_prev_log_level);
38 }
39
40 private:
41 const BCLog::Level m_prev_log_level;
42 const decltype(CreateSock) m_create_sock_orig;
43 };
44
45 BOOST_FIXTURE_TEST_SUITE(i2p_tests, EnvTestingSetup)
46
47 BOOST_AUTO_TEST_CASE(unlimited_recv)
48 {
49 CreateSock = [](int, int, int) {
50 return std::make_unique<StaticContentsSock>(std::string(i2p::sam::MAX_MSG_SIZE + 1, 'a'));
51 };
52
53 CThreadInterrupt interrupt;
54 const std::optional<CService> addr{Lookup("127.0.0.1", 9000, false)};
55 const Proxy sam_proxy(addr.value(), false);
56 i2p::sam::Session session(gArgs.GetDataDirNet() / "test_i2p_private_key", sam_proxy, &interrupt);
57
58 {
59 ASSERT_DEBUG_LOG("Creating persistent SAM session");
60 ASSERT_DEBUG_LOG("too many bytes without a terminator");
61
62 i2p::Connection conn;
63 bool proxy_error;
64 BOOST_REQUIRE(!session.Connect(CService{}, conn, proxy_error));
65 }
66 }
67
68 BOOST_AUTO_TEST_CASE(listen_ok_accept_fail)
69 {
70 size_t num_sockets{0};
71 CreateSock = [&num_sockets](int, int, int) {
72 // clang-format off
73 ++num_sockets;
74 // First socket is the control socket for creating the session.
75 if (num_sockets == 1) {
76 return std::make_unique<StaticContentsSock>(
77 // reply to HELLO
78 "HELLO REPLY RESULT=OK VERSION=3.1\n"
79 // reply to DEST GENERATE
80 "DEST REPLY PUB=WnGOLXRBqHQhdVjFlWqRxJwz9hxx~2~wGc2Vplta1KhacY4tdEGodCF1WMWVapHEnDP2HHH~b~AZzZWmW1rUqFpxji10Qah0IXVYxZVqkcScM~Yccf9v8BnNlaZbWtSoWnGOLXRBqHQhdVjFlWqRxJwz9hxx~2~wGc2Vplta1KhacY4tdEGodCF1WMWVapHEnDP2HHH~b~AZzZWmW1rUqFpxji10Qah0IXVYxZVqkcScM~Yccf9v8BnNlaZbWtSoWnGOLXRBqHQhdVjFlWqRxJwz9hxx~2~wGc2Vplta1KhacY4tdEGodCF1WMWVapHEnDP2HHH~b~AZzZWmW1rUqFpxji10Qah0IXVYxZVqkcScM~Yccf9v8BnNlaZbWtSoWnGOLXRBqHQhdVjFlWqRxJwz9hxx~2~wGc2Vplta1KhacY4tdEGodCF1WMWVapHEnDP2HHH~b~AZzZWmW1rUqLE4SD-yjT48UNI7qiTUfIPiDitCoiTTz2cr4QGfw89rBQAEAAcAAA== PRIV=WnGOLXRBqHQhdVjFlWqRxJwz9hxx~2~wGc2Vplta1KhacY4tdEGodCF1WMWVapHEnDP2HHH~b~AZzZWmW1rUqFpxji10Qah0IXVYxZVqkcScM~Yccf9v8BnNlaZbWtSoWnGOLXRBqHQhdVjFlWqRxJwz9hxx~2~wGc2Vplta1KhacY4tdEGodCF1WMWVapHEnDP2HHH~b~AZzZWmW1rUqFpxji10Qah0IXVYxZVqkcScM~Yccf9v8BnNlaZbWtSoWnGOLXRBqHQhdVjFlWqRxJwz9hxx~2~wGc2Vplta1KhacY4tdEGodCF1WMWVapHEnDP2HHH~b~AZzZWmW1rUqFpxji10Qah0IXVYxZVqkcScM~Yccf9v8BnNlaZbWtSoWnGOLXRBqHQhdVjFlWqRxJwz9hxx~2~wGc2Vplta1KhacY4tdEGodCF1WMWVapHEnDP2HHH~b~AZzZWmW1rUqLE4SD-yjT48UNI7qiTUfIPiDitCoiTTz2cr4QGfw89rBQAEAAcAAOvuCIKTyv5f~1QgGq7XQl-IqBULTB5WzB3gw5yGPtd1p0AeoADrq1ccZggLPQ4ZLUsGK-HVw373rcTfvxrcuwenqVjiN4tbbYLWtP7xXGWj6fM6HyORhU63GphrjEePpMUHDHXd3o7pWGM-ieVVQSK~1MzF9P93pQWI3Do52EeNAayz4HbpPjNhVBzG1hUEFwznfPmUZBPuaOR4-uBm1NEWEuONlNOCctE4-U0Ukh94z-Qb55U5vXjR5G4apmBblr68t6Wm1TKlzpgFHzSqLryh3stWqrOKY1H0z9eZ2z1EkHFOpD5LyF6nf51e-lV7HLMl44TYzoEHK8RRVodtLcW9lacVdBpv~tOzlZERIiDziZODPETENZMz5oy9DQ7UUw==\n"
81 // reply to SESSION CREATE
82 "SESSION STATUS RESULT=OK\n"
83 // dummy to avoid reporting EOF on the socket
84 "a"
85 );
86 }
87 // Subsequent sockets are for recreating the session or for listening and accepting incoming connections.
88 if (num_sockets % 2 == 0) {
89 // Replies to Listen() and Accept()
90 return std::make_unique<StaticContentsSock>(
91 // reply to HELLO
92 "HELLO REPLY RESULT=OK VERSION=3.1\n"
93 // reply to STREAM ACCEPT
94 "STREAM STATUS RESULT=OK\n"
95 // continued reply to STREAM ACCEPT, violating the protocol described at
96 // https://geti2p.net/en/docs/api/samv3#Accept%20Response
97 // should be base64, something like
98 // "IchV608baDoXbqzQKSqFDmTXPVgoDbPAhZJvNRXXxi4hyFXrTxtoOhdurNApKoUOZNc9WCgNs8CFkm81FdfGLiHIVetPG2g6F26s0CkqhQ5k1z1YKA2zwIWSbzUV18YuIchV608baDoXbqzQKSqFDmTXPVgoDbPAhZJvNRXXxi4hyFXrTxtoOhdurNApKoUOZNc9WCgNs8CFkm81FdfGLiHIVetPG2g6F26s0CkqhQ5k1z1YKA2zwIWSbzUV18YuIchV608baDoXbqzQKSqFDmTXPVgoDbPAhZJvNRXXxi4hyFXrTxtoOhdurNApKoUOZNc9WCgNs8CFkm81FdfGLiHIVetPG2g6F26s0CkqhQ5k1z1YKA2zwIWSbzUV18YuIchV608baDoXbqzQKSqFDmTXPVgoDbPAhZJvNRXXxi4hyFXrTxtoOhdurNApKoUOZNc9WCgNs8CFkm81FdfGLlSreVaCuCS5sdb-8ToWULWP7kt~lRPDeUNxQMq3cRSBBQAEAAcAAA==\n"
99 "STREAM STATUS RESULT=I2P_ERROR MESSAGE=\"Session was closed\"\n"
100 );
101 } else {
102 // Another control socket, but without creating a destination (it is cached in the session).
103 return std::make_unique<StaticContentsSock>(
104 // reply to HELLO
105 "HELLO REPLY RESULT=OK VERSION=3.1\n"
106 // reply to SESSION CREATE
107 "SESSION STATUS RESULT=OK\n"
108 // dummy to avoid reporting EOF on the socket
109 "a"
110 );
111 }
112 // clang-format on
113 };
114
115 CThreadInterrupt interrupt;
116 const CService addr{in6_addr(IN6ADDR_LOOPBACK_INIT), /*port=*/7656};
117 const Proxy sam_proxy(addr, false);
118 i2p::sam::Session session(gArgs.GetDataDirNet() / "test_i2p_private_key",
119 sam_proxy,
120 &interrupt);
121
122 i2p::Connection conn;
123 for (size_t i = 0; i < 5; ++i) {
124 ASSERT_DEBUG_LOG("Creating persistent SAM session");
125 ASSERT_DEBUG_LOG("Persistent SAM session" /* ... created */);
126 ASSERT_DEBUG_LOG("Error accepting");
127 ASSERT_DEBUG_LOG("Destroying SAM session");
128 BOOST_REQUIRE(session.Listen(conn));
129 BOOST_REQUIRE(!session.Accept(conn));
130 }
131 }
132
133 BOOST_AUTO_TEST_CASE(damaged_private_key)
134 {
135 CreateSock = [](int, int, int) {
136 return std::make_unique<StaticContentsSock>("HELLO REPLY RESULT=OK VERSION=3.1\n"
137 "SESSION STATUS RESULT=OK DESTINATION=\n");
138 };
139
140 const auto i2p_private_key_file = m_args.GetDataDirNet() / "test_i2p_private_key_damaged";
141
142 for (const auto& [file_contents, expected_error] : std::vector<std::tuple<std::string, std::string>>{
143 {"", "The private key is too short (0 < 387)"},
144
145 {"abcd", "The private key is too short (4 < 387)"},
146
147 {std::string(386, '\0'), "The private key is too short (386 < 387)"},
148
149 {std::string(385, '\0') + '\0' + '\1',
150 "Certificate length (1) designates that the private key should be 388 bytes, but it is only "
151 "387 bytes"},
152
153 {std::string(385, '\0') + '\0' + '\5' + "abcd",
154 "Certificate length (5) designates that the private key should be 392 bytes, but it is only "
155 "391 bytes"}}) {
156 BOOST_REQUIRE(WriteBinaryFile(i2p_private_key_file, file_contents));
157
158 CThreadInterrupt interrupt;
159 const CService addr{in6_addr(IN6ADDR_LOOPBACK_INIT), /*port=*/7656};
160 const Proxy sam_proxy{addr, false};
161 i2p::sam::Session session(i2p_private_key_file, sam_proxy, &interrupt);
162
163 {
164 ASSERT_DEBUG_LOG("Creating persistent SAM session");
165 ASSERT_DEBUG_LOG(expected_error);
166
167 i2p::Connection conn;
168 bool proxy_error;
169 BOOST_CHECK(!session.Connect(CService{}, conn, proxy_error));
170 }
171 }
172 }
173
174 BOOST_AUTO_TEST_SUITE_END()
175