OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/renderer/prerender/prerender_dispatcher.h" | 5 #include "chrome/renderer/prerender/prerender_dispatcher.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "chrome/common/prerender_messages.h" | 8 #include "chrome/common/prerender_messages.h" |
9 #include "chrome/renderer/prerender/prerender_extra_data.h" | 9 #include "chrome/renderer/prerender/prerender_extra_data.h" |
| 10 #include "content/public/common/referrer.h" |
10 #include "content/public/renderer/render_thread.h" | 11 #include "content/public/renderer/render_thread.h" |
| 12 #include "content/public/renderer/render_view.h" |
11 #include "googleurl/src/gurl.h" | 13 #include "googleurl/src/gurl.h" |
| 14 #include "third_party/WebKit/Source/Platform/chromium/public/WebString.h" |
| 15 #include "third_party/WebKit/Source/Platform/chromium/public/WebURL.h" |
12 #include "third_party/WebKit/Source/Platform/chromium/public/WebPrerenderingSupp
ort.h" | 16 #include "third_party/WebKit/Source/Platform/chromium/public/WebPrerenderingSupp
ort.h" |
13 | 17 |
14 namespace prerender { | 18 namespace prerender { |
15 | 19 |
| 20 using WebKit::WebPrerender; |
| 21 using WebKit::WebPrerenderingSupport; |
| 22 |
16 PrerenderDispatcher::PrerenderDispatcher() { | 23 PrerenderDispatcher::PrerenderDispatcher() { |
17 WebKit::WebPrerenderingSupport::initialize(this); | 24 WebPrerenderingSupport::initialize(this); |
18 } | 25 } |
19 | 26 |
20 PrerenderDispatcher::~PrerenderDispatcher() { | 27 PrerenderDispatcher::~PrerenderDispatcher() { |
21 WebKit::WebPrerenderingSupport::shutdown(); | 28 WebPrerenderingSupport::shutdown(); |
22 } | 29 } |
23 | 30 |
24 bool PrerenderDispatcher::IsPrerenderURL(const GURL& url) const { | 31 bool PrerenderDispatcher::IsPrerenderURL(const GURL& url) const { |
25 return prerender_urls_.find(url) != prerender_urls_.end(); | 32 for (std::map<int, WebKit::WebPrerender*>::const_iterator |
26 } | 33 it = prerenders_.begin(); |
27 | 34 it != prerenders_.end(); ++it) { |
28 void PrerenderDispatcher::OnAddPrerenderURL(const GURL& url) { | 35 if (it->second->url() == url) |
29 PrerenderMap::iterator it = prerender_urls_.find(url); | 36 return true; |
30 if (it != prerender_urls_.end()) { | |
31 DCHECK_LT(0, it->second); | |
32 it->second += 1; | |
33 } else { | |
34 prerender_urls_[url] = 1; | |
35 } | 37 } |
36 } | 38 for (std::multimap<int, GURL>::const_iterator it = prerender_aliases_.begin(); |
37 | 39 it != prerender_aliases_.end(); ++it) { |
38 void PrerenderDispatcher::OnRemovePrerenderURL(const GURL& url) { | 40 if (it->second == url) |
39 PrerenderMap::iterator it = prerender_urls_.find(url); | 41 return true; |
40 // This is possible with a spurious remove. | 42 } |
41 // TODO(cbentzel): We'd also want to send the map of active prerenders when | 43 return false; |
42 // creating a new render process, so the Add/Remove go relative to that. | |
43 // This may not be that big of a deal in practice, since the newly created tab | |
44 // is unlikely to go to the prerendered page. | |
45 if (it == prerender_urls_.end()) | |
46 return; | |
47 DCHECK_LT(0, it->second); | |
48 it->second -= 1; | |
49 if (it->second == 0) | |
50 prerender_urls_.erase(it); | |
51 } | 44 } |
52 | 45 |
53 bool PrerenderDispatcher::OnControlMessageReceived( | 46 bool PrerenderDispatcher::OnControlMessageReceived( |
54 const IPC::Message& message) { | 47 const IPC::Message& message) { |
55 bool handled = true; | 48 bool handled = true; |
56 IPC_BEGIN_MESSAGE_MAP(PrerenderDispatcher, message) | 49 IPC_BEGIN_MESSAGE_MAP(PrerenderDispatcher, message) |
57 IPC_MESSAGE_HANDLER(PrerenderMsg_AddPrerenderURL, OnAddPrerenderURL) | 50 IPC_MESSAGE_HANDLER(PrerenderMsg_StartedPrerender, OnStartedPrerender) |
58 IPC_MESSAGE_HANDLER(PrerenderMsg_RemovePrerenderURL, OnRemovePrerenderURL) | 51 IPC_MESSAGE_HANDLER(PrerenderMsg_AddAliasToPrerender, OnAddAliasToPrerender) |
| 52 IPC_MESSAGE_HANDLER(PrerenderMsg_StoppedPrerender, OnStoppedPrerender) |
59 IPC_MESSAGE_UNHANDLED(handled = false) | 53 IPC_MESSAGE_UNHANDLED(handled = false) |
60 IPC_END_MESSAGE_MAP() | 54 IPC_END_MESSAGE_MAP() |
61 | 55 |
62 return handled; | 56 return handled; |
63 } | 57 } |
64 | 58 |
65 void PrerenderDispatcher::add(const WebKit::WebPrerender& prerender) { | 59 void PrerenderDispatcher::OnStartedPrerender(int prerender_id) { |
| 60 running_prerenders_.insert(prerender_id); |
| 61 |
| 62 std::map<int, WebPrerender*>::iterator it = prerenders_.find(prerender_id); |
| 63 if (it != prerenders_.end()) |
| 64 it->second->didStartPrerender(); |
| 65 } |
| 66 |
| 67 void PrerenderDispatcher::OnAddAliasToPrerender(int prerender_id, |
| 68 const GURL& url) { |
| 69 prerender_aliases_.insert( |
| 70 std::multimap<int, GURL>::value_type(prerender_id, url)); |
| 71 } |
| 72 |
| 73 void PrerenderDispatcher::OnStoppedPrerender(int prerender_id) { |
| 74 // It's possible we will receive a spurious remove, so we don't crash in that |
| 75 // situation. |
| 76 |
| 77 // TODO(cbentzel): We'd also want to send the map of active prerenders when |
| 78 // creating a new render process, so the Add/Remove go relative to that. |
| 79 // This may not be that big of a deal in practice, since the newly created tab |
| 80 // is unlikely to go to the prerendered page. |
| 81 |
| 82 running_prerenders_.erase(prerender_id); |
| 83 prerender_aliases_.erase(prerender_id); |
| 84 |
| 85 std::map<int, WebPrerender*>::iterator it = prerenders_.find(prerender_id); |
| 86 if (it != prerenders_.end()) |
| 87 it->second->didStopPrerender(); |
| 88 } |
| 89 |
| 90 void PrerenderDispatcher::add(WebPrerender* prerender) { |
66 const PrerenderExtraData& extra_data = | 91 const PrerenderExtraData& extra_data = |
67 PrerenderExtraData::FromPrerender(prerender); | 92 PrerenderExtraData::FromPrerender(*prerender); |
| 93 prerenders_.insert(std::map<int, WebPrerender*>::value_type( |
| 94 extra_data.prerender_id(), prerender)); |
68 content::RenderThread::Get()->Send(new PrerenderHostMsg_AddLinkRelPrerender( | 95 content::RenderThread::Get()->Send(new PrerenderHostMsg_AddLinkRelPrerender( |
69 extra_data.prerender_id(), GURL(prerender.url()), | 96 extra_data.prerender_id(), GURL(prerender->url()), |
70 content::Referrer(GURL(prerender.referrer()), prerender.referrerPolicy()), | 97 content::Referrer(GURL(prerender->referrer()), |
| 98 prerender->referrerPolicy()), |
71 extra_data.size(), extra_data.render_view_route_id())); | 99 extra_data.size(), extra_data.render_view_route_id())); |
72 } | 100 } |
73 | 101 |
74 void PrerenderDispatcher::cancel(const WebKit::WebPrerender& prerender) { | 102 void PrerenderDispatcher::cancel(WebPrerender* prerender) { |
75 const PrerenderExtraData& extra_data = | 103 const PrerenderExtraData& extra_data = |
76 PrerenderExtraData::FromPrerender(prerender); | 104 PrerenderExtraData::FromPrerender(*prerender); |
| 105 std::map<int, WebPrerender*>::iterator it = |
| 106 prerenders_.find(extra_data.prerender_id()); |
| 107 if (it != prerenders_.end()) |
| 108 prerenders_.erase(it); |
77 content::RenderThread::Get()->Send( | 109 content::RenderThread::Get()->Send( |
78 new PrerenderHostMsg_CancelLinkRelPrerender(extra_data.prerender_id())); | 110 new PrerenderHostMsg_CancelLinkRelPrerender(extra_data.prerender_id())); |
79 } | 111 } |
80 | 112 |
81 void PrerenderDispatcher::abandon(const WebKit::WebPrerender& prerender) { | 113 void PrerenderDispatcher::abandon(WebPrerender* prerender) { |
82 const PrerenderExtraData& extra_data = | 114 const PrerenderExtraData& extra_data = |
83 PrerenderExtraData::FromPrerender(prerender); | 115 PrerenderExtraData::FromPrerender(*prerender); |
| 116 std::map<int, WebPrerender*>::iterator it = |
| 117 prerenders_.find(extra_data.prerender_id()); |
| 118 if (it != prerenders_.end()) |
| 119 prerenders_.erase(it); |
84 content::RenderThread::Get()->Send( | 120 content::RenderThread::Get()->Send( |
85 new PrerenderHostMsg_AbandonLinkRelPrerender(extra_data.prerender_id())); | 121 new PrerenderHostMsg_AbandonLinkRelPrerender(extra_data.prerender_id())); |
86 } | 122 } |
87 | 123 |
88 } // namespace prerender | 124 } // namespace prerender |
OLD | NEW |