blob: 2c326cb7522fafb4d8967effcb20e96d97ba3379 [file] [log] [blame]
// Copyright (c) 2022 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "quiche/quic/load_balancer/load_balancer_config.h"
#include "quiche/quic/load_balancer/load_balancer_server_id.h"
namespace quic {
// Manages QUIC-LB configurations to extract a server ID from a properly
// encoded connection ID, usually on behalf of a load balancer.
class QUIC_EXPORT_PRIVATE LoadBalancerDecoder {
// Returns false if the config_id codepoint is already occupied.
bool AddConfig(const LoadBalancerConfig& config);
// Remove support for a config. Does nothing if there is no config for
// |config_id|. Does nothing and creates a bug if |config_id| is greater than
// 2.
void DeleteConfig(const uint8_t config_id);
// Extract a server ID from |connection_id|. If there is no config for the
// codepoint, |connection_id| is too short, or there's a decrypt error,
// returns empty. Will accept |connection_id| that is longer than necessary
// without error.
absl::optional<LoadBalancerServerId> GetServerId(
const QuicConnectionId& connection_id) const;
// Returns the config ID stored in the first two bits of |connection_id|, or
// empty if |connection_id| is empty.
static absl::optional<uint8_t> GetConfigId(
const QuicConnectionId& connection_id);
// Decoders can support up to 3 configs at once.
absl::optional<LoadBalancerConfig> config_[kNumLoadBalancerConfigs];
} // namespace quic