Index: chrome/browser/ui/webui/vr_shell/vr_shell_ui_message_handler.cc |
diff --git a/chrome/browser/ui/webui/vr_shell/vr_shell_ui_message_handler.cc b/chrome/browser/ui/webui/vr_shell/vr_shell_ui_message_handler.cc |
index c970d84b09a05184467e30984398c8985f42c988..c9c6e510dcb924268e08e433b1f8fb1c7a321997 100644 |
--- a/chrome/browser/ui/webui/vr_shell/vr_shell_ui_message_handler.cc |
+++ b/chrome/browser/ui/webui/vr_shell/vr_shell_ui_message_handler.cc |
@@ -10,11 +10,13 @@ |
#include "base/bind_helpers.h" |
#include "base/callback.h" |
#include "base/values.h" |
+#include "chrome/browser/android/vr_shell/ui_interface.h" |
#include "chrome/browser/android/vr_shell/ui_scene.h" |
#include "chrome/browser/android/vr_shell/vr_shell.h" |
#include "content/public/browser/web_ui.h" |
-VrShellUIMessageHandler::VrShellUIMessageHandler() = default; |
+VrShellUIMessageHandler::VrShellUIMessageHandler() : |
bshe
2016/10/19 21:02:40
nit: : move colon to next line
cjgrant
2016/10/20 15:25:39
Done.
|
+ weak_ptr_factory_(this) {} |
VrShellUIMessageHandler::~VrShellUIMessageHandler() = default; |
@@ -36,6 +38,8 @@ void VrShellUIMessageHandler::HandleDomLoaded(const base::ListValue* args) { |
if (!vr_shell_) |
return; |
+ vr_shell_->GetUiInterface()->SetUiMessageHandler( |
+ weak_ptr_factory_.GetWeakPtr()); |
vr_shell_->OnDomContentsLoaded(); |
} |