diff options
| author | bitflash-maintainer <noreply@bitflash.invalid> | 2026-08-09 02:43:44 +0000 |
|---|---|---|
| committer | bitflash-maintainer <noreply@bitflash.invalid> | 2026-08-23 04:37:36 +0000 |
| commit | 857f57e962b24f51621ddac188fa6e42d0ac6f41 (patch) | |
| tree | c993ae79cc008b1a82d3b45b29c3dc26a24a3db9 /src/test_rvaccept.cpp | |
| parent | 89964636055e0856b2ff138edc4b7f45e0099f1b (diff) | |
pool: wait to be paired before wrapping the rendezvous socket
Upstream #30 split rendezvous registration from waiting for a dial, so
RvServiceRegister now returns as soon as the relay lists the service and
no longer consumes the relay's 0x01 pairing byte. It updated its own
caller in net.cpp and left this one behind.
AcceptOneFn went straight from RvServiceRegister to BtfServiceWrap, so
BtfServiceWrap read that 0x01 as the first byte of the miner's 32-byte
ephemeral key and every byte after it was shifted by one. The X25519
handshake has no confirmation step, so it completed against a garbage
shared secret, the tunnel was handed to the Stratum event loop, and
nothing failed loudly: the pool logged a miner arriving and then dropped
it as "never authorized", while the miner logged a pool that closed on it
with gotSubscribeAck=0. Loopback miners were unaffected, which is exactly
what made it look like a relay problem.
Block in RvServiceWaitPaired before wrapping, and close and re-register if
the relay drops us first.
test_rvaccept.cpp covers both shapes against a real bitflashd on loopback:
the pre-fix order establishes a tunnel that carries no bytes, the fixed
order round-trips. Also corrects the btftunnel.h contract comment, which
still described the pre-#30 behaviour and is what the old code was written
against -- upstream's own test_relayproc.cpp, test_btftunnel.cpp and
test_btfrv.cpp are still on that stale contract and fail the same way.
(cherry picked from commit 3e3a36b2bf1e6ffb4861507089b743688619377f)
Diffstat (limited to 'src/test_rvaccept.cpp')
| -rw-r--r-- | src/test_rvaccept.cpp | 180 |
1 files changed, 180 insertions, 0 deletions
diff --git a/src/test_rvaccept.cpp b/src/test_rvaccept.cpp new file mode 100644 index 0000000..afa9046 --- /dev/null +++ b/src/test_rvaccept.cpp | |||
| @@ -0,0 +1,180 @@ | |||
| 1 | // Local regression test for the pool's rendezvous accept path. | ||
| 2 | // | ||
| 3 | // Reproduces the exact failure that took remote miners off the pool: since | ||
| 4 | // upstream #30, RvServiceRegister returns as soon as the relay lists the | ||
| 5 | // service and no longer consumes the relay's 0x01 pairing byte. A service that | ||
| 6 | // goes straight from RvServiceRegister to BtfServiceWrap therefore reads that | ||
| 7 | // 0x01 as the first byte of the client's 32-byte ephemeral key, shifting the | ||
| 8 | // whole stream by one. X25519 has no confirmation step, so the handshake | ||
| 9 | // "succeeds" against a garbage shared secret and the corruption only shows up | ||
| 10 | // as a tunnel that never carries a valid byte. | ||
| 11 | // | ||
| 12 | // Build: g++ -std=gnu++14 -O2 -pthread test_rvaccept.cpp btftunnel.cpp btfrv.cpp \ | ||
| 13 | // btfchan.cpp -lsodium -o test_rvaccept | ||
| 14 | // | ||
| 15 | // Build: g++ -std=gnu++14 -O2 -pthread test_rvaccept.cpp btftunnel.cpp \ | ||
| 16 | // btfrv.cpp btfchan.cpp -lsodium -o test_rvaccept | ||
| 17 | // | ||
| 18 | // Run against a bitflashd relay already listening on 127.0.0.1:<port>. | ||
| 19 | // test_rvaccept broken <port> -- the pre-fix pool code path, expected to FAIL | ||
| 20 | // test_rvaccept fixed <port> -- with RvServiceWaitPaired, expected to PASS | ||
| 21 | |||
| 22 | #include "btftunnel.h" | ||
| 23 | #include "btfrv.h" | ||
| 24 | #include "btfchan.h" | ||
| 25 | |||
| 26 | #include <sys/socket.h> | ||
| 27 | #include <sys/time.h> | ||
| 28 | #include <unistd.h> | ||
| 29 | #include <cstdio> | ||
| 30 | #include <cstdlib> | ||
| 31 | #include <cstring> | ||
| 32 | #include <string> | ||
| 33 | #include <vector> | ||
| 34 | #include <thread> | ||
| 35 | #include <chrono> | ||
| 36 | |||
| 37 | using namespace btf; | ||
| 38 | |||
| 39 | static const char* HOST = "127.0.0.1"; | ||
| 40 | static unsigned short PORT = 18446; | ||
| 41 | static const char* MSG = "mining.subscribe stand-in payload"; | ||
| 42 | |||
| 43 | static void SetTimeout(int fd, int secs) | ||
| 44 | { | ||
| 45 | struct timeval tv; | ||
| 46 | tv.tv_sec = secs; | ||
| 47 | tv.tv_usec = 0; | ||
| 48 | setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)); | ||
| 49 | setsockopt(fd, SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(tv)); | ||
| 50 | } | ||
| 51 | |||
| 52 | static bool RecvN(int fd, void* buf, int n) | ||
| 53 | { | ||
| 54 | char* p = (char*)buf; | ||
| 55 | int off = 0; | ||
| 56 | while (off < n) { | ||
| 57 | int r = recv(fd, p + off, n - off, 0); | ||
| 58 | if (r <= 0) return false; | ||
| 59 | off += r; | ||
| 60 | } | ||
| 61 | return true; | ||
| 62 | } | ||
| 63 | |||
| 64 | static bool SendN(int fd, const void* buf, int n) | ||
| 65 | { | ||
| 66 | const char* p = (const char*)buf; | ||
| 67 | int off = 0; | ||
| 68 | while (off < n) { | ||
| 69 | int r = send(fd, p + off, n - off, 0); | ||
| 70 | if (r <= 0) return false; | ||
| 71 | off += r; | ||
| 72 | } | ||
| 73 | return true; | ||
| 74 | } | ||
| 75 | |||
| 76 | static bool g_serviceEcho = false; | ||
| 77 | static std::string g_serviceSaw; | ||
| 78 | |||
| 79 | // The pool's AcceptOneFn, in both shapes. | ||
| 80 | static void ServiceThread(bool fWaitPaired, | ||
| 81 | std::vector<unsigned char> id, | ||
| 82 | std::vector<unsigned char> enc_sk, | ||
| 83 | int len) | ||
| 84 | { | ||
| 85 | RvSocket rv = RvServiceRegister(HOST, PORT, id.data()); | ||
| 86 | if (rv == RV_INVALID) { printf(" service: RvServiceRegister failed\n"); return; } | ||
| 87 | |||
| 88 | if (fWaitPaired) { | ||
| 89 | if (!RvServiceWaitPaired(rv)) { | ||
| 90 | printf(" service: RvServiceWaitPaired failed\n"); | ||
| 91 | RvClose(rv); | ||
| 92 | return; | ||
| 93 | } | ||
| 94 | printf(" service: paired (0x01 consumed by RvServiceWaitPaired)\n"); | ||
| 95 | } else { | ||
| 96 | printf(" service: skipped RvServiceWaitPaired (pre-fix behaviour)\n"); | ||
| 97 | } | ||
| 98 | |||
| 99 | btf_socket_t app = BtfServiceWrap(rv, enc_sk.data()); | ||
| 100 | if (app == INVALID_SOCKET) { printf(" service: BtfServiceWrap failed\n"); return; } | ||
| 101 | printf(" service: BtfServiceWrap returned a tunnel\n"); | ||
| 102 | |||
| 103 | SetTimeout(app, 5); | ||
| 104 | std::vector<char> buf(len); | ||
| 105 | if (!RecvN(app, buf.data(), len)) { | ||
| 106 | printf(" service: never received the client's plaintext\n"); | ||
| 107 | close(app); | ||
| 108 | return; | ||
| 109 | } | ||
| 110 | g_serviceSaw.assign(buf.data(), len); | ||
| 111 | g_serviceEcho = SendN(app, buf.data(), len); | ||
| 112 | close(app); | ||
| 113 | } | ||
| 114 | |||
| 115 | int main(int argc, char* argv[]) | ||
| 116 | { | ||
| 117 | bool fWaitPaired = (argc >= 2 && strcmp(argv[1], "fixed") == 0); | ||
| 118 | if (argc >= 3) PORT = (unsigned short)atoi(argv[2]); | ||
| 119 | |||
| 120 | if (!TunnelInit()) { printf("TunnelInit failed\n"); return 2; } | ||
| 121 | |||
| 122 | unsigned char enc_pk[32], enc_sk[32]; | ||
| 123 | ChanKeypair(enc_pk, enc_sk); | ||
| 124 | |||
| 125 | std::vector<unsigned char> id(32); | ||
| 126 | for (int i = 0; i < 32; i++) id[i] = (unsigned char)(i * 7 + 13); | ||
| 127 | |||
| 128 | const std::string msg(MSG); | ||
| 129 | const int len = (int)msg.size(); | ||
| 130 | |||
| 131 | printf(" mode: %s relay: %s:%u\n", fWaitPaired ? "fixed" : "broken", HOST, PORT); | ||
| 132 | |||
| 133 | std::thread svc(ServiceThread, fWaitPaired, id, | ||
| 134 | std::vector<unsigned char>(enc_sk, enc_sk + 32), len); | ||
| 135 | std::this_thread::sleep_for(std::chrono::milliseconds(500)); | ||
| 136 | |||
| 137 | int nFail = 0; | ||
| 138 | |||
| 139 | btf_socket_t app = BtfClientTunnel(HOST, PORT, id.data(), enc_pk); | ||
| 140 | if (app == INVALID_SOCKET) { | ||
| 141 | printf(" client: BtfClientTunnel failed\n"); | ||
| 142 | nFail++; | ||
| 143 | } else { | ||
| 144 | printf(" client: tunnel established\n"); | ||
| 145 | SetTimeout(app, 5); | ||
| 146 | if (!SendN(app, msg.data(), len)) { printf(" client: send failed\n"); nFail++; } | ||
| 147 | std::vector<char> echo(len); | ||
| 148 | if (!RecvN(app, echo.data(), len)) { | ||
| 149 | printf(" client: no echo came back\n"); | ||
| 150 | nFail++; | ||
| 151 | } else if (std::string(echo.data(), len) != msg) { | ||
| 152 | printf(" client: echo did not match what was sent\n"); | ||
| 153 | nFail++; | ||
| 154 | } else { | ||
| 155 | printf(" client: round-tripped the payload intact\n"); | ||
| 156 | } | ||
| 157 | close(app); | ||
| 158 | } | ||
| 159 | |||
| 160 | svc.join(); | ||
| 161 | |||
| 162 | if (g_serviceSaw != msg) { | ||
| 163 | printf(" service did not see the correct plaintext (saw %d bytes)\n", | ||
| 164 | (int)g_serviceSaw.size()); | ||
| 165 | nFail++; | ||
| 166 | } | ||
| 167 | if (!g_serviceEcho) { printf(" service could not echo\n"); nFail++; } | ||
| 168 | |||
| 169 | bool fPassed = (nFail == 0); | ||
| 170 | printf(" RESULT: %s\n", fPassed ? "PASS -- tunnel carries plaintext end to end" | ||
| 171 | : "FAIL -- tunnel never carried a valid byte"); | ||
| 172 | |||
| 173 | // A broken run is the expected outcome pre-fix, so exit 0 when the | ||
| 174 | // observed behaviour matches the expectation for the mode under test. | ||
| 175 | bool fAsExpected = (fWaitPaired ? fPassed : !fPassed); | ||
| 176 | printf(" EXPECTED %s -> %s\n", | ||
| 177 | fWaitPaired ? "PASS" : "FAIL", | ||
| 178 | fAsExpected ? "as expected" : "UNEXPECTED"); | ||
| 179 | return fAsExpected ? 0 : 1; | ||
| 180 | } | ||