quic settings
This commit is contained in:
parent
0827e1f566
commit
675476dfea
5 changed files with 175 additions and 9 deletions
130
tests/ShouldHonourQUICSettings/main.cpp
Normal file
130
tests/ShouldHonourQUICSettings/main.cpp
Normal file
|
|
@ -0,0 +1,130 @@
|
|||
//SPDX-License-Identifier: LGPL-3.0-only
|
||||
//SPDX-FileCopyrightText: Copyright (C) 2026 Catcrafts®
|
||||
|
||||
// QUICSettings has to actually reach the msquic Configuration. The knob with
|
||||
// observable behaviour is peerUnidiStreamCount: it caps how many streams the
|
||||
// peer may open toward us, so the listener here opens more unidi streams than
|
||||
// the previously hardcoded limit of 16 allowed. Against a build that ignores
|
||||
// the setting the extra streams never arrive and this test times out.
|
||||
|
||||
import Crafter.Network;
|
||||
import Crafter.Thread;
|
||||
import std;
|
||||
using namespace Crafter;
|
||||
|
||||
namespace {
|
||||
int g_failures = 0;
|
||||
void Check(bool cond, std::string_view what) {
|
||||
std::println(" {}: {}", cond ? "ok" : "FAIL", what);
|
||||
if (!cond) ++g_failures;
|
||||
}
|
||||
|
||||
constexpr std::uint16_t port = 9187;
|
||||
constexpr std::string_view alpn = "f3d/test-settings";
|
||||
|
||||
// Above the old hardcoded cap of 16, below the 32 we grant below.
|
||||
constexpr std::size_t streamCount = 24;
|
||||
}
|
||||
|
||||
int main() {
|
||||
ThreadPool::Start();
|
||||
|
||||
std::mutex mtx;
|
||||
std::condition_variable cv;
|
||||
std::set<int> received;
|
||||
|
||||
QUICServerCredentials serverCreds;
|
||||
serverCreds.selfSigned = true;
|
||||
|
||||
// The server-side streams must outlive their SendSync, so park them here
|
||||
// rather than letting them destruct at the end of each loop iteration.
|
||||
std::mutex serverMtx;
|
||||
std::vector<QUICStream> serverStreams;
|
||||
serverStreams.reserve(streamCount);
|
||||
std::string openError;
|
||||
|
||||
ListenerQUIC listener(port, std::string(alpn), serverCreds, [&](ClientQUIC* peer) {
|
||||
std::vector<QUICStream> opened;
|
||||
opened.reserve(streamCount);
|
||||
try {
|
||||
// Open every stream *before* sending on any of them, so all of
|
||||
// them are live at once and the peer's unidi credit has to cover
|
||||
// the whole count simultaneously. Opening and finishing one at a
|
||||
// time recycles credit as each stream closes, which passes even
|
||||
// at the old cap of 16 and would test nothing.
|
||||
for (std::size_t i = 0; i < streamCount; ++i) {
|
||||
opened.push_back(peer->OpenStream(/*unidirectional=*/true));
|
||||
}
|
||||
// Leave the send side open — a finished stream returns its credit.
|
||||
for (std::size_t i = 0; i < streamCount; ++i) {
|
||||
std::string message = std::format("{}", i);
|
||||
opened[i].SendSync(message.data(), static_cast<std::uint32_t>(message.size()),
|
||||
/*finish=*/false);
|
||||
}
|
||||
} catch (std::exception& e) {
|
||||
std::lock_guard lock(serverMtx);
|
||||
openError = e.what();
|
||||
return;
|
||||
}
|
||||
std::lock_guard lock(serverMtx);
|
||||
serverStreams = std::move(opened);
|
||||
});
|
||||
listener.ListenAsyncAsync();
|
||||
|
||||
try {
|
||||
QUICClientCredentials clientCreds;
|
||||
clientCreds.insecureNoServerValidation = true;
|
||||
// The knob under test. Without it this defaults to 16 and the
|
||||
// listener stalls partway through its 24 streams.
|
||||
clientCreds.settings.peerUnidiStreamCount = 32;
|
||||
|
||||
ClientQUIC client(std::string("localhost"), port, std::string(alpn), clientCreds);
|
||||
|
||||
// Streams opened before this callback lands are queued and drained
|
||||
// into it, so registering after connect is safe.
|
||||
std::mutex clientMtx;
|
||||
std::vector<QUICStream> clientStreams;
|
||||
clientStreams.reserve(streamCount);
|
||||
client.OnStream([&](QUICStream stream) {
|
||||
try {
|
||||
// One chunk only: the peer deliberately leaves the send
|
||||
// side open so the stream keeps holding its credit.
|
||||
std::vector<char> got = stream.RecieveSync();
|
||||
int index = std::stoi(std::string(got.begin(), got.end()));
|
||||
{
|
||||
std::lock_guard lock(mtx);
|
||||
received.insert(index);
|
||||
}
|
||||
cv.notify_all();
|
||||
std::lock_guard lock(clientMtx);
|
||||
clientStreams.push_back(std::move(stream));
|
||||
} catch (...) {
|
||||
// A stream that fails simply never counts toward the total.
|
||||
}
|
||||
});
|
||||
|
||||
std::size_t got = 0;
|
||||
{
|
||||
std::unique_lock lock(mtx);
|
||||
cv.wait_for(lock, std::chrono::seconds(20),
|
||||
[&] { return received.size() == streamCount; });
|
||||
got = received.size();
|
||||
}
|
||||
|
||||
{
|
||||
std::lock_guard lock(serverMtx);
|
||||
Check(openError.empty(),
|
||||
openError.empty() ? "listener opened every stream"
|
||||
: std::format("listener failed to open: {}", openError));
|
||||
}
|
||||
Check(got == streamCount,
|
||||
std::format("received {} of {} peer-initiated unidi streams", got, streamCount));
|
||||
|
||||
// msquic's RegistrationClose blocks on connections the listener still
|
||||
// owns, so skip the static-dtor path the way the other QUIC tests do.
|
||||
std::_Exit(g_failures ? 1 : 0);
|
||||
} catch (std::exception& e) {
|
||||
std::println("{}", e.what());
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue