Add Resource Handler for creating Streams to forward to extensions
BUG=171585
Review URL: https://chromiumcodereview.appspot.com/12645004
git-svn-id: svn://svn.chromium.org/chrome/trunk/src@189172 0039d316-1c4b-4281-b951-d872f2087c98
diff --git a/content/browser/loader/stream_resource_handler.cc b/content/browser/loader/stream_resource_handler.cc
new file mode 100644
index 0000000..581b944
--- /dev/null
+++ b/content/browser/loader/stream_resource_handler.cc
@@ -0,0 +1,118 @@
+// 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 "content/browser/loader/stream_resource_handler.h"
+
+#include "base/guid.h"
+#include "base/logging.h"
+#include "content/browser/streams/stream.h"
+#include "content/browser/streams/stream_registry.h"
+#include "content/public/browser/resource_controller.h"
+#include "content/public/common/url_constants.h"
+#include "net/base/io_buffer.h"
+#include "net/url_request/url_request_status.h"
+
+namespace content {
+
+StreamResourceHandler::StreamResourceHandler(
+ net::URLRequest* request,
+ StreamRegistry* registry,
+ const GURL& security_origin)
+ : request_(request),
+ read_buffer_(NULL) {
+ // TODO(zork): Find a way to share this with the blob URL creation in WebKit.
+ GURL url(std::string(chrome::kBlobScheme) + ":" +
+ security_origin.spec() + base::GenerateGUID());
+ stream_ = new Stream(registry, this, security_origin, url);
+}
+
+StreamResourceHandler::~StreamResourceHandler() {
+ stream_->RemoveWriteObserver(this);
+}
+
+bool StreamResourceHandler::OnUploadProgress(int request_id,
+ uint64 position,
+ uint64 size) {
+ return true;
+}
+
+bool StreamResourceHandler::OnRequestRedirected(int request_id,
+ const GURL& url,
+ ResourceResponse* resp,
+ bool* defer) {
+ return true;
+}
+
+bool StreamResourceHandler::OnResponseStarted(int request_id,
+ ResourceResponse* resp,
+ bool* defer) {
+ return true;
+}
+
+bool StreamResourceHandler::OnWillStart(int request_id,
+ const GURL& url,
+ bool* defer) {
+ return true;
+}
+
+bool StreamResourceHandler::OnWillRead(int request_id,
+ net::IOBuffer** buf,
+ int* buf_size,
+ int min_size) {
+ static const int kReadBufSize = 32768;
+
+ DCHECK(buf && buf_size);
+ if (!read_buffer_)
+ read_buffer_ = new net::IOBuffer(kReadBufSize);
+ *buf = read_buffer_.get();
+ *buf_size = kReadBufSize;
+
+ return true;
+}
+
+bool StreamResourceHandler::OnReadCompleted(int request_id,
+ int bytes_read,
+ bool* defer) {
+ if (!bytes_read)
+ return true;
+
+ // We have more data to read.
+ DCHECK(read_buffer_);
+
+ // Release the ownership of the buffer, and store a reference
+ // to it. A new one will be allocated in OnWillRead().
+ net::IOBuffer* buffer = NULL;
+ read_buffer_.swap(&buffer);
+ stream_->AddData(buffer, bytes_read);
+
+ if (!stream_->can_add_data())
+ *defer = true;
+
+ return true;
+}
+
+bool StreamResourceHandler::OnResponseCompleted(
+ int request_id,
+ const net::URLRequestStatus& status,
+ const std::string& sec_info) {
+ stream_->Finalize();
+ return status.status() == net::URLRequestStatus::SUCCESS;
+}
+
+void StreamResourceHandler::OnDataDownloaded(
+ int request_id,
+ int bytes_downloaded) {
+ NOTREACHED();
+}
+
+void StreamResourceHandler::OnSpaceAvailable(Stream* stream) {
+ controller()->Resume();
+}
+
+void StreamResourceHandler::OnClose(Stream* stream) {
+ controller()->Cancel();
+}
+
+} // namespace content
+
