net/tools/quic/quic_client_session.h

   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 // A client specific QuicSession subclass.
   6
   7 #ifndef NET_TOOLS_QUIC_QUIC_CLIENT_SESSION_H_
   8 #define NET_TOOLS_QUIC_QUIC_CLIENT_SESSION_H_
   9
  10 #include <string>
  11
  12 #include "base/basictypes.h"
  13 #include "net/quic/quic_client_session_base.h"
  14 #include "net/quic/quic_crypto_client_stream.h"
  15 #include "net/quic/quic_protocol.h"
  16 #include "net/tools/quic/quic_spdy_client_stream.h"
  17
  18 namespace net {
  19
  20 class QuicConnection;
  21 class QuicServerId;
  22 class ReliableQuicStream;
  23
  24 namespace tools {
  25
  26 class QuicClientSession : public QuicClientSessionBase {
  27  public:
  28   QuicClientSession(const QuicConfig& config, QuicConnection* connection);
  29   ~QuicClientSession() override;
  30
  31   void InitializeSession(const QuicServerId& server_id,
  32                          QuicCryptoClientConfig* config);
  33
  34   // QuicSession methods:
  35   QuicSpdyClientStream* CreateOutgoingDataStream() override;
  36   QuicCryptoClientStream* GetCryptoStream() override;
  37
  38   // QuicClientSessionBase methods:
  39   void OnProofValid(const QuicCryptoClientConfig::CachedState& cached) override;
  40   void OnProofVerifyDetailsAvailable(
  41       const ProofVerifyDetails& verify_details) override;
  42
  43   // Performs a crypto handshake with the server.
  44   void CryptoConnect();
  45
  46   // Returns the number of client hello messages that have been sent on the
  47   // crypto stream. If the handshake has completed then this is one greater
  48   // than the number of round-trips needed for the handshake.
  49   int GetNumSentClientHellos() const;
  50
  51   void set_respect_goaway(bool respect_goaway) {
  52     respect_goaway_ = respect_goaway;
  53   }
  54
  55  protected:
  56   // QuicSession methods:
  57   QuicDataStream* CreateIncomingDataStream(QuicStreamId id) override;
  58
  59  private:
  60   scoped_ptr<QuicCryptoClientStream> crypto_stream_;
  61
  62   // If this is set to false, the client will ignore server GOAWAYs and allow
  63   // the creation of streams regardless of the high chance they will fail.
  64   bool respect_goaway_;
  65
  66   DISALLOW_COPY_AND_ASSIGN(QuicClientSession);
  67 };
  68
  69 }  // namespace tools
  70 }  // namespace net
  71
  72 #endif  // NET_TOOLS_QUIC_QUIC_CLIENT_SESSION_H_