1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/search/iframe_source.h"
6
7 #include "base/memory/ref_counted_memory.h"
8 #include "base/strings/string_piece.h"
9 #include "base/strings/string_util.h"
10 #include "chrome/browser/search/instant_io_context.h"
11 #include "chrome/common/url_constants.h"
12 #include "content/public/browser/render_view_host.h"
13 #include "content/public/browser/web_contents.h"
14 #include "grit/browser_resources.h"
15 #include "net/url_request/url_request.h"
16 #include "ui/base/resource/resource_bundle.h"
17 #include "url/gurl.h"
18
IframeSource()19 IframeSource::IframeSource() {
20 }
21
~IframeSource()22 IframeSource::~IframeSource() {
23 }
24
GetMimeType(const std::string & path_and_query) const25 std::string IframeSource::GetMimeType(
26 const std::string& path_and_query) const {
27 std::string path(GURL("chrome-search://host/" + path_and_query).path());
28 if (EndsWith(path, ".js", false))
29 return "application/javascript";
30 if (EndsWith(path, ".png", false))
31 return "image/png";
32 if (EndsWith(path, ".css", false))
33 return "text/css";
34 if (EndsWith(path, ".html", false))
35 return "text/html";
36 return "";
37 }
38
ShouldServiceRequest(const net::URLRequest * request) const39 bool IframeSource::ShouldServiceRequest(
40 const net::URLRequest* request) const {
41 const std::string& path = request->url().path();
42 return InstantIOContext::ShouldServiceRequest(request) &&
43 request->url().SchemeIs(chrome::kChromeSearchScheme) &&
44 request->url().host() == GetSource() &&
45 ServesPath(path);
46 }
47
ShouldDenyXFrameOptions() const48 bool IframeSource::ShouldDenyXFrameOptions() const {
49 return false;
50 }
51
GetOrigin(int render_process_id,int render_view_id,std::string * origin) const52 bool IframeSource::GetOrigin(
53 int render_process_id,
54 int render_view_id,
55 std::string* origin) const {
56 content::RenderViewHost* rvh =
57 content::RenderViewHost::FromID(render_process_id, render_view_id);
58 if (rvh == NULL)
59 return false;
60 content::WebContents* contents =
61 content::WebContents::FromRenderViewHost(rvh);
62 if (contents == NULL)
63 return false;
64 *origin = contents->GetURL().GetOrigin().spec();
65 // Origin should not include a trailing slash. That is part of the path.
66 base::TrimString(*origin, "/", origin);
67 return true;
68 }
69
SendResource(int resource_id,const content::URLDataSource::GotDataCallback & callback)70 void IframeSource::SendResource(
71 int resource_id,
72 const content::URLDataSource::GotDataCallback& callback) {
73 scoped_refptr<base::RefCountedStaticMemory> response(
74 ResourceBundle::GetSharedInstance().LoadDataResourceBytes(resource_id));
75 callback.Run(response.get());
76 }
77
SendJSWithOrigin(int resource_id,int render_process_id,int render_view_id,const content::URLDataSource::GotDataCallback & callback)78 void IframeSource::SendJSWithOrigin(
79 int resource_id,
80 int render_process_id,
81 int render_view_id,
82 const content::URLDataSource::GotDataCallback& callback) {
83 std::string origin;
84 if (!GetOrigin(render_process_id, render_view_id, &origin)) {
85 callback.Run(NULL);
86 return;
87 }
88
89 base::StringPiece template_js =
90 ResourceBundle::GetSharedInstance().GetRawDataResource(resource_id);
91 std::string response(template_js.as_string());
92 ReplaceFirstSubstringAfterOffset(&response, 0, "{{ORIGIN}}", origin);
93 callback.Run(base::RefCountedString::TakeString(&response));
94 }
95