216 lines
9.2 KiB
C++
216 lines
9.2 KiB
C++
/*
|
|
* Copyright 2016 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef WEBRTC_API_QUICDATACHANNEL_H_
|
|
#define WEBRTC_API_QUICDATACHANNEL_H_
|
|
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
|
|
#include "webrtc/api/datachannelinterface.h"
|
|
#include "webrtc/base/asyncinvoker.h"
|
|
#include "webrtc/base/sigslot.h"
|
|
#include "webrtc/base/thread.h"
|
|
|
|
namespace cricket {
|
|
class QuicTransportChannel;
|
|
class ReliableQuicStream;
|
|
class TransportChannel;
|
|
} // namepsace cricket
|
|
|
|
namespace net {
|
|
// TODO(mikescarlett): Make this uint64_t once QUIC uses 64-bit ids.
|
|
typedef uint32_t QuicStreamId;
|
|
} // namespace net
|
|
|
|
namespace rtc {
|
|
class CopyOnWriteBuffer;
|
|
} // namespace rtc
|
|
|
|
namespace webrtc {
|
|
|
|
// Encodes a QUIC message header with the data channel ID and message ID, then
|
|
// stores the result in |header|.
|
|
void WriteQuicDataChannelMessageHeader(int data_channel_id,
|
|
uint64_t message_id,
|
|
rtc::CopyOnWriteBuffer* header);
|
|
|
|
// Decodes the data channel ID and message ID from the initial data received by
|
|
// an incoming QUIC stream. The data channel ID is output to |data_channel_id|,
|
|
// the message ID is output to |message_id|, and the number of bytes read is
|
|
// output to |bytes_read|. Returns false if either ID cannot be read.
|
|
bool ParseQuicDataMessageHeader(const char* data,
|
|
size_t len,
|
|
int* data_channel_id,
|
|
uint64_t* message_id,
|
|
size_t* bytes_read);
|
|
|
|
// QuicDataChannel is an implementation of DataChannelInterface based on the
|
|
// QUIC protocol. It uses a QuicTransportChannel to establish encryption and
|
|
// transfer data, and a QuicDataTransport to receive incoming messages at
|
|
// the correct data channel. Currently this class implements unordered, reliable
|
|
// delivery and does not send an "OPEN" message.
|
|
//
|
|
// Each time a message is sent:
|
|
//
|
|
// - The QuicDataChannel prepends it with the data channel id and message id.
|
|
// The QuicTransportChannel creates a ReliableQuicStream, then the
|
|
// ReliableQuicStream sends the message with a FIN.
|
|
//
|
|
// - The remote QuicSession creates a ReliableQuicStream to receive the data.
|
|
// The remote QuicDataTransport dispatches the ReliableQuicStream to the
|
|
// QuicDataChannel with the same id as this data channel.
|
|
//
|
|
// - The remote QuicDataChannel queues data from the ReliableQuicStream. Once
|
|
// it receives a QUIC stream frame with a FIN, it provides the message to the
|
|
// DataChannelObserver.
|
|
//
|
|
// TODO(mikescarlett): Implement ordered delivery, unreliable delivery, and
|
|
// an OPEN message similar to the one for SCTP.
|
|
class QuicDataChannel : public rtc::RefCountedObject<DataChannelInterface>,
|
|
public sigslot::has_slots<> {
|
|
public:
|
|
// Message stores buffered data from the incoming QUIC stream. The QUIC stream
|
|
// is provided so that remaining data can be received from the remote peer.
|
|
struct Message {
|
|
uint64_t id;
|
|
rtc::CopyOnWriteBuffer buffer;
|
|
cricket::ReliableQuicStream* stream;
|
|
};
|
|
|
|
QuicDataChannel(rtc::Thread* signaling_thread,
|
|
rtc::Thread* worker_thread,
|
|
const std::string& label,
|
|
const DataChannelInit& config);
|
|
~QuicDataChannel() override;
|
|
|
|
// DataChannelInterface overrides.
|
|
std::string label() const override { return label_; }
|
|
bool reliable() const override { return true; }
|
|
bool ordered() const override { return false; }
|
|
uint16_t maxRetransmitTime() const override { return -1; }
|
|
uint16_t maxRetransmits() const override { return -1; }
|
|
bool negotiated() const override { return false; }
|
|
int id() const override { return id_; }
|
|
DataState state() const override { return state_; }
|
|
uint64_t buffered_amount() const override { return buffered_amount_; }
|
|
std::string protocol() const override { return protocol_; }
|
|
void RegisterObserver(DataChannelObserver* observer) override;
|
|
void UnregisterObserver() override;
|
|
void Close() override;
|
|
bool Send(const DataBuffer& buffer) override;
|
|
|
|
// Called from QuicDataTransport to set the QUIC transport channel that the
|
|
// QuicDataChannel sends messages with. Returns false if a different QUIC
|
|
// transport channel is already set or |channel| is NULL.
|
|
//
|
|
// The QUIC transport channel is not set in the constructor to allow creating
|
|
// the QuicDataChannel before the PeerConnection has a QUIC transport channel,
|
|
// such as before the session description is not set.
|
|
bool SetTransportChannel(cricket::QuicTransportChannel* channel);
|
|
|
|
// Called from QuicDataTransport when an incoming ReliableQuicStream is
|
|
// receiving a message received for this data channel. Once this function is
|
|
// called, |message| is owned by the QuicDataChannel and should not be
|
|
// accessed by the QuicDataTransport.
|
|
void OnIncomingMessage(Message&& message);
|
|
|
|
// Methods for testing.
|
|
// Gets the number of outgoing QUIC streams with write blocked data that are
|
|
// currently open for this data channel and are not finished writing a
|
|
// message. This is equivalent to the size of |write_blocked_quic_streams_|.
|
|
size_t GetNumWriteBlockedStreams() const;
|
|
// Gets the number of incoming QUIC streams with buffered data that are
|
|
// currently open for this data channel and are not finished receiving a
|
|
// message. This is equivalent to the size of |incoming_quic_messages_|.
|
|
size_t GetNumIncomingStreams() const;
|
|
|
|
private:
|
|
// Callbacks from ReliableQuicStream.
|
|
// Called when an incoming QUIC stream in |incoming_quic_messages_| has
|
|
// received a QUIC stream frame.
|
|
void OnDataReceived(net::QuicStreamId stream_id,
|
|
const char* data,
|
|
size_t len);
|
|
// Called when a write blocked QUIC stream that has been added to
|
|
// |write_blocked_quic_streams_| is closed.
|
|
void OnWriteBlockedStreamClosed(net::QuicStreamId stream_id, int error);
|
|
// Called when an incoming QUIC stream that has been added to
|
|
// |incoming_quic_messages_| is closed.
|
|
void OnIncomingQueuedStreamClosed(net::QuicStreamId stream_id, int error);
|
|
// Called when a write blocked QUIC stream in |write_blocked_quic_streams_|
|
|
// has written previously queued data.
|
|
void OnQueuedBytesWritten(net::QuicStreamId stream_id,
|
|
uint64_t queued_bytes_written);
|
|
|
|
// Callbacks from |quic_transport_channel_|.
|
|
void OnReadyToSend(cricket::TransportChannel* channel);
|
|
void OnConnectionClosed();
|
|
|
|
// Worker thread methods.
|
|
// Sends the data buffer to the remote peer using an outgoing QUIC stream.
|
|
// Returns true if the data buffer can be successfully sent, or if it is
|
|
// queued to be sent later.
|
|
bool Send_w(const DataBuffer& buffer);
|
|
// Connects the |quic_transport_channel_| signals to this QuicDataChannel,
|
|
// then returns the new QuicDataChannel state.
|
|
DataState SetTransportChannel_w();
|
|
// Closes the QUIC streams associated with this QuicDataChannel.
|
|
void Close_w();
|
|
// Sets |buffered_amount_|.
|
|
void SetBufferedAmount_w(uint64_t buffered_amount);
|
|
|
|
// Signaling thread methods.
|
|
// Triggers QuicDataChannelObserver::OnMessage when a message from the remote
|
|
// peer is ready to be read.
|
|
void OnMessage_s(const DataBuffer& received_data);
|
|
// Triggers QuicDataChannel::OnStateChange if the state change is valid.
|
|
// Otherwise does nothing if |state| == |state_| or |state| != kClosed when
|
|
// the data channel is closing.
|
|
void SetState_s(DataState state);
|
|
// Triggers QuicDataChannelObserver::OnBufferedAmountChange when the total
|
|
// buffered data changes for a QUIC stream.
|
|
void OnBufferedAmountChange_s(uint64_t buffered_amount);
|
|
|
|
// QUIC transport channel which owns the QUIC session. It is used to create
|
|
// a QUIC stream for sending outgoing messages.
|
|
cricket::QuicTransportChannel* quic_transport_channel_ = nullptr;
|
|
// Signaling thread for DataChannelInterface methods.
|
|
rtc::Thread* const signaling_thread_;
|
|
// Worker thread for sending data and |quic_transport_channel_| callbacks.
|
|
rtc::Thread* const worker_thread_;
|
|
rtc::AsyncInvoker invoker_;
|
|
// Map of QUIC stream ID => ReliableQuicStream* for write blocked QUIC
|
|
// streams.
|
|
std::unordered_map<net::QuicStreamId, cricket::ReliableQuicStream*>
|
|
write_blocked_quic_streams_;
|
|
// Map of QUIC stream ID => Message for each incoming QUIC stream.
|
|
std::unordered_map<net::QuicStreamId, Message> incoming_quic_messages_;
|
|
// Handles received data from the remote peer and data channel state changes.
|
|
DataChannelObserver* observer_ = nullptr;
|
|
// QuicDataChannel ID.
|
|
int id_;
|
|
// Connectivity state of the QuicDataChannel.
|
|
DataState state_;
|
|
// Total bytes that are buffered among the QUIC streams.
|
|
uint64_t buffered_amount_;
|
|
// Counter for number of sent messages that is used for message IDs.
|
|
uint64_t next_message_id_;
|
|
|
|
// Variables for application use.
|
|
const std::string& label_;
|
|
const std::string& protocol_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_API_QUICDATACHANNEL_H_
|