Index: headless/public/util/in_memory_request_job.cc |
diff --git a/headless/public/util/in_memory_request_job.cc b/headless/public/util/in_memory_request_job.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..5fa841519800e64819b218acbd7f5fd0a3422cf4 |
--- /dev/null |
+++ b/headless/public/util/in_memory_request_job.cc |
@@ -0,0 +1,65 @@ |
+// Copyright 2016 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 "headless/public/util/in_memory_request_job.h" |
+ |
+#include "base/threading/thread_task_runner_handle.h" |
+#include "headless/public/util/in_memory_protocol_handler.h" |
+#include "net/base/io_buffer.h" |
+#include "net/base/net_errors.h" |
+#include "net/url_request/url_request_status.h" |
+ |
+namespace headless { |
+ |
+InMemoryRequestJob::InMemoryRequestJob( |
+ net::URLRequest* request, |
+ net::NetworkDelegate* network_delegate, |
+ const InMemoryProtocolHandler::Response* response) |
+ : net::URLRequestJob(request, network_delegate), |
+ response_(response), |
+ data_offset_(0), |
+ weak_factory_(this) {} |
+ |
+InMemoryRequestJob::~InMemoryRequestJob() {} |
+ |
+void InMemoryRequestJob::Start() { |
+ // Start reading asynchronously so that all error reporting and data |
+ // callbacks happen as they would for network requests. |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
+ FROM_HERE, |
+ base::Bind(&InMemoryRequestJob::StartAsync, weak_factory_.GetWeakPtr())); |
+} |
+ |
+void InMemoryRequestJob::StartAsync() { |
+ if (!response_) { |
+ NotifyStartError(net::URLRequestStatus(net::URLRequestStatus::FAILED, |
+ net::ERR_FILE_NOT_FOUND)); |
+ return; |
+ } |
+ NotifyHeadersComplete(); |
+} |
+ |
+void InMemoryRequestJob::Kill() { |
+ weak_factory_.InvalidateWeakPtrs(); |
+ URLRequestJob::Kill(); |
+} |
+ |
+int InMemoryRequestJob::ReadRawData(net::IOBuffer* buf, int buf_size) { |
+ DCHECK(response_); |
+ DCHECK_LE(data_offset_, response_->data.size()); |
+ int remaining = |
+ base::checked_cast<int>(response_->data.size() - data_offset_); |
+ if (buf_size > remaining) |
+ buf_size = remaining; |
+ memcpy(buf->data(), response_->data.data() + data_offset_, buf_size); |
+ data_offset_ += buf_size; |
+ return buf_size; |
+} |
+ |
+bool InMemoryRequestJob::GetMimeType(std::string* mime_type) const { |
+ *mime_type = response_->mime_type; |
+ return true; |
+} |
+ |
+} // namespace headless |