Home | History | Annotate | Download | only in task 
      1 /*
      2  * Copyright 2014, 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.managedprovisioning.task;
     18 
     19 import android.app.admin.DevicePolicyManager;
     20 import android.content.ComponentName;
     21 import android.content.Context;
     22 import android.content.Intent;
     23 import android.content.pm.ActivityInfo;
     24 import android.content.pm.PackageInfo;
     25 import android.content.pm.PackageManager;
     26 import android.content.pm.PackageManager.NameNotFoundException;
     27 import android.text.TextUtils;
     28 
     29 import com.android.managedprovisioning.ProvisionLogger;
     30 
     31 public class SetDevicePolicyTask {
     32     public static final int ERROR_PACKAGE_NOT_INSTALLED = 0;
     33     public static final int ERROR_NO_RECEIVER = 1;
     34     public static final int ERROR_OTHER = 2;
     35 
     36     private final Callback mCallback;
     37     private final Context mContext;
     38     private final String mPackageName;
     39     private final String mOwner;
     40 
     41     private String mAdminReceiver;
     42     private PackageManager mPackageManager;
     43     private DevicePolicyManager mDevicePolicyManager;
     44 
     45     public SetDevicePolicyTask(Context context, String packageName, String owner,
     46             Callback callback) {
     47         mCallback = callback;
     48         mContext = context;
     49         mPackageName = packageName;
     50         mOwner = owner;
     51         mPackageManager = mContext.getPackageManager();
     52         mDevicePolicyManager = (DevicePolicyManager) mContext.
     53                 getSystemService(Context.DEVICE_POLICY_SERVICE);
     54     }
     55 
     56     public void run() {
     57         // Check whether package is installed and find the admin receiver.
     58         if (isPackageInstalled()) {
     59             enableDevicePolicyApp();
     60             setActiveAdmin();
     61             setDeviceOwner();
     62             mCallback.onSuccess();
     63         }
     64     }
     65 
     66     private boolean isPackageInstalled() {
     67         try {
     68             PackageInfo pi = mPackageManager.getPackageInfo(mPackageName,
     69                     PackageManager.GET_RECEIVERS);
     70             for (ActivityInfo ai : pi.receivers) {
     71                 if (!TextUtils.isEmpty(ai.permission) &&
     72                         ai.permission.equals(android.Manifest.permission.BIND_DEVICE_ADMIN)) {
     73                     mAdminReceiver = ai.name;
     74                     return true;
     75                 }
     76             }
     77             mCallback.onError(ERROR_NO_RECEIVER);
     78             return false;
     79         } catch (NameNotFoundException e) {
     80             mCallback.onError(ERROR_PACKAGE_NOT_INSTALLED);
     81             return false;
     82         }
     83     }
     84 
     85     private void enableDevicePolicyApp() {
     86         int enabledSetting = mPackageManager
     87                 .getApplicationEnabledSetting(mPackageName);
     88         if (enabledSetting != PackageManager.COMPONENT_ENABLED_STATE_DEFAULT) {
     89             mPackageManager.setApplicationEnabledSetting(mPackageName,
     90                     PackageManager.COMPONENT_ENABLED_STATE_DEFAULT, 0);
     91         }
     92     }
     93 
     94     public void setActiveAdmin() {
     95         ProvisionLogger.logd("Setting " + mPackageName + " as active admin.");
     96         ComponentName component = new ComponentName(mPackageName, mAdminReceiver);
     97         mDevicePolicyManager.setActiveAdmin(component, true);
     98     }
     99 
    100     public void setDeviceOwner() {
    101         ProvisionLogger.logd("Setting " + mPackageName + " as device owner " + mOwner + ".");
    102         if (!mDevicePolicyManager.isDeviceOwner(mPackageName)) {
    103             mDevicePolicyManager.setDeviceOwner(mPackageName, mOwner);
    104         }
    105     }
    106 
    107     public abstract static class Callback {
    108         public abstract void onSuccess();
    109         public abstract void onError(int errorCode);
    110     }
    111 }
    112