| /* |
| * Copyright (C) 2024 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.android.art.tests; |
| |
| import com.android.csuite.core.ApkInstaller.ApkInstallerException; |
| import com.android.csuite.core.TestUtils; |
| import com.android.csuite.tests.AppLaunchTest; |
| import com.android.tradefed.config.Option; |
| import com.android.tradefed.device.DeviceNotAvailableException; |
| import com.android.tradefed.result.LogDataType; |
| import com.android.tradefed.util.CommandResult; |
| import com.android.tradefed.util.CommandStatus; |
| |
| import org.junit.After; |
| import org.junit.Assert; |
| |
| import java.io.File; |
| |
| /** A test that gets imgdiag data after launching an app. */ |
| public class AppLaunchImgdiagTest extends AppLaunchTest { |
| @Option( |
| name = "imgdiag-out-path", |
| description = "Path to directory containing imgdiag output files.") |
| private String mImgdiagOutPath; |
| |
| @After |
| public void tearDown() throws DeviceNotAvailableException, ApkInstallerException { |
| String[] packagePids = |
| getDevice().executeShellCommand("pidof " + mPackageName).strip().split(" "); |
| |
| for (String targetPid : packagePids) { |
| targetPid = targetPid.strip(); |
| Assert.assertFalse(targetPid.isEmpty()); |
| |
| String outFileName = String.format("imgdiag_%s_%s.txt", mPackageName, targetPid); |
| String outFilePath = new File(mImgdiagOutPath, outFileName).getAbsolutePath(); |
| |
| String imgdiagCmd = getImgdiagRunCmd(targetPid, outFilePath); |
| CommandResult res = getDevice().executeShellV2Command(imgdiagCmd); |
| Assert.assertEquals( |
| String.format( |
| "Failed to run imgdiag:\n%s\nResult:\n%s", imgdiagCmd, res.toString()), |
| CommandStatus.SUCCESS, |
| res.getStatus()); |
| |
| File imgdiagFile = getDevice().pullFile(outFilePath); |
| TestUtils testUtils = TestUtils.getInstance(getTestInformation(), mLogData); |
| testUtils |
| .getTestArtifactReceiver() |
| .addTestArtifact(outFileName, LogDataType.HOST_LOG, imgdiagFile); |
| } |
| |
| super.tearDown(); |
| } |
| |
| /** |
| * Constructs the command line string to run the `imgdiag` tool for gathering dirty image |
| * objects. |
| * |
| * @param targetPid The process ID of the target process to analyze. |
| * @param outFilePath The absolute file path on the device where the imgdiag data will be saved. |
| * @return The complete `imgdiag` command that can be run with `adb shell`. |
| */ |
| public static String getImgdiagRunCmd(String targetPid, String outFilePath) { |
| // There may be more than one process with "zygote64" name when a new app |
| // is forked. Use `pgrep -f zygote64 -o` to get the oldest process. |
| return String.format( |
| "imgdiag --zygote-diff-pid=`pgrep -f zygote64 -o` --image-diff-pid=%s" |
| + " --output=%s" |
| + " --dump-dirty-objects --boot-image=" |
| + "/data/misc/apexdata/com.android.art/dalvik-cache/boot.art", |
| targetPid, outFilePath); |
| } |
| } |