net/tools/quic/quic_server_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 server specific QuicSession subclass.
   6
   7 #ifndef NET_TOOLS_QUIC_QUIC_SERVER_SESSION_H_
   8 #define NET_TOOLS_QUIC_QUIC_SERVER_SESSION_H_
   9
  10 #include <set>
  11 #include <vector>
  12
  13 #include "base/containers/hash_tables.h"
  14 #include "base/memory/scoped_ptr.h"
  15 #include "net/quic/quic_crypto_server_stream.h"
  16 #include "net/quic/quic_protocol.h"
  17 #include "net/quic/quic_session.h"
  18
  19 namespace net {
  20
  21 class QuicConfig;
  22 class QuicConnection;
  23 class QuicCryptoServerConfig;
  24 class ReliableQuicStream;
  25
  26 namespace tools {
  27
  28 // An interface from the session to the entity owning the session.
  29 // This lets the session notify its owner (the Dispatcher) when the connection
  30 // is closed.
  31 class QuicSessionOwner {
  32  public:
  33   virtual ~QuicSessionOwner() {}
  34
  35   virtual void OnConnectionClose(QuicGuid guid, QuicErrorCode error) = 0;
  36 };
  37
  38 class QuicServerSession : public QuicSession {
  39  public:
  40   QuicServerSession(const QuicConfig& config,
  41                     QuicConnection *connection,
  42                     QuicSessionOwner* owner);
  43
  44   // Override the base class to notify the owner of the connection close.
  45   virtual void ConnectionClose(QuicErrorCode error, bool from_peer) OVERRIDE;
  46
  47   virtual ~QuicServerSession();
  48
  49   virtual void InitializeSession(const QuicCryptoServerConfig& crypto_config);
  50
  51   const QuicCryptoServerStream* crypto_stream() { return crypto_stream_.get(); }
  52
  53  protected:
  54   // QuicSession methods:
  55   virtual ReliableQuicStream* CreateIncomingReliableStream(
  56       QuicStreamId id) OVERRIDE;
  57   virtual ReliableQuicStream* CreateOutgoingReliableStream() OVERRIDE;
  58   virtual QuicCryptoServerStream* GetCryptoStream() OVERRIDE;
  59
  60   // If we should create an incoming stream, returns true. Otherwise
  61   // does error handling, including communicating the error to the client and
  62   // possibly closing the connection, and returns false.
  63   virtual bool ShouldCreateIncomingReliableStream(QuicStreamId id);
  64
  65   virtual QuicCryptoServerStream* CreateQuicCryptoServerStream(
  66     const QuicCryptoServerConfig& crypto_config);
  67
  68  private:
  69   scoped_ptr<QuicCryptoServerStream> crypto_stream_;
  70   QuicSessionOwner* owner_;
  71
  72   DISALLOW_COPY_AND_ASSIGN(QuicServerSession);
  73 };
  74
  75 }  // namespace tools
  76 }  // namespace net
  77
  78 #endif  // NET_TOOLS_QUIC_QUIC_SERVER_SESSION_H_