Merge "Hold a wakelock while retrieving reports"

import android.os.IPullAtomCallback;

import android.os.IStatsManagerService;

import android.os.IStatsd;

import android.os.PowerManager;

import android.os.Process;

import android.os.RemoteException;

import android.util.ArrayMap;

    @Override

    public byte[] getData(long key, String packageName) throws IllegalStateException {

        enforceDumpAndUsageStatsPermission(packageName);

        PowerManager powerManager = (PowerManager)

                mContext.getSystemService(Context.POWER_SERVICE);

        PowerManager.WakeLock wl = powerManager.newWakeLock(PowerManager.PARTIAL_WAKE_LOCK,

                /*tag=*/ StatsManagerService.class.getCanonicalName());

        int callingUid = Binder.getCallingUid();

        final long token = Binder.clearCallingIdentity();

        wl.acquire();

        try {

            IStatsd statsd = waitForStatsd();

            if (statsd != null) {

            Log.e(TAG, "Failed to getData with statsd");

            throw new IllegalStateException(e.getMessage(), e);

        } finally {

            wl.release();

            Binder.restoreCallingIdentity(token);

        }

        throw new IllegalStateException("Failed to connect to statsd to getData");