OLD | NEW |
| (Empty) |
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 QuicSession, which demuxes a single connection to individual streams. | |
6 | |
7 #ifndef NET_QUIC_QUIC_SESSION_H_ | |
8 #define NET_QUIC_QUIC_SESSION_H_ | |
9 | |
10 #include <vector> | |
11 | |
12 #include "base/compiler_specific.h" | |
13 #include "base/hash_tables.h" | |
14 #include "net/base/ip_endpoint.h" | |
15 #include "net/quic/quic_connection.h" | |
16 #include "net/quic/quic_crypto_stream.h" | |
17 #include "net/quic/quic_packet_creator.h" | |
18 #include "net/quic/quic_protocol.h" | |
19 #include "net/quic/reliable_quic_stream.h" | |
20 | |
21 namespace net { | |
22 | |
23 class NET_EXPORT_PRIVATE QuicSession : public QuicConnectionVisitorInterface { | |
24 public: | |
25 QuicSession(QuicConnection* connection, bool is_server); | |
26 | |
27 virtual ~QuicSession(); | |
28 | |
29 // QuicConnectionVisitorInterface methods: | |
30 virtual bool OnPacket(const IPEndPoint& self_address, | |
31 const IPEndPoint& peer_address, | |
32 const QuicPacketHeader& header, | |
33 const std::vector<QuicStreamFrame>& frame) OVERRIDE; | |
34 virtual void OnRstStream(const QuicRstStreamFrame& frame) OVERRIDE; | |
35 virtual void ConnectionClose(QuicErrorCode error, bool from_peer) OVERRIDE; | |
36 // Not needed for HTTP. | |
37 virtual void OnAck(AckedPackets acked_packets) OVERRIDE {} | |
38 | |
39 // Called by streams when they want to write data to the peer. | |
40 virtual int WriteData(QuicStreamId id, base::StringPiece data, | |
41 QuicStreamOffset offset, bool fin); | |
42 // Called by streams when they want to close the stream in both directions. | |
43 void SendRstStream(QuicStreamId id, | |
44 QuicErrorCode error, | |
45 QuicStreamOffset offset); | |
46 | |
47 // Removes the stream associated with 'stream_id' from the active stream map. | |
48 virtual void CloseStream(QuicStreamId stream_id); | |
49 | |
50 // Returns true once the crypto handshake is complete. | |
51 virtual bool IsHandshakeComplete(); | |
52 | |
53 QuicConnection* connection() { return connection_.get(); } | |
54 size_t num_active_requests() const { return stream_map_.size(); } | |
55 const IPEndPoint& peer_address() const { | |
56 return connection_->peer_address(); | |
57 } | |
58 | |
59 QuicPacketCreator::Options* options() { return connection()->options(); } | |
60 | |
61 // Returns the number of currently open streams, including those which have | |
62 // been implicitly created. | |
63 virtual size_t GetNumOpenStreams(); | |
64 | |
65 protected: | |
66 // Creates a new stream, owned by the caller, to handle a peer-initiated | |
67 // stream. Returns NULL if max streams have already been opened. | |
68 virtual ReliableQuicStream* CreateIncomingReliableStream(QuicStreamId id) = 0; | |
69 | |
70 // Create a new stream, owned by the caller, to handle a locally-initiated | |
71 // stream. Returns NULL if max streams have already been opened. | |
72 virtual ReliableQuicStream* CreateOutgoingReliableStream() = 0; | |
73 | |
74 // Return the reserved crypto stream. | |
75 virtual QuicCryptoStream* GetCryptoStream() = 0; | |
76 | |
77 // Adds 'stream' to the active stream map. | |
78 void ActivateStream(ReliableQuicStream* stream); | |
79 | |
80 // Returns the stream id for a new stream. | |
81 QuicStreamId GetNextStreamId(); | |
82 | |
83 // Returns true if the stream existed previously and has been closed. | |
84 bool IsClosedStream(QuicStreamId id); | |
85 | |
86 ReliableQuicStream* GetIncomingReliableStream(QuicStreamId stream_id); | |
87 | |
88 size_t get_max_open_streams() const { | |
89 return max_open_streams_; | |
90 } | |
91 | |
92 private: | |
93 friend class QuicSessionPeer; | |
94 | |
95 typedef base::hash_map<QuicStreamId, ReliableQuicStream*> ReliableStreamMap; | |
96 | |
97 ReliableQuicStream* GetStream(const QuicStreamId stream_id); | |
98 | |
99 scoped_ptr<QuicConnection> connection_; | |
100 | |
101 // Returns the maximum number of streams this connection can open. | |
102 const size_t max_open_streams_; | |
103 | |
104 // Map from StreamId to pointers to streams that are owned by the caller. | |
105 ReliableStreamMap stream_map_; | |
106 QuicStreamId next_stream_id_; | |
107 bool is_server_; | |
108 | |
109 // Set of stream ids that have been "implicitly created" by receipt | |
110 // of a stream id larger than the next expected stream id. | |
111 base::hash_set<QuicStreamId> implicitly_created_streams_; | |
112 QuicStreamId largest_peer_created_stream_id_; | |
113 }; | |
114 | |
115 } // namespace net | |
116 | |
117 #endif // NET_QUIC_QUIC_SESSION_H_ | |
OLD | NEW |