Update security method after switching user

Bug: 15445037
Change-Id: If296dd3ab6510c335aba7ffdce5cdd923ceada6b

1 files changed, 7 insertions, 0 deletions

diff --git a/packages/Keyguard/src/com/android/keyguard/KeyguardSimpleHostView.java b/packages/Keyguard/src/com/android/keyguard/KeyguardSimpleHostView.java
index 3f6ced6..bc159cb 100644
--- a/packages/Keyguard/src/com/android/keyguard/KeyguardSimpleHostView.java
+++ b/packages/Keyguard/src/com/android/keyguard/KeyguardSimpleHostView.java
@@ -25,6 +25,7 @@ public class KeyguardSimpleHostView extends KeyguardViewBase {
 
     public KeyguardSimpleHostView(Context context, AttributeSet attrs) {
         super(context, attrs);
+        KeyguardUpdateMonitor.getInstance(context).registerCallback(mUpdateCallback);
     }
 
     @Override
@@ -62,4 +63,10 @@ public class KeyguardSimpleHostView extends KeyguardViewBase {
         // TODO Auto-generated method stub
     }
 
+    private KeyguardUpdateMonitorCallback mUpdateCallback = new KeyguardUpdateMonitorCallback() {
+        @Override
+        public void onUserSwitchComplete(int userId) {
+            getSecurityContainer().showPrimarySecurityScreen(false /* turning off */);
+        }
+    };
 }