net/quic/crypto/crypto_handshake_message.cc

   1 // Copyright (c) 2013 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/quic/crypto/crypto_handshake_message.h"
   6
   7 #include "base/strings/stringprintf.h"
   8 #include "base/strings/string_number_conversions.h"
   9 #include "net/quic/crypto/crypto_framer.h"
  10 #include "net/quic/crypto/crypto_protocol.h"
  11 #include "net/quic/quic_socket_address_coder.h"
  12 #include "net/quic/quic_utils.h"
  13
  14 using base::StringPiece;
  15 using base::StringPrintf;
  16 using std::string;
  17 using std::vector;
  18
  19 namespace net {
  20
  21 CryptoHandshakeMessage::CryptoHandshakeMessage()
  22     : tag_(0),
  23       minimum_size_(0) {}
  24
  25 CryptoHandshakeMessage::CryptoHandshakeMessage(
  26     const CryptoHandshakeMessage& other)
  27     : tag_(other.tag_),
  28       tag_value_map_(other.tag_value_map_),
  29       minimum_size_(other.minimum_size_) {
  30   // Don't copy serialized_. scoped_ptr doesn't have a copy constructor.
  31   // The new object can lazily reconstruct serialized_.
  32 }
  33
  34 CryptoHandshakeMessage::~CryptoHandshakeMessage() {}
  35
  36 CryptoHandshakeMessage& CryptoHandshakeMessage::operator=(
  37     const CryptoHandshakeMessage& other) {
  38   tag_ = other.tag_;
  39   tag_value_map_ = other.tag_value_map_;
  40   // Don't copy serialized_. scoped_ptr doesn't have an assignment operator.
  41   // However, invalidate serialized_.
  42   serialized_.reset();
  43   minimum_size_ = other.minimum_size_;
  44   return *this;
  45 }
  46
  47 void CryptoHandshakeMessage::Clear() {
  48   tag_ = 0;
  49   tag_value_map_.clear();
  50   minimum_size_ = 0;
  51   serialized_.reset();
  52 }
  53
  54 const QuicData& CryptoHandshakeMessage::GetSerialized() const {
  55   if (!serialized_.get()) {
  56     serialized_.reset(CryptoFramer::ConstructHandshakeMessage(*this));
  57   }
  58   return *serialized_;
  59 }
  60
  61 void CryptoHandshakeMessage::MarkDirty() {
  62   serialized_.reset();
  63 }
  64
  65 void CryptoHandshakeMessage::SetTaglist(QuicTag tag, ...) {
  66   // Warning, if sizeof(QuicTag) > sizeof(int) then this function will break
  67   // because the terminating 0 will only be promoted to int.
  68   static_assert(sizeof(QuicTag) <= sizeof(int),
  69                 "crypto tag may not be larger than int or varargs will break");
  70
  71   vector<QuicTag> tags;
  72   va_list ap;
  73
  74   va_start(ap, tag);
  75   for (;;) {
  76     QuicTag list_item = va_arg(ap, QuicTag);
  77     if (list_item == 0) {
  78       break;
  79     }
  80     tags.push_back(list_item);
  81   }
  82
  83   // Because of the way that we keep tags in memory, we can copy the contents
  84   // of the vector and get the correct bytes in wire format. See
  85   // crypto_protocol.h. This assumes that the system is little-endian.
  86   SetVector(tag, tags);
  87
  88   va_end(ap);
  89 }
  90
  91 void CryptoHandshakeMessage::SetStringPiece(QuicTag tag, StringPiece value) {
  92   tag_value_map_[tag] = value.as_string();
  93 }
  94
  95 void CryptoHandshakeMessage::Erase(QuicTag tag) {
  96   tag_value_map_.erase(tag);
  97 }
  98
  99 QuicErrorCode CryptoHandshakeMessage::GetTaglist(QuicTag tag,
 100                                                  const QuicTag** out_tags,
 101                                                  size_t* out_len) const {
 102   QuicTagValueMap::const_iterator it = tag_value_map_.find(tag);
 103   QuicErrorCode ret = QUIC_NO_ERROR;
 104
 105   if (it == tag_value_map_.end()) {
 106     ret = QUIC_CRYPTO_MESSAGE_PARAMETER_NOT_FOUND;
 107   } else if (it->second.size() % sizeof(QuicTag) != 0) {
 108     ret = QUIC_INVALID_CRYPTO_MESSAGE_PARAMETER;
 109   }
 110
 111   if (ret != QUIC_NO_ERROR) {
 112     *out_tags = nullptr;
 113     *out_len = 0;
 114     return ret;
 115   }
 116
 117   *out_tags = reinterpret_cast<const QuicTag*>(it->second.data());
 118   *out_len = it->second.size() / sizeof(QuicTag);
 119   return ret;
 120 }
 121
 122 bool CryptoHandshakeMessage::GetStringPiece(QuicTag tag,
 123                                             StringPiece* out) const {
 124   QuicTagValueMap::const_iterator it = tag_value_map_.find(tag);
 125   if (it == tag_value_map_.end()) {
 126     return false;
 127   }
 128   *out = it->second;
 129   return true;
 130 }
 131
 132 QuicErrorCode CryptoHandshakeMessage::GetNthValue24(QuicTag tag,
 133                                                     unsigned index,
 134                                                     StringPiece* out) const {
 135   StringPiece value;
 136   if (!GetStringPiece(tag, &value)) {
 137     return QUIC_CRYPTO_MESSAGE_PARAMETER_NOT_FOUND;
 138   }
 139
 140   for (unsigned i = 0;; i++) {
 141     if (value.empty()) {
 142       return QUIC_CRYPTO_MESSAGE_INDEX_NOT_FOUND;
 143     }
 144     if (value.size() < 3) {
 145       return QUIC_INVALID_CRYPTO_MESSAGE_PARAMETER;
 146     }
 147
 148     const unsigned char* data =
 149         reinterpret_cast<const unsigned char*>(value.data());
 150     size_t size = static_cast<size_t>(data[0]) |
 151                   (static_cast<size_t>(data[1]) << 8) |
 152                   (static_cast<size_t>(data[2]) << 16);
 153     value.remove_prefix(3);
 154
 155     if (value.size() < size) {
 156       return QUIC_INVALID_CRYPTO_MESSAGE_PARAMETER;
 157     }
 158
 159     if (i == index) {
 160       *out = StringPiece(value.data(), size);
 161       return QUIC_NO_ERROR;
 162     }
 163
 164     value.remove_prefix(size);
 165   }
 166 }
 167
 168 QuicErrorCode CryptoHandshakeMessage::GetUint32(QuicTag tag,
 169                                                 uint32* out) const {
 170   return GetPOD(tag, out, sizeof(uint32));
 171 }
 172
 173 QuicErrorCode CryptoHandshakeMessage::GetUint64(QuicTag tag,
 174                                                 uint64* out) const {
 175   return GetPOD(tag, out, sizeof(uint64));
 176 }
 177
 178 size_t CryptoHandshakeMessage::size() const {
 179   size_t ret = sizeof(QuicTag) +
 180                sizeof(uint16) /* number of entries */ +
 181                sizeof(uint16) /* padding */;
 182   ret += (sizeof(QuicTag) + sizeof(uint32) /* end offset */) *
 183          tag_value_map_.size();
 184   for (QuicTagValueMap::const_iterator i = tag_value_map_.begin();
 185        i != tag_value_map_.end(); ++i) {
 186     ret += i->second.size();
 187   }
 188
 189   return ret;
 190 }
 191
 192 void CryptoHandshakeMessage::set_minimum_size(size_t min_bytes) {
 193   if (min_bytes == minimum_size_) {
 194     return;
 195   }
 196   serialized_.reset();
 197   minimum_size_ = min_bytes;
 198 }
 199
 200 size_t CryptoHandshakeMessage::minimum_size() const {
 201   return minimum_size_;
 202 }
 203
 204 string CryptoHandshakeMessage::DebugString() const {
 205   return DebugStringInternal(0);
 206 }
 207
 208 QuicErrorCode CryptoHandshakeMessage::GetPOD(
 209     QuicTag tag, void* out, size_t len) const {
 210   QuicTagValueMap::const_iterator it = tag_value_map_.find(tag);
 211   QuicErrorCode ret = QUIC_NO_ERROR;
 212
 213   if (it == tag_value_map_.end()) {
 214     ret = QUIC_CRYPTO_MESSAGE_PARAMETER_NOT_FOUND;
 215   } else if (it->second.size() != len) {
 216     ret = QUIC_INVALID_CRYPTO_MESSAGE_PARAMETER;
 217   }
 218
 219   if (ret != QUIC_NO_ERROR) {
 220     memset(out, 0, len);
 221     return ret;
 222   }
 223
 224   memcpy(out, it->second.data(), len);
 225   return ret;
 226 }
 227
 228 string CryptoHandshakeMessage::DebugStringInternal(size_t indent) const {
 229   string ret = string(2 * indent, ' ') + QuicUtils::TagToString(tag_) + "<\n";
 230   ++indent;
 231   for (QuicTagValueMap::const_iterator it = tag_value_map_.begin();
 232        it != tag_value_map_.end(); ++it) {
 233     ret += string(2 * indent, ' ') + QuicUtils::TagToString(it->first) + ": ";
 234
 235     bool done = false;
 236     switch (it->first) {
 237       case kICSL:
 238       case kCFCW:
 239       case kSFCW:
 240       case kIRTT:
 241       case kMSPC:
 242       case kSRBF:
 243       case kSWND:
 244         // uint32 value
 245         if (it->second.size() == 4) {
 246           uint32 value;
 247           memcpy(&value, it->second.data(), sizeof(value));
 248           ret += base::UintToString(value);
 249           done = true;
 250         }
 251         break;
 252       case kKEXS:
 253       case kAEAD:
 254       case kCOPT:
 255       case kPDMD:
 256       case kVER:
 257         // tag lists
 258         if (it->second.size() % sizeof(QuicTag) == 0) {
 259           for (size_t j = 0; j < it->second.size(); j += sizeof(QuicTag)) {
 260             QuicTag tag;
 261             memcpy(&tag, it->second.data() + j, sizeof(tag));
 262             if (j > 0) {
 263               ret += ",";
 264             }
 265             ret += "'" + QuicUtils::TagToString(tag) + "'";
 266           }
 267           done = true;
 268         }
 269         break;
 270       case kCADR:
 271         // IP address and port
 272         if (!it->second.empty()) {
 273           QuicSocketAddressCoder decoder;
 274           if (decoder.Decode(it->second.data(), it->second.size())) {
 275             ret += IPAddressToStringWithPort(decoder.ip(), decoder.port());
 276             done = true;
 277           }
 278         }
 279         break;
 280       case kSCFG:
 281         // nested messages.
 282         if (!it->second.empty()) {
 283           scoped_ptr<CryptoHandshakeMessage> msg(
 284               CryptoFramer::ParseMessage(it->second));
 285           if (msg.get()) {
 286             ret += "\n";
 287             ret += msg->DebugStringInternal(indent + 1);
 288
 289             done = true;
 290           }
 291         }
 292         break;
 293       case kPAD:
 294         ret += StringPrintf("(%d bytes of padding)",
 295                             static_cast<int>(it->second.size()));
 296         done = true;
 297         break;
 298       case kSNI:
 299       case kUAID:
 300         ret += "\"" + it->second + "\"";
 301         done = true;
 302         break;
 303     }
 304
 305     if (!done) {
 306       // If there's no specific format for this tag, or the value is invalid,
 307       // then just use hex.
 308       ret += "0x" + base::HexEncode(it->second.data(), it->second.size());
 309     }
 310     ret += "\n";
 311   }
 312   --indent;
 313   ret += string(2 * indent, ' ') + ">";
 314   return ret;
 315 }
 316
 317 }  // namespace net