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/net_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 in_port)
  20     : host_(in_host), port_(in_port) {}
  21
  22 // static
  23 HostPortPair HostPortPair::FromURL(const GURL& url) {
  24   return HostPortPair(url.HostNoBrackets(),
  25                       static_cast<uint16>(url.EffectiveIntPort()));
  26 }
  27
  28 // static
  29 HostPortPair HostPortPair::FromIPEndPoint(const IPEndPoint& ipe) {
  30   return HostPortPair(ipe.ToStringWithoutPort(), ipe.port());
  31 }
  32
  33 HostPortPair HostPortPair::FromString(const std::string& str) {
  34   std::vector<std::string> key_port;
  35   base::SplitString(str, ':', &key_port);
  36   if (key_port.size() != 2)
  37     return HostPortPair();
  38   int port;
  39   if (!base::StringToInt(key_port[1], &port))
  40     return HostPortPair();
  41   if (!IsPortValid(port))
  42     return HostPortPair();
  43   HostPortPair host_port_pair;
  44   host_port_pair.set_host(key_port[0]);
  45   host_port_pair.set_port(static_cast<uint16>(port));
  46   return host_port_pair;
  47 }
  48
  49 std::string HostPortPair::ToString() const {
  50   std::string ret(HostForURL());
  51   ret += ':';
  52   ret += base::IntToString(port_);
  53   return ret;
  54 }
  55
  56 std::string HostPortPair::HostForURL() const {
  57   // TODO(rtenneti): Add support for |host| to have '\0'.
  58   if (host_.find('\0') != std::string::npos) {
  59     std::string host_for_log(host_);
  60     size_t nullpos;
  61     while ((nullpos = host_for_log.find('\0')) != std::string::npos) {
  62       host_for_log.replace(nullpos, 1, "%00");
  63     }
  64     LOG(DFATAL) << "Host has a null char: " << host_for_log;
  65   }
  66   // Check to see if the host is an IPv6 address.  If so, added brackets.
  67   if (host_.find(':') != std::string::npos) {
  68     DCHECK_NE(host_[0], '[');
  69     return base::StringPrintf("[%s]", host_.c_str());
  70   }
  71
  72   return host_;
  73 }
  74
  75 }  // namespace net