Fix crash in InputImeEventRouter::RegisterImeExtension

Bug: 1147192,1147526
Test: Resolves reproducible crash on a test device

Change-Id: Ie9dadf1b1165e32c45fa2878ca7cd11be823b384
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2527845Reviewed-by: Shu Chen <shuchen@chromium.org>
Reviewed-by: Jeffrey Kardatzke <jkardatzke@google.com>
Commit-Queue: Shu Chen <shuchen@chromium.org>
Auto-Submit: Alexander Alekseev <alemate@chromium.org>
Cr-Commit-Position: refs/heads/master@{#826160}

chrome/browser/extensions/api/input_ime/input_ime_api_chromeos.cc

@@ -586,14 +586,22 @@ bool InputImeEventRouter::RegisterImeExtension(
   }
 
   Profile* profile = GetProfile();
-
   // TODO(https://crbug.com/1140236): Investigate whether profile selection
   // is really needed.
-  if (chromeos::input_method::InputMethodManager::Get()
-              ->GetActiveIMEState()
-              ->GetUIStyle() ==
-          chromeos::input_method::InputMethodManager::UIStyle::kLogin &&
-      profile->HasPrimaryOTRProfile()) {
+  bool is_login = false;
+  // When Chrome starts with the Login screen, sometimes active IME State was
+  // not set yet. It's asynchronous process. So we need a fallback for that.
+  scoped_refptr<chromeos::input_method::InputMethodManager::State>
+      active_state = chromeos::input_method::InputMethodManager::Get()
+                         ->GetActiveIMEState();
+  if (active_state) {
+    is_login = active_state->GetUIStyle() ==
+               chromeos::input_method::InputMethodManager::UIStyle::kLogin;
+  } else {
+    is_login = chromeos::ProfileHelper::IsSigninProfile(profile);
+  }
+
+  if (is_login && profile->HasPrimaryOTRProfile()) {
     profile = profile->GetPrimaryOTRProfile();
   }