Add Resource Handler for creating Streams to forward to extensions

content/browser/loader/buffered_resource_handler.cc

 // Copyright (c) 2012 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/buffered_resource_handler.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/metrics/histogram.h"
 #include "base/string_util.h"
 #include "content/browser/download/download_resource_handler.h"
 #include "content/browser/download/download_stats.h"
 #include "content/browser/loader/certificate_resource_handler.h"
 #include "content/browser/loader/resource_dispatcher_host_impl.h"
 #include "content/browser/loader/resource_request_info_impl.h"
+#include "content/browser/loader/stream_resource_handler.h"
 #include "content/browser/plugin_service_impl.h"
+#include "content/browser/resource_context_impl.h"
+#include "content/browser/streams/stream.h"
+#include "content/browser/streams/stream_context.h"
+#include "content/browser/streams/stream_registry.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/download_id.h"
 #include "content/public/browser/download_save_info.h"
 #include "content/public/browser/resource_context.h"
 #include "content/public/browser/resource_dispatcher_host_delegate.h"
 #include "content/public/common/resource_response.h"
 #include "net/base/io_buffer.h"
 #include "net/base/mime_sniffer.h"
 #include "net/base/mime_util.h"
 #include "net/base/net_errors.h"
@@ skipping 284 matching lines @@
   }
 
   if (!info->allow_download())
     return true;
 
   bool must_download = MustDownload();
   if (!must_download) {
     if (net::IsSupportedMimeType(mime_type))
       return true;
 
+    GURL security_origin;
+    if (host_->delegate() &&
+        host_->delegate()->ShouldCreateStream(info->GetContext(),
+                                              request_->url(),
+                                              mime_type,
+                                              &security_origin)) {
+      StreamContext* stream_context =
+          GetStreamContextForResourceContext(info->GetContext());
+
+      scoped_ptr<StreamResourceHandler> handler(
+          new StreamResourceHandler(request_,
+                                    stream_context->registry(),
+                                    security_origin));
+
+      info->set_is_stream(true);
+      host_->delegate()->OnStreamCreated(
+          info->GetContext(),
+          info->GetChildID(),
+          info->GetRouteID(),
+          handler->stream()->url(),
+          mime_type,
+          request_->url());
+      return UseAlternateNextHandler(
+          (scoped_ptr<ResourceHandler>(handler.release())).Pass());
+    }
+
 #if defined(ENABLE_PLUGINS)
     bool stale;
     bool has_plugin = HasSupportingPlugin(&stale);
     if (stale) {
       // Refresh the plugins asynchronously.
       PluginServiceImpl::GetInstance()->GetPlugins(
           base::Bind(&BufferedResourceHandler::OnPluginsLoaded,
                      weak_ptr_factory_.GetWeakPtr()));
       *defer = true;
       return true;
@@ skipping 125 matching lines @@
     const std::vector<webkit::WebPluginInfo>& plugins) {
   bool defer = false;
   if (!ProcessResponse(&defer)) {
     controller()->Cancel();
   } else if (!defer) {
     controller()->Resume();
   }
 }
 
 }  // namespace content