1 /* 2 * Copyright (C) 2011 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.cts.verifier.admin; 18 19 import com.android.cts.verifier.managedprovisioning.DeviceAdminTestReceiver; 20 import com.android.cts.verifier.PassFailButtons; 21 import com.android.cts.verifier.R; 22 23 import android.app.AlertDialog; 24 import android.app.KeyguardManager; 25 import android.app.admin.DevicePolicyManager; 26 import android.content.BroadcastReceiver; 27 import android.content.Context; 28 import android.content.Intent; 29 import android.content.IntentFilter; 30 import android.os.Bundle; 31 import android.view.View; 32 import android.view.View.OnClickListener; 33 import android.widget.Button; 34 35 public class ScreenLockTestActivity extends PassFailButtons.Activity { 36 37 private static final int ADD_DEVICE_ADMIN_REQUEST_CODE = 1; 38 39 private ScreenOffReceiver mReceiver; 40 41 private Button mForceLockButton; 42 43 private DevicePolicyManager mDevicePolicyManager; 44 45 private KeyguardManager mKeyguardManager; 46 47 @Override onCreate(Bundle savedInstanceState)48 protected void onCreate(Bundle savedInstanceState) { 49 super.onCreate(savedInstanceState); 50 setContentView(R.layout.da_screen_lock_main); 51 setPassFailButtonClickListeners(); 52 setInfoResources(R.string.da_screen_lock_test, R.string.da_screen_lock_info, -1); 53 54 mDevicePolicyManager = (DevicePolicyManager) getSystemService(DEVICE_POLICY_SERVICE); 55 mKeyguardManager = (KeyguardManager) getSystemService(KEYGUARD_SERVICE); 56 57 getPassButton().setEnabled(false); 58 59 mForceLockButton = (Button) findViewById(R.id.da_force_lock_button); 60 mForceLockButton.setOnClickListener(new OnClickListener() { 61 @Override 62 public void onClick(View v) { 63 sendAddDeviceAdminIntent(); 64 } 65 }); 66 67 mReceiver = new ScreenOffReceiver(); 68 IntentFilter filter = new IntentFilter(); 69 filter.addAction(Intent.ACTION_SCREEN_OFF); 70 registerReceiver(mReceiver, filter); 71 } 72 sendAddDeviceAdminIntent()73 private void sendAddDeviceAdminIntent() { 74 Intent intent = new Intent(DevicePolicyManager.ACTION_ADD_DEVICE_ADMIN); 75 intent.putExtra(DevicePolicyManager.EXTRA_DEVICE_ADMIN, 76 DeviceAdminTestReceiver.getReceiverComponentName()); 77 startActivityForResult(intent, ADD_DEVICE_ADMIN_REQUEST_CODE); 78 } 79 80 @Override onActivityResult(int requestCode, int resultCode, Intent data)81 protected void onActivityResult(int requestCode, int resultCode, Intent data) { 82 super.onActivityResult(requestCode, resultCode, data); 83 switch (requestCode) { 84 case ADD_DEVICE_ADMIN_REQUEST_CODE: 85 handleAddDeviceAdminResult(resultCode, data); 86 break; 87 } 88 } 89 handleAddDeviceAdminResult(int resultCode, Intent data)90 private void handleAddDeviceAdminResult(int resultCode, Intent data) { 91 if (resultCode == RESULT_OK) { 92 mDevicePolicyManager.lockNow(); 93 } 94 } 95 96 @Override onDestroy()97 protected void onDestroy() { 98 super.onDestroy(); 99 if (mDevicePolicyManager.isAdminActive( 100 DeviceAdminTestReceiver.getReceiverComponentName())) { 101 mDevicePolicyManager.removeActiveAdmin( 102 DeviceAdminTestReceiver.getReceiverComponentName()); 103 } 104 unregisterReceiver(mReceiver); 105 } 106 107 private class ScreenOffReceiver extends BroadcastReceiver { 108 109 private static final int LOCK_CHECK_DELAY = 1000; 110 111 @Override onReceive(Context context, Intent intent)112 public void onReceive(Context context, Intent intent) { 113 mForceLockButton.postDelayed(new Runnable() { 114 @Override 115 public void run() { 116 boolean lockSuccess = mKeyguardManager.inKeyguardRestrictedInputMode(); 117 getPassButton().setEnabled(lockSuccess); 118 119 int iconId = lockSuccess 120 ? android.R.drawable.ic_dialog_info 121 : android.R.drawable.ic_dialog_alert; 122 int messageId = lockSuccess 123 ? R.string.da_lock_success 124 : R.string.da_lock_error; 125 new AlertDialog.Builder(ScreenLockTestActivity.this) 126 .setTitle(R.string.da_screen_lock_test) 127 .setMessage(messageId) 128 .setIcon(iconId) 129 .setPositiveButton(android.R.string.ok, null) 130 .show(); 131 } 132 }, LOCK_CHECK_DELAY); 133 } 134 } 135 } 136