blob: f949b123adc8ea059a21e2e043681a8d7ce09ce0 [file] [log] [blame]
Colin Cross43f08db2018-11-12 10:13:39 -08001// Copyright 2018 Google Inc. All rights reserved.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package java
16
17import (
Jiakai Zhangca9bc982021-09-09 08:09:41 +000018 "path/filepath"
Jiakai Zhang51b2a8b2023-06-26 16:47:38 +010019 "sort"
Jiakai Zhangca9bc982021-09-09 08:09:41 +000020 "strings"
21
Spandan Das3dbda182024-05-20 22:23:10 +000022 "github.com/google/blueprint/proptools"
23
Colin Cross43f08db2018-11-12 10:13:39 -080024 "android/soong/android"
25 "android/soong/dexpreopt"
26)
27
Jiakai Zhangca9bc982021-09-09 08:09:41 +000028type DexpreopterInterface interface {
Jiakai Zhang81e46812023-02-08 21:56:07 +080029 // True if the java module is to be dexed and installed on devices.
30 // Structs that embed dexpreopter must implement this.
31 IsInstallable() bool
32
33 // True if dexpreopt is disabled for the java module.
Spandan Dase21a8d42024-01-23 23:56:29 +000034 dexpreoptDisabled(ctx android.BaseModuleContext, libraryName string) bool
Jiakai Zhang81e46812023-02-08 21:56:07 +080035
36 // If the java module is to be installed into an APEX, this list contains information about the
37 // dexpreopt outputs to be installed on devices. Note that these dexpreopt outputs are installed
38 // outside of the APEX.
Jiakai Zhangca9bc982021-09-09 08:09:41 +000039 DexpreoptBuiltInstalledForApex() []dexpreopterInstall
Jiakai Zhang81e46812023-02-08 21:56:07 +080040
41 // The Make entries to install the dexpreopt outputs. Derived from
42 // `DexpreoptBuiltInstalledForApex`.
Jiakai Zhangca9bc982021-09-09 08:09:41 +000043 AndroidMkEntriesForApex() []android.AndroidMkEntries
Jiakai Zhang81e46812023-02-08 21:56:07 +080044
45 // See `dexpreopter.outputProfilePathOnHost`.
46 OutputProfilePathOnHost() android.Path
Jiakai Zhangca9bc982021-09-09 08:09:41 +000047}
48
49type dexpreopterInstall struct {
50 // A unique name to distinguish an output from others for the same java library module. Usually in
51 // the form of `<arch>-<encoded-path>.odex/vdex/art`.
52 name string
53
54 // The name of the input java module.
55 moduleName string
56
57 // The path to the dexpreopt output on host.
58 outputPathOnHost android.Path
59
60 // The directory on the device for the output to install to.
61 installDirOnDevice android.InstallPath
62
63 // The basename (the last segment of the path) for the output to install as.
64 installFileOnDevice string
65}
66
67// The full module name of the output in the makefile.
68func (install *dexpreopterInstall) FullModuleName() string {
69 return install.moduleName + install.SubModuleName()
70}
71
72// The sub-module name of the output in the makefile (the name excluding the java module name).
73func (install *dexpreopterInstall) SubModuleName() string {
74 return "-dexpreopt-" + install.name
Martin Stjernholm6d415272020-01-31 17:10:36 +000075}
76
Jiakai Zhang6decef92022-01-12 17:56:19 +000077// Returns Make entries for installing the file.
78//
79// This function uses a value receiver rather than a pointer receiver to ensure that the object is
80// safe to use in `android.AndroidMkExtraEntriesFunc`.
81func (install dexpreopterInstall) ToMakeEntries() android.AndroidMkEntries {
82 return android.AndroidMkEntries{
83 Class: "ETC",
Jiakai Zhang6decef92022-01-12 17:56:19 +000084 OutputFile: android.OptionalPathForPath(install.outputPathOnHost),
85 ExtraEntries: []android.AndroidMkExtraEntriesFunc{
86 func(ctx android.AndroidMkExtraEntriesContext, entries *android.AndroidMkEntries) {
Spandan Das2069c3f2023-12-06 19:40:24 +000087 entries.SetString("LOCAL_MODULE", install.FullModuleName())
Jiakai Zhang6decef92022-01-12 17:56:19 +000088 entries.SetString("LOCAL_MODULE_PATH", install.installDirOnDevice.String())
89 entries.SetString("LOCAL_INSTALLED_MODULE_STEM", install.installFileOnDevice)
90 entries.SetString("LOCAL_NOT_AVAILABLE_FOR_PLATFORM", "false")
91 },
92 },
93 }
94}
95
Spandan Das2069c3f2023-12-06 19:40:24 +000096type Dexpreopter struct {
97 dexpreopter
98}
99
Colin Cross43f08db2018-11-12 10:13:39 -0800100type dexpreopter struct {
Jiakai Zhang9c4dc192023-02-09 00:09:24 +0800101 dexpreoptProperties DexpreoptProperties
102 importDexpreoptProperties ImportDexpreoptProperties
Colin Cross43f08db2018-11-12 10:13:39 -0800103
Spandan Das0727ba72024-02-13 16:37:43 +0000104 // If true, the dexpreopt rules will not be generated
105 // Unlike Dex_preopt.Enabled which is user-facing,
106 // shouldDisableDexpreopt is a mutated propery.
107 shouldDisableDexpreopt bool
108
Colin Cross70dda7e2019-10-01 22:05:35 -0700109 installPath android.InstallPath
Jaewoong Jungccbb3932019-04-15 09:48:31 -0700110 uncompressedDex bool
111 isSDKLibrary bool
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000112 isApp bool
Jaewoong Jungccbb3932019-04-15 09:48:31 -0700113 isTest bool
Jaewoong Jungccbb3932019-04-15 09:48:31 -0700114 isPresignedPrebuilt bool
Colin Crossfa9bfcd2021-11-10 16:42:38 -0800115 preventInstall bool
Colin Cross43f08db2018-11-12 10:13:39 -0800116
Ulya Trafimovich8cbc5d22020-11-03 15:15:46 +0000117 manifestFile android.Path
Ulya Trafimovich8c35fcf2021-02-17 16:23:28 +0000118 statusFile android.WritablePath
Ulya Trafimovich8cbc5d22020-11-03 15:15:46 +0000119 enforceUsesLibs bool
120 classLoaderContexts dexpreopt.ClassLoaderContextMap
Colin Cross50ddcc42019-05-16 12:28:22 -0700121
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000122 // See the `dexpreopt` function for details.
123 builtInstalled string
124 builtInstalledForApex []dexpreopterInstall
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000125
Jeongik Chac6246672021-04-08 00:00:19 +0900126 // The config is used for two purposes:
127 // - Passing dexpreopt information about libraries from Soong to Make. This is needed when
128 // a <uses-library> is defined in Android.bp, but used in Android.mk (see dex_preopt_config_merger.py).
129 // Note that dexpreopt.config might be needed even if dexpreopt is disabled for the library itself.
130 // - Dexpreopt post-processing (using dexpreopt artifacts from a prebuilt system image to incrementally
131 // dexpreopt another partition).
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000132 configPath android.WritablePath
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800133
Jiakai Zhang81e46812023-02-08 21:56:07 +0800134 // The path to the profile on host that dexpreopter generates. This is used as the input for
135 // dex2oat.
136 outputProfilePathOnHost android.Path
137
138 // The path to the profile that dexpreopter accepts. It must be in the binary format. If this is
139 // set, it overrides the profile settings in `dexpreoptProperties`.
140 inputProfilePathOnHost android.Path
Spandan Das3dbda182024-05-20 22:23:10 +0000141
142 // The path to the profile that matches the dex optimized by r8/d8. It is in text format. If this is
143 // set, it will be converted to a binary profile which will be subsequently used for dexpreopt.
144 rewrittenProfile android.Path
Colin Cross43f08db2018-11-12 10:13:39 -0800145}
146
147type DexpreoptProperties struct {
148 Dex_preopt struct {
Nicolas Geoffrayc1bf7242019-10-18 14:51:38 +0100149 // If false, prevent dexpreopting. Defaults to true.
Colin Cross43f08db2018-11-12 10:13:39 -0800150 Enabled *bool
151
152 // If true, generate an app image (.art file) for this module.
153 App_image *bool
154
155 // If true, use a checked-in profile to guide optimization. Defaults to false unless
156 // a matching profile is set or a profile is found in PRODUCT_DEX_PREOPT_PROFILE_DIR
157 // that matches the name of this module, in which case it is defaulted to true.
158 Profile_guided *bool
159
160 // If set, provides the path to profile relative to the Android.bp file. If not set,
161 // defaults to searching for a file that matches the name of this module in the default
162 // profile location set by PRODUCT_DEX_PREOPT_PROFILE_DIR, or empty if not found.
Colin Crossde4e4e62019-04-26 10:52:32 -0700163 Profile *string `android:"path"`
Spandan Das3dbda182024-05-20 22:23:10 +0000164
165 // If set to true, r8/d8 will use `profile` as input to generate a new profile that matches
166 // the optimized dex.
167 // The new profile will be subsequently used as the profile to dexpreopt the dex file.
168 Enable_profile_rewriting *bool
Colin Cross43f08db2018-11-12 10:13:39 -0800169 }
Jiakai Zhang9c4dc192023-02-09 00:09:24 +0800170
171 Dex_preopt_result struct {
172 // True if profile-guided optimization is actually enabled.
173 Profile_guided bool
174 } `blueprint:"mutated"`
175}
176
177type ImportDexpreoptProperties struct {
178 Dex_preopt struct {
179 // If true, use the profile in the prebuilt APEX to guide optimization. Defaults to false.
180 Profile_guided *bool
181 }
Colin Cross43f08db2018-11-12 10:13:39 -0800182}
183
Ulya Trafimovich6cf2c0c2020-04-24 12:15:20 +0100184func init() {
185 dexpreopt.DexpreoptRunningInSoong = true
186}
187
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000188func isApexVariant(ctx android.BaseModuleContext) bool {
Colin Crossff694a82023-12-13 15:54:49 -0800189 apexInfo, _ := android.ModuleProvider(ctx, android.ApexInfoProvider)
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000190 return !apexInfo.IsForPlatform()
191}
192
Jiakai Zhang28bc9a82021-12-20 15:08:57 +0000193func forPrebuiltApex(ctx android.BaseModuleContext) bool {
Colin Crossff694a82023-12-13 15:54:49 -0800194 apexInfo, _ := android.ModuleProvider(ctx, android.ApexInfoProvider)
Jiakai Zhang28bc9a82021-12-20 15:08:57 +0000195 return apexInfo.ForPrebuiltApex
196}
197
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000198// For apex variant of modules, this returns true on the source variant if the prebuilt apex
199// has been selected using apex_contributions.
200// The prebuilt apex will be responsible for generating the dexpreopt rules of the deapexed java lib.
201func disableSourceApexVariant(ctx android.BaseModuleContext) bool {
202 if !isApexVariant(ctx) {
203 return false // platform variant
204 }
205 apexInfo, _ := android.ModuleProvider(ctx, android.ApexInfoProvider)
206 psi := android.PrebuiltSelectionInfoMap{}
Jihoon Kanga3a05462024-04-05 00:36:44 +0000207 ctx.VisitDirectDeps(func(am android.Module) {
208 if prebuiltSelectionInfo, ok := android.OtherModuleProvider(ctx, am, android.PrebuiltSelectionInfoProvider); ok {
209 psi = prebuiltSelectionInfo
210 }
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000211 })
212 // Find the apex variant for this module
Spandan Dased7a0302024-08-26 18:06:25 +0000213 var apexVariantsWithoutTestApexes []string
214 if apexInfo.BaseApexName != "" {
215 // This is a transitive dependency of an override_apex
216 apexVariantsWithoutTestApexes = []string{apexInfo.BaseApexName}
217 } else {
218 _, apexVariantsWithoutTestApexes, _ = android.ListSetDifference(apexInfo.InApexVariants, apexInfo.TestApexes)
219 }
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000220 disableSource := false
221 // find the selected apexes
222 for _, apexVariant := range apexVariantsWithoutTestApexes {
Spandan Dased7a0302024-08-26 18:06:25 +0000223 if len(psi.GetSelectedModulesForApiDomain(apexVariant)) > 0 {
224 // If the apex_contribution for this api domain is non-empty, disable the source variant
225 disableSource = true
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000226 }
227 }
228 return disableSource
229}
230
Jiakai Zhangcf61e3c2023-05-08 16:28:38 +0000231// Returns whether dexpreopt is applicable to the module.
232// When it returns true, neither profile nor dexpreopt artifacts will be generated.
Spandan Dase21a8d42024-01-23 23:56:29 +0000233func (d *dexpreopter) dexpreoptDisabled(ctx android.BaseModuleContext, libName string) bool {
Colin Cross38310bb2021-12-01 10:34:14 -0800234 if !ctx.Device() {
Colin Cross43f08db2018-11-12 10:13:39 -0800235 return true
236 }
237
Colin Cross43f08db2018-11-12 10:13:39 -0800238 if d.isTest {
239 return true
240 }
241
242 if !BoolDefault(d.dexpreoptProperties.Dex_preopt.Enabled, true) {
243 return true
244 }
245
Spandan Das0727ba72024-02-13 16:37:43 +0000246 if d.shouldDisableDexpreopt {
247 return true
248 }
249
Jiakai Zhang28bc9a82021-12-20 15:08:57 +0000250 // If the module is from a prebuilt APEX, it shouldn't be installable, but it can still be
251 // dexpreopted.
252 if !ctx.Module().(DexpreopterInterface).IsInstallable() && !forPrebuiltApex(ctx) {
Martin Stjernholm6d415272020-01-31 17:10:36 +0000253 return true
254 }
255
Colin Cross38310bb2021-12-01 10:34:14 -0800256 if !android.IsModulePreferred(ctx.Module()) {
257 return true
258 }
259
Spandan Dase21a8d42024-01-23 23:56:29 +0000260 if _, isApex := android.ModuleProvider(ctx, android.ApexBundleInfoProvider); isApex {
261 // dexpreopt rules for system server jars can be generated in the ModuleCtx of prebuilt apexes
262 return false
263 }
264
Colin Cross38310bb2021-12-01 10:34:14 -0800265 global := dexpreopt.GetGlobalConfig(ctx)
266
Spandan Dase21a8d42024-01-23 23:56:29 +0000267 // Use the libName argument to determine if the library being dexpreopt'd is a system server jar
268 // ctx.ModuleName() is not safe. In case of prebuilt apexes, the dexpreopt rules of system server jars
269 // are created in the ctx object of the top-level prebuilt apex.
270 isApexSystemServerJar := global.AllApexSystemServerJars(ctx).ContainsJar(libName)
271
272 if _, isApex := android.ModuleProvider(ctx, android.ApexBundleInfoProvider); isApex || isApexVariant(ctx) {
273 // dexpreopt rules for system server jars can be generated in the ModuleCtx of prebuilt apexes
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800274 if !isApexSystemServerJar {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000275 return true
276 }
Spandan Das50801e22024-05-13 18:29:45 +0000277 ai, _ := android.ModuleProvider(ctx, android.ApexInfoProvider)
278 allApexInfos := []android.ApexInfo{}
279 if allApexInfosProvider, ok := android.ModuleProvider(ctx, android.AllApexInfoProvider); ok {
280 allApexInfos = allApexInfosProvider.ApexInfos
281 }
282 if len(allApexInfos) > 0 && !ai.MinSdkVersion.EqualTo(allApexInfos[0].MinSdkVersion) {
283 // Apex system server jars are dexpreopted and installed on to the system image.
284 // Since we can have BigAndroid and Go variants of system server jar providing apexes,
285 // and these two variants can have different min_sdk_versions, hide one of the apex variants
286 // from make to prevent collisions.
287 //
288 // Unlike cc, min_sdk_version does not have an effect on the build actions of java libraries.
289 ctx.Module().MakeUninstallable()
290 }
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000291 } else {
292 // Don't preopt the platform variant of an APEX system server jar to avoid conflicts.
Jiakai Zhang389a6472021-12-14 18:54:06 +0000293 if isApexSystemServerJar {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000294 return true
295 }
Yo Chiangdbdf8f92020-01-09 19:00:27 +0800296 }
297
Colin Cross43f08db2018-11-12 10:13:39 -0800298 // TODO: contains no java code
299
300 return false
301}
302
Martin Stjernholm6d415272020-01-31 17:10:36 +0000303func dexpreoptToolDepsMutator(ctx android.BottomUpMutatorContext) {
Spandan Dase21a8d42024-01-23 23:56:29 +0000304 if _, isApex := android.ModuleProvider(ctx, android.ApexBundleInfoProvider); isApex && dexpreopt.IsDex2oatNeeded(ctx) {
305 // prebuilt apexes can genererate rules to dexpreopt deapexed jars
306 // Add a dex2oat dep aggressively on _every_ apex module
307 dexpreopt.RegisterToolDeps(ctx)
308 return
309 }
310 if d, ok := ctx.Module().(DexpreopterInterface); !ok || d.dexpreoptDisabled(ctx, android.RemoveOptionalPrebuiltPrefix(ctx.ModuleName())) || !dexpreopt.IsDex2oatNeeded(ctx) {
Martin Stjernholm6d415272020-01-31 17:10:36 +0000311 return
312 }
313 dexpreopt.RegisterToolDeps(ctx)
314}
315
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000316// Returns the install path of the dex jar of a module.
317//
318// Do not rely on `ApexInfo.ApexVariationName` because it can be something like "apex1000", rather
319// than the `name` in the path `/apex/<name>` as suggested in its comment.
320//
321// This function is on a best-effort basis. It cannot handle the case where an APEX jar is not a
322// system server jar, which is fine because we currently only preopt system server jars for APEXes.
323func (d *dexpreopter) getInstallPath(
Spandan Dase21a8d42024-01-23 23:56:29 +0000324 ctx android.ModuleContext, libName string, defaultInstallPath android.InstallPath) android.InstallPath {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000325 global := dexpreopt.GetGlobalConfig(ctx)
Spandan Dase21a8d42024-01-23 23:56:29 +0000326 if global.AllApexSystemServerJars(ctx).ContainsJar(libName) {
327 dexLocation := dexpreopt.GetSystemServerDexLocation(ctx, global, libName)
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000328 return android.PathForModuleInPartitionInstall(ctx, "", strings.TrimPrefix(dexLocation, "/"))
329 }
Spandan Dase21a8d42024-01-23 23:56:29 +0000330 if !d.dexpreoptDisabled(ctx, libName) && isApexVariant(ctx) &&
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000331 filepath.Base(defaultInstallPath.PartitionDir()) != "apex" {
332 ctx.ModuleErrorf("unable to get the install path of the dex jar for dexpreopt")
333 }
334 return defaultInstallPath
Nicolas Geoffrayfa6e9ec2019-02-12 13:12:16 +0000335}
336
Spandan Das2069c3f2023-12-06 19:40:24 +0000337// DexpreoptPrebuiltApexSystemServerJars generates the dexpreopt artifacts from a jar file that has been deapexed from a prebuilt apex
338func (d *Dexpreopter) DexpreoptPrebuiltApexSystemServerJars(ctx android.ModuleContext, libraryName string, di *android.DeapexerInfo) {
339 // A single prebuilt apex can have multiple apex system jars
340 // initialize the output path for this dex jar
341 dc := dexpreopt.GetGlobalConfig(ctx)
342 d.installPath = android.PathForModuleInPartitionInstall(ctx, "", strings.TrimPrefix(dexpreopt.GetSystemServerDexLocation(ctx, dc, libraryName), "/"))
343 // generate the rules for creating the .odex and .vdex files for this system server jar
Spandan Das5be63332023-12-13 00:06:32 +0000344 dexJarFile := di.PrebuiltExportPath(ApexRootRelativePathToJavaLib(libraryName))
Spandan Das2ea84dd2024-01-25 22:12:50 +0000345
346 d.inputProfilePathOnHost = nil // reset: TODO(spandandas): Make dexpreopter stateless
347 if android.InList(libraryName, di.GetDexpreoptProfileGuidedExportedModuleNames()) {
348 // Set the profile path to guide optimization
349 prof := di.PrebuiltExportPath(ApexRootRelativePathToJavaLib(libraryName) + ".prof")
350 if prof == nil {
351 ctx.ModuleErrorf("Could not find a .prof file in this prebuilt apex")
352 }
353 d.inputProfilePathOnHost = prof
354 }
355
Spandan Dase21a8d42024-01-23 23:56:29 +0000356 d.dexpreopt(ctx, libraryName, dexJarFile)
Spandan Das2069c3f2023-12-06 19:40:24 +0000357}
358
Colin Cross7707b242024-07-26 12:02:36 -0700359func (d *dexpreopter) dexpreopt(ctx android.ModuleContext, libName string, dexJarFile android.Path) {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000360 global := dexpreopt.GetGlobalConfig(ctx)
361
Martin Stjernholm6d415272020-01-31 17:10:36 +0000362 // TODO(b/148690468): The check on d.installPath is to bail out in cases where
363 // the dexpreopter struct hasn't been fully initialized before we're called,
364 // e.g. in aar.go. This keeps the behaviour that dexpreopting is effectively
365 // disabled, even if installable is true.
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000366 if d.installPath.Base() == "." {
367 return
368 }
369
370 dexLocation := android.InstallPathToOnDevicePath(ctx, d.installPath)
371
Spandan Dase21a8d42024-01-23 23:56:29 +0000372 providesUsesLib := libName
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000373 if ulib, ok := ctx.Module().(ProvidesUsesLib); ok {
374 name := ulib.ProvidesUsesLib()
375 if name != nil {
376 providesUsesLib = *name
377 }
378 }
379
Jeongik Cha4b073cd2021-06-08 11:35:00 +0900380 // If it is test, make config files regardless of its dexpreopt setting.
Jeongik Chac6246672021-04-08 00:00:19 +0900381 // The config files are required for apps defined in make which depend on the lib.
Spandan Dase21a8d42024-01-23 23:56:29 +0000382 if d.isTest && d.dexpreoptDisabled(ctx, libName) {
Jaewoong Jung4b97a562020-12-17 09:43:28 -0800383 return
Nicolas Geoffrayfa6e9ec2019-02-12 13:12:16 +0000384 }
385
Spandan Dase21a8d42024-01-23 23:56:29 +0000386 isSystemServerJar := global.AllSystemServerJars(ctx).ContainsJar(libName)
Ulya Trafimovich9023b022021-03-22 16:02:28 +0000387
Colin Cross44df5812019-02-15 23:06:46 -0800388 bootImage := defaultBootImageConfig(ctx)
Jiakai Zhangb8796202023-03-06 19:16:48 +0000389 // When `global.PreoptWithUpdatableBcp` is true, `bcpForDexpreopt` below includes the mainline
390 // boot jars into bootclasspath, so we should include the mainline boot image as well because it's
391 // generated from those jars.
392 if global.PreoptWithUpdatableBcp {
393 bootImage = mainlineBootImageConfig(ctx)
394 }
Jiakai Zhang02669e82021-09-11 03:44:06 +0000395 dexFiles, dexLocations := bcpForDexpreopt(ctx, global.PreoptWithUpdatableBcp)
Ulya Trafimovich9023b022021-03-22 16:02:28 +0000396
David Srbeckyc177ebe2020-02-18 20:43:06 +0000397 targets := ctx.MultiTargets()
398 if len(targets) == 0 {
Colin Cross43f08db2018-11-12 10:13:39 -0800399 // assume this is a java library, dexpreopt for all arches for now
400 for _, target := range ctx.Config().Targets[android.Android] {
dimitry1f33e402019-03-26 12:39:31 +0100401 if target.NativeBridge == android.NativeBridgeDisabled {
David Srbeckyc177ebe2020-02-18 20:43:06 +0000402 targets = append(targets, target)
dimitry1f33e402019-03-26 12:39:31 +0100403 }
Colin Cross43f08db2018-11-12 10:13:39 -0800404 }
Spandan Dase21a8d42024-01-23 23:56:29 +0000405 if isSystemServerJar && libName != "com.android.location.provider" {
Jiakai Zhang2fbc3552022-11-28 15:38:23 +0000406 // If the module is a system server jar, only preopt for the primary arch because the jar can
407 // only be loaded by system server. "com.android.location.provider" is a special case because
408 // it's also used by apps as a shared library.
David Srbeckyc177ebe2020-02-18 20:43:06 +0000409 targets = targets[:1]
Colin Cross43f08db2018-11-12 10:13:39 -0800410 }
411 }
Colin Cross43f08db2018-11-12 10:13:39 -0800412
David Srbeckyc177ebe2020-02-18 20:43:06 +0000413 var archs []android.ArchType
Colin Cross69f59a32019-02-15 10:39:37 -0800414 var images android.Paths
Ulya Trafimovich4d2eeed2019-11-08 10:54:21 +0000415 var imagesDeps []android.OutputPaths
David Srbeckyc177ebe2020-02-18 20:43:06 +0000416 for _, target := range targets {
417 archs = append(archs, target.Arch.ArchType)
418 variant := bootImage.getVariant(target)
Jeongik Chaa5969092021-05-07 18:53:21 +0900419 images = append(images, variant.imagePathOnHost)
David Srbeckyc177ebe2020-02-18 20:43:06 +0000420 imagesDeps = append(imagesDeps, variant.imagesDeps)
Colin Crossc7e40aa2019-02-08 21:37:00 -0800421 }
David Srbeckyab994982020-03-30 17:24:13 +0100422 // The image locations for all Android variants are identical.
Jeongik Cha4dda75e2021-04-27 23:56:44 +0900423 hostImageLocations, deviceImageLocations := bootImage.getAnyAndroidVariant().imageLocations()
Colin Crossc7e40aa2019-02-08 21:37:00 -0800424
Colin Cross43f08db2018-11-12 10:13:39 -0800425 var profileClassListing android.OptionalPath
Nicolas Geoffraye7102422019-07-24 13:19:29 +0100426 var profileBootListing android.OptionalPath
Colin Cross43f08db2018-11-12 10:13:39 -0800427 profileIsTextListing := false
Spandan Das2ea84dd2024-01-25 22:12:50 +0000428
Jiakai Zhang81e46812023-02-08 21:56:07 +0800429 if d.inputProfilePathOnHost != nil {
430 profileClassListing = android.OptionalPathForPath(d.inputProfilePathOnHost)
431 } else if BoolDefault(d.dexpreoptProperties.Dex_preopt.Profile_guided, true) && !forPrebuiltApex(ctx) {
Spandan Das3dbda182024-05-20 22:23:10 +0000432 // If enable_profile_rewriting is set, use the rewritten profile instead of the checked-in profile
433 if d.EnableProfileRewriting() {
434 profileClassListing = android.OptionalPathForPath(d.GetRewrittenProfile())
435 profileIsTextListing = true
436 } else if profile := d.GetProfile(); profile != "" {
437 // If dex_preopt.profile_guided is not set, default it based on the existence of the
438 // dexprepot.profile option or the profile class listing.
Colin Cross43f08db2018-11-12 10:13:39 -0800439 profileClassListing = android.OptionalPathForPath(
Spandan Das3dbda182024-05-20 22:23:10 +0000440 android.PathForModuleSrc(ctx, profile))
Nicolas Geoffraye7102422019-07-24 13:19:29 +0100441 profileBootListing = android.ExistentPathForSource(ctx,
Spandan Das3dbda182024-05-20 22:23:10 +0000442 ctx.ModuleDir(), profile+"-boot")
Colin Cross43f08db2018-11-12 10:13:39 -0800443 profileIsTextListing = true
Dan Willemsen78d51b02020-06-24 16:33:31 -0700444 } else if global.ProfileDir != "" {
Colin Cross43f08db2018-11-12 10:13:39 -0800445 profileClassListing = android.ExistentPathForSource(ctx,
Spandan Dase21a8d42024-01-23 23:56:29 +0000446 global.ProfileDir, libName+".prof")
Colin Cross43f08db2018-11-12 10:13:39 -0800447 }
448 }
449
Jiakai Zhang9c4dc192023-02-09 00:09:24 +0800450 d.dexpreoptProperties.Dex_preopt_result.Profile_guided = profileClassListing.Valid()
451
Spandan Das2069c3f2023-12-06 19:40:24 +0000452 // A single apex can have multiple system server jars
453 // Use the dexJar to create a unique scope for each
454 dexJarStem := strings.TrimSuffix(dexJarFile.Base(), dexJarFile.Ext())
455
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000456 // Full dexpreopt config, used to create dexpreopt build rules.
Martin Stjernholm8d80cee2020-01-31 17:44:54 +0000457 dexpreoptConfig := &dexpreopt.ModuleConfig{
Spandan Dase21a8d42024-01-23 23:56:29 +0000458 Name: libName,
Victor Hsiehd181c8b2019-01-29 13:00:33 -0800459 DexLocation: dexLocation,
Spandan Dase21a8d42024-01-23 23:56:29 +0000460 BuildPath: android.PathForModuleOut(ctx, "dexpreopt", dexJarStem, libName+".jar").OutputPath,
Colin Cross69f59a32019-02-15 10:39:37 -0800461 DexPath: dexJarFile,
Jeongik Cha33a3a812021-04-15 09:12:49 +0900462 ManifestPath: android.OptionalPathForPath(d.manifestFile),
Victor Hsiehd181c8b2019-01-29 13:00:33 -0800463 UncompressedDex: d.uncompressedDex,
464 HasApkLibraries: false,
465 PreoptFlags: nil,
Colin Cross43f08db2018-11-12 10:13:39 -0800466
Colin Cross69f59a32019-02-15 10:39:37 -0800467 ProfileClassListing: profileClassListing,
Colin Cross43f08db2018-11-12 10:13:39 -0800468 ProfileIsTextListing: profileIsTextListing,
Nicolas Geoffraye7102422019-07-24 13:19:29 +0100469 ProfileBootListing: profileBootListing,
Colin Cross43f08db2018-11-12 10:13:39 -0800470
Ulya Trafimovich8c35fcf2021-02-17 16:23:28 +0000471 EnforceUsesLibrariesStatusFile: dexpreopt.UsesLibrariesStatusFile(ctx),
472 EnforceUsesLibraries: d.enforceUsesLibs,
473 ProvidesUsesLibrary: providesUsesLib,
474 ClassLoaderContexts: d.classLoaderContexts,
Colin Cross43f08db2018-11-12 10:13:39 -0800475
Jeongik Cha4dda75e2021-04-27 23:56:44 +0900476 Archs: archs,
477 DexPreoptImagesDeps: imagesDeps,
478 DexPreoptImageLocationsOnHost: hostImageLocations,
479 DexPreoptImageLocationsOnDevice: deviceImageLocations,
Colin Cross43f08db2018-11-12 10:13:39 -0800480
Ulya Trafimovich9023b022021-03-22 16:02:28 +0000481 PreoptBootClassPathDexFiles: dexFiles.Paths(),
Vladimir Marko40139d62020-02-06 15:14:29 +0000482 PreoptBootClassPathDexLocations: dexLocations,
Colin Cross800fe132019-02-11 14:21:24 -0800483
Colin Cross43f08db2018-11-12 10:13:39 -0800484 NoCreateAppImage: !BoolDefault(d.dexpreoptProperties.Dex_preopt.App_image, true),
485 ForceCreateAppImage: BoolDefault(d.dexpreoptProperties.Dex_preopt.App_image, false),
486
Jaewoong Jungccbb3932019-04-15 09:48:31 -0700487 PresignedPrebuilt: d.isPresignedPrebuilt,
Colin Cross43f08db2018-11-12 10:13:39 -0800488 }
489
Spandan Das2069c3f2023-12-06 19:40:24 +0000490 d.configPath = android.PathForModuleOut(ctx, "dexpreopt", dexJarStem, "dexpreopt.config")
Jeongik Chac6246672021-04-08 00:00:19 +0900491 dexpreopt.WriteModuleConfig(ctx, dexpreoptConfig, d.configPath)
492
Spandan Dase21a8d42024-01-23 23:56:29 +0000493 if d.dexpreoptDisabled(ctx, libName) {
Jeongik Chac6246672021-04-08 00:00:19 +0900494 return
495 }
496
497 globalSoong := dexpreopt.GetGlobalSoongConfig(ctx)
498
Jiakai Zhang51b2a8b2023-06-26 16:47:38 +0100499 // The root "product_packages.txt" is generated by `build/make/core/Makefile`. It contains a list
500 // of all packages that are installed on the device. We use `grep` to filter the list by the app's
501 // dependencies to create a per-app list, and use `rsync --checksum` to prevent the file's mtime
502 // from being changed if the contents don't change. This avoids unnecessary dexpreopt reruns.
Jiakai Zhanga4496782023-05-17 16:57:30 +0100503 productPackages := android.PathForModuleInPartitionInstall(ctx, "", "product_packages.txt")
Spandan Das2069c3f2023-12-06 19:40:24 +0000504 appProductPackages := android.PathForModuleOut(ctx, "dexpreopt", dexJarStem, "product_packages.txt")
Jiakai Zhang51b2a8b2023-06-26 16:47:38 +0100505 appProductPackagesStaging := appProductPackages.ReplaceExtension(ctx, "txt.tmp")
506 clcNames, _ := dexpreopt.ComputeClassLoaderContextDependencies(dexpreoptConfig.ClassLoaderContexts)
507 sort.Strings(clcNames) // The order needs to be deterministic.
508 productPackagesRule := android.NewRuleBuilder(pctx, ctx)
509 if len(clcNames) > 0 {
510 productPackagesRule.Command().
511 Text("grep -F -x").
512 FlagForEachArg("-e ", clcNames).
513 Input(productPackages).
514 FlagWithOutput("> ", appProductPackagesStaging).
515 Text("|| true")
516 } else {
517 productPackagesRule.Command().
518 Text("rm -f").Output(appProductPackagesStaging).
519 Text("&&").
520 Text("touch").Output(appProductPackagesStaging)
521 }
522 productPackagesRule.Command().
523 Text("rsync --checksum").
524 Input(appProductPackagesStaging).
525 Output(appProductPackages)
Spandan Das2069c3f2023-12-06 19:40:24 +0000526 productPackagesRule.Restat().Build("product_packages."+dexJarStem, "dexpreopt product_packages")
Jiakai Zhanga4496782023-05-17 16:57:30 +0100527
Spandan Das5ae65ee2024-04-16 22:03:26 +0000528 // Prebuilts are active, do not copy the dexpreopt'd source javalib to out/soong/system_server_dexjars
529 // The javalib from the deapexed prebuilt will be copied to this location.
530 // TODO (b/331665856): Implement a principled solution for this.
Spandan Das50801e22024-05-13 18:29:45 +0000531 copyApexSystemServerJarDex := !disableSourceApexVariant(ctx) && !ctx.Module().IsHideFromMake()
Jiakai Zhanga4496782023-05-17 16:57:30 +0100532 dexpreoptRule, err := dexpreopt.GenerateDexpreoptRule(
Spandan Das5ae65ee2024-04-16 22:03:26 +0000533 ctx, globalSoong, global, dexpreoptConfig, appProductPackages, copyApexSystemServerJarDex)
Colin Cross43f08db2018-11-12 10:13:39 -0800534 if err != nil {
535 ctx.ModuleErrorf("error generating dexpreopt rule: %s", err.Error())
Jaewoong Jung4b97a562020-12-17 09:43:28 -0800536 return
Colin Cross43f08db2018-11-12 10:13:39 -0800537 }
538
Spandan Das2069c3f2023-12-06 19:40:24 +0000539 dexpreoptRule.Build("dexpreopt"+"."+dexJarStem, "dexpreopt")
Colin Cross43f08db2018-11-12 10:13:39 -0800540
Spandan Das2069c3f2023-12-06 19:40:24 +0000541 // The current ctx might be of a deapexer module created by a prebuilt apex
542 // Use the path of the dex file to determine the library name
543 isApexSystemServerJar := global.AllApexSystemServerJars(ctx).ContainsJar(dexJarStem)
Jiakai Zhang389a6472021-12-14 18:54:06 +0000544
Justin Yun613bdc52024-06-12 21:32:10 +0900545 dexpreoptPartition := d.installPath.Partition()
546 // dexpreoptPartition is set to empty for dexpreopts of system APEX and system_other.
547 // In case of system APEX, however, we can set it to "system" manually.
548 // TODO(b/346662300): Let dexpreopter generate the installPath for dexpreopt files instead of
549 // using the dex location to generate the installPath.
550 if isApexSystemServerJar {
551 dexpreoptPartition = "system"
552 }
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700553 for _, install := range dexpreoptRule.Installs() {
554 // Remove the "/" prefix because the path should be relative to $ANDROID_PRODUCT_OUT.
555 installDir := strings.TrimPrefix(filepath.Dir(install.To), "/")
Justin Yun613bdc52024-06-12 21:32:10 +0900556 partition := dexpreoptPartition
Justin Yun22c8aca2024-06-05 20:25:03 +0900557 if strings.HasPrefix(installDir, partition+"/") {
558 installDir = strings.TrimPrefix(installDir, partition+"/")
559 } else {
560 // If the partition for the installDir is different from the install partition, set the
561 // partition empty to install the dexpreopt files to the desired partition.
562 // TODO(b/346439786): Define and use the dexpreopt module type to avoid this mismatch.
563 partition = ""
564 }
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700565 installBase := filepath.Base(install.To)
566 arch := filepath.Base(installDir)
Justin Yun22c8aca2024-06-05 20:25:03 +0900567 installPath := android.PathForModuleInPartitionInstall(ctx, partition, installDir)
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800568 isProfile := strings.HasSuffix(installBase, ".prof")
569
570 if isProfile {
Jiakai Zhang81e46812023-02-08 21:56:07 +0800571 d.outputProfilePathOnHost = install.From
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800572 }
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700573
Jiakai Zhang389a6472021-12-14 18:54:06 +0000574 if isApexSystemServerJar {
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800575 // Profiles are handled separately because they are installed into the APEX.
576 if !isProfile {
577 // APEX variants of java libraries are hidden from Make, so their dexpreopt
578 // outputs need special handling. Currently, for APEX variants of java
579 // libraries, only those in the system server classpath are handled here.
580 // Preopting of boot classpath jars in the ART APEX are handled in
581 // java/dexpreopt_bootjars.go, and other APEX jars are not preopted.
582 // The installs will be handled by Make as sub-modules of the java library.
Spandan Das746161d2024-08-21 22:47:53 +0000583 di := dexpreopterInstall{
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800584 name: arch + "-" + installBase,
Spandan Dase21a8d42024-01-23 23:56:29 +0000585 moduleName: libName,
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800586 outputPathOnHost: install.From,
587 installDirOnDevice: installPath,
588 installFileOnDevice: installBase,
Spandan Das746161d2024-08-21 22:47:53 +0000589 }
590 ctx.InstallFile(di.installDirOnDevice, di.installFileOnDevice, di.outputPathOnHost)
591 d.builtInstalledForApex = append(d.builtInstalledForApex, di)
592
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800593 }
Colin Crossfa9bfcd2021-11-10 16:42:38 -0800594 } else if !d.preventInstall {
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700595 ctx.InstallFile(installPath, installBase, install.From)
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000596 }
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700597 }
598
Jiakai Zhang389a6472021-12-14 18:54:06 +0000599 if !isApexSystemServerJar {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000600 d.builtInstalled = dexpreoptRule.Installs().String()
601 }
602}
603
Justin Yun22c8aca2024-06-05 20:25:03 +0900604func getModuleInstallPathInfo(ctx android.ModuleContext, fullInstallPath string) (android.InstallPath, string, string) {
605 installPath := android.PathForModuleInstall(ctx)
606 installDir, installBase := filepath.Split(strings.TrimPrefix(fullInstallPath, "/"))
607
608 if !strings.HasPrefix(installDir, installPath.Partition()+"/") {
609 // Return empty filename if the install partition is not for the target image.
610 return installPath, "", ""
611 }
612 relDir, err := filepath.Rel(installPath.Partition(), installDir)
613 if err != nil {
614 panic(err)
615 }
616 return installPath, relDir, installBase
617}
618
Spandan Das29207b52024-07-30 23:28:17 +0000619// installFile will install the file if `install` path and the target install partition are the same.
620func installFile(ctx android.ModuleContext, install android.RuleBuilderInstall) {
Justin Yun22c8aca2024-06-05 20:25:03 +0900621 installPath, relDir, name := getModuleInstallPathInfo(ctx, install.To)
622 // Empty name means the install partition is not for the target image.
623 // For the system image, files for "apex" and "system_other" are skipped here.
624 // The skipped "apex" files are for testing only, for example,
625 // "/apex/art_boot_images/javalib/x86/boot.vdex".
626 // TODO(b/320196894): Files for "system_other" are skipped because soong creates the system
627 // image only for now.
628 if name != "" {
Spandan Das29207b52024-07-30 23:28:17 +0000629 ctx.InstallFile(installPath.Join(ctx, relDir), name, install.From)
Justin Yun22c8aca2024-06-05 20:25:03 +0900630 }
631}
632
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000633func (d *dexpreopter) DexpreoptBuiltInstalledForApex() []dexpreopterInstall {
634 return d.builtInstalledForApex
635}
636
637func (d *dexpreopter) AndroidMkEntriesForApex() []android.AndroidMkEntries {
638 var entries []android.AndroidMkEntries
639 for _, install := range d.builtInstalledForApex {
Jiakai Zhang6decef92022-01-12 17:56:19 +0000640 entries = append(entries, install.ToMakeEntries())
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000641 }
642 return entries
Colin Cross43f08db2018-11-12 10:13:39 -0800643}
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800644
Jiakai Zhang81e46812023-02-08 21:56:07 +0800645func (d *dexpreopter) OutputProfilePathOnHost() android.Path {
646 return d.outputProfilePathOnHost
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800647}
Spandan Das0727ba72024-02-13 16:37:43 +0000648
649func (d *dexpreopter) disableDexpreopt() {
650 d.shouldDisableDexpreopt = true
651}
Spandan Das3dbda182024-05-20 22:23:10 +0000652
653func (d *dexpreopter) EnableProfileRewriting() bool {
654 return proptools.Bool(d.dexpreoptProperties.Dex_preopt.Enable_profile_rewriting)
655}
656
657func (d *dexpreopter) GetProfile() string {
658 return proptools.String(d.dexpreoptProperties.Dex_preopt.Profile)
659}
660
661func (d *dexpreopter) GetProfileGuided() bool {
662 return proptools.Bool(d.dexpreoptProperties.Dex_preopt.Profile_guided)
663}
664
665func (d *dexpreopter) GetRewrittenProfile() android.Path {
666 return d.rewrittenProfile
667}
668
669func (d *dexpreopter) SetRewrittenProfile(p android.Path) {
670 d.rewrittenProfile = p
671}