mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-12-03 02:36:09 +03:00
60 lines
1.5 KiB
C++
60 lines
1.5 KiB
C++
// Copyright (c) 2013 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/spdy/spdy_read_queue.h"
|
|
|
|
#include <algorithm>
|
|
#include <utility>
|
|
|
|
#include "base/logging.h"
|
|
#include "net/spdy/spdy_buffer.h"
|
|
|
|
namespace net {
|
|
|
|
SpdyReadQueue::SpdyReadQueue() : total_size_(0) {}
|
|
|
|
SpdyReadQueue::~SpdyReadQueue() {
|
|
Clear();
|
|
}
|
|
|
|
bool SpdyReadQueue::IsEmpty() const {
|
|
DCHECK_EQ(queue_.empty(), total_size_ == 0);
|
|
return queue_.empty();
|
|
}
|
|
|
|
size_t SpdyReadQueue::GetTotalSize() const {
|
|
return total_size_;
|
|
}
|
|
|
|
void SpdyReadQueue::Enqueue(std::unique_ptr<SpdyBuffer> buffer) {
|
|
DCHECK_GT(buffer->GetRemainingSize(), 0u);
|
|
total_size_ += buffer->GetRemainingSize();
|
|
queue_.push_back(std::move(buffer));
|
|
}
|
|
|
|
size_t SpdyReadQueue::Dequeue(char* out, size_t len) {
|
|
DCHECK_GT(len, 0u);
|
|
size_t bytes_copied = 0;
|
|
while (!queue_.empty() && bytes_copied < len) {
|
|
SpdyBuffer* buffer = queue_.front().get();
|
|
size_t bytes_to_copy =
|
|
std::min(len - bytes_copied, buffer->GetRemainingSize());
|
|
memcpy(out + bytes_copied, buffer->GetRemainingData(), bytes_to_copy);
|
|
bytes_copied += bytes_to_copy;
|
|
if (bytes_to_copy == buffer->GetRemainingSize())
|
|
queue_.pop_front();
|
|
else
|
|
buffer->Consume(bytes_to_copy);
|
|
}
|
|
total_size_ -= bytes_copied;
|
|
return bytes_copied;
|
|
}
|
|
|
|
void SpdyReadQueue::Clear() {
|
|
queue_.clear();
|
|
total_size_ = 0;
|
|
}
|
|
|
|
} // namespace net
|