| // 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 QUICHE_QUIC_TEST_TOOLS_LIMITED_MTU_TEST_WRITER_H_ |
| #define QUICHE_QUIC_TEST_TOOLS_LIMITED_MTU_TEST_WRITER_H_ |
| |
| #include "quiche/quic/core/quic_packet_writer_wrapper.h" |
| #include "quiche/quic/core/quic_packets.h" |
| |
| namespace quic { |
| namespace test { |
| |
| // Simulates a connection over a link with fixed MTU. Drops packets which |
| // exceed the MTU and passes the rest of them as-is. |
| class LimitedMtuTestWriter : public QuicPacketWriterWrapper { |
| public: |
| explicit LimitedMtuTestWriter(QuicByteCount mtu); |
| LimitedMtuTestWriter(const LimitedMtuTestWriter&) = delete; |
| LimitedMtuTestWriter& operator=(const LimitedMtuTestWriter&) = delete; |
| ~LimitedMtuTestWriter() override; |
| |
| // Inherited from QuicPacketWriterWrapper. |
| WriteResult WritePacket(const char* buffer, size_t buf_len, |
| const QuicIpAddress& self_address, |
| const QuicSocketAddress& peer_address, |
| PerPacketOptions* options) override; |
| |
| private: |
| QuicByteCount mtu_; |
| }; |
| |
| } // namespace test |
| } // namespace quic |
| |
| #endif // QUICHE_QUIC_TEST_TOOLS_LIMITED_MTU_TEST_WRITER_H_ |