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