1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef NET_BASE_PROXY_CHAIN_H_
#define NET_BASE_PROXY_CHAIN_H_
#include <stdint.h>
#include <ostream>
#include <string>
#include <tuple>
#include <vector>
#include "base/strings/string_piece.h"
#include "net/base/host_port_pair.h"
#include "net/base/net_export.h"
#include "net/base/proxy_server.h"
#include "third_party/abseil-cpp/absl/types/optional.h"
namespace net {
// ProxyChain represents a chain of ProxyServers. A chain with multiple proxy
// servers means that a single connection will go through all of the proxies in
// order, using a tunnel through the first proxy to connect to the second, etc.
// A "direct" connection is a chain of length zero.
//
// TODO(crbug.com/1491092): Initial implementations of proxy chaining may, in
// fact, not tunnel through the last proxy in the ProxyChain if the destination
// is http.
class NET_EXPORT ProxyChain {
public:
// Constructs an invalid ProxyChain.
ProxyChain();
ProxyChain(const ProxyChain& other); // Copy constructor
ProxyChain(ProxyChain&& other) noexcept; // Move constructor
ProxyChain& operator=(const ProxyChain& other); // Copy assignment operator
ProxyChain& operator=(
ProxyChain&& other) noexcept; // Move assignment operator
ProxyChain(ProxyServer::Scheme scheme, const HostPortPair& host_port_pair);
explicit ProxyChain(std::vector<ProxyServer> proxy_server_list);
explicit ProxyChain(ProxyServer proxy_server);
~ProxyChain(); // Destructor declaration
// Creates a single-proxy ProxyChain, validating and canonicalizing input.
// Port is optional and, if not provided, will be replaced with the default
// port for the given scheme. Accepts IPv6 literal `host`s with surrounding
// brackets (URL format) or without (HostPortPair format). On invalid input,
// result will be a `SCHEME_INVALID` ProxyChain.
//
// Must not be called with `SCHEME_INVALID` or `SCHEME_DIRECT`. Use
// `ProxyChain()` or `Direct()` respectively to create an invalid or direct
// ProxyChain.
static ProxyChain FromSchemeHostAndPort(ProxyServer::Scheme scheme,
base::StringPiece host,
base::StringPiece port_str) {
return ProxyChain(
ProxyServer::FromSchemeHostAndPort(scheme, host, port_str));
}
static ProxyChain FromSchemeHostAndPort(ProxyServer::Scheme scheme,
base::StringPiece host,
absl::optional<uint16_t> port) {
return ProxyChain(ProxyServer::FromSchemeHostAndPort(scheme, host, port));
}
// Create a "direct" proxy chain, which includes no proxy servers.
static ProxyChain Direct() { return ProxyChain(ProxyServer::Direct()); }
// Get ProxyServer at index in chain. This is not valid for direct or invalid
// proxy chains.
const ProxyServer& GetProxyServer(size_t chain_index) const;
// Get the ProxyServers in this chain. This must not be called on invalid
// proxy chains. An empty vector is returned for direct proxy chains.
const std::vector<ProxyServer>& proxy_servers() const;
// Get the ProxyServers in this chain, or `nullopt` if the chain is not valid.
const absl::optional<std::vector<ProxyServer>>& proxy_servers_if_valid()
const {
return proxy_server_list_;
}
// Get the single ProxyServer equivalent to this chain. This must not be
// called for multi-proxy chains.
// TODO(crbug.com/1491092): Remove this method.
const ProxyServer& proxy_server() const;
// Returns number of proxy servers in chain.
size_t length() const {
return proxy_server_list_.has_value() ? proxy_server_list_.value().size()
: 0;
}
// Returns true if this chain contains more than one proxy.
bool is_multi_proxy() const {
return proxy_server_list_.has_value()
? proxy_server_list_.value().size() > 1
: false;
}
// Returns true if this chain contains exactly one proxy.
bool is_single_proxy() const {
return proxy_server_list_.has_value()
? proxy_server_list_.value().size() == 1
: false;
}
// Returns true if this is a direct (equivalently, zero-proxy) chain.
bool is_direct() const {
return proxy_server_list_.has_value() ? proxy_server_list_.value().empty()
: false;
}
// Returns true if a proxy server list is available .
bool IsValid() const { return proxy_server_list_.has_value(); }
bool operator==(const ProxyChain& other) const {
return proxy_server_list_ == other.proxy_server_list_;
}
bool operator!=(const ProxyChain& other) const { return !(*this == other); }
// Comparator function so this can be placed in a std::map.
bool operator<(const ProxyChain& other) const {
return proxy_server_list_ < other.proxy_server_list_;
}
std::string ToDebugString() const;
private:
absl::optional<std::vector<ProxyServer>> proxy_server_list_;
// Returns true if this chain is valid.
bool IsValidInternal() const;
};
NET_EXPORT_PRIVATE std::ostream& operator<<(std::ostream& os,
const ProxyChain& proxy_chain);
// A HostPortProxyPair holds a host/port destination and a ProxyChain describing
// how that destination is reached.
typedef std::pair<HostPortPair, ProxyChain> HostPortProxyPair;
} // namespace net
#endif // NET_BASE_PROXY_CHAIN_H_
|