naiveproxy/net/base/host_port_pair.cc
2018-08-14 22:19:20 +00:00

84 lines
2.5 KiB
C++

// Copyright (c) 2012 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 "net/base/host_port_pair.h"
#include "base/logging.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_split.h"
#include "base/strings/string_util.h"
#include "base/strings/stringprintf.h"
#include "base/trace_event/memory_usage_estimator.h"
#include "net/base/ip_endpoint.h"
#include "net/base/parse_number.h"
#include "net/base/port_util.h"
#include "url/gurl.h"
namespace net {
HostPortPair::HostPortPair() : port_(0) {}
HostPortPair::HostPortPair(const std::string& in_host, uint16_t in_port)
: host_(in_host), port_(in_port) {
}
// static
HostPortPair HostPortPair::FromURL(const GURL& url) {
return HostPortPair(url.HostNoBrackets(),
static_cast<uint16_t>(url.EffectiveIntPort()));
}
// static
HostPortPair HostPortPair::FromIPEndPoint(const IPEndPoint& ipe) {
return HostPortPair(ipe.ToStringWithoutPort(), ipe.port());
}
// static
HostPortPair HostPortPair::FromString(const std::string& str) {
std::vector<base::StringPiece> key_port = base::SplitStringPiece(
str, ":", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
if (key_port.size() != 2)
return HostPortPair();
int port;
if (!ParseInt32(key_port[1], ParseIntFormat::NON_NEGATIVE, &port))
return HostPortPair();
if (!IsPortValid(port))
return HostPortPair();
HostPortPair host_port_pair;
host_port_pair.set_host(key_port[0].as_string());
host_port_pair.set_port(static_cast<uint16_t>(port));
return host_port_pair;
}
std::string HostPortPair::ToString() const {
std::string ret(HostForURL());
ret += ':';
ret += base::UintToString(port_);
return ret;
}
std::string HostPortPair::HostForURL() const {
// TODO(rtenneti): Add support for |host| to have '\0'.
if (host_.find('\0') != std::string::npos) {
std::string host_for_log(host_);
size_t nullpos;
while ((nullpos = host_for_log.find('\0')) != std::string::npos) {
host_for_log.replace(nullpos, 1, "%00");
}
LOG(DFATAL) << "Host has a null char: " << host_for_log;
}
// Check to see if the host is an IPv6 address. If so, added brackets.
if (host_.find(':') != std::string::npos) {
DCHECK_NE(host_[0], '[');
return base::StringPrintf("[%s]", host_.c_str());
}
return host_;
}
size_t HostPortPair::EstimateMemoryUsage() const {
return base::trace_event::EstimateMemoryUsage(host_);
}
} // namespace net