Landing Recent QUIC changes until 2015-11-09 20:32 UTC

net/tools/quic/test_tools/quic_test_server.h

 // Copyright (c) 2015 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.
 
 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_SERVER_H_
 #define NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_SERVER_H_
 
 #include <string>
 
 #include "base/basictypes.h"
@@ skipping 31 matching lines @@
   // Factory for creating QuicServerStreams.
   class StreamFactory {
    public:
     virtual ~StreamFactory() {}
 
     // Returns a new stream owned by the caller.
     virtual QuicSpdyServerStream* CreateStream(QuicStreamId id,
                                                QuicSpdySession* session) = 0;
   };
 
+  class CryptoStreamFactory {
+   public:
+    virtual ~CryptoStreamFactory() {}
+
+    // Returns a new QuicCryptoServerStreamBase owned by the caller
+    virtual QuicCryptoServerStreamBase* CreateCryptoStream(
+        const QuicCryptoServerConfig* crypto_config,
+        QuicSpdySession* session) = 0;
+  };
+
   explicit QuicTestServer(ProofSource* proof_source);
   QuicTestServer(ProofSource* proof_source,
                  const QuicConfig& config,
                  const QuicVersionVector& supported_versions);
 
   // Create a custom dispatcher which creates custom sessions.
   QuicDispatcher* CreateQuicDispatcher() override;
 
   // Sets a custom session factory, owned by the caller, for easy custom
-  // session logic. This is incompatible with setting a stream factory.
+  // session logic. This is incompatible with setting a stream factory or a
+  // crypto stream factory.
   void SetSessionFactory(SessionFactory* factory);
 
   // Sets a custom stream factory, owned by the caller, for easy custom
   // stream logic. This is incompatible with setting a session factory.
   void SetSpdyStreamFactory(StreamFactory* factory);
+
+  // Sets a custom crypto stream factory, owned by the caller, for easy custom
+  // crypto logic.  This is incompatible with setting a session factory.
+  void SetCryptoStreamFactory(CryptoStreamFactory* factory);
 };
 
 // Useful test sessions for the QuicTestServer.
 
 // Test session which sends a GOAWAY immedaitely on creation, before crypto
 // credentials have even been established.
 class ImmediateGoAwaySession : public QuicServerSession {
  public:
   ImmediateGoAwaySession(const QuicConfig& config,
                          QuicConnection* connection,
                          QuicServerSessionVisitor* visitor,
                          const QuicCryptoServerConfig* crypto_config);
 };
 
 }  // namespace test
 
 }  // namespace tools
 
 }  // namespace net
 
 #endif  // NET_TOOLS_QUIC_TEST_TOOLS_QUIC_TEST_SERVER_H_