Add the client address to the Public Reset packet.

net/quic/crypto/crypto_framer.cc

 // 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/quic/crypto/crypto_framer.h"
 
-#include "net/quic/crypto/crypto_handshake.h"
+#include "net/quic/crypto/crypto_protocol.h"
 #include "net/quic/quic_data_reader.h"
 #include "net/quic/quic_data_writer.h"
 
 using base::StringPiece;
 using std::make_pair;
 using std::pair;
 using std::vector;
 
 namespace net {
 
 namespace {
 
 const size_t kQuicTagSize = sizeof(uint32);
 const size_t kCryptoEndOffsetSize = sizeof(uint32);
 const size_t kNumEntriesSize = sizeof(uint16);
 
 // OneShotVisitor is a framer visitor that records a single handshake message.
 class OneShotVisitor : public CryptoFramerVisitorInterface {
  public:
   explicit OneShotVisitor(CryptoHandshakeMessage* out)
       : out_(out),
-        error_(false) {
+        error_(false),
+        done_(false) {
   }
 
   virtual void OnError(CryptoFramer* framer) OVERRIDE { error_ = true; }
 
   virtual void OnHandshakeMessage(
       const CryptoHandshakeMessage& message) OVERRIDE {
     *out_ = message;
+    done_ = true;
   }
 
   bool error() const { return error_; }
 
+  bool done() const { return done_; }
+
  private:
   CryptoHandshakeMessage* const out_;
   bool error_;
+  bool done_;
 };
 
 }  // namespace
 
 CryptoFramer::CryptoFramer()
     : visitor_(NULL),
       num_entries_(0),
       values_len_(0) {
   Clear();
 }
 
 CryptoFramer::~CryptoFramer() {}
 
 // static
 CryptoHandshakeMessage* CryptoFramer::ParseMessage(StringPiece in) {
   scoped_ptr<CryptoHandshakeMessage> msg(new CryptoHandshakeMessage);
   OneShotVisitor visitor(msg.get());
   CryptoFramer framer;
 
   framer.set_visitor(&visitor);
-  if (!framer.ProcessInput(in) || visitor.error() ||
+  if (!framer.ProcessInput(in) || visitor.error() || !visitor.done() ||
       framer.InputBytesRemaining()) {
     return NULL;
   }
 
   return msg.release();
 }
 
 bool CryptoFramer::ProcessInput(StringPiece input) {
   DCHECK_EQ(QUIC_NO_ERROR, error_);
   if (error_ != QUIC_NO_ERROR) {
@@ skipping 208 matching lines @@
   }
   *end_offset += pad_length;
   if (!writer->WriteUInt32(*end_offset)) {
     DCHECK(false) << "Failed to write end offset.";
     return false;
   }
   return true;
 }
 
 }  // namespace net