Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 10 additions & 4 deletions core/connection.c
Original file line number Diff line number Diff line change
Expand Up @@ -1607,10 +1607,16 @@ QuicConnHandshakeConfigure(
QUIC_CID_HASH_ENTRY,
Link);
LocalTP.Flags |= QUIC_TP_FLAG_STATELESS_RESET_TOKEN;
QuicBindingGenerateStatelessResetToken(
Connection->Paths[0].Binding,
SourceCID->CID.Data,
LocalTP.StatelessResetToken);
Status =
QuicBindingGenerateStatelessResetToken(
Connection->Paths[0].Binding,
SourceCID->CID.Data,
LocalTP.StatelessResetToken);
if (QUIC_FAILED(Status)) {
EventWriteQuicConnErrorStatus(Connection, Status,
"QuicBindingGenerateStatelessResetToken");
goto Error;
}

if (Connection->AckDelayExponent != QUIC_DEFAULT_ACK_DELAY_EXPONENT) {
LocalTP.Flags |= QUIC_TP_FLAG_ACK_DELAY_EXPONENT;
Expand Down
2 changes: 1 addition & 1 deletion inc/quic_platform_winuser.h
Original file line number Diff line number Diff line change
Expand Up @@ -236,7 +236,7 @@ _Post_writable_byte_size_(ByteCount)
DECLSPEC_ALLOCATOR
void*
QuicAlloc(
_In_ SIZE_T ByteCount
_In_ size_t ByteCount
);

void
Expand Down
3 changes: 2 additions & 1 deletion platform/tls_openssl.c
Original file line number Diff line number Diff line change
Expand Up @@ -2039,7 +2039,8 @@ QuicHashCompute(
InputLength,
Hash->Salt,
Hash->SaltLength,
Hash);
Hash) != 0 ?
QUIC_STATUS_SUCCESS : QUIC_STATUS_INTERNAL_ERROR;
}

static
Expand Down
3 changes: 1 addition & 2 deletions platform/tls_stub.c
Original file line number Diff line number Diff line change
Expand Up @@ -1407,7 +1407,6 @@ QuicHashCompute(
UNREFERENCED_PARAMETER(Hash);
UNREFERENCED_PARAMETER(Input);
UNREFERENCED_PARAMETER(InputLength);
UNREFERENCED_PARAMETER(OutputLength);
UNREFERENCED_PARAMETER(Output);
QuicZeroMemory(Output, OutputLength);
return QUIC_STATUS_SUCCESS;
}
99 changes: 80 additions & 19 deletions test/bin/quic_gtest.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,10 @@

#include "quic_gtest.h"

#ifdef QUIC_LOGS_WPP
#include "quic_gtest.tmh"
#endif

QUIC_API_V1* MsQuic;
HQUIC Registration;
QUIC_SEC_CONFIG_PARAMS* SelfSignedCertParams;
Expand Down Expand Up @@ -98,19 +102,45 @@ LogTestFailure(
va_start(Args, Format);
(void)_vsnprintf_s(Buffer, sizeof(Buffer), _TRUNCATE, Format, Args);
va_end(Args);
LogError("[test] FAILURE - %s:%d - %s", File, Line, Buffer);
GTEST_MESSAGE_AT_(File, Line, Buffer, ::testing::TestPartResult::kFatalFailure);
}

struct TestLogger {
const char* TestName;
TestLogger(const char* Name) : TestName(Name) {
LogInfo("[test] START %s", TestName);
}
~TestLogger() {
LogInfo("[test] END %s", TestName);
}
};

template<class T>
struct TestLoggerT {
const char* TestName;
TestLoggerT(const char* Name, const T& Params) : TestName(Name) {
std::ostringstream stream; stream << Params;
LogInfo("[test] START %s, %s", TestName, stream.str().c_str());
}
~TestLoggerT() {
LogInfo("[test] END %s", TestName);
}
};

TEST(ParameterValidation, ValidateApi) {
TestLogger Logger("QuicTestValidateApi");
QuicTestValidateApi();
}

TEST(ParameterValidation, ValidateRegistration) {
TestLogger Logger("QuicTestValidateRegistration");
QuicTestValidateRegistration();
}

#if _WIN32
TEST(ParameterValidation, ValidateServerSecConfig) {
TestLogger Logger("QuicTestValidateServerSecConfig");
QUIC_CERTIFICATE_HASH_STORE CertHashStore = { QUIC_CERTIFICATE_HASH_STORE_FLAG_NONE };
memcpy(CertHashStore.ShaHash, SelfSignedCertParams->Thumbprint, sizeof(CertHashStore.ShaHash));
memcpy(CertHashStore.StoreName, "My", 2);
Expand All @@ -120,58 +150,72 @@ TEST(ParameterValidation, ValidateServerSecConfig) {
#endif // _WIN32

TEST(ParameterValidation, ValidateSession) {
TestLogger Logger("QuicTestValidateSession");
QuicTestValidateSession();
}

TEST(ParameterValidation, ValidateListener) {
TestLogger Logger("QuicTestValidateListener");
QuicTestValidateListener();
}

TEST(ParameterValidation, ValidateConnection) {
TestLogger Logger("QuicTestValidateConnection");
QuicTestValidateConnection();
}

TEST_P(WithBool, ValidateStream) {
TestLoggerT<ParamType> Logger("QuicTestValidateStream", GetParam());
QuicTestValidateStream(GetParam());
}

TEST(Basic, CreateListener) {
TestLogger Logger("QuicTestCreateListener");
QuicTestCreateListener();
}

TEST(Basic, StartListener) {
TestLogger Logger("QuicTestStartListener");
QuicTestStartListener();
}

TEST_P(WithFamilyArgs, StartListenerImplicit) {
TestLoggerT<ParamType> Logger("QuicTestStartListenerImplicit", GetParam());
QuicTestStartListenerImplicit(GetParam().Family);
}

TEST(Basic, StartTwoListeners) {
TestLogger Logger("QuicTestStartTwoListeners");
QuicTestStartTwoListeners();
}

TEST(Basic, StartTwoListenersSameALPN) {
TestLogger Logger("QuicTestStartTwoListenersSameALPN");
QuicTestStartTwoListenersSameALPN();
}

TEST_P(WithFamilyArgs, StartListenerExplicit) {
TestLoggerT<ParamType> Logger("QuicTestStartListenerImplicit", GetParam());
QuicTestStartListenerExplicit(GetParam().Family);
}

TEST(Basic, CreateConnection) {
TestLogger Logger("QuicTestCreateConnection");
QuicTestCreateConnection();
}

TEST_P(WithFamilyArgs, BindConnectionImplicit) {
TestLoggerT<ParamType> Logger("QuicTestBindConnectionImplicit", GetParam());
QuicTestBindConnectionImplicit(GetParam().Family);
}

TEST_P(WithFamilyArgs, BindConnectionExplicit) {
TestLoggerT<ParamType> Logger("QuicTestBindConnectionExplicit", GetParam());
QuicTestBindConnectionExplicit(GetParam().Family);
}

TEST_P(WithHandshakeArgs1, Connect) {
TestLoggerT<ParamType> Logger("QuicTestConnect", GetParam());
QuicTestConnect(
GetParam().Family,
GetParam().ServerStatelessRetry,
Expand All @@ -180,11 +224,11 @@ TEST_P(WithHandshakeArgs1, Connect) {
false, // ChangeMaxStreamID
GetParam().MultipleALPNs,
false, // AsyncSecConfig
GetParam().MultiPacketClientInitial
);
GetParam().MultiPacketClientInitial);
}

TEST_P(WithHandshakeArgs2, OldVersion) {
TestLoggerT<ParamType> Logger("QuicTestConnect", GetParam());
QuicTestConnect(
GetParam().Family,
GetParam().ServerStatelessRetry,
Expand All @@ -193,15 +237,16 @@ TEST_P(WithHandshakeArgs2, OldVersion) {
false, // ChangeMaxStreamID
false, // MultipleALPNs
false, // AsyncSecConfig
false // MultiPacketClientInitial
);
false); // MultiPacketClientInitial
}

TEST_P(WithFamilyArgs, VersionNegotiation) {
TestLoggerT<ParamType> Logger("QuicTestVersionNegotiation", GetParam());
QuicTestVersionNegotiation(GetParam().Family);
}

TEST_P(WithFamilyArgs, Rebind) {
TestLoggerT<ParamType> Logger("QuicTestConnect", GetParam());
QuicTestConnect(
GetParam().Family,
false, // ServerStatelessRetry
Expand All @@ -210,11 +255,11 @@ TEST_P(WithFamilyArgs, Rebind) {
false, // ChangeMaxStreamID
false, // MultipleALPNs
false, // AsyncSecConfig
false // MultiPacketClientInitial
);
false); // MultiPacketClientInitial
}

TEST_P(WithFamilyArgs, ChangeMaxStreamIDs) {
TestLoggerT<ParamType> Logger("QuicTestConnect", GetParam());
QuicTestConnect(
GetParam().Family,
false, // ServerStatelessRetry
Expand All @@ -223,11 +268,11 @@ TEST_P(WithFamilyArgs, ChangeMaxStreamIDs) {
true, // ChangeMaxStreamID
false, // MultipleALPNs
false, // AsyncSecConfig
false // MultiPacketClientInitial
);
false); // MultiPacketClientInitial
}

TEST_P(WithHandshakeArgs1, AsyncSecurityConfig) {
TestLoggerT<ParamType> Logger("QuicTestConnect", GetParam());
QuicTestConnect(
GetParam().Family,
GetParam().ServerStatelessRetry,
Expand All @@ -236,27 +281,31 @@ TEST_P(WithHandshakeArgs1, AsyncSecurityConfig) {
false, // ChangeMaxStreamID
GetParam().MultipleALPNs,
true, // AsyncSecConfig
false // MultiPacketClientInitial
);
false); // MultiPacketClientInitial
}

TEST_P(WithFamilyArgs, Unreachable) {
TestLoggerT<ParamType> Logger("QuicTestConnectUnreachable", GetParam());
QuicTestConnectUnreachable(GetParam().Family);
}

TEST_P(WithFamilyArgs, BadALPN) {
TestLoggerT<ParamType> Logger("QuicTestConnectBadAlpn", GetParam());
QuicTestConnectBadAlpn(GetParam().Family);
}

TEST_P(WithFamilyArgs, BadSNI) {
TestLoggerT<ParamType> Logger("QuicTestConnectBadSni", GetParam());
QuicTestConnectBadSni(GetParam().Family);
}

TEST_P(WithFamilyArgs, ServerRejected) {
TestLoggerT<ParamType> Logger("QuicTestConnectServerRejected", GetParam());
QuicTestConnectServerRejected(GetParam().Family);
}

TEST_P(WithSendArgs1, Send) {
TestLoggerT<ParamType> Logger("QuicTestConnectAndPing", GetParam());
QuicTestConnectAndPing(
GetParam().Family,
GetParam().Length,
Expand All @@ -270,8 +319,7 @@ TEST_P(WithSendArgs1, Send) {
false, // ServerRejectZeroRtt
GetParam().UseSendBuffer,
GetParam().UnidirectionalStreams,
GetParam().ServerInitiatedStreams
);
GetParam().ServerInitiatedStreams);
}

#ifndef QUIC_0RTT_UNSUPPORTED
Expand All @@ -280,6 +328,7 @@ TEST_P(WithSendArgs1, Send) {
#endif

TEST_P(WithSendArgs2, SendLarge) {
TestLoggerT<ParamType> Logger("QuicTestConnectAndPing", GetParam());
QuicTestConnectAndPing(
GetParam().Family,
100000000llu,
Expand All @@ -293,11 +342,11 @@ TEST_P(WithSendArgs2, SendLarge) {
false, // ServerRejectZeroRtt
GetParam().UseSendBuffer,
false, // UnidirectionalStreams
false // ServerInitiatedStreams
);
false); // ServerInitiatedStreams
}

TEST_P(WithSendArgs3, SendIntermittently) {
TestLoggerT<ParamType> Logger("QuicTestConnectAndPing", GetParam());
QuicTestConnectAndPing(
GetParam().Family,
GetParam().Length,
Expand All @@ -311,46 +360,58 @@ TEST_P(WithSendArgs3, SendIntermittently) {
false, // ServerRejectZeroRtt
GetParam().UseSendBuffer,
false, // UnidirectionalStreams
false // ServerInitiatedStreams
);
false); // ServerInitiatedStreams
}

TEST_P(WithBool, IdleTimeout) {
TestLoggerT<ParamType> Logger("QuicTestConnectAndIdle", GetParam());
QuicTestConnectAndIdle(GetParam());
}

TEST(Misc, ServerDisconnect) {
TestLogger Logger("QuicTestServerDisconnect");
QuicTestServerDisconnect();
}

TEST(Misc, ClientDisconnect) {
TestLogger Logger("QuicTestClientDisconnect");
QuicTestClientDisconnect(false); // TODO - Support true, when race condition is fixed.
}

TEST_P(WithKeyUpdateArgs1, KeyUpdate) {
TestLoggerT<ParamType> Logger("QuicTestKeyUpdate", GetParam());
QuicTestKeyUpdate(
GetParam().Family,
GetParam().KeyUpdate == 0 ? 5 : 1, // Iterations
0, // KeyUpdateBytes
GetParam().KeyUpdate == 0, // UseKeyUpdateBytes
GetParam().KeyUpdate & 1, // ClientKeyUpdate
GetParam().KeyUpdate & 2 // ServerKeyUpdate
);
GetParam().KeyUpdate & 2); // ServerKeyUpdate
}

TEST_P(WithAbortiveArgs, AbortiveShutdown) {
TestLoggerT<ParamType> Logger("QuicAbortiveTransfers", GetParam());
QuicAbortiveTransfers(GetParam().Family, GetParam().Flags);
}

TEST_P(WithCidUpdateArgs, CidUpdate) {
TestLoggerT<ParamType> Logger("QuicTestCidUpdate", GetParam());
QuicTestCidUpdate(GetParam().Family, GetParam().Iterations);
}

TEST_P(WithReceiveResumeArgs, ReceiveResume) {
QuicTestReceiveResume(GetParam().Family, GetParam().SendBytes, GetParam().ConsumeBytes, GetParam().ShutdownType, GetParam().PauseType, GetParam().PauseFirst);
TestLoggerT<ParamType> Logger("QuicTestReceiveResume", GetParam());
QuicTestReceiveResume(
GetParam().Family,
GetParam().SendBytes,
GetParam().ConsumeBytes,
GetParam().ShutdownType,
GetParam().PauseType,
GetParam().PauseFirst);
}

TEST_P(WithReceiveResumeNoDataArgs, ReceiveResumeNoData) {
TestLoggerT<ParamType> Logger("QuicTestReceiveResumeNoData", GetParam());
QuicTestReceiveResumeNoData(GetParam().Family, GetParam().ShutdownType);
}

Expand Down
3 changes: 2 additions & 1 deletion test/bin/quic_gtest.h
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
#include <quic_platform.h>
#include <MsQuicTests.h>
#include <msquichelper.h>
#include "quic_trace.h"
#undef min // gtest headers conflict with previous definitions of min/max.
#undef max
#include "gtest/gtest.h"
Expand Down Expand Up @@ -189,7 +190,7 @@ struct KeyUpdateArgs1 {
static ::std::vector<KeyUpdateArgs1> Generate() {
::std::vector<KeyUpdateArgs1> list;
for (int Family : { 4, 6 })
for (int KeyUpdate : { 1, 2, 3, 4 })
for (int KeyUpdate : { 0, 1, 2, 3 })
list.push_back({ Family, KeyUpdate });
return list;
}
Expand Down
Loading