Fix omnibox security icons.

chrome/browser/resources/vr_shell/vr_shell_ui.js

 // Copyright 2016 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.
 
 var vrShellUi = (function() {
   'use strict';
 
   let scene = new ui.Scene();
   let sceneManager;
 
@@ skipping 270 matching lines @@
       update.setVisible(false);
       scene.updateElement(this.domUiElement.uiElementId, update);
       this.nativeState.visible = false;
 
       // Listen to the end of transitions, so that the box can be natively
       // hidden after it finishes hiding itself.
       document.querySelector('#omni').addEventListener('transitionend',
           this.onAnimationDone.bind(this));
     }
 
+    getSecurityIconElementId(level) {
+      // See security_state.h and getSecurityIconResource() for this mapping.
+      switch (level) {
+        case 0: // NONE
+        case 1: // HTTP_SHOW_WARNING
+        case 4: // SECURITY_WARNING
+          return '#omni-info-icon';
+        case 2: // SECURE:
+        case 3: // EV_SECURE:
+          return '#omni-lock-icon';
+        case 5: // SECURE_WITH_POLICY_INSTALLED_CERT (ChromeOS only)
+        case 6: // DANGEROUS
+        default:
+          return '#omni-warning-icon';
+      }
+    }
+
     setEnabled(enabled) {
       this.enabled = enabled;
       this.resetVisibilityTimer();
       this.updateState();
     }
 
     setLoading(loading) {
       this.loading = loading;
       this.resetVisibilityTimer();
       this.updateState();
@@ skipping 39 matching lines @@
       if (e.propertyName == 'opacity' && !this.visibleAfterTransition) {
         this.setNativeVisibility(false);
       }
     }
 
     updateState() {
       if (!this.enabled) {
         this.setNativeVisibility(false);
         return;
       }
-      let secure = this.level == 2 || this.level == 3;
-      document.querySelector('#omni-secure-icon').style.display =
-          (secure ? 'block' : 'none');
-      document.querySelector('#omni-insecure-icon').style.display =
-          (secure ? 'none' : 'block');
+
+      document.querySelector('#omni-warning-icon').style.display = 'none';
+      document.querySelector('#omni-info-icon').style.display = 'none';
+      document.querySelector('#omni-lock-icon').style.display = 'none';
+      let icon = this.getSecurityIconElementId(this.level);
+      document.querySelector(icon).style.display = 'block';
 
       let state = 'idle';
       this.visibleAfterTransition = true;
       if (this.loading) {
         state = 'loading';
       } else if (this.visibilityTimeout > 0 && !this.visibilityTimer) {
         state = 'hide';
         this.visibleAfterTransition = false;
       }
       document.querySelector('#omni').className = state;
@@ skipping 93 matching lines @@
     scene.flush();
   }
 
   return {
     initialize: initialize,
     command: command,
   };
 })();
 
 document.addEventListener('DOMContentLoaded', vrShellUi.initialize);