Initial PlzNavigate RDH-side logic.

content/browser/loader/navigation_resource_handler.cc

+// Copyright 2014 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 "content/browser/loader/navigation_resource_handler.h"
+
+#include "base/logging.h"
+#include "content/browser/loader/navigation_url_loader_core.h"
+#include "content/browser/loader/resource_request_info_impl.h"
+#include "content/browser/resource_context_impl.h"
+#include "content/browser/streams/stream.h"
+#include "content/browser/streams/stream_context.h"
+#include "content/public/browser/resource_controller.h"
+#include "content/public/browser/stream_handle.h"
+#include "content/public/common/resource_response.h"
+#include "net/base/net_errors.h"
+#include "net/url_request/url_request.h"
+
+namespace content {
+
+NavigationResourceHandler::NavigationResourceHandler(
+    net::URLRequest* request,
+    NavigationURLLoaderCore* core)
+    : ResourceHandler(request),
+      core_(core) {
+  core_->set_resource_handler(this);
+}
+
+NavigationResourceHandler::~NavigationResourceHandler() {
+  if (core_.get()) {
+    core_->RequestFailedOnIOThread(net::ERR_ABORTED);
+    core_->set_resource_handler(NULL);
+    core_ = NULL;
+  }
+}
+
+void NavigationResourceHandler::Cancel() {
+  controller()->Cancel();
+
+  if (core_.get()) {
+    // At this point, the core is detached from the UI-thread loader, so there
+    // is no use in signaling back in OnResponseCompleted.
+    core_->set_resource_handler(NULL);
+    core_ = NULL;
+  }
+}
+
+void NavigationResourceHandler::FollowRedirect() {
+  controller()->Resume();
+}
+
+void NavigationResourceHandler::SetController(ResourceController* controller) {
+  writer_.set_controller(controller);
+  ResourceHandler::SetController(controller);
+}
+
+bool NavigationResourceHandler::OnUploadProgress(uint64 position,
+                                                 uint64 size) {
+  return true;
+}
+
+bool NavigationResourceHandler::OnRequestRedirected(
+    const net::RedirectInfo& redirect_info,
+    ResourceResponse* response,
+    bool* defer) {
+  // TODO(davidben): Perform a CSP check here, and anything else that would have
+  // been done renderer-side.
+  core_->RequestRedirectedOnIOThread(redirect_info, response);
+  *defer = true;
+  return true;
+}
+
+bool NavigationResourceHandler::OnResponseStarted(ResourceResponse* response,
+                                                  bool* defer) {
+  DCHECK(core_.get());
+
+  ResourceRequestInfoImpl* info = GetRequestInfo();
+
+  // If the BufferedResourceHandler intercepted this request and converted it
+  // into a download, it will still call OnResponseStarted and immediately
+  // cancel. Ignore the call; OnReadCompleted will happen shortly.
+  //
+  // TODO(davidben): Move the dispatch out of BufferedResourceHandler. Perhaps
+  // all the way to the UI thread. Downloads, user certificates, etc., should be
+  // dispatched at the navigation layer.
+  if (info->IsDownload() || info->is_stream())
+    return true;
+
+  StreamContext* stream_context =
+      GetStreamContextForResourceContext(info->GetContext());
+  writer_.InitializeStream(stream_context->registry(),
+                           request()->url().GetOrigin());
+
+  // Detach from the loader; at this point, the request is now owned by the
+  // StreamHandle.
+  core_->ResponseStartedOnIOThread(response, writer_.stream()->CreateHandle());
+  core_->set_resource_handler(NULL);
+  core_ = NULL;
+  return true;
+}
+
+bool NavigationResourceHandler::OnWillStart(const GURL& url, bool* defer) {
+  return true;
+}
+
+bool NavigationResourceHandler::OnBeforeNetworkStart(const GURL& url,
+                                                     bool* defer) {
+  return true;
+}
+
+bool NavigationResourceHandler::OnWillRead(scoped_refptr<net::IOBuffer>* buf,
+                                           int* buf_size,
+                                           int min_size) {
+  writer_.OnWillRead(buf, buf_size, min_size);
+  return true;
+}
+
+bool NavigationResourceHandler::OnReadCompleted(int bytes_read, bool* defer) {
+  writer_.OnReadCompleted(bytes_read, defer);
+  return true;
+}
+
+void NavigationResourceHandler::OnResponseCompleted(
+    const net::URLRequestStatus& status,
+    const std::string& security_info,
+    bool* defer) {
+  // If the request has already committed, close the stream and leave it as-is.
+  //
+  // TODO(davidben): Is the final status of a navigation request meaningful?
+  // Perhaps we should eventually extend the streams interface to push the
+  // status down. (If there was an error partway through loading the top-level.)
+  if (writer_.stream()) {
+    writer_.Finalize();
+    return;
+  }
+
+  if (core_.get()) {
+    DCHECK_NE(net::OK, status.error());
+    core_->RequestFailedOnIOThread(status.error());
+    core_->set_resource_handler(NULL);
+    core_ = NULL;
+  }
+}
+
+void NavigationResourceHandler::OnDataDownloaded(int bytes_downloaded) {
+  NOTREACHED();
+}
+
+}  // namespace content