net/base/host_port_pair.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/base/host_port_pair.h"
   6
   7 #include "base/logging.h"
   8 #include "base/strings/string_number_conversions.h"
   9 #include "base/strings/string_split.h"
  10 #include "base/strings/string_util.h"
  11 #include "base/strings/stringprintf.h"
  12 #include "net/base/ip_endpoint.h"
  13 #include "net/base/port_util.h"
  14 #include "url/gurl.h"
  15
  16 namespace net {
  17
  18 HostPortPair::HostPortPair() : port_(0) {}
  19 HostPortPair::HostPortPair(const std::string& in_host, uint16_t in_port)
  20     : host_(in_host), port_(in_port) {
  21 }
  22
  23 // static
  24 HostPortPair HostPortPair::FromURL(const GURL& url) {
  25   return HostPortPair(url.HostNoBrackets(),
  26                       static_cast<uint16_t>(url.EffectiveIntPort()));
  27 }
  28
  29 // static
  30 HostPortPair HostPortPair::FromIPEndPoint(const IPEndPoint& ipe) {
  31   return HostPortPair(ipe.ToStringWithoutPort(), ipe.port());
  32 }
  33
  34 HostPortPair HostPortPair::FromString(const std::string& str) {
  35   std::vector<base::StringPiece> key_port = base::SplitStringPiece(
  36       str, ":", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
  37   if (key_port.size() != 2)
  38     return HostPortPair();
  39   int port;
  40   if (!base::StringToInt(key_port[1], &port))
  41     return HostPortPair();
  42   if (!IsPortValid(port))
  43     return HostPortPair();
  44   HostPortPair host_port_pair;
  45   host_port_pair.set_host(key_port[0].as_string());
  46   host_port_pair.set_port(static_cast<uint16_t>(port));
  47   return host_port_pair;
  48 }
  49
  50 std::string HostPortPair::ToString() const {
  51   std::string ret(HostForURL());
  52   ret += ':';
  53   ret += base::UintToString(port_);
  54   return ret;
  55 }
  56
  57 std::string HostPortPair::HostForURL() const {
  58   // TODO(rtenneti): Add support for |host| to have '\0'.
  59   if (host_.find('\0') != std::string::npos) {
  60     std::string host_for_log(host_);
  61     size_t nullpos;
  62     while ((nullpos = host_for_log.find('\0')) != std::string::npos) {
  63       host_for_log.replace(nullpos, 1, "%00");
  64     }
  65     LOG(DFATAL) << "Host has a null char: " << host_for_log;
  66   }
  67   // Check to see if the host is an IPv6 address.  If so, added brackets.
  68   if (host_.find(':') != std::string::npos) {
  69     DCHECK_NE(host_[0], '[');
  70     return base::StringPrintf("[%s]", host_.c_str());
  71   }
  72
  73   return host_;
  74 }
  75
  76 }  // namespace net