Adds support for headless chrome embedder mojo services

headless/public/util/kv_map_request_job.cc

+// 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/kv_map_request_job.h"
+
+#include "base/threading/thread_task_runner_handle.h"
+#include "headless/public/util/kv_map_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 {
+
+KVMapRequestJob::KVMapRequestJob(net::URLRequest* request,
+                                 net::NetworkDelegate* network_delegate,
+                                 const KVMapProtocolHandler::Response* response)
+    : net::URLRequestJob(request, network_delegate),
+      response_(response),
+      data_offset_(0),
+      weak_factory_(this) {}
+
+KVMapRequestJob::~KVMapRequestJob() {}
+
+void KVMapRequestJob::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(&KVMapRequestJob::StartAsync, weak_factory_.GetWeakPtr()));
+}
+
+void KVMapRequestJob::StartAsync() {
+  if (!response_) {
+    NotifyStartError(net::URLRequestStatus(net::URLRequestStatus::FAILED,
+                                           net::ERR_FILE_NOT_FOUND));
+    return;
+  }
+  NotifyHeadersComplete();
+}
+
+void KVMapRequestJob::Kill() {
+  weak_factory_.InvalidateWeakPtrs();
+  URLRequestJob::Kill();
+}
+
+int KVMapRequestJob::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 KVMapRequestJob::GetMimeType(std::string* mime_type) const {
+  *mime_type = response_->mime_type;
+  return true;
+}
+
+}  // namespace headless