mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-12-01 01:36:09 +03:00
59 lines
1.9 KiB
C
59 lines
1.9 KiB
C
|
// Copyright (c) 2017 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 NET_QUIC_TEST_TOOLS_SIMPLE_DATA_PRODUCER_H_
|
||
|
|
#define NET_QUIC_TEST_TOOLS_SIMPLE_DATA_PRODUCER_H_
|
||
|
|
|
||
|
|
#include "net/quic/core/quic_simple_buffer_allocator.h"
|
||
|
|
#include "net/quic/core/quic_stream_frame_data_producer.h"
|
||
|
|
#include "net/quic/core/quic_stream_send_buffer.h"
|
||
|
|
#include "net/quic/core/stream_notifier_interface.h"
|
||
|
|
#include "net/quic/platform/api/quic_containers.h"
|
||
|
|
|
||
|
|
namespace net {
|
||
|
|
|
||
|
|
namespace test {
|
||
|
|
|
||
|
|
// A simple data producer which copies stream data into a map from stream
|
||
|
|
// id to send buffer.
|
||
|
|
class SimpleDataProducer : public QuicStreamFrameDataProducer,
|
||
|
|
public StreamNotifierInterface {
|
||
|
|
public:
|
||
|
|
SimpleDataProducer();
|
||
|
|
~SimpleDataProducer() override;
|
||
|
|
|
||
|
|
void SaveStreamData(QuicStreamId id,
|
||
|
|
const struct iovec* iov,
|
||
|
|
int iov_count,
|
||
|
|
size_t iov_offset,
|
||
|
|
QuicStreamOffset offset,
|
||
|
|
QuicByteCount data_length);
|
||
|
|
|
||
|
|
// QuicStreamFrameDataProducer
|
||
|
|
bool WriteStreamData(QuicStreamId id,
|
||
|
|
QuicStreamOffset offset,
|
||
|
|
QuicByteCount data_length,
|
||
|
|
QuicDataWriter* writer) override;
|
||
|
|
|
||
|
|
// StreamNotifierInterface methods:
|
||
|
|
void OnStreamFrameAcked(const QuicStreamFrame& frame,
|
||
|
|
QuicTime::Delta ack_delay_time) override;
|
||
|
|
void OnStreamFrameRetransmitted(const QuicStreamFrame& frame) override {}
|
||
|
|
void OnStreamFrameDiscarded(const QuicStreamFrame& frame) override;
|
||
|
|
|
||
|
|
private:
|
||
|
|
using SendBufferMap =
|
||
|
|
QuicUnorderedMap<QuicStreamId, std::unique_ptr<QuicStreamSendBuffer>>;
|
||
|
|
|
||
|
|
SimpleBufferAllocator allocator_;
|
||
|
|
|
||
|
|
SendBufferMap send_buffer_map_;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace test
|
||
|
|
|
||
|
|
} // namespace net
|
||
|
|
|
||
|
|
#endif // NET_QUIC_TEST_TOOLS_SIMPLE_DATA_PRODUCER_H_
|