src/net/quic/quic_crypto_server_stream.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 #ifndef NET_QUIC_QUIC_CRYPTO_SERVER_STREAM_H_
   6 #define NET_QUIC_QUIC_CRYPTO_SERVER_STREAM_H_
   7
   8 #include <string>
   9
  10 #include "net/quic/crypto/crypto_handshake.h"
  11 #include "net/quic/crypto/quic_crypto_server_config.h"
  12 #include "net/quic/quic_config.h"
  13 #include "net/quic/quic_crypto_stream.h"
  14
  15 namespace net {
  16
  17 class CryptoHandshakeMessage;
  18 class QuicCryptoServerConfig;
  19 class QuicSession;
  20
  21 namespace test {
  22 class CryptoTestUtils;
  23 }  // namespace test
  24
  25 class NET_EXPORT_PRIVATE QuicCryptoServerStream : public QuicCryptoStream {
  26  public:
  27   QuicCryptoServerStream(const QuicCryptoServerConfig& crypto_config,
  28                          QuicSession* session);
  29   explicit QuicCryptoServerStream(QuicSession* session);
  30   virtual ~QuicCryptoServerStream();
  31
  32   // Cancel any outstanding callbacks, such as asynchronous validation of client
  33   // hello.
  34   void CancelOutstandingCallbacks();
  35
  36   // CryptoFramerVisitorInterface implementation
  37   virtual void OnHandshakeMessage(
  38       const CryptoHandshakeMessage& message) OVERRIDE;
  39
  40   // GetBase64SHA256ClientChannelID sets |*output| to the base64 encoded,
  41   // SHA-256 hash of the client's ChannelID key and returns true, if the client
  42   // presented a ChannelID. Otherwise it returns false.
  43   bool GetBase64SHA256ClientChannelID(std::string* output) const;
  44
  45   uint8 num_handshake_messages() const { return num_handshake_messages_; }
  46
  47  protected:
  48   virtual QuicErrorCode ProcessClientHello(
  49       const CryptoHandshakeMessage& message,
  50       const ValidateClientHelloResultCallback::Result& result,
  51       CryptoHandshakeMessage* reply,
  52       std::string* error_details);
  53
  54  private:
  55   friend class test::CryptoTestUtils;
  56
  57   class ValidateCallback : public ValidateClientHelloResultCallback {
  58    public:
  59     explicit ValidateCallback(QuicCryptoServerStream* parent);
  60     // To allow the parent to detach itself from the callback before deletion.
  61     void Cancel();
  62
  63     // From ValidateClientHelloResultCallback
  64     virtual void RunImpl(const CryptoHandshakeMessage& client_hello,
  65                          const Result& result) OVERRIDE;
  66
  67    private:
  68     QuicCryptoServerStream* parent_;
  69
  70     DISALLOW_COPY_AND_ASSIGN(ValidateCallback);
  71   };
  72
  73   // Invoked by ValidateCallback::RunImpl once initial validation of
  74   // the client hello is complete.  Finishes processing of the client
  75   // hello message and handles handshake success/failure.
  76   void FinishProcessingHandshakeMessage(
  77       const CryptoHandshakeMessage& message,
  78       const ValidateClientHelloResultCallback::Result& result);
  79
  80   // crypto_config_ contains crypto parameters for the handshake.
  81   const QuicCryptoServerConfig& crypto_config_;
  82
  83   // Pointer to the active callback that will receive the result of
  84   // the client hello validation request and forward it to
  85   // FinishProcessingHandshakeMessage for processing.  NULL if no
  86   // handshake message is being validated.
  87   ValidateCallback* validate_client_hello_cb_;
  88
  89   uint8 num_handshake_messages_;
  90
  91   DISALLOW_COPY_AND_ASSIGN(QuicCryptoServerStream);
  92 };
  93
  94 }  // namespace net
  95
  96 #endif  // NET_QUIC_QUIC_CRYPTO_SERVER_STREAM_H_