| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/quic_session.h" | 5 #include "net/quic/quic_session.h" |
| 6 | 6 |
| 7 #include "base/stl_util.h" | 7 #include "base/stl_util.h" |
| 8 #include "net/quic/quic_connection.h" | 8 #include "net/quic/quic_connection.h" |
| 9 | 9 |
| 10 using base::StringPiece; | 10 using base::StringPiece; |
| 11 using base::hash_map; | 11 using base::hash_map; |
| 12 using base::hash_set; | 12 using base::hash_set; |
| 13 using std::vector; | 13 using std::vector; |
| 14 | 14 |
| 15 namespace net { | 15 namespace net { |
| 16 | 16 |
| 17 #define ENDPOINT (is_server_ ? "Server: " : " Client: ") | |
| 18 | |
| 19 // We want to make sure we delete any closed streams in a safe manner. | 17 // We want to make sure we delete any closed streams in a safe manner. |
| 20 // To avoid deleting a stream in mid-operation, we have a simple shim between | 18 // To avoid deleting a stream in mid-operation, we have a simple shim between |
| 21 // us and the stream, so we can delete any streams when we return from | 19 // us and the stream, so we can delete any streams when we return from |
| 22 // processing. | 20 // processing. |
| 23 // | 21 // |
| 24 // We could just override the base methods, but this makes it easier to make | 22 // We could just override the base methods, but this makes it easier to make |
| 25 // sure we don't miss any. | 23 // sure we don't miss any. |
| 26 class VisitorShim : public QuicConnectionVisitorInterface { | 24 class VisitorShim : public QuicConnectionVisitorInterface { |
| 27 public: | 25 public: |
| 28 explicit VisitorShim(QuicSession* session) : session_(session) {} | 26 explicit VisitorShim(QuicSession* session) : session_(session) {} |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 69 QuicSession::QuicSession(QuicConnection* connection, | 67 QuicSession::QuicSession(QuicConnection* connection, |
| 70 const QuicConfig& config, | 68 const QuicConfig& config, |
| 71 bool is_server) | 69 bool is_server) |
| 72 : connection_(connection), | 70 : connection_(connection), |
| 73 visitor_shim_(new VisitorShim(this)), | 71 visitor_shim_(new VisitorShim(this)), |
| 74 config_(config), | 72 config_(config), |
| 75 max_open_streams_(kDefaultMaxStreamsPerConnection), | 73 max_open_streams_(kDefaultMaxStreamsPerConnection), |
| 76 next_stream_id_(is_server ? 2 : 3), | 74 next_stream_id_(is_server ? 2 : 3), |
| 77 is_server_(is_server), | 75 is_server_(is_server), |
| 78 largest_peer_created_stream_id_(0), | 76 largest_peer_created_stream_id_(0), |
| 79 error_(QUIC_NO_ERROR), | |
| 80 goaway_received_(false), | 77 goaway_received_(false), |
| 81 goaway_sent_(false) { | 78 goaway_sent_(false) { |
| 82 set_max_open_streams(config_.max_streams_per_connection()); | 79 set_max_open_streams(config_.max_streams_per_connection()); |
| 83 | 80 |
| 84 connection_->set_visitor(visitor_shim_.get()); | 81 connection_->set_visitor(visitor_shim_.get()); |
| 85 connection_->SetIdleNetworkTimeout(config_.idle_connection_state_lifetime()); | 82 connection_->SetIdleNetworkTimeout(config_.idle_connection_state_lifetime()); |
| 86 connection_->SetOverallConnectionTimeout( | 83 connection_->SetOverallConnectionTimeout( |
| 87 config_.max_time_before_crypto_handshake()); | 84 config_.max_time_before_crypto_handshake()); |
| 88 // TODO(satyamshekhar): Set congestion control and ICSL also. | 85 // TODO(satyamshekhar): Set congestion control and ICSL also. |
| 89 } | 86 } |
| 90 | 87 |
| 91 QuicSession::~QuicSession() { | 88 QuicSession::~QuicSession() { |
| 92 STLDeleteElements(&closed_streams_); | 89 STLDeleteElements(&closed_streams_); |
| 93 STLDeleteValues(&stream_map_); | 90 STLDeleteValues(&stream_map_); |
| 94 } | 91 } |
| 95 | 92 |
| 96 bool QuicSession::OnPacket(const IPEndPoint& self_address, | 93 bool QuicSession::OnPacket(const IPEndPoint& self_address, |
| 97 const IPEndPoint& peer_address, | 94 const IPEndPoint& peer_address, |
| 98 const QuicPacketHeader& header, | 95 const QuicPacketHeader& header, |
| 99 const vector<QuicStreamFrame>& frames) { | 96 const vector<QuicStreamFrame>& frames) { |
| 100 if (header.public_header.guid != connection()->guid()) { | 97 if (header.public_header.guid != connection()->guid()) { |
| 101 DLOG(INFO) << ENDPOINT << "Got packet header for invalid GUID: " | 98 DLOG(INFO) << "Got packet header for invalid GUID: " |
| 102 << header.public_header.guid; | 99 << header.public_header.guid; |
| 103 return false; | 100 return false; |
| 104 } | 101 } |
| 105 for (size_t i = 0; i < frames.size(); ++i) { | 102 for (size_t i = 0; i < frames.size(); ++i) { |
| 106 // TODO(rch) deal with the error case of stream id 0 | 103 // TODO(rch) deal with the error case of stream id 0 |
| 107 if (IsClosedStream(frames[i].stream_id)) continue; | 104 if (IsClosedStream(frames[i].stream_id)) continue; |
| 108 | 105 |
| 109 ReliableQuicStream* stream = GetStream(frames[i].stream_id); | 106 ReliableQuicStream* stream = GetStream(frames[i].stream_id); |
| 110 if (stream == NULL) return false; | 107 if (stream == NULL) return false; |
| 111 if (!stream->WillAcceptStreamFrame(frames[i])) return false; | 108 if (!stream->WillAcceptStreamFrame(frames[i])) return false; |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 145 } | 142 } |
| 146 stream->OnStreamReset(frame.error_code); | 143 stream->OnStreamReset(frame.error_code); |
| 147 } | 144 } |
| 148 | 145 |
| 149 void QuicSession::OnGoAway(const QuicGoAwayFrame& frame) { | 146 void QuicSession::OnGoAway(const QuicGoAwayFrame& frame) { |
| 150 DCHECK(frame.last_good_stream_id < next_stream_id_); | 147 DCHECK(frame.last_good_stream_id < next_stream_id_); |
| 151 goaway_received_ = true; | 148 goaway_received_ = true; |
| 152 } | 149 } |
| 153 | 150 |
| 154 void QuicSession::ConnectionClose(QuicErrorCode error, bool from_peer) { | 151 void QuicSession::ConnectionClose(QuicErrorCode error, bool from_peer) { |
| 155 if (error_ == QUIC_NO_ERROR) { | |
| 156 error_ = error; | |
| 157 } | |
| 158 | |
| 159 while (stream_map_.size() != 0) { | 152 while (stream_map_.size() != 0) { |
| 160 ReliableStreamMap::iterator it = stream_map_.begin(); | 153 ReliableStreamMap::iterator it = stream_map_.begin(); |
| 161 QuicStreamId id = it->first; | 154 QuicStreamId id = it->first; |
| 162 it->second->ConnectionClose(error, from_peer); | 155 it->second->ConnectionClose(error, from_peer); |
| 163 // The stream should call CloseStream as part of ConnectionClose. | 156 // The stream should call CloseStream as part of ConnectionClose. |
| 164 if (stream_map_.find(id) != stream_map_.end()) { | 157 if (stream_map_.find(id) != stream_map_.end()) { |
| 165 LOG(DFATAL) << ENDPOINT << "Stream failed to close under ConnectionClose"; | 158 LOG(DFATAL) << "Stream failed to close under ConnectionClose"; |
| 166 CloseStream(id); | 159 CloseStream(id); |
| 167 } | 160 } |
| 168 } | 161 } |
| 169 } | 162 } |
| 170 | 163 |
| 171 bool QuicSession::OnCanWrite() { | 164 bool QuicSession::OnCanWrite() { |
| 172 // We latch this here rather than doing a traditional loop, because streams | 165 // We latch this here rather than doing a traditional loop, because streams |
| 173 // may be modifying the list as we loop. | 166 // may be modifying the list as we loop. |
| 174 int remaining_writes = write_blocked_streams_.NumObjects(); | 167 int remaining_writes = write_blocked_streams_.NumObjects(); |
| 175 | 168 |
| (...skipping 25 matching lines...) Expand all Loading... |
| 201 connection_->SendRstStream(id, error); | 194 connection_->SendRstStream(id, error); |
| 202 CloseStream(id); | 195 CloseStream(id); |
| 203 } | 196 } |
| 204 | 197 |
| 205 void QuicSession::SendGoAway(QuicErrorCode error_code, const string& reason) { | 198 void QuicSession::SendGoAway(QuicErrorCode error_code, const string& reason) { |
| 206 goaway_sent_ = true; | 199 goaway_sent_ = true; |
| 207 connection_->SendGoAway(error_code, largest_peer_created_stream_id_, reason); | 200 connection_->SendGoAway(error_code, largest_peer_created_stream_id_, reason); |
| 208 } | 201 } |
| 209 | 202 |
| 210 void QuicSession::CloseStream(QuicStreamId stream_id) { | 203 void QuicSession::CloseStream(QuicStreamId stream_id) { |
| 211 DLOG(INFO) << ENDPOINT << "Closing stream " << stream_id; | 204 DLOG(INFO) << "Closing stream " << stream_id; |
| 212 | 205 |
| 213 ReliableStreamMap::iterator it = stream_map_.find(stream_id); | 206 ReliableStreamMap::iterator it = stream_map_.find(stream_id); |
| 214 if (it == stream_map_.end()) { | 207 if (it == stream_map_.end()) { |
| 215 DLOG(INFO) << ENDPOINT << "Stream is already closed: " << stream_id; | 208 DLOG(INFO) << "Stream is already closed: " << stream_id; |
| 216 return; | 209 return; |
| 217 } | 210 } |
| 218 ReliableQuicStream* stream = it->second; | 211 ReliableQuicStream* stream = it->second; |
| 219 closed_streams_.push_back(it->second); | 212 closed_streams_.push_back(it->second); |
| 220 stream_map_.erase(it); | 213 stream_map_.erase(it); |
| 221 stream->OnClose(); | 214 stream->OnClose(); |
| 222 } | 215 } |
| 223 | 216 |
| 224 bool QuicSession::IsEncryptionEstablished() { | 217 bool QuicSession::IsEncryptionEstablished() { |
| 225 return GetCryptoStream()->encryption_established(); | 218 return GetCryptoStream()->encryption_established(); |
| (...skipping 11 matching lines...) Expand all Loading... |
| 237 break; | 230 break; |
| 238 | 231 |
| 239 case ENCRYPTION_REESTABLISHED: | 232 case ENCRYPTION_REESTABLISHED: |
| 240 // Retransmit originally packets that were sent, since they can't be | 233 // Retransmit originally packets that were sent, since they can't be |
| 241 // decrypted by the peer. | 234 // decrypted by the peer. |
| 242 connection_->RetransmitUnackedPackets( | 235 connection_->RetransmitUnackedPackets( |
| 243 QuicConnection::INITIAL_ENCRYPTION_ONLY); | 236 QuicConnection::INITIAL_ENCRYPTION_ONLY); |
| 244 break; | 237 break; |
| 245 | 238 |
| 246 case HANDSHAKE_CONFIRMED: | 239 case HANDSHAKE_CONFIRMED: |
| 247 LOG_IF(DFATAL, !config_.negotiated()) << ENDPOINT | 240 LOG_IF(DFATAL, !config_.negotiated()) |
| 248 << "Handshake confirmed without parameter negotiation."; | 241 << "Handshake confirmed without parameter negotiation."; |
| 249 connection_->SetIdleNetworkTimeout( | 242 connection_->SetIdleNetworkTimeout( |
| 250 config_.idle_connection_state_lifetime()); | 243 config_.idle_connection_state_lifetime()); |
| 251 connection_->SetOverallConnectionTimeout(QuicTime::Delta::Infinite()); | 244 connection_->SetOverallConnectionTimeout(QuicTime::Delta::Infinite()); |
| 252 max_open_streams_ = config_.max_streams_per_connection(); | 245 max_open_streams_ = config_.max_streams_per_connection(); |
| 253 break; | 246 break; |
| 254 | 247 |
| 255 default: | 248 default: |
| 256 LOG(ERROR) << ENDPOINT << "Got unknown handshake event: " << event; | 249 LOG(ERROR) << "Got unknown handshake event: " << event; |
| 257 } | 250 } |
| 258 } | 251 } |
| 259 | 252 |
| 260 QuicConfig* QuicSession::config() { | 253 QuicConfig* QuicSession::config() { |
| 261 return &config_; | 254 return &config_; |
| 262 } | 255 } |
| 263 | 256 |
| 264 void QuicSession::ActivateStream(ReliableQuicStream* stream) { | 257 void QuicSession::ActivateStream(ReliableQuicStream* stream) { |
| 265 DLOG(INFO) << ENDPOINT << "num_streams: " << stream_map_.size() | 258 DLOG(INFO) << "num_streams: " << stream_map_.size() |
| 266 << ". activating " << stream->id(); | 259 << ". activating " << stream->id(); |
| 267 DCHECK(stream_map_.count(stream->id()) == 0); | 260 DCHECK(stream_map_.count(stream->id()) == 0); |
| 268 stream_map_[stream->id()] = stream; | 261 stream_map_[stream->id()] = stream; |
| 269 } | 262 } |
| 270 | 263 |
| 271 QuicStreamId QuicSession::GetNextStreamId() { | 264 QuicStreamId QuicSession::GetNextStreamId() { |
| 272 QuicStreamId id = next_stream_id_; | 265 QuicStreamId id = next_stream_id_; |
| 273 next_stream_id_ += 2; | 266 next_stream_id_ += 2; |
| 274 return id; | 267 return id; |
| 275 } | 268 } |
| 276 | 269 |
| (...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 366 QuicStreamId stream_id) { | 359 QuicStreamId stream_id) { |
| 367 decompression_blocked_streams_[header_id] = stream_id; | 360 decompression_blocked_streams_[header_id] = stream_id; |
| 368 } | 361 } |
| 369 | 362 |
| 370 void QuicSession::PostProcessAfterData() { | 363 void QuicSession::PostProcessAfterData() { |
| 371 STLDeleteElements(&closed_streams_); | 364 STLDeleteElements(&closed_streams_); |
| 372 closed_streams_.clear(); | 365 closed_streams_.clear(); |
| 373 } | 366 } |
| 374 | 367 |
| 375 } // namespace net | 368 } // namespace net |
| OLD | NEW |