mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-24 14:26:09 +03:00
97 lines
2.5 KiB
C++
97 lines
2.5 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.
|
|
|
|
#include "net/quic/quartc/quartc_stream.h"
|
|
|
|
#include "net/quic/platform/api/quic_string_piece.h"
|
|
|
|
namespace net {
|
|
|
|
QuartcStream::QuartcStream(QuicStreamId id, QuicSession* session)
|
|
: QuicStream(id, session) {}
|
|
QuartcStream::~QuartcStream() {}
|
|
|
|
void QuartcStream::OnDataAvailable() {
|
|
struct iovec iov;
|
|
while (sequencer()->GetReadableRegions(&iov, 1) == 1) {
|
|
DCHECK(delegate_);
|
|
delegate_->OnReceived(this, reinterpret_cast<const char*>(iov.iov_base),
|
|
iov.iov_len);
|
|
sequencer()->MarkConsumed(iov.iov_len);
|
|
}
|
|
// All the data has been received if the sequencer is closed.
|
|
// Notify the delegate by calling the callback function one more time with
|
|
// iov_len = 0.
|
|
if (sequencer()->IsClosed()) {
|
|
OnFinRead();
|
|
delegate_->OnReceived(this, reinterpret_cast<const char*>(iov.iov_base), 0);
|
|
}
|
|
}
|
|
|
|
void QuartcStream::OnClose() {
|
|
QuicStream::OnClose();
|
|
DCHECK(delegate_);
|
|
delegate_->OnClose(this);
|
|
}
|
|
|
|
void QuartcStream::OnCanWrite() {
|
|
QuicStream::OnCanWrite();
|
|
DCHECK(delegate_);
|
|
// Don't call the delegate if the write-side is closed or a fin is buffered.
|
|
// It is already done with this stream.
|
|
if (!write_side_closed() && !fin_buffered()) {
|
|
delegate_->OnCanWrite(this);
|
|
}
|
|
}
|
|
|
|
uint32_t QuartcStream::stream_id() {
|
|
return id();
|
|
}
|
|
|
|
uint64_t QuartcStream::bytes_written() {
|
|
return stream_bytes_written();
|
|
}
|
|
|
|
bool QuartcStream::fin_sent() {
|
|
return QuicStream::fin_sent();
|
|
}
|
|
|
|
int QuartcStream::stream_error() {
|
|
return QuicStream::stream_error();
|
|
}
|
|
|
|
int QuartcStream::connection_error() {
|
|
return QuicStream::connection_error();
|
|
}
|
|
|
|
void QuartcStream::Write(const char* data,
|
|
size_t size,
|
|
const WriteParameters& param) {
|
|
struct iovec iov = {const_cast<char*>(data), size};
|
|
WritevData(&iov, 1, param.fin);
|
|
}
|
|
|
|
void QuartcStream::FinishWriting() {
|
|
WriteOrBufferData(QuicStringPiece(nullptr, 0), true, nullptr);
|
|
}
|
|
|
|
void QuartcStream::FinishReading() {
|
|
QuicStream::StopReading();
|
|
}
|
|
|
|
void QuartcStream::Close() {
|
|
QuicStream::session()->CloseStream(id());
|
|
}
|
|
|
|
void QuartcStream::SetDelegate(QuartcStreamInterface::Delegate* delegate) {
|
|
if (delegate_) {
|
|
LOG(WARNING) << "The delegate for Stream " << id()
|
|
<< " has already been set.";
|
|
}
|
|
delegate_ = delegate;
|
|
DCHECK(delegate_);
|
|
}
|
|
|
|
} // namespace net
|