| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/tools/quic/quic_server_session.h" | 5 #include "net/tools/quic/quic_server_session.h" |
| 6 | 6 |
| 7 | |
| 8 #include "net/quic/crypto/quic_crypto_server_config.h" | 7 #include "net/quic/crypto/quic_crypto_server_config.h" |
| 9 #include "net/quic/crypto/quic_random.h" | 8 #include "net/quic/crypto/quic_random.h" |
| 10 #include "net/quic/quic_connection.h" | 9 #include "net/quic/quic_connection.h" |
| 11 #include "net/quic/test_tools/quic_connection_peer.h" | 10 #include "net/quic/test_tools/quic_connection_peer.h" |
| 11 #include "net/quic/test_tools/quic_data_stream_peer.h" |
| 12 #include "net/quic/test_tools/quic_test_utils.h" | 12 #include "net/quic/test_tools/quic_test_utils.h" |
| 13 #include "net/quic/test_tools/reliable_quic_stream_peer.h" | |
| 14 #include "net/tools/epoll_server/epoll_server.h" | 13 #include "net/tools/epoll_server/epoll_server.h" |
| 15 #include "net/tools/quic/quic_spdy_server_stream.h" | 14 #include "net/tools/quic/quic_spdy_server_stream.h" |
| 16 #include "net/tools/quic/test_tools/quic_test_utils.h" | 15 #include "net/tools/quic/test_tools/quic_test_utils.h" |
| 17 #include "testing/gmock/include/gmock/gmock.h" | 16 #include "testing/gmock/include/gmock/gmock.h" |
| 18 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 19 | 18 |
| 20 using __gnu_cxx::vector; | 19 using __gnu_cxx::vector; |
| 21 using net::test::MockConnection; | 20 using net::test::MockConnection; |
| 22 using net::test::QuicConnectionPeer; | 21 using net::test::QuicConnectionPeer; |
| 23 using net::test::ReliableQuicStreamPeer; | 22 using net::test::QuicDataStreamPeer; |
| 24 using testing::_; | 23 using testing::_; |
| 25 using testing::StrictMock; | 24 using testing::StrictMock; |
| 26 | 25 |
| 27 namespace net { | 26 namespace net { |
| 28 namespace tools { | 27 namespace tools { |
| 29 namespace test { | 28 namespace test { |
| 30 | 29 |
| 31 class QuicServerSessionPeer { | 30 class QuicServerSessionPeer { |
| 32 public: | 31 public: |
| 33 static ReliableQuicStream* GetIncomingReliableStream( | 32 static QuicDataStream* GetIncomingReliableStream( |
| 34 QuicServerSession* s, QuicStreamId id) { | 33 QuicServerSession* s, QuicStreamId id) { |
| 35 return s->GetIncomingReliableStream(id); | 34 return s->GetIncomingReliableStream(id); |
| 36 } | 35 } |
| 37 static ReliableQuicStream* GetStream(QuicServerSession* s, QuicStreamId id) { | 36 static QuicDataStream* GetDataStream(QuicServerSession* s, QuicStreamId id) { |
| 38 return s->GetStream(id); | 37 return s->GetDataStream(id); |
| 39 } | 38 } |
| 40 }; | 39 }; |
| 41 | 40 |
| 42 class CloseOnDataStream : public ReliableQuicStream { | 41 class CloseOnDataStream : public QuicDataStream { |
| 43 public: | 42 public: |
| 44 CloseOnDataStream(QuicStreamId id, QuicSession* session) | 43 CloseOnDataStream(QuicStreamId id, QuicSession* session) |
| 45 : ReliableQuicStream(id, session) { | 44 : QuicDataStream(id, session) { |
| 46 } | 45 } |
| 47 | 46 |
| 48 virtual bool OnStreamFrame(const QuicStreamFrame& frame) OVERRIDE { | 47 virtual bool OnStreamFrame(const QuicStreamFrame& frame) OVERRIDE { |
| 49 session()->MarkDecompressionBlocked(1, id()); | 48 session()->MarkDecompressionBlocked(1, id()); |
| 50 session()->CloseStream(id()); | 49 session()->CloseStream(id()); |
| 51 return true; | 50 return true; |
| 52 } | 51 } |
| 53 | 52 |
| 54 virtual uint32 ProcessData(const char* data, uint32 data_len) OVERRIDE { | 53 virtual uint32 ProcessData(const char* data, uint32 data_len) OVERRIDE { |
| 55 return 0; | 54 return 0; |
| 56 } | 55 } |
| 57 }; | 56 }; |
| 58 | 57 |
| 59 class TestQuicQuicServerSession : public QuicServerSession { | 58 class TestQuicQuicServerSession : public QuicServerSession { |
| 60 public: | 59 public: |
| 61 TestQuicQuicServerSession(const QuicConfig& config, | 60 TestQuicQuicServerSession(const QuicConfig& config, |
| 62 QuicConnection* connection, | 61 QuicConnection* connection, |
| 63 QuicSessionOwner* owner) | 62 QuicSessionOwner* owner) |
| 64 : QuicServerSession(config, connection, owner), | 63 : QuicServerSession(config, connection, owner), |
| 65 close_stream_on_data_(false) { | 64 close_stream_on_data_(false) { |
| 66 } | 65 } |
| 67 | 66 |
| 68 virtual ReliableQuicStream* CreateIncomingReliableStream( | 67 virtual QuicDataStream* CreateIncomingDataStream( |
| 69 QuicStreamId id) OVERRIDE { | 68 QuicStreamId id) OVERRIDE { |
| 70 if (!ShouldCreateIncomingReliableStream(id)) { | 69 if (!ShouldCreateIncomingDataStream(id)) { |
| 71 return NULL; | 70 return NULL; |
| 72 } | 71 } |
| 73 if (close_stream_on_data_) { | 72 if (close_stream_on_data_) { |
| 74 return new CloseOnDataStream(id, this); | 73 return new CloseOnDataStream(id, this); |
| 75 } else { | 74 } else { |
| 76 return new QuicSpdyServerStream(id, this); | 75 return new QuicSpdyServerStream(id, this); |
| 77 } | 76 } |
| 78 } | 77 } |
| 79 | 78 |
| 80 void CloseStreamOnData() { | 79 void CloseStreamOnData() { |
| (...skipping 16 matching lines...) Expand all Loading... |
| 97 config_.set_max_streams_per_connection(3, 3); | 96 config_.set_max_streams_per_connection(3, 3); |
| 98 | 97 |
| 99 connection_ = new MockConnection(guid_, IPEndPoint(), 0, &eps_, true); | 98 connection_ = new MockConnection(guid_, IPEndPoint(), 0, &eps_, true); |
| 100 session_.reset(new TestQuicQuicServerSession( | 99 session_.reset(new TestQuicQuicServerSession( |
| 101 config_, connection_, &owner_)); | 100 config_, connection_, &owner_)); |
| 102 session_->InitializeSession(crypto_config_); | 101 session_->InitializeSession(crypto_config_); |
| 103 visitor_ = QuicConnectionPeer::GetVisitor(connection_); | 102 visitor_ = QuicConnectionPeer::GetVisitor(connection_); |
| 104 } | 103 } |
| 105 | 104 |
| 106 void MarkHeadersReadForStream(QuicStreamId id) { | 105 void MarkHeadersReadForStream(QuicStreamId id) { |
| 107 ReliableQuicStream* stream = QuicServerSessionPeer::GetStream( | 106 QuicDataStream* stream = QuicServerSessionPeer::GetDataStream( |
| 108 session_.get(), id); | 107 session_.get(), id); |
| 109 ASSERT_TRUE(stream != NULL); | 108 ASSERT_TRUE(stream != NULL); |
| 110 ReliableQuicStreamPeer::SetHeadersDecompressed(stream, true); | 109 QuicDataStreamPeer::SetHeadersDecompressed(stream, true); |
| 111 } | 110 } |
| 112 | 111 |
| 113 QuicGuid guid_; | 112 QuicGuid guid_; |
| 114 EpollServer eps_; | 113 EpollServer eps_; |
| 115 StrictMock<MockQuicSessionOwner> owner_; | 114 StrictMock<MockQuicSessionOwner> owner_; |
| 116 MockConnection* connection_; | 115 MockConnection* connection_; |
| 117 QuicConfig config_; | 116 QuicConfig config_; |
| 118 QuicCryptoServerConfig crypto_config_; | 117 QuicCryptoServerConfig crypto_config_; |
| 119 scoped_ptr<TestQuicQuicServerSession> session_; | 118 scoped_ptr<TestQuicQuicServerSession> session_; |
| 120 QuicConnectionVisitorInterface* visitor_; | 119 QuicConnectionVisitorInterface* visitor_; |
| (...skipping 111 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 232 // Incoming streams on the server session must be odd. | 231 // Incoming streams on the server session must be odd. |
| 233 EXPECT_EQ(NULL, | 232 EXPECT_EQ(NULL, |
| 234 QuicServerSessionPeer::GetIncomingReliableStream( | 233 QuicServerSessionPeer::GetIncomingReliableStream( |
| 235 session_.get(), 2)); | 234 session_.get(), 2)); |
| 236 } | 235 } |
| 237 | 236 |
| 238 } // namespace | 237 } // namespace |
| 239 } // namespace test | 238 } // namespace test |
| 240 } // namespace tools | 239 } // namespace tools |
| 241 } // namespace net | 240 } // namespace net |
| OLD | NEW |