xref: /aosp_15_r20/art/libartbase/base/file_utils_test.cc (revision 795d594fd825385562da6b089ea9b2033f3abf5a)
1*795d594fSAndroid Build Coastguard Worker /*
2*795d594fSAndroid Build Coastguard Worker  * Copyright (C) 2011 The Android Open Source Project
3*795d594fSAndroid Build Coastguard Worker  *
4*795d594fSAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*795d594fSAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*795d594fSAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*795d594fSAndroid Build Coastguard Worker  *
8*795d594fSAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*795d594fSAndroid Build Coastguard Worker  *
10*795d594fSAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*795d594fSAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*795d594fSAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*795d594fSAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*795d594fSAndroid Build Coastguard Worker  * limitations under the License.
15*795d594fSAndroid Build Coastguard Worker  */
16*795d594fSAndroid Build Coastguard Worker 
17*795d594fSAndroid Build Coastguard Worker #include "base/file_utils.h"
18*795d594fSAndroid Build Coastguard Worker 
19*795d594fSAndroid Build Coastguard Worker #include <libgen.h>
20*795d594fSAndroid Build Coastguard Worker #include <stdlib.h>
21*795d594fSAndroid Build Coastguard Worker 
22*795d594fSAndroid Build Coastguard Worker #include <optional>
23*795d594fSAndroid Build Coastguard Worker #include <vector>
24*795d594fSAndroid Build Coastguard Worker 
25*795d594fSAndroid Build Coastguard Worker #include "base/stl_util.h"
26*795d594fSAndroid Build Coastguard Worker #include "common_art_test.h"
27*795d594fSAndroid Build Coastguard Worker 
28*795d594fSAndroid Build Coastguard Worker namespace art {
29*795d594fSAndroid Build Coastguard Worker 
30*795d594fSAndroid Build Coastguard Worker static constexpr const char kAndroidWifiApexDefaultPath[] = "/apex/com.android.wifi";
31*795d594fSAndroid Build Coastguard Worker 
32*795d594fSAndroid Build Coastguard Worker namespace {
33*795d594fSAndroid Build Coastguard Worker class ScopedOverrideDalvikCacheSubDirectory {
34*795d594fSAndroid Build Coastguard Worker  public:
ScopedOverrideDalvikCacheSubDirectory(const char * override)35*795d594fSAndroid Build Coastguard Worker   explicit ScopedOverrideDalvikCacheSubDirectory(const char* override) {
36*795d594fSAndroid Build Coastguard Worker     OverrideDalvikCacheSubDirectory(override);
37*795d594fSAndroid Build Coastguard Worker   }
38*795d594fSAndroid Build Coastguard Worker 
~ScopedOverrideDalvikCacheSubDirectory()39*795d594fSAndroid Build Coastguard Worker   ~ScopedOverrideDalvikCacheSubDirectory() { OverrideDalvikCacheSubDirectory("dalvik-cache"); }
40*795d594fSAndroid Build Coastguard Worker 
41*795d594fSAndroid Build Coastguard Worker  private:
42*795d594fSAndroid Build Coastguard Worker   DISALLOW_COPY_AND_ASSIGN(ScopedOverrideDalvikCacheSubDirectory);
43*795d594fSAndroid Build Coastguard Worker };
44*795d594fSAndroid Build Coastguard Worker }  // namespace
45*795d594fSAndroid Build Coastguard Worker 
46*795d594fSAndroid Build Coastguard Worker class FileUtilsTest : public CommonArtTest {};
47*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetDalvikCacheFilename)48*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetDalvikCacheFilename) {
49*795d594fSAndroid Build Coastguard Worker   std::string name;
50*795d594fSAndroid Build Coastguard Worker   std::string error;
51*795d594fSAndroid Build Coastguard Worker 
52*795d594fSAndroid Build Coastguard Worker   EXPECT_TRUE(GetDalvikCacheFilename("/system/app/Foo.apk", "/foo", &name, &error)) << error;
53*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/foo/system@[email protected]@classes.dex", name);
54*795d594fSAndroid Build Coastguard Worker 
55*795d594fSAndroid Build Coastguard Worker   EXPECT_TRUE(GetDalvikCacheFilename("/data/app/foo-1.apk", "/foo", &name, &error)) << error;
56*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/foo/data@[email protected]@classes.dex", name);
57*795d594fSAndroid Build Coastguard Worker 
58*795d594fSAndroid Build Coastguard Worker   EXPECT_TRUE(GetDalvikCacheFilename("/system/framework/core.jar", "/foo", &name, &error)) << error;
59*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/foo/system@[email protected]@classes.dex", name);
60*795d594fSAndroid Build Coastguard Worker 
61*795d594fSAndroid Build Coastguard Worker   EXPECT_TRUE(GetDalvikCacheFilename("/system/framework/boot.art", "/foo", &name, &error)) << error;
62*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/foo/system@[email protected]", name);
63*795d594fSAndroid Build Coastguard Worker 
64*795d594fSAndroid Build Coastguard Worker   EXPECT_TRUE(GetDalvikCacheFilename("/system/framework/boot.oat", "/foo", &name, &error)) << error;
65*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/foo/system@[email protected]", name);
66*795d594fSAndroid Build Coastguard Worker }
67*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetSystemImageFilename)68*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetSystemImageFilename) {
69*795d594fSAndroid Build Coastguard Worker   EXPECT_STREQ("/system/framework/arm/boot.art",
70*795d594fSAndroid Build Coastguard Worker                GetSystemImageFilename("/system/framework/boot.art", InstructionSet::kArm).c_str());
71*795d594fSAndroid Build Coastguard Worker }
72*795d594fSAndroid Build Coastguard Worker 
73*795d594fSAndroid Build Coastguard Worker // TODO(dsrbecky): b/160885380: This test is failing in eng-prod because libartbase
74*795d594fSAndroid Build Coastguard Worker //                              is loaded from different path (under testcases).
TEST_F(FileUtilsTest,DISABLED_GetAndroidRootSafe)75*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, DISABLED_GetAndroidRootSafe) {
76*795d594fSAndroid Build Coastguard Worker   std::string error_msg;
77*795d594fSAndroid Build Coastguard Worker 
78*795d594fSAndroid Build Coastguard Worker   // We don't expect null returns for most cases, so don't check and let std::string crash.
79*795d594fSAndroid Build Coastguard Worker 
80*795d594fSAndroid Build Coastguard Worker   // CommonArtTest sets ANDROID_ROOT, so expect this to be the same.
81*795d594fSAndroid Build Coastguard Worker   std::string android_root = GetAndroidRootSafe(&error_msg);
82*795d594fSAndroid Build Coastguard Worker   std::string android_root_env = getenv("ANDROID_ROOT");
83*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(android_root, android_root_env) << error_msg;
84*795d594fSAndroid Build Coastguard Worker 
85*795d594fSAndroid Build Coastguard Worker   // Set ANDROID_ROOT to something else (but the directory must exist). So use dirname.
86*795d594fSAndroid Build Coastguard Worker   UniqueCPtr<char> root_dup(strdup(android_root_env.c_str()));
87*795d594fSAndroid Build Coastguard Worker   char* dir = dirname(root_dup.get());
88*795d594fSAndroid Build Coastguard Worker   ASSERT_EQ(0, setenv("ANDROID_ROOT", dir, /* overwrite */ 1));
89*795d594fSAndroid Build Coastguard Worker   std::string android_root2 = GetAndroidRootSafe(&error_msg);
90*795d594fSAndroid Build Coastguard Worker   EXPECT_STREQ(dir, android_root2.c_str()) << error_msg;
91*795d594fSAndroid Build Coastguard Worker 
92*795d594fSAndroid Build Coastguard Worker   // Set a bogus value for ANDROID_ROOT. This should be an error.
93*795d594fSAndroid Build Coastguard Worker   ASSERT_EQ(0, setenv("ANDROID_ROOT", "/this/is/obviously/bogus", /* overwrite */ 1));
94*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetAndroidRootSafe(&error_msg), "");
95*795d594fSAndroid Build Coastguard Worker   error_msg = "";
96*795d594fSAndroid Build Coastguard Worker 
97*795d594fSAndroid Build Coastguard Worker   // Inferring the Android Root from the location of libartbase only works on host.
98*795d594fSAndroid Build Coastguard Worker   if (!kIsTargetBuild) {
99*795d594fSAndroid Build Coastguard Worker     // Unset ANDROID_ROOT and see that it still returns something (as libartbase code is running).
100*795d594fSAndroid Build Coastguard Worker     ASSERT_EQ(0, unsetenv("ANDROID_ROOT"));
101*795d594fSAndroid Build Coastguard Worker     std::string android_root3 = GetAndroidRootSafe(&error_msg);
102*795d594fSAndroid Build Coastguard Worker     // This should be the same as the other root (modulo realpath), otherwise the test setup is
103*795d594fSAndroid Build Coastguard Worker     // broken. On non-bionic. On bionic we can be running with a different libartbase that lives
104*795d594fSAndroid Build Coastguard Worker     // outside of ANDROID_ROOT.
105*795d594fSAndroid Build Coastguard Worker     UniqueCPtr<char> real_root3(realpath(android_root3.c_str(), nullptr));
106*795d594fSAndroid Build Coastguard Worker #if !defined(__BIONIC__) || defined(__ANDROID__)
107*795d594fSAndroid Build Coastguard Worker     UniqueCPtr<char> real_root(realpath(android_root.c_str(), nullptr));
108*795d594fSAndroid Build Coastguard Worker     EXPECT_STREQ(real_root.get(), real_root3.get()) << error_msg;
109*795d594fSAndroid Build Coastguard Worker #else
110*795d594fSAndroid Build Coastguard Worker     EXPECT_STRNE(real_root3.get(), "") << error_msg;
111*795d594fSAndroid Build Coastguard Worker #endif
112*795d594fSAndroid Build Coastguard Worker   }
113*795d594fSAndroid Build Coastguard Worker 
114*795d594fSAndroid Build Coastguard Worker   // Reset ANDROID_ROOT, as other things may depend on it.
115*795d594fSAndroid Build Coastguard Worker   ASSERT_EQ(0, setenv("ANDROID_ROOT", android_root_env.c_str(), /* overwrite */ 1));
116*795d594fSAndroid Build Coastguard Worker }
117*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetArtRootSafe)118*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetArtRootSafe) {
119*795d594fSAndroid Build Coastguard Worker   std::string error_msg;
120*795d594fSAndroid Build Coastguard Worker   std::string android_art_root;
121*795d594fSAndroid Build Coastguard Worker   std::string android_art_root_env;
122*795d594fSAndroid Build Coastguard Worker 
123*795d594fSAndroid Build Coastguard Worker   // TODO(b/130295968): Re-enable this part when the directory exists on host
124*795d594fSAndroid Build Coastguard Worker   if (kIsTargetBuild) {
125*795d594fSAndroid Build Coastguard Worker     // We don't expect null returns for most cases, so don't check and let std::string crash.
126*795d594fSAndroid Build Coastguard Worker 
127*795d594fSAndroid Build Coastguard Worker     // CommonArtTest sets ANDROID_ART_ROOT, so expect this to be the same.
128*795d594fSAndroid Build Coastguard Worker     android_art_root = GetArtRootSafe(&error_msg);
129*795d594fSAndroid Build Coastguard Worker     android_art_root_env = getenv("ANDROID_ART_ROOT");
130*795d594fSAndroid Build Coastguard Worker     EXPECT_EQ(android_art_root, android_art_root_env) << error_msg;
131*795d594fSAndroid Build Coastguard Worker 
132*795d594fSAndroid Build Coastguard Worker     // Set ANDROID_ART_ROOT to something else (but the directory must exist). So use dirname.
133*795d594fSAndroid Build Coastguard Worker     UniqueCPtr<char> root_dup(strdup(android_art_root_env.c_str()));
134*795d594fSAndroid Build Coastguard Worker     char* dir = dirname(root_dup.get());
135*795d594fSAndroid Build Coastguard Worker     ASSERT_EQ(0, setenv("ANDROID_ART_ROOT", dir, /* overwrite */ 1));
136*795d594fSAndroid Build Coastguard Worker     std::string android_art_root2 = GetArtRootSafe(&error_msg);
137*795d594fSAndroid Build Coastguard Worker     EXPECT_STREQ(dir, android_art_root2.c_str()) << error_msg;
138*795d594fSAndroid Build Coastguard Worker   }
139*795d594fSAndroid Build Coastguard Worker 
140*795d594fSAndroid Build Coastguard Worker   // Set a bogus value for ANDROID_ART_ROOT. This should be an error.
141*795d594fSAndroid Build Coastguard Worker   ASSERT_EQ(0, setenv("ANDROID_ART_ROOT", "/this/is/obviously/bogus", /* overwrite */ 1));
142*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtRootSafe(&error_msg), "");
143*795d594fSAndroid Build Coastguard Worker 
144*795d594fSAndroid Build Coastguard Worker   // Inferring the ART root from the location of libartbase only works on target.
145*795d594fSAndroid Build Coastguard Worker   if (kIsTargetBuild) {
146*795d594fSAndroid Build Coastguard Worker     // Disabled for now, as we cannot reliably use `GetRootContainingLibartbase`
147*795d594fSAndroid Build Coastguard Worker     // to find the ART root on target yet (see comment in `GetArtRootSafe`).
148*795d594fSAndroid Build Coastguard Worker     //
149*795d594fSAndroid Build Coastguard Worker     // TODO(b/129534335): Re-enable this part of the test on target when the
150*795d594fSAndroid Build Coastguard Worker     // only instance of libartbase is the one from the ART APEX.
151*795d594fSAndroid Build Coastguard Worker     if ((false)) {
152*795d594fSAndroid Build Coastguard Worker       // Unset ANDROID_ART_ROOT and see that it still returns something (as
153*795d594fSAndroid Build Coastguard Worker       // libartbase code is running).
154*795d594fSAndroid Build Coastguard Worker       ASSERT_EQ(0, unsetenv("ANDROID_ART_ROOT"));
155*795d594fSAndroid Build Coastguard Worker       std::string android_art_root3 = GetArtRootSafe(&error_msg);
156*795d594fSAndroid Build Coastguard Worker       // This should be the same as the other root (modulo realpath), otherwise
157*795d594fSAndroid Build Coastguard Worker       // the test setup is broken. On non-bionic. On bionic we can be running
158*795d594fSAndroid Build Coastguard Worker       // with a different libartbase that lives outside of ANDROID_ART_ROOT.
159*795d594fSAndroid Build Coastguard Worker       UniqueCPtr<char> real_root3(realpath(android_art_root3.c_str(), nullptr));
160*795d594fSAndroid Build Coastguard Worker #if !defined(__BIONIC__) || defined(__ANDROID__)
161*795d594fSAndroid Build Coastguard Worker       UniqueCPtr<char> real_root(realpath(android_art_root.c_str(), nullptr));
162*795d594fSAndroid Build Coastguard Worker       EXPECT_STREQ(real_root.get(), real_root3.get()) << error_msg;
163*795d594fSAndroid Build Coastguard Worker #else
164*795d594fSAndroid Build Coastguard Worker       EXPECT_STRNE(real_root3.get(), "") << error_msg;
165*795d594fSAndroid Build Coastguard Worker #endif
166*795d594fSAndroid Build Coastguard Worker     }
167*795d594fSAndroid Build Coastguard Worker   }
168*795d594fSAndroid Build Coastguard Worker 
169*795d594fSAndroid Build Coastguard Worker   // Reset ANDROID_ART_ROOT, as other things may depend on it.
170*795d594fSAndroid Build Coastguard Worker   ASSERT_EQ(0, setenv("ANDROID_ART_ROOT", android_art_root_env.c_str(), /* overwrite */ 1));
171*795d594fSAndroid Build Coastguard Worker }
172*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,ReplaceFileExtension)173*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, ReplaceFileExtension) {
174*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/directory/file.vdex", ReplaceFileExtension("/directory/file.oat", "vdex"));
175*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/.directory/file.vdex", ReplaceFileExtension("/.directory/file.oat", "vdex"));
176*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/directory/file.vdex", ReplaceFileExtension("/directory/file", "vdex"));
177*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/.directory/file.vdex", ReplaceFileExtension("/.directory/file", "vdex"));
178*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/directory/file.vdex", ReplaceFileExtension("/directory/file.oat", ".vdex"));
179*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/directory/file.vdex", ReplaceFileExtension("/directory/file", ".vdex"));
180*795d594fSAndroid Build Coastguard Worker }
181*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,ArtApexDataPath)182*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, ArtApexDataPath) {
183*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable no_env("ART_APEX_DATA");
184*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(kArtApexDataDefaultPath, GetArtApexData());
185*795d594fSAndroid Build Coastguard Worker 
186*795d594fSAndroid Build Coastguard Worker   setenv("ART_APEX_DATA", "/path/from/env", /* overwrite */ 1);
187*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("/path/from/env", GetArtApexData());
188*795d594fSAndroid Build Coastguard Worker }
189*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetApexDataOatFilename)190*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetApexDataOatFilename) {
191*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable android_root("ANDROID_ROOT");
192*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable i18n_root("ANDROID_I18N_ROOT");
193*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable art_apex_data("ART_APEX_DATA");
194*795d594fSAndroid Build Coastguard Worker 
195*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() + "/dalvik-cache/arm/boot-beep.oat",
196*795d594fSAndroid Build Coastguard Worker             GetApexDataOatFilename("/product/javalib/beep.jar", InstructionSet::kArm));
197*795d594fSAndroid Build Coastguard Worker 
198*795d594fSAndroid Build Coastguard Worker   const std::string art_apex_jar = std::string {kAndroidArtApexDefaultPath} + "/javalib/some.jar";
199*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(std::string{}, GetApexDataOatFilename(art_apex_jar, InstructionSet::kArm));
200*795d594fSAndroid Build Coastguard Worker 
201*795d594fSAndroid Build Coastguard Worker   const std::string i18n_jar = std::string{kAndroidI18nApexDefaultPath} + "/javalib/core-icu4j.jar";
202*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(std::string{}, GetApexDataOatFilename(i18n_jar, InstructionSet::kArm));
203*795d594fSAndroid Build Coastguard Worker 
204*795d594fSAndroid Build Coastguard Worker   const std::string system_jar_apexdata_oat = GetArtApexData() + "/dalvik-cache/x86/boot-lace.oat";
205*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(system_jar_apexdata_oat,
206*795d594fSAndroid Build Coastguard Worker             GetApexDataOatFilename("/system/framework/lace.jar", InstructionSet::kX86));
207*795d594fSAndroid Build Coastguard Worker }
208*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetApexDataOdexFilename)209*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetApexDataOdexFilename) {
210*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable android_root("ANDROID_ROOT");
211*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable art_apex_data("ART_APEX_DATA");
212*795d594fSAndroid Build Coastguard Worker 
213*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() + "/dalvik-cache/arm/data@[email protected]",
214*795d594fSAndroid Build Coastguard Worker             GetApexDataOdexFilename("/data/some/code.dex", InstructionSet::kArm));
215*795d594fSAndroid Build Coastguard Worker 
216*795d594fSAndroid Build Coastguard Worker   const std::string art_apex_jar = std::string {kAndroidArtApexDefaultPath} + "/javalib/some.jar";
217*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(
218*795d594fSAndroid Build Coastguard Worker       GetArtApexData() + "/dalvik-cache/arm/apex@com.android.art@[email protected]@classes.odex",
219*795d594fSAndroid Build Coastguard Worker       GetApexDataOdexFilename(art_apex_jar, InstructionSet::kArm));
220*795d594fSAndroid Build Coastguard Worker 
221*795d594fSAndroid Build Coastguard Worker   const std::string i18n_jar = std::string{kAndroidI18nApexDefaultPath} + "/javalib/core-icu4j.jar";
222*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() +
223*795d594fSAndroid Build Coastguard Worker                 "/dalvik-cache/arm/apex@com.android.i18n@[email protected]@classes.odex",
224*795d594fSAndroid Build Coastguard Worker             GetApexDataOdexFilename(i18n_jar, InstructionSet::kArm));
225*795d594fSAndroid Build Coastguard Worker 
226*795d594fSAndroid Build Coastguard Worker   const std::string system_jar_apexdata_odex =
227*795d594fSAndroid Build Coastguard Worker       GetArtApexData() + "/dalvik-cache/x86/system@[email protected]@classes.odex";
228*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(system_jar_apexdata_odex,
229*795d594fSAndroid Build Coastguard Worker             GetApexDataOdexFilename("/system/framework/cookie.jar", InstructionSet::kX86));
230*795d594fSAndroid Build Coastguard Worker }
231*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetApexDataBootImage)232*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetApexDataBootImage) {
233*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable android_root("ANDROID_ROOT");
234*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable art_apex_data("ART_APEX_DATA");
235*795d594fSAndroid Build Coastguard Worker 
236*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(std::string{},
237*795d594fSAndroid Build Coastguard Worker             GetApexDataBootImage(std::string{kAndroidI18nApexDefaultPath} + "/javalib/bar.jar"));
238*795d594fSAndroid Build Coastguard Worker 
239*795d594fSAndroid Build Coastguard Worker   // Check image location has the prefix "boot-" in front of the basename of dex location and
240*795d594fSAndroid Build Coastguard Worker   // that image suffix is .art.
241*795d594fSAndroid Build Coastguard Worker   const std::string system_jar = "/system/framework/disk.jar";
242*795d594fSAndroid Build Coastguard Worker   const std::string boot_image = GetApexDataBootImage(system_jar);
243*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() + "/dalvik-cache/boot-disk.art", boot_image);
244*795d594fSAndroid Build Coastguard Worker 
245*795d594fSAndroid Build Coastguard Worker   // Check the image filename corresponds to the oat file for the same system jar.
246*795d594fSAndroid Build Coastguard Worker   const InstructionSet isa = InstructionSet::kArm64;
247*795d594fSAndroid Build Coastguard Worker   const std::string boot_image_filename = GetSystemImageFilename(boot_image.c_str(), isa);
248*795d594fSAndroid Build Coastguard Worker   const std::string accompanying_oat_file = ReplaceFileExtension(boot_image_filename, "oat");
249*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(accompanying_oat_file, GetApexDataOatFilename(system_jar, isa));
250*795d594fSAndroid Build Coastguard Worker }
251*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetApexDataImage)252*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetApexDataImage) {
253*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable android_root("ANDROID_ROOT");
254*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable art_apex_data("ART_APEX_DATA");
255*795d594fSAndroid Build Coastguard Worker 
256*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(
257*795d594fSAndroid Build Coastguard Worker       GetArtApexData() + "/dalvik-cache/apex@com.android.wifi@lib@[email protected]@classes.art",
258*795d594fSAndroid Build Coastguard Worker       GetApexDataImage(std::string {kAndroidWifiApexDefaultPath} + "/lib/javalib/bar.jar"));
259*795d594fSAndroid Build Coastguard Worker 
260*795d594fSAndroid Build Coastguard Worker   // Check image has basename of dex location with the .art suffix.
261*795d594fSAndroid Build Coastguard Worker   const char* jar = "/system/framework/mcguffin/test.jar";
262*795d594fSAndroid Build Coastguard Worker   const std::string image = GetApexDataImage(jar);
263*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() + "/dalvik-cache/system@framework@[email protected]@classes.art",
264*795d594fSAndroid Build Coastguard Worker             image);
265*795d594fSAndroid Build Coastguard Worker 
266*795d594fSAndroid Build Coastguard Worker   // Check the image filename corresponds to the .odex file for the same system jar.
267*795d594fSAndroid Build Coastguard Worker   const InstructionSet isa = InstructionSet::kX86_64;
268*795d594fSAndroid Build Coastguard Worker   const std::string image_filename = GetSystemImageFilename(image.c_str(), isa);
269*795d594fSAndroid Build Coastguard Worker   const std::string accompanying_odex_file = ReplaceFileExtension(image_filename, "odex");
270*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(accompanying_odex_file, GetApexDataOdexFilename(jar, isa));
271*795d594fSAndroid Build Coastguard Worker }
272*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetApexDataDalvikCacheFilename)273*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetApexDataDalvikCacheFilename) {
274*795d594fSAndroid Build Coastguard Worker   const std::string apex_jar = std::string {kAndroidWifiApexDefaultPath} + "/lib/javalib/bar.jar";
275*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() +
276*795d594fSAndroid Build Coastguard Worker                 "/dalvik-cache/x86_64/apex@com.android.wifi@lib@[email protected]@classes.art",
277*795d594fSAndroid Build Coastguard Worker             GetApexDataDalvikCacheFilename(apex_jar, InstructionSet::kX86_64, "art"));
278*795d594fSAndroid Build Coastguard Worker 
279*795d594fSAndroid Build Coastguard Worker   // Check dalvik-cache filename follows convention.
280*795d594fSAndroid Build Coastguard Worker   const std::string non_apex_jar = "/vendor/javalib/test.jar";
281*795d594fSAndroid Build Coastguard Worker   const std::string art_filename =
282*795d594fSAndroid Build Coastguard Worker       GetApexDataDalvikCacheFilename(non_apex_jar, InstructionSet::kArm, "art");
283*795d594fSAndroid Build Coastguard Worker   CHECK_EQ(GetArtApexData() + "/dalvik-cache/arm/vendor@[email protected]@classes.art",
284*795d594fSAndroid Build Coastguard Worker            art_filename);
285*795d594fSAndroid Build Coastguard Worker 
286*795d594fSAndroid Build Coastguard Worker   // Check ".art", ".odex" and ".vdex" filenames are the same with the appropriate extensions
287*795d594fSAndroid Build Coastguard Worker   // substituted.
288*795d594fSAndroid Build Coastguard Worker   const std::string odex_filename =
289*795d594fSAndroid Build Coastguard Worker       GetApexDataDalvikCacheFilename(non_apex_jar, InstructionSet::kArm, "odex");
290*795d594fSAndroid Build Coastguard Worker   CHECK_EQ(odex_filename, ReplaceFileExtension(art_filename, "odex"));
291*795d594fSAndroid Build Coastguard Worker   const std::string vdex_filename =
292*795d594fSAndroid Build Coastguard Worker       GetApexDataDalvikCacheFilename(non_apex_jar, InstructionSet::kArm, "vdex");
293*795d594fSAndroid Build Coastguard Worker   CHECK_EQ(vdex_filename, ReplaceFileExtension(art_filename, "vdex"));
294*795d594fSAndroid Build Coastguard Worker }
295*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,OverrideDalvikCacheSubDirectory)296*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, OverrideDalvikCacheSubDirectory) {
297*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable android_root("ANDROID_ROOT");
298*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable i18n_root("ANDROID_I18N_ROOT");
299*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable art_apex_data("ART_APEX_DATA");
300*795d594fSAndroid Build Coastguard Worker 
301*795d594fSAndroid Build Coastguard Worker   ScopedOverrideDalvikCacheSubDirectory dalvik_cache("overridden-cache");
302*795d594fSAndroid Build Coastguard Worker 
303*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() + "/overridden-cache/arm/boot-beep.oat",
304*795d594fSAndroid Build Coastguard Worker             GetApexDataOatFilename("/product/javalib/beep.jar", InstructionSet::kArm));
305*795d594fSAndroid Build Coastguard Worker 
306*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() + "/overridden-cache/arm/data@[email protected]",
307*795d594fSAndroid Build Coastguard Worker             GetApexDataOdexFilename("/data/some/code.dex", InstructionSet::kArm));
308*795d594fSAndroid Build Coastguard Worker 
309*795d594fSAndroid Build Coastguard Worker   const std::string system_jar = "/system/framework/disk.jar";
310*795d594fSAndroid Build Coastguard Worker   const std::string boot_image = GetApexDataBootImage(system_jar);
311*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() + "/overridden-cache/boot-disk.art", boot_image);
312*795d594fSAndroid Build Coastguard Worker 
313*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(
314*795d594fSAndroid Build Coastguard Worker       GetArtApexData() + "/overridden-cache/apex@com.android.wifi@lib@[email protected]@classes.art",
315*795d594fSAndroid Build Coastguard Worker       GetApexDataImage(std::string {kAndroidWifiApexDefaultPath} + "/lib/javalib/bar.jar"));
316*795d594fSAndroid Build Coastguard Worker 
317*795d594fSAndroid Build Coastguard Worker   const std::string apex_jar = std::string {kAndroidWifiApexDefaultPath} + "/lib/javalib/bar.jar";
318*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(GetArtApexData() +
319*795d594fSAndroid Build Coastguard Worker                 "/overridden-cache/x86_64/apex@com.android.wifi@lib@[email protected]@classes.art",
320*795d594fSAndroid Build Coastguard Worker             GetApexDataDalvikCacheFilename(apex_jar, InstructionSet::kX86_64, "art"));
321*795d594fSAndroid Build Coastguard Worker }
322*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,GetSystemOdexFilenameForApex)323*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, GetSystemOdexFilenameForApex) {
324*795d594fSAndroid Build Coastguard Worker   ScopedUnsetEnvironmentVariable android_root("ANDROID_ROOT");
325*795d594fSAndroid Build Coastguard Worker 
326*795d594fSAndroid Build Coastguard Worker   const std::string apex_jar = std::string {kAndroidArtApexDefaultPath} + "/javalib/some.jar";
327*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ(
328*795d594fSAndroid Build Coastguard Worker       GetSystemExtRoot() + "/framework/oat/arm/apex@com.android.art@[email protected]@classes.odex",
329*795d594fSAndroid Build Coastguard Worker       GetSystemOdexFilenameForApex(apex_jar, InstructionSet::kArm));
330*795d594fSAndroid Build Coastguard Worker }
331*795d594fSAndroid Build Coastguard Worker 
TEST_F(FileUtilsTest,ApexNameFromLocation)332*795d594fSAndroid Build Coastguard Worker TEST_F(FileUtilsTest, ApexNameFromLocation) {
333*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("", ApexNameFromLocation(""));
334*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("", ApexNameFromLocation("/apex/com.android.foo"));
335*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("", ApexNameFromLocation("/apex//something"));
336*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("com.android.foo", ApexNameFromLocation("/apex/com.android.foo/"));
337*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("", ApexNameFromLocation("apex/com.android.foo/"));
338*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("foo", ApexNameFromLocation("/apex/foo/something.jar"));
339*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("", ApexNameFromLocation("/bar/foo/baz"));
340*795d594fSAndroid Build Coastguard Worker   EXPECT_EQ("", ApexNameFromLocation("/apexx/foo/baz"));
341*795d594fSAndroid Build Coastguard Worker }
342*795d594fSAndroid Build Coastguard Worker 
343*795d594fSAndroid Build Coastguard Worker }  // namespace art
344