Home | History | Annotate | Download | only in storageapp 
      1 /*
      2  * Copyright (C) 2017 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.storageapp;
     18 
     19 import static com.android.cts.storageapp.Utils.TAG;
     20 import static com.android.cts.storageapp.Utils.makeUniqueFile;
     21 import static com.android.cts.storageapp.Utils.useFallocate;
     22 import static com.android.cts.storageapp.Utils.useWrite;
     23 
     24 import android.app.Activity;
     25 import android.content.BroadcastReceiver;
     26 import android.content.Context;
     27 import android.content.Intent;
     28 import android.os.Bundle;
     29 import android.os.storage.StorageManager;
     30 import android.system.Os;
     31 import android.util.Log;
     32 
     33 import java.io.File;
     34 import java.util.Objects;
     35 import java.util.UUID;
     36 
     37 public class UtilsReceiver extends BroadcastReceiver {
     38     public static final String EXTRA_FRACTION = "fraction";
     39     public static final String EXTRA_BYTES = "bytes";
     40     public static final String EXTRA_TIME = "time";
     41 
     42     @Override
     43     public void onReceive(Context context, Intent intent) {
     44         final Bundle res = doAllocation(context, intent.getExtras());
     45         if (res != null) {
     46             setResultCode(Activity.RESULT_OK);
     47             setResultExtras(res);
     48         } else {
     49             setResultCode(Activity.RESULT_CANCELED);
     50         }
     51     }
     52 
     53     public static Bundle doAllocation(Context context, Bundle extras) {
     54         final StorageManager sm = context.getSystemService(StorageManager.class);
     55 
     56         long allocated = 0;
     57         try {
     58             // When shared storage is backed by internal, then pivot our cache
     59             // files between the two locations to ensure clearing logic works.
     60             final File intDir = context.getCacheDir();
     61             final File extDir = context.getExternalCacheDir();
     62             final UUID intUuid = sm.getUuidForPath(intDir);
     63             final UUID extUuid = sm.getUuidForPath(extDir);
     64 
     65             Log.d(TAG, "Found internal " + intUuid + " and external " + extUuid);
     66             final boolean doPivot = Objects.equals(intUuid, extUuid);
     67 
     68             final double fraction = extras.getDouble(EXTRA_FRACTION, 0);
     69             final long quota = sm.getCacheQuotaBytes(intUuid);
     70             final long bytes = (long) (quota * fraction);
     71             final long time = extras.getLong(EXTRA_TIME, System.currentTimeMillis());
     72 
     73             int i = 0;
     74             while (allocated < bytes) {
     75                 final File target;
     76                 if (doPivot) {
     77                     target = (i++ % 2) == 0 ? intDir : extDir;
     78                 } else {
     79                     target = intDir;
     80                 }
     81 
     82                 final File f = makeUniqueFile(target);
     83                 final long size = 1024 * 1024;
     84                 if (target == intDir) {
     85                     useFallocate(f, size);
     86                 } else {
     87                     useWrite(f, size);
     88                 }
     89                 f.setLastModified(time);
     90                 allocated += Os.stat(f.getAbsolutePath()).st_blocks * 512;
     91             }
     92 
     93             Log.d(TAG, "Quota " + quota + ", target " + bytes + ", allocated " + allocated);
     94 
     95             final Bundle res = new Bundle();
     96             res.putLong(EXTRA_BYTES, allocated);
     97             return res;
     98         } catch (Exception e) {
     99             Log.e(TAG, "Failed to allocate cache files", e);
    100             return null;
    101         }
    102     }
    103 }
    104