Home | History | Annotate | Download | only in am 
      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.server.am;
     18 
     19 import android.app.IInstrumentationWatcher;
     20 import android.app.IUiAutomationConnection;
     21 import android.content.ComponentName;
     22 import android.content.pm.ApplicationInfo;
     23 import android.os.Bundle;
     24 import android.util.PrintWriterPrinter;
     25 import android.util.proto.ProtoOutputStream;
     26 
     27 import java.io.PrintWriter;
     28 import java.util.ArrayList;
     29 import java.util.Arrays;
     30 
     31 class ActiveInstrumentation {
     32     final ActivityManagerService mService;
     33 
     34     // Class installed to instrument app
     35     ComponentName mClass;
     36 
     37     // All process names that should be instrumented
     38     String[] mTargetProcesses;
     39 
     40     // The application being instrumented
     41     ApplicationInfo mTargetInfo;
     42 
     43     // Where to save profiling
     44     String mProfileFile;
     45 
     46     // Who is waiting
     47     IInstrumentationWatcher mWatcher;
     48 
     49     // Connection to use the UI introspection APIs.
     50     IUiAutomationConnection mUiAutomationConnection;
     51 
     52     // As given to us
     53     Bundle mArguments;
     54 
     55     // Any intermediate results that have been collected.
     56     Bundle mCurResults;
     57 
     58     // Copy of instrumentationClass.
     59     ComponentName mResultClass;
     60 
     61     // Contains all running processes that have active instrumentation.
     62     final ArrayList<ProcessRecord> mRunningProcesses = new ArrayList<>();
     63 
     64     // Set to true when we have told the watcher the instrumentation is finished.
     65     boolean mFinished;
     66 
     67     ActiveInstrumentation(ActivityManagerService service) {
     68         mService = service;
     69     }
     70 
     71     void removeProcess(ProcessRecord proc) {
     72         mFinished = true;
     73         mRunningProcesses.remove(proc);
     74         if (mRunningProcesses.size() == 0) {
     75             mService.mActiveInstrumentation.remove(this);
     76         }
     77     }
     78 
     79     public String toString() {
     80         StringBuilder sb = new StringBuilder(128);
     81         sb.append("ActiveInstrumentation{");
     82         sb.append(Integer.toHexString(System.identityHashCode(this)));
     83         sb.append(' ');
     84         sb.append(mClass.toShortString());
     85         if (mFinished) {
     86             sb.append(" FINISHED");
     87         }
     88         sb.append(" ");
     89         sb.append(mRunningProcesses.size());
     90         sb.append(" procs");
     91         sb.append('}');
     92         return sb.toString();
     93     }
     94 
     95     void dump(PrintWriter pw, String prefix) {
     96         pw.print(prefix); pw.print("mClass="); pw.print(mClass);
     97         pw.print(" mFinished="); pw.println(mFinished);
     98         pw.print(prefix); pw.println("mRunningProcesses:");
     99         for (int i=0; i<mRunningProcesses.size(); i++) {
    100             pw.print(prefix); pw.print("  #"); pw.print(i); pw.print(": ");
    101             pw.println(mRunningProcesses.get(i));
    102         }
    103         pw.print(prefix); pw.print("mTargetProcesses=");
    104         pw.println(Arrays.toString(mTargetProcesses));
    105         pw.print(prefix); pw.print("mTargetInfo=");
    106         pw.println(mTargetInfo);
    107         if (mTargetInfo != null) {
    108             mTargetInfo.dump(new PrintWriterPrinter(pw), prefix + "  ", 0);
    109         }
    110         if (mProfileFile != null) {
    111             pw.print(prefix); pw.print("mProfileFile="); pw.println(mProfileFile);
    112         }
    113         if (mWatcher != null) {
    114             pw.print(prefix); pw.print("mWatcher="); pw.println(mWatcher);
    115         }
    116         if (mUiAutomationConnection != null) {
    117             pw.print(prefix); pw.print("mUiAutomationConnection=");
    118             pw.println(mUiAutomationConnection);
    119         }
    120         pw.print(prefix); pw.print("mArguments=");
    121         pw.println(mArguments);
    122     }
    123 
    124     void writeToProto(ProtoOutputStream proto, long fieldId) {
    125         long token = proto.start(fieldId);
    126         mClass.writeToProto(proto, ActiveInstrumentationProto.CLASS);
    127         proto.write(ActiveInstrumentationProto.FINISHED, mFinished);
    128         for (int i=0; i<mRunningProcesses.size(); i++) {
    129             mRunningProcesses.get(i).writeToProto(proto,
    130                     ActiveInstrumentationProto.RUNNING_PROCESSES);
    131         }
    132         for (String p : mTargetProcesses) {
    133             proto.write(ActiveInstrumentationProto.TARGET_PROCESSES, p);
    134         }
    135         if (mTargetInfo != null) {
    136             mTargetInfo.writeToProto(proto, ActiveInstrumentationProto.TARGET_INFO);
    137         }
    138         proto.write(ActiveInstrumentationProto.PROFILE_FILE, mProfileFile);
    139         proto.write(ActiveInstrumentationProto.WATCHER, mWatcher.toString());
    140         proto.write(ActiveInstrumentationProto.UI_AUTOMATION_CONNECTION,
    141                 mUiAutomationConnection.toString());
    142         proto.write(ActiveInstrumentationProto.ARGUMENTS, mArguments.toString());
    143         proto.end(token);
    144     }
    145 }
    146