Change LoadBalancerEncoder::Create to return unique_ptr.
More convenient in light of changes to who owns instances of LoadBalancerEncoder
PiperOrigin-RevId: 471573005
diff --git a/quiche/quic/load_balancer/load_balancer_encoder.cc b/quiche/quic/load_balancer/load_balancer_encoder.cc
index bb9608b..5b0a613 100644
--- a/quiche/quic/load_balancer/load_balancer_encoder.cc
+++ b/quiche/quic/load_balancer/load_balancer_encoder.cc
@@ -47,7 +47,7 @@
constexpr uint8_t kLoadBalancerLengthMask = 0x3f;
constexpr uint8_t kLoadBalancerUnroutableConfigId = 0xc0;
-absl::optional<LoadBalancerEncoder> LoadBalancerEncoder::Create(
+std::unique_ptr<LoadBalancerEncoder> LoadBalancerEncoder::Create(
QuicRandom &random, LoadBalancerEncoderVisitorInterface *const visitor,
const bool len_self_encoded, const uint8_t unroutable_connection_id_len) {
if (unroutable_connection_id_len == 0 ||
@@ -56,10 +56,10 @@
QUIC_BUG(quic_bug_435375038_01)
<< "Invalid unroutable_connection_id_len = "
<< static_cast<int>(unroutable_connection_id_len);
- return absl::optional<LoadBalancerEncoder>();
+ return nullptr;
}
- return LoadBalancerEncoder(random, visitor, len_self_encoded,
- unroutable_connection_id_len);
+ return absl::WrapUnique(new LoadBalancerEncoder(
+ random, visitor, len_self_encoded, unroutable_connection_id_len));
}
bool LoadBalancerEncoder::UpdateConfig(const LoadBalancerConfig &config,
diff --git a/quiche/quic/load_balancer/load_balancer_encoder.h b/quiche/quic/load_balancer/load_balancer_encoder.h
index b48bb19..1405d62 100644
--- a/quiche/quic/load_balancer/load_balancer_encoder.h
+++ b/quiche/quic/load_balancer/load_balancer_encoder.h
@@ -60,7 +60,7 @@
// random. |unroutable_connection_id_length| specifies the length of
// connection IDs to be generated when there is no active config. It must not
// be 0 and must not be larger than the RFC9000 maximum of 20.
- static absl::optional<LoadBalancerEncoder> Create(
+ static std::unique_ptr<LoadBalancerEncoder> Create(
QuicRandom& random, LoadBalancerEncoderVisitorInterface* const visitor,
const bool len_self_encoded,
const uint8_t unroutable_connection_id_len = kLoadBalancerUnroutableLen);
diff --git a/quiche/quic/load_balancer/load_balancer_encoder_test.cc b/quiche/quic/load_balancer/load_balancer_encoder_test.cc
index f1df860..0ea7aa0 100644
--- a/quiche/quic/load_balancer/load_balancer_encoder_test.cc
+++ b/quiche/quic/load_balancer/load_balancer_encoder_test.cc
@@ -119,18 +119,18 @@
TEST_F(LoadBalancerEncoderTest, BadUnroutableLength) {
EXPECT_QUIC_BUG(
- EXPECT_FALSE(
- LoadBalancerEncoder::Create(random_, nullptr, false, 0).has_value()),
+ EXPECT_EQ(LoadBalancerEncoder::Create(random_, nullptr, false, 0),
+ nullptr),
"Invalid unroutable_connection_id_len = 0");
EXPECT_QUIC_BUG(
- EXPECT_FALSE(
- LoadBalancerEncoder::Create(random_, nullptr, false, 21).has_value()),
+ EXPECT_EQ(LoadBalancerEncoder::Create(random_, nullptr, false, 21),
+ nullptr),
"Invalid unroutable_connection_id_len = 21");
}
TEST_F(LoadBalancerEncoderTest, BadServerIdLength) {
auto encoder = LoadBalancerEncoder::Create(random_, nullptr, true);
- EXPECT_TRUE(encoder.has_value());
+ EXPECT_NE(encoder, nullptr);
// Expects a 3 byte server ID and got 4.
auto config = LoadBalancerConfig::CreateUnencrypted(1, 3, 4);
EXPECT_TRUE(config.has_value());
@@ -143,8 +143,7 @@
TEST_F(LoadBalancerEncoderTest, FailToUpdateConfigWithSameId) {
TestLoadBalancerEncoderVisitor visitor;
auto encoder = LoadBalancerEncoder::Create(random_, &visitor, true);
- EXPECT_TRUE(encoder.has_value());
- EXPECT_TRUE(encoder.has_value());
+ EXPECT_NE(encoder, nullptr);
auto config = LoadBalancerConfig::CreateUnencrypted(1, 3, 4);
EXPECT_TRUE(config.has_value());
EXPECT_TRUE(encoder->UpdateConfig(*config, MakeServerId(kServerId, 3)));
@@ -199,7 +198,7 @@
};
random_.AddNextValues(0, 0x75c2699c);
auto encoder = LoadBalancerEncoder::Create(random_, nullptr, true, 8);
- EXPECT_TRUE(encoder.has_value());
+ EXPECT_NE(encoder, nullptr);
auto config = LoadBalancerConfig::Create(config_id, server_id_len, nonce_len,
absl::string_view(raw_key));
EXPECT_TRUE(config.has_value());
@@ -248,7 +247,7 @@
};
for (const auto &test : test_vectors) {
auto encoder = LoadBalancerEncoder::Create(random_, nullptr, true, 8);
- EXPECT_TRUE(encoder.has_value());
+ EXPECT_NE(encoder, nullptr);
random_.AddNextValues(kNonceHigh, kNonceLow);
EXPECT_TRUE(encoder->UpdateConfig(test.config, test.server_id));
EXPECT_EQ(encoder->GenerateConnectionId(), test.connection_id);
@@ -259,7 +258,7 @@
const uint8_t server_id_len = 3;
TestLoadBalancerEncoderVisitor visitor;
auto encoder = LoadBalancerEncoder::Create(random_, &visitor, true, 8);
- EXPECT_TRUE(encoder.has_value());
+ EXPECT_NE(encoder, nullptr);
auto config = LoadBalancerConfig::Create(0, server_id_len, 4, kKey);
EXPECT_TRUE(config.has_value());
EXPECT_TRUE(
@@ -279,7 +278,7 @@
TEST_F(LoadBalancerEncoderTest, UnroutableConnectionId) {
random_.AddNextValues(0x83, kNonceHigh);
auto encoder = LoadBalancerEncoder::Create(random_, nullptr, false);
- EXPECT_TRUE(encoder.has_value());
+ EXPECT_NE(encoder, nullptr);
EXPECT_EQ(encoder->num_nonces_left(), 0);
auto connection_id = encoder->GenerateConnectionId();
// The first byte is the config_id (0xc0) xored with (0x83 & 0x3f).
@@ -290,7 +289,7 @@
TEST_F(LoadBalancerEncoderTest, NonDefaultUnroutableConnectionIdLength) {
auto encoder = LoadBalancerEncoder::Create(random_, nullptr, true, 9);
- EXPECT_TRUE(encoder.has_value());
+ EXPECT_NE(encoder, nullptr);
QuicConnectionId connection_id = encoder->GenerateConnectionId();
EXPECT_EQ(connection_id.length(), 9);
}
@@ -298,7 +297,7 @@
TEST_F(LoadBalancerEncoderTest, DeleteConfigWhenNoConfigExists) {
TestLoadBalancerEncoderVisitor visitor;
auto encoder = LoadBalancerEncoder::Create(random_, &visitor, true);
- EXPECT_TRUE(encoder.has_value());
+ EXPECT_NE(encoder, nullptr);
encoder->DeleteConfig();
EXPECT_EQ(visitor.num_deletes(), 0u);
}
