Chromium Code Reviews| Index: extensions/browser/extension_function_dispatcher.cc |
| diff --git a/extensions/browser/extension_function_dispatcher.cc b/extensions/browser/extension_function_dispatcher.cc |
| index 93a8b24c3b660e06693e848ba34715164f215b49..7bb79bdd625c16061edff7ce141b9314c0df18e6 100644 |
| --- a/extensions/browser/extension_function_dispatcher.cc |
| +++ b/extensions/browser/extension_function_dispatcher.cc |
| @@ -31,6 +31,7 @@ |
| #include "content/public/browser/web_contents_observer.h" |
| #include "content/public/common/result_codes.h" |
| #include "extensions/browser/api_activity_monitor.h" |
| +#include "extensions/browser/bad_message.h" |
| #include "extensions/browser/extension_function_registry.h" |
| #include "extensions/browser/extension_registry.h" |
| #include "extensions/browser/extension_system.h" |
| @@ -79,43 +80,19 @@ struct Static { |
| }; |
| base::LazyInstance<Static> g_global_io_data = LAZY_INSTANCE_INITIALIZER; |
| -// Kills the specified process because it sends us a malformed message. |
| -// Track the specific function's |histogram_value|, as this may indicate a bug |
| -// in that API's implementation on the renderer. |
| -void KillBadMessageSender(const base::Process& process, |
| - functions::HistogramValue histogram_value) { |
| - // The renderer has done validation before sending extension api requests. |
| - // Therefore, we should never receive a request that is invalid in a way |
| - // that JSON validation in the renderer should have caught. It could be an |
| - // attacker trying to exploit the browser, so we crash the renderer instead. |
| - LOG(ERROR) << "Terminating renderer because of malformed extension message."; |
| - if (content::RenderProcessHost::run_renderer_in_process()) { |
| - // In single process mode it is better if we don't suicide but just crash. |
| - CHECK(false); |
| - return; |
| - } |
| - |
| - NOTREACHED(); |
| +void LogBadMessage(functions::HistogramValue histogram_value) { |
| content::RecordAction(base::UserMetricsAction("BadMessageTerminate_EFD")); |
| + // Track the specific function's |histogram_value|, as this may indicate a |
| + // bug in that API's implementation on the renderer. |
|
Devlin
2017/01/13 02:45:50
s/on the renderer/. We flag something as a bad me
lazyboy
2017/01/13 03:04:53
Done.
|
| UMA_HISTOGRAM_ENUMERATION("Extensions.BadMessageFunctionName", |
| histogram_value, functions::ENUM_BOUNDARY); |
| - if (process.IsValid()) |
| - process.Terminate(content::RESULT_CODE_KILLED_BAD_MESSAGE, false); |
| } |
| -void KillBadMessageSenderRPH(content::RenderProcessHost* sender_process_host, |
| - functions::HistogramValue histogram_value) { |
| - base::Process peer_process = |
| - content::RenderProcessHost::run_renderer_in_process() |
| - ? base::Process::Current() |
| - : base::Process::DeprecatedGetProcessFromHandle( |
| - sender_process_host->GetHandle()); |
| - KillBadMessageSender(peer_process, histogram_value); |
| -} |
| +using BadMessageReceivedFunc = base::Callback<void(functions::HistogramValue)>; |
| void CommonResponseCallback(IPC::Sender* ipc_sender, |
|
Devlin
2017/01/13 02:45:50
instead of BadMessageReceivedFunc, could we just h
lazyboy
2017/01/13 03:04:53
Ty, Done.
|
| int routing_id, |
| - const base::Process& peer_process, |
| + const BadMessageReceivedFunc& bad_message_func, |
| int request_id, |
| ExtensionFunction::ResponseType type, |
| const base::ListValue& results, |
| @@ -124,7 +101,7 @@ void CommonResponseCallback(IPC::Sender* ipc_sender, |
| DCHECK(ipc_sender); |
| if (type == ExtensionFunction::BAD_MESSAGE) { |
| - KillBadMessageSender(peer_process, histogram_value); |
| + bad_message_func.Run(histogram_value); |
| return; |
| } |
| @@ -133,6 +110,18 @@ void CommonResponseCallback(IPC::Sender* ipc_sender, |
| error)); |
| } |
| +void SendBadMessageToFilter(content::BrowserMessageFilter* filter, |
| + functions::HistogramValue histogram_value) { |
| + LogBadMessage(histogram_value); |
| + bad_message::ReceivedBadMessage(filter, bad_message::EFD_BAD_MESSAGE); |
| +} |
| + |
| +void SendBadMessageToProcess(content::RenderProcessHost* rph, |
| + functions::HistogramValue histogram_value) { |
| + LogBadMessage(histogram_value); |
| + bad_message::ReceivedBadMessage(rph, bad_message::EFD_BAD_MESSAGE); |
| +} |
| + |
| void IOThreadResponseCallback( |
| const base::WeakPtr<IOThreadExtensionMessageFilter>& ipc_sender, |
| int routing_id, |
| @@ -144,10 +133,13 @@ void IOThreadResponseCallback( |
| if (!ipc_sender.get()) |
| return; |
| - base::Process peer_process = |
| - base::Process::DeprecatedGetProcessFromHandle(ipc_sender->PeerHandle()); |
| - CommonResponseCallback(ipc_sender.get(), routing_id, peer_process, request_id, |
| - type, results, error, histogram_value); |
| + // Extract the pointer |filter_ptr| to make the base::Bind call below happy. |
| + IOThreadExtensionMessageFilter* filter_ptr = ipc_sender.get(); |
| + BadMessageReceivedFunc bad_message_received_func = |
| + base::Bind(&SendBadMessageToFilter, filter_ptr); |
| + CommonResponseCallback(ipc_sender.get(), routing_id, |
| + bad_message_received_func, request_id, type, results, |
| + error, histogram_value); |
| } |
| } // namespace |
| @@ -195,15 +187,12 @@ class ExtensionFunctionDispatcher::UIThreadResponseCallbackWrapper |
| const base::ListValue& results, |
| const std::string& error, |
| functions::HistogramValue histogram_value) { |
| - base::Process process = |
| - content::RenderProcessHost::run_renderer_in_process() |
| - ? base::Process::Current() |
| - : base::Process::DeprecatedGetProcessFromHandle( |
| - render_frame_host_->GetProcess()->GetHandle()); |
| + BadMessageReceivedFunc bad_message_received_func = |
| + base::Bind(&SendBadMessageToProcess, render_frame_host_->GetProcess()); |
| CommonResponseCallback(render_frame_host_, |
| render_frame_host_->GetRoutingID(), |
| - process, request_id, type, results, error, |
| - histogram_value); |
| + bad_message_received_func, request_id, type, results, |
| + error, histogram_value); |
| } |
| base::WeakPtr<ExtensionFunctionDispatcher> dispatcher_; |
| @@ -266,7 +255,7 @@ class ExtensionFunctionDispatcher::UIThreadWorkerResponseCallbackWrapper |
| content::RenderProcessHost* sender = |
| content::RenderProcessHost::FromID(render_process_id_); |
| if (type == ExtensionFunction::BAD_MESSAGE) { |
| - KillBadMessageSenderRPH(sender, histogram_value); |
| + SendBadMessageToProcess(sender, histogram_value); |
| return; |
| } |
| DCHECK(sender); |