Check internal display RootWindow before getting the compositor.

ash/wallpaper/wallpaper_controller.cc

 // Copyright (c) 2012 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 "ash/wallpaper/wallpaper_controller.h"
 
 #include <string>
 #include <utility>
 
 #include "ash/ash_switches.h"
@@ skipping 398 matching lines @@
 bool WallpaperController::MoveToUnlockedContainer() {
   if (!locked_)
     return false;
 
   locked_ = false;
   return ReparentWallpaper(GetWallpaperContainerId(false));
 }
 
 void WallpaperController::GetInternalDisplayCompositorLock() {
   if (display::Display::HasInternalDisplay()) {
-    compositor_lock_ =
-        Shell::Get()
-            ->window_tree_host_manager()
-            ->GetRootWindowForDisplayId(display::Display::InternalDisplayId())
-            ->layer()
-            ->GetCompositor()
-            ->GetCompositorLock(this, base::TimeDelta::FromMilliseconds(
-                                          kCompositorLockTimeoutMs));
+    aura::Window* root_window =
+        Shell::Get()->window_tree_host_manager()->GetRootWindowForDisplayId(
+            display::Display::InternalDisplayId());

[oshima, 2017/05/11 21:03:45]

can you use ShellPort::GetRootWindowForDisplayId ?

[wutao, 2017/05/11 21:34:35]

sg.
BTW, It seems the code wraps aura::Window with WmWindow and then I need to get
aura::Window again. 

[sadrul, 2017/05/11 23:41:12]

Actually, we are trying to get rid of WmWindow. So directly using aura::Window
here is better.

https://bugs.chromium.org/p/chromium/issues/detail?id=671246

+    if (root_window) {
+      compositor_lock_ =
+          root_window->layer()->GetCompositor()->GetCompositorLock(
+              this,
+              base::TimeDelta::FromMilliseconds(kCompositorLockTimeoutMs));
+    }
   }
 }
 
 void WallpaperController::CompositorLockTimedOut() {
   compositor_lock_.reset();
 }
 
 }  // namespace ash