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   // CryptoFramerVisitorInterface implementation
  33   virtual void OnHandshakeMessage(
  34       const CryptoHandshakeMessage& message) OVERRIDE;
  35
  36   // GetBase64SHA256ClientChannelID sets |*output| to the base64 encoded,
  37   // SHA-256 hash of the client's ChannelID key and returns true, if the client
  38   // presented a ChannelID. Otherwise it returns false.
  39   bool GetBase64SHA256ClientChannelID(std::string* output) const;
  40
  41  protected:
  42   virtual QuicErrorCode ProcessClientHello(
  43       const CryptoHandshakeMessage& message,
  44       const ValidateClientHelloResultCallback::Result& result,
  45       CryptoHandshakeMessage* reply,
  46       std::string* error_details);
  47
  48  private:
  49   friend class test::CryptoTestUtils;
  50
  51   class ValidateCallback : public ValidateClientHelloResultCallback {
  52    public:
  53     explicit ValidateCallback(QuicCryptoServerStream* parent);
  54     // To allow the parent to detach itself from the callback before deletion.
  55     void Cancel();
  56
  57     // From ValidateClientHelloResultCallback
  58     virtual void RunImpl(const CryptoHandshakeMessage& client_hello,
  59                          const Result& result) OVERRIDE;
  60
  61    private:
  62     QuicCryptoServerStream* parent_;
  63
  64     DISALLOW_COPY_AND_ASSIGN(ValidateCallback);
  65   };
  66
  67   // Invoked by ValidateCallback::RunImpl once initial validation of
  68   // the client hello is complete.  Finishes processing of the client
  69   // hello message and handles handshake success/failure.
  70   void FinishProcessingHandshakeMessage(
  71       const CryptoHandshakeMessage& message,
  72       const ValidateClientHelloResultCallback::Result& result);
  73
  74   // crypto_config_ contains crypto parameters for the handshake.
  75   const QuicCryptoServerConfig& crypto_config_;
  76
  77   // Pointer to the active callback that will receive the result of
  78   // the client hello validation request and forward it to
  79   // FinishProcessingHandshakeMessage for processing.  NULL if no
  80   // handshake message is being validated.
  81   ValidateCallback* validate_client_hello_cb_;
  82 };
  83
  84 }  // namespace net
  85
  86 #endif  // NET_QUIC_QUIC_CRYPTO_SERVER_STREAM_H_