Add support for redirecting ResourceHandlers to a blob: URL

content/browser/streams/stream_registry.cc

+// 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/streams/stream_registry.h"
+
+#include <string>
+
+#include "base/guid.h"
+#include "base/stl_util.h"
+#include "base/string_number_conversions.h"
+#include "content/browser/streams/stream.h"
+#include "content/public/common/url_constants.h"
+
+namespace content {
+
+StreamRegistry::StreamRegistry() {
+}
+
+StreamRegistry::~StreamRegistry() {
+}
+
+void StreamRegistry::RegisterStream(scoped_refptr<Stream> stream) {
+  GURL url(std::string(chrome::kBlobScheme) + ":" +
+           stream->security_origin().spec() +
+           base::GenerateGUID());
+  stream->SetStreamUrl(url);
+  streams_[url] = stream;
+}
+
+scoped_refptr<Stream> StreamRegistry::GetStream(const GURL& url) {
+  std::map<GURL, scoped_refptr<Stream> >::iterator stream = streams_.find(url);
+  if (stream != streams_.end()) {
+    return stream->second;
+  } else {
+    return NULL;
+  }
+}
+
+void StreamRegistry::OnStreamConsumed(Stream* stream) {
+  streams_.erase(stream->stream_url());
+}
+
+}  // namespace content