blob: 15e40ba0d09590e3e8ee4ffd1c80bcebe6bb3488 [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.
Colin Cross388c6612025-01-28 14:00:12 -080039 ApexSystemServerDexpreoptInstalls() []DexpreopterInstall
Jiakai Zhang81e46812023-02-08 21:56:07 +080040
Colin Cross388c6612025-01-28 14:00:12 -080041 // ApexSystemServerDexJars returns the list of dex jars if this is an apex system server jar.
42 ApexSystemServerDexJars() android.Paths
Jiakai Zhang81e46812023-02-08 21:56:07 +080043
44 // See `dexpreopter.outputProfilePathOnHost`.
45 OutputProfilePathOnHost() android.Path
Jiakai Zhangca9bc982021-09-09 08:09:41 +000046}
47
Colin Cross388c6612025-01-28 14:00:12 -080048type DexpreopterInstall struct {
Jiakai Zhangca9bc982021-09-09 08:09:41 +000049 // The path to the dexpreopt output on host.
Colin Cross388c6612025-01-28 14:00:12 -080050 OutputPathOnHost android.Path
Jiakai Zhangca9bc982021-09-09 08:09:41 +000051
52 // The directory on the device for the output to install to.
Colin Cross388c6612025-01-28 14:00:12 -080053 InstallDirOnDevice android.InstallPath
Jiakai Zhangca9bc982021-09-09 08:09:41 +000054
55 // The basename (the last segment of the path) for the output to install as.
Colin Cross388c6612025-01-28 14:00:12 -080056 InstallFileOnDevice string
Jihoon Kangd4063812025-01-24 00:25:30 +000057}
58
Spandan Das2069c3f2023-12-06 19:40:24 +000059type Dexpreopter struct {
60 dexpreopter
61}
62
Colin Cross43f08db2018-11-12 10:13:39 -080063type dexpreopter struct {
Jiakai Zhang9c4dc192023-02-09 00:09:24 +080064 dexpreoptProperties DexpreoptProperties
65 importDexpreoptProperties ImportDexpreoptProperties
Colin Cross43f08db2018-11-12 10:13:39 -080066
Spandan Das0727ba72024-02-13 16:37:43 +000067 // If true, the dexpreopt rules will not be generated
68 // Unlike Dex_preopt.Enabled which is user-facing,
69 // shouldDisableDexpreopt is a mutated propery.
70 shouldDisableDexpreopt bool
71
Colin Cross70dda7e2019-10-01 22:05:35 -070072 installPath android.InstallPath
Jaewoong Jungccbb3932019-04-15 09:48:31 -070073 uncompressedDex bool
74 isSDKLibrary bool
Ulya Trafimovich76b08522021-01-14 17:52:43 +000075 isApp bool
Jaewoong Jungccbb3932019-04-15 09:48:31 -070076 isTest bool
Jaewoong Jungccbb3932019-04-15 09:48:31 -070077 isPresignedPrebuilt bool
Colin Crossfa9bfcd2021-11-10 16:42:38 -080078 preventInstall bool
Colin Cross43f08db2018-11-12 10:13:39 -080079
Ulya Trafimovich8cbc5d22020-11-03 15:15:46 +000080 manifestFile android.Path
Ulya Trafimovich8c35fcf2021-02-17 16:23:28 +000081 statusFile android.WritablePath
Ulya Trafimovich8cbc5d22020-11-03 15:15:46 +000082 enforceUsesLibs bool
83 classLoaderContexts dexpreopt.ClassLoaderContextMap
Colin Cross50ddcc42019-05-16 12:28:22 -070084
Jiakai Zhangca9bc982021-09-09 08:09:41 +000085 // See the `dexpreopt` function for details.
Colin Cross388c6612025-01-28 14:00:12 -080086 builtInstalled string
87 apexSystemServerDexpreoptInstalls []DexpreopterInstall
88 apexSystemServerDexJars android.Paths
Ulya Trafimovich76b08522021-01-14 17:52:43 +000089
Jeongik Chac6246672021-04-08 00:00:19 +090090 // The config is used for two purposes:
91 // - Passing dexpreopt information about libraries from Soong to Make. This is needed when
92 // a <uses-library> is defined in Android.bp, but used in Android.mk (see dex_preopt_config_merger.py).
93 // Note that dexpreopt.config might be needed even if dexpreopt is disabled for the library itself.
94 // - Dexpreopt post-processing (using dexpreopt artifacts from a prebuilt system image to incrementally
95 // dexpreopt another partition).
Ulya Trafimovich76b08522021-01-14 17:52:43 +000096 configPath android.WritablePath
Jiakai Zhang3317ce72023-02-08 01:19:19 +080097
Jiakai Zhang81e46812023-02-08 21:56:07 +080098 // The path to the profile on host that dexpreopter generates. This is used as the input for
99 // dex2oat.
100 outputProfilePathOnHost android.Path
101
102 // The path to the profile that dexpreopter accepts. It must be in the binary format. If this is
103 // set, it overrides the profile settings in `dexpreoptProperties`.
104 inputProfilePathOnHost android.Path
Spandan Das3dbda182024-05-20 22:23:10 +0000105
106 // The path to the profile that matches the dex optimized by r8/d8. It is in text format. If this is
107 // set, it will be converted to a binary profile which will be subsequently used for dexpreopt.
108 rewrittenProfile android.Path
Colin Cross43f08db2018-11-12 10:13:39 -0800109}
110
111type DexpreoptProperties struct {
112 Dex_preopt struct {
Nicolas Geoffrayc1bf7242019-10-18 14:51:38 +0100113 // If false, prevent dexpreopting. Defaults to true.
Cole Fausteb032462024-09-19 11:12:54 -0700114 Enabled proptools.Configurable[bool] `android:"replace_instead_of_append"`
Colin Cross43f08db2018-11-12 10:13:39 -0800115
116 // If true, generate an app image (.art file) for this module.
Cole Fausteb032462024-09-19 11:12:54 -0700117 App_image proptools.Configurable[bool] `android:"replace_instead_of_append"`
Colin Cross43f08db2018-11-12 10:13:39 -0800118
119 // If true, use a checked-in profile to guide optimization. Defaults to false unless
120 // a matching profile is set or a profile is found in PRODUCT_DEX_PREOPT_PROFILE_DIR
121 // that matches the name of this module, in which case it is defaulted to true.
Cole Fausteb032462024-09-19 11:12:54 -0700122 Profile_guided proptools.Configurable[bool] `android:"replace_instead_of_append"`
Colin Cross43f08db2018-11-12 10:13:39 -0800123
124 // If set, provides the path to profile relative to the Android.bp file. If not set,
125 // defaults to searching for a file that matches the name of this module in the default
126 // profile location set by PRODUCT_DEX_PREOPT_PROFILE_DIR, or empty if not found.
Cole Fausteb032462024-09-19 11:12:54 -0700127 Profile proptools.Configurable[string] `android:"path,replace_instead_of_append"`
Spandan Das3dbda182024-05-20 22:23:10 +0000128
129 // If set to true, r8/d8 will use `profile` as input to generate a new profile that matches
130 // the optimized dex.
131 // The new profile will be subsequently used as the profile to dexpreopt the dex file.
Cole Fausteb032462024-09-19 11:12:54 -0700132 Enable_profile_rewriting proptools.Configurable[bool] `android:"replace_instead_of_append"`
Colin Cross43f08db2018-11-12 10:13:39 -0800133 }
Jiakai Zhang9c4dc192023-02-09 00:09:24 +0800134
135 Dex_preopt_result struct {
136 // True if profile-guided optimization is actually enabled.
137 Profile_guided bool
138 } `blueprint:"mutated"`
139}
140
141type ImportDexpreoptProperties struct {
142 Dex_preopt struct {
143 // If true, use the profile in the prebuilt APEX to guide optimization. Defaults to false.
144 Profile_guided *bool
145 }
Colin Cross43f08db2018-11-12 10:13:39 -0800146}
147
Ulya Trafimovich6cf2c0c2020-04-24 12:15:20 +0100148func init() {
149 dexpreopt.DexpreoptRunningInSoong = true
150}
151
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000152func isApexVariant(ctx android.BaseModuleContext) bool {
Colin Crossff694a82023-12-13 15:54:49 -0800153 apexInfo, _ := android.ModuleProvider(ctx, android.ApexInfoProvider)
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000154 return !apexInfo.IsForPlatform()
155}
156
Jiakai Zhang28bc9a82021-12-20 15:08:57 +0000157func forPrebuiltApex(ctx android.BaseModuleContext) bool {
Colin Crossff694a82023-12-13 15:54:49 -0800158 apexInfo, _ := android.ModuleProvider(ctx, android.ApexInfoProvider)
Jiakai Zhang28bc9a82021-12-20 15:08:57 +0000159 return apexInfo.ForPrebuiltApex
160}
161
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000162// For apex variant of modules, this returns true on the source variant if the prebuilt apex
163// has been selected using apex_contributions.
164// The prebuilt apex will be responsible for generating the dexpreopt rules of the deapexed java lib.
165func disableSourceApexVariant(ctx android.BaseModuleContext) bool {
166 if !isApexVariant(ctx) {
167 return false // platform variant
168 }
169 apexInfo, _ := android.ModuleProvider(ctx, android.ApexInfoProvider)
170 psi := android.PrebuiltSelectionInfoMap{}
Yu Liu7eebf8b2025-01-17 00:23:57 +0000171 ctx.VisitDirectDepsProxy(func(am android.ModuleProxy) {
Jihoon Kanga3a05462024-04-05 00:36:44 +0000172 if prebuiltSelectionInfo, ok := android.OtherModuleProvider(ctx, am, android.PrebuiltSelectionInfoProvider); ok {
173 psi = prebuiltSelectionInfo
174 }
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000175 })
Spandan Das003452f2024-09-06 00:56:25 +0000176
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000177 // Find the apex variant for this module
Colin Crossaf4c8562024-11-20 15:07:58 -0800178 apexVariants := []string{}
Spandan Dased7a0302024-08-26 18:06:25 +0000179 if apexInfo.BaseApexName != "" {
180 // This is a transitive dependency of an override_apex
Colin Crossaf4c8562024-11-20 15:07:58 -0800181 apexVariants = append(apexVariants, apexInfo.BaseApexName)
Spandan Dased7a0302024-08-26 18:06:25 +0000182 } else {
Colin Crossaf4c8562024-11-20 15:07:58 -0800183 apexVariants = append(apexVariants, apexInfo.InApexVariants...)
Spandan Das003452f2024-09-06 00:56:25 +0000184 }
185 if apexInfo.ApexAvailableName != "" {
Colin Crossaf4c8562024-11-20 15:07:58 -0800186 apexVariants = append(apexVariants, apexInfo.ApexAvailableName)
Spandan Dased7a0302024-08-26 18:06:25 +0000187 }
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000188 disableSource := false
189 // find the selected apexes
Colin Crossaf4c8562024-11-20 15:07:58 -0800190 for _, apexVariant := range apexVariants {
Spandan Dased7a0302024-08-26 18:06:25 +0000191 if len(psi.GetSelectedModulesForApiDomain(apexVariant)) > 0 {
192 // If the apex_contribution for this api domain is non-empty, disable the source variant
193 disableSource = true
Spandan Dasa8afdcb2024-02-29 06:40:16 +0000194 }
195 }
196 return disableSource
197}
198
Jiakai Zhangcf61e3c2023-05-08 16:28:38 +0000199// Returns whether dexpreopt is applicable to the module.
200// When it returns true, neither profile nor dexpreopt artifacts will be generated.
Spandan Dase21a8d42024-01-23 23:56:29 +0000201func (d *dexpreopter) dexpreoptDisabled(ctx android.BaseModuleContext, libName string) bool {
Colin Cross38310bb2021-12-01 10:34:14 -0800202 if !ctx.Device() {
Colin Cross43f08db2018-11-12 10:13:39 -0800203 return true
204 }
205
Colin Cross43f08db2018-11-12 10:13:39 -0800206 if d.isTest {
207 return true
208 }
209
Cole Fausteb032462024-09-19 11:12:54 -0700210 if !d.dexpreoptProperties.Dex_preopt.Enabled.GetOrDefault(ctx, true) {
Colin Cross43f08db2018-11-12 10:13:39 -0800211 return true
212 }
213
Spandan Das0727ba72024-02-13 16:37:43 +0000214 if d.shouldDisableDexpreopt {
215 return true
216 }
217
Jiakai Zhang28bc9a82021-12-20 15:08:57 +0000218 // If the module is from a prebuilt APEX, it shouldn't be installable, but it can still be
219 // dexpreopted.
220 if !ctx.Module().(DexpreopterInterface).IsInstallable() && !forPrebuiltApex(ctx) {
Martin Stjernholm6d415272020-01-31 17:10:36 +0000221 return true
222 }
223
Colin Cross38310bb2021-12-01 10:34:14 -0800224 if !android.IsModulePreferred(ctx.Module()) {
225 return true
226 }
227
Spandan Dase21a8d42024-01-23 23:56:29 +0000228 if _, isApex := android.ModuleProvider(ctx, android.ApexBundleInfoProvider); isApex {
229 // dexpreopt rules for system server jars can be generated in the ModuleCtx of prebuilt apexes
230 return false
231 }
232
Colin Cross38310bb2021-12-01 10:34:14 -0800233 global := dexpreopt.GetGlobalConfig(ctx)
234
Spandan Dase21a8d42024-01-23 23:56:29 +0000235 // Use the libName argument to determine if the library being dexpreopt'd is a system server jar
236 // ctx.ModuleName() is not safe. In case of prebuilt apexes, the dexpreopt rules of system server jars
237 // are created in the ctx object of the top-level prebuilt apex.
238 isApexSystemServerJar := global.AllApexSystemServerJars(ctx).ContainsJar(libName)
239
240 if _, isApex := android.ModuleProvider(ctx, android.ApexBundleInfoProvider); isApex || isApexVariant(ctx) {
241 // dexpreopt rules for system server jars can be generated in the ModuleCtx of prebuilt apexes
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800242 if !isApexSystemServerJar {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000243 return true
244 }
245 } else {
246 // Don't preopt the platform variant of an APEX system server jar to avoid conflicts.
Jiakai Zhang389a6472021-12-14 18:54:06 +0000247 if isApexSystemServerJar {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000248 return true
249 }
Yo Chiangdbdf8f92020-01-09 19:00:27 +0800250 }
251
Colin Cross43f08db2018-11-12 10:13:39 -0800252 // TODO: contains no java code
253
254 return false
255}
256
Martin Stjernholm6d415272020-01-31 17:10:36 +0000257func dexpreoptToolDepsMutator(ctx android.BottomUpMutatorContext) {
Spandan Dase21a8d42024-01-23 23:56:29 +0000258 if _, isApex := android.ModuleProvider(ctx, android.ApexBundleInfoProvider); isApex && dexpreopt.IsDex2oatNeeded(ctx) {
259 // prebuilt apexes can genererate rules to dexpreopt deapexed jars
260 // Add a dex2oat dep aggressively on _every_ apex module
261 dexpreopt.RegisterToolDeps(ctx)
262 return
263 }
264 if d, ok := ctx.Module().(DexpreopterInterface); !ok || d.dexpreoptDisabled(ctx, android.RemoveOptionalPrebuiltPrefix(ctx.ModuleName())) || !dexpreopt.IsDex2oatNeeded(ctx) {
Martin Stjernholm6d415272020-01-31 17:10:36 +0000265 return
266 }
267 dexpreopt.RegisterToolDeps(ctx)
268}
269
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000270// Returns the install path of the dex jar of a module.
271//
272// Do not rely on `ApexInfo.ApexVariationName` because it can be something like "apex1000", rather
273// than the `name` in the path `/apex/<name>` as suggested in its comment.
274//
275// This function is on a best-effort basis. It cannot handle the case where an APEX jar is not a
276// system server jar, which is fine because we currently only preopt system server jars for APEXes.
277func (d *dexpreopter) getInstallPath(
Spandan Dase21a8d42024-01-23 23:56:29 +0000278 ctx android.ModuleContext, libName string, defaultInstallPath android.InstallPath) android.InstallPath {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000279 global := dexpreopt.GetGlobalConfig(ctx)
Spandan Dase21a8d42024-01-23 23:56:29 +0000280 if global.AllApexSystemServerJars(ctx).ContainsJar(libName) {
281 dexLocation := dexpreopt.GetSystemServerDexLocation(ctx, global, libName)
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000282 return android.PathForModuleInPartitionInstall(ctx, "", strings.TrimPrefix(dexLocation, "/"))
283 }
Spandan Dase21a8d42024-01-23 23:56:29 +0000284 if !d.dexpreoptDisabled(ctx, libName) && isApexVariant(ctx) &&
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000285 filepath.Base(defaultInstallPath.PartitionDir()) != "apex" {
286 ctx.ModuleErrorf("unable to get the install path of the dex jar for dexpreopt")
287 }
288 return defaultInstallPath
Nicolas Geoffrayfa6e9ec2019-02-12 13:12:16 +0000289}
290
Spandan Das2069c3f2023-12-06 19:40:24 +0000291// DexpreoptPrebuiltApexSystemServerJars generates the dexpreopt artifacts from a jar file that has been deapexed from a prebuilt apex
292func (d *Dexpreopter) DexpreoptPrebuiltApexSystemServerJars(ctx android.ModuleContext, libraryName string, di *android.DeapexerInfo) {
293 // A single prebuilt apex can have multiple apex system jars
294 // initialize the output path for this dex jar
295 dc := dexpreopt.GetGlobalConfig(ctx)
296 d.installPath = android.PathForModuleInPartitionInstall(ctx, "", strings.TrimPrefix(dexpreopt.GetSystemServerDexLocation(ctx, dc, libraryName), "/"))
297 // generate the rules for creating the .odex and .vdex files for this system server jar
Spandan Das5be63332023-12-13 00:06:32 +0000298 dexJarFile := di.PrebuiltExportPath(ApexRootRelativePathToJavaLib(libraryName))
Spandan Das419b3c62025-01-13 21:46:01 +0000299 if dexJarFile == nil {
300 ctx.ModuleErrorf(
301 `Could not find library %s in prebuilt apex %s.
302Please make sure that the value of PRODUCT_APEX_(SYSTEM_SERVER|STANDALONE_SYSTEM_SERVER)_JARS is correct`, libraryName, ctx.ModuleName())
303 }
Spandan Das2ea84dd2024-01-25 22:12:50 +0000304 d.inputProfilePathOnHost = nil // reset: TODO(spandandas): Make dexpreopter stateless
305 if android.InList(libraryName, di.GetDexpreoptProfileGuidedExportedModuleNames()) {
306 // Set the profile path to guide optimization
307 prof := di.PrebuiltExportPath(ApexRootRelativePathToJavaLib(libraryName) + ".prof")
308 if prof == nil {
309 ctx.ModuleErrorf("Could not find a .prof file in this prebuilt apex")
310 }
311 d.inputProfilePathOnHost = prof
312 }
313
Spandan Dase21a8d42024-01-23 23:56:29 +0000314 d.dexpreopt(ctx, libraryName, dexJarFile)
Spandan Das2069c3f2023-12-06 19:40:24 +0000315}
316
Colin Cross7707b242024-07-26 12:02:36 -0700317func (d *dexpreopter) dexpreopt(ctx android.ModuleContext, libName string, dexJarFile android.Path) {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000318 global := dexpreopt.GetGlobalConfig(ctx)
319
Martin Stjernholm6d415272020-01-31 17:10:36 +0000320 // TODO(b/148690468): The check on d.installPath is to bail out in cases where
321 // the dexpreopter struct hasn't been fully initialized before we're called,
322 // e.g. in aar.go. This keeps the behaviour that dexpreopting is effectively
323 // disabled, even if installable is true.
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000324 if d.installPath.Base() == "." {
325 return
326 }
327
328 dexLocation := android.InstallPathToOnDevicePath(ctx, d.installPath)
329
Spandan Dase21a8d42024-01-23 23:56:29 +0000330 providesUsesLib := libName
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000331 if ulib, ok := ctx.Module().(ProvidesUsesLib); ok {
332 name := ulib.ProvidesUsesLib()
333 if name != nil {
334 providesUsesLib = *name
335 }
336 }
337
Jeongik Cha4b073cd2021-06-08 11:35:00 +0900338 // If it is test, make config files regardless of its dexpreopt setting.
Jeongik Chac6246672021-04-08 00:00:19 +0900339 // The config files are required for apps defined in make which depend on the lib.
Spandan Dase21a8d42024-01-23 23:56:29 +0000340 if d.isTest && d.dexpreoptDisabled(ctx, libName) {
Jaewoong Jung4b97a562020-12-17 09:43:28 -0800341 return
Nicolas Geoffrayfa6e9ec2019-02-12 13:12:16 +0000342 }
343
Spandan Dase21a8d42024-01-23 23:56:29 +0000344 isSystemServerJar := global.AllSystemServerJars(ctx).ContainsJar(libName)
Ulya Trafimovich9023b022021-03-22 16:02:28 +0000345
Colin Cross44df5812019-02-15 23:06:46 -0800346 bootImage := defaultBootImageConfig(ctx)
Jiakai Zhangb8796202023-03-06 19:16:48 +0000347 // When `global.PreoptWithUpdatableBcp` is true, `bcpForDexpreopt` below includes the mainline
348 // boot jars into bootclasspath, so we should include the mainline boot image as well because it's
349 // generated from those jars.
350 if global.PreoptWithUpdatableBcp {
351 bootImage = mainlineBootImageConfig(ctx)
352 }
Jiakai Zhang02669e82021-09-11 03:44:06 +0000353 dexFiles, dexLocations := bcpForDexpreopt(ctx, global.PreoptWithUpdatableBcp)
Ulya Trafimovich9023b022021-03-22 16:02:28 +0000354
David Srbeckyc177ebe2020-02-18 20:43:06 +0000355 targets := ctx.MultiTargets()
356 if len(targets) == 0 {
Colin Cross43f08db2018-11-12 10:13:39 -0800357 // assume this is a java library, dexpreopt for all arches for now
358 for _, target := range ctx.Config().Targets[android.Android] {
dimitry1f33e402019-03-26 12:39:31 +0100359 if target.NativeBridge == android.NativeBridgeDisabled {
David Srbeckyc177ebe2020-02-18 20:43:06 +0000360 targets = append(targets, target)
dimitry1f33e402019-03-26 12:39:31 +0100361 }
Colin Cross43f08db2018-11-12 10:13:39 -0800362 }
Spandan Dase21a8d42024-01-23 23:56:29 +0000363 if isSystemServerJar && libName != "com.android.location.provider" {
Jiakai Zhang2fbc3552022-11-28 15:38:23 +0000364 // If the module is a system server jar, only preopt for the primary arch because the jar can
365 // only be loaded by system server. "com.android.location.provider" is a special case because
366 // it's also used by apps as a shared library.
David Srbeckyc177ebe2020-02-18 20:43:06 +0000367 targets = targets[:1]
Colin Cross43f08db2018-11-12 10:13:39 -0800368 }
369 }
Colin Cross43f08db2018-11-12 10:13:39 -0800370
David Srbeckyc177ebe2020-02-18 20:43:06 +0000371 var archs []android.ArchType
Colin Cross69f59a32019-02-15 10:39:37 -0800372 var images android.Paths
Ulya Trafimovich4d2eeed2019-11-08 10:54:21 +0000373 var imagesDeps []android.OutputPaths
David Srbeckyc177ebe2020-02-18 20:43:06 +0000374 for _, target := range targets {
375 archs = append(archs, target.Arch.ArchType)
376 variant := bootImage.getVariant(target)
Jeongik Chaa5969092021-05-07 18:53:21 +0900377 images = append(images, variant.imagePathOnHost)
David Srbeckyc177ebe2020-02-18 20:43:06 +0000378 imagesDeps = append(imagesDeps, variant.imagesDeps)
Colin Crossc7e40aa2019-02-08 21:37:00 -0800379 }
David Srbeckyab994982020-03-30 17:24:13 +0100380 // The image locations for all Android variants are identical.
Jeongik Cha4dda75e2021-04-27 23:56:44 +0900381 hostImageLocations, deviceImageLocations := bootImage.getAnyAndroidVariant().imageLocations()
Colin Crossc7e40aa2019-02-08 21:37:00 -0800382
Colin Cross43f08db2018-11-12 10:13:39 -0800383 var profileClassListing android.OptionalPath
Nicolas Geoffraye7102422019-07-24 13:19:29 +0100384 var profileBootListing android.OptionalPath
Colin Cross43f08db2018-11-12 10:13:39 -0800385 profileIsTextListing := false
Spandan Das2ea84dd2024-01-25 22:12:50 +0000386
Jiakai Zhang81e46812023-02-08 21:56:07 +0800387 if d.inputProfilePathOnHost != nil {
388 profileClassListing = android.OptionalPathForPath(d.inputProfilePathOnHost)
Cole Fausteb032462024-09-19 11:12:54 -0700389 } else if d.dexpreoptProperties.Dex_preopt.Profile_guided.GetOrDefault(ctx, true) && !forPrebuiltApex(ctx) {
Spandan Das3dbda182024-05-20 22:23:10 +0000390 // If enable_profile_rewriting is set, use the rewritten profile instead of the checked-in profile
Cole Fausteb032462024-09-19 11:12:54 -0700391 if d.EnableProfileRewriting(ctx) {
Spandan Das3dbda182024-05-20 22:23:10 +0000392 profileClassListing = android.OptionalPathForPath(d.GetRewrittenProfile())
393 profileIsTextListing = true
Cole Fausteb032462024-09-19 11:12:54 -0700394 } else if profile := d.GetProfile(ctx); profile != "" {
Spandan Das3dbda182024-05-20 22:23:10 +0000395 // If dex_preopt.profile_guided is not set, default it based on the existence of the
396 // dexprepot.profile option or the profile class listing.
Colin Cross43f08db2018-11-12 10:13:39 -0800397 profileClassListing = android.OptionalPathForPath(
Spandan Das3dbda182024-05-20 22:23:10 +0000398 android.PathForModuleSrc(ctx, profile))
Nicolas Geoffraye7102422019-07-24 13:19:29 +0100399 profileBootListing = android.ExistentPathForSource(ctx,
Spandan Das3dbda182024-05-20 22:23:10 +0000400 ctx.ModuleDir(), profile+"-boot")
Colin Cross43f08db2018-11-12 10:13:39 -0800401 profileIsTextListing = true
Dan Willemsen78d51b02020-06-24 16:33:31 -0700402 } else if global.ProfileDir != "" {
Colin Cross43f08db2018-11-12 10:13:39 -0800403 profileClassListing = android.ExistentPathForSource(ctx,
Spandan Dase21a8d42024-01-23 23:56:29 +0000404 global.ProfileDir, libName+".prof")
Colin Cross43f08db2018-11-12 10:13:39 -0800405 }
406 }
407
Jiakai Zhang9c4dc192023-02-09 00:09:24 +0800408 d.dexpreoptProperties.Dex_preopt_result.Profile_guided = profileClassListing.Valid()
409
Spandan Das2069c3f2023-12-06 19:40:24 +0000410 // A single apex can have multiple system server jars
411 // Use the dexJar to create a unique scope for each
412 dexJarStem := strings.TrimSuffix(dexJarFile.Base(), dexJarFile.Ext())
413
Cole Fausteb032462024-09-19 11:12:54 -0700414 appImage := d.dexpreoptProperties.Dex_preopt.App_image.Get(ctx)
415
Ulya Trafimovich76b08522021-01-14 17:52:43 +0000416 // Full dexpreopt config, used to create dexpreopt build rules.
Martin Stjernholm8d80cee2020-01-31 17:44:54 +0000417 dexpreoptConfig := &dexpreopt.ModuleConfig{
Spandan Dase21a8d42024-01-23 23:56:29 +0000418 Name: libName,
Victor Hsiehd181c8b2019-01-29 13:00:33 -0800419 DexLocation: dexLocation,
Spandan Dase21a8d42024-01-23 23:56:29 +0000420 BuildPath: android.PathForModuleOut(ctx, "dexpreopt", dexJarStem, libName+".jar").OutputPath,
Colin Cross69f59a32019-02-15 10:39:37 -0800421 DexPath: dexJarFile,
Jeongik Cha33a3a812021-04-15 09:12:49 +0900422 ManifestPath: android.OptionalPathForPath(d.manifestFile),
Victor Hsiehd181c8b2019-01-29 13:00:33 -0800423 UncompressedDex: d.uncompressedDex,
424 HasApkLibraries: false,
425 PreoptFlags: nil,
Colin Cross43f08db2018-11-12 10:13:39 -0800426
Colin Cross69f59a32019-02-15 10:39:37 -0800427 ProfileClassListing: profileClassListing,
Colin Cross43f08db2018-11-12 10:13:39 -0800428 ProfileIsTextListing: profileIsTextListing,
Nicolas Geoffraye7102422019-07-24 13:19:29 +0100429 ProfileBootListing: profileBootListing,
Colin Cross43f08db2018-11-12 10:13:39 -0800430
Ulya Trafimovich8c35fcf2021-02-17 16:23:28 +0000431 EnforceUsesLibrariesStatusFile: dexpreopt.UsesLibrariesStatusFile(ctx),
432 EnforceUsesLibraries: d.enforceUsesLibs,
433 ProvidesUsesLibrary: providesUsesLib,
434 ClassLoaderContexts: d.classLoaderContexts,
Colin Cross43f08db2018-11-12 10:13:39 -0800435
Jeongik Cha4dda75e2021-04-27 23:56:44 +0900436 Archs: archs,
437 DexPreoptImagesDeps: imagesDeps,
438 DexPreoptImageLocationsOnHost: hostImageLocations,
439 DexPreoptImageLocationsOnDevice: deviceImageLocations,
Colin Cross43f08db2018-11-12 10:13:39 -0800440
Ulya Trafimovich9023b022021-03-22 16:02:28 +0000441 PreoptBootClassPathDexFiles: dexFiles.Paths(),
Vladimir Marko40139d62020-02-06 15:14:29 +0000442 PreoptBootClassPathDexLocations: dexLocations,
Colin Cross800fe132019-02-11 14:21:24 -0800443
Cole Fausteb032462024-09-19 11:12:54 -0700444 NoCreateAppImage: !appImage.GetOrDefault(true),
445 ForceCreateAppImage: appImage.GetOrDefault(false),
Colin Cross43f08db2018-11-12 10:13:39 -0800446
Jaewoong Jungccbb3932019-04-15 09:48:31 -0700447 PresignedPrebuilt: d.isPresignedPrebuilt,
Colin Cross43f08db2018-11-12 10:13:39 -0800448 }
449
Spandan Das950deca2024-10-01 18:35:23 +0000450 if ctx.Config().InstallApexSystemServerDexpreoptSamePartition() {
451 dexpreoptConfig.ApexPartition = android.PathForModuleInstall(ctx).Partition()
452 } else {
453 dexpreoptConfig.ApexPartition = "system"
454 }
455
Spandan Das2069c3f2023-12-06 19:40:24 +0000456 d.configPath = android.PathForModuleOut(ctx, "dexpreopt", dexJarStem, "dexpreopt.config")
Jeongik Chac6246672021-04-08 00:00:19 +0900457 dexpreopt.WriteModuleConfig(ctx, dexpreoptConfig, d.configPath)
Colin Crossa6182ab2024-08-21 10:47:44 -0700458 ctx.CheckbuildFile(d.configPath)
Jeongik Chac6246672021-04-08 00:00:19 +0900459
Spandan Dase21a8d42024-01-23 23:56:29 +0000460 if d.dexpreoptDisabled(ctx, libName) {
Jeongik Chac6246672021-04-08 00:00:19 +0900461 return
462 }
463
464 globalSoong := dexpreopt.GetGlobalSoongConfig(ctx)
465
Jiakai Zhang51b2a8b2023-06-26 16:47:38 +0100466 // The root "product_packages.txt" is generated by `build/make/core/Makefile`. It contains a list
467 // of all packages that are installed on the device. We use `grep` to filter the list by the app's
468 // dependencies to create a per-app list, and use `rsync --checksum` to prevent the file's mtime
469 // from being changed if the contents don't change. This avoids unnecessary dexpreopt reruns.
Jiakai Zhanga4496782023-05-17 16:57:30 +0100470 productPackages := android.PathForModuleInPartitionInstall(ctx, "", "product_packages.txt")
Spandan Das2069c3f2023-12-06 19:40:24 +0000471 appProductPackages := android.PathForModuleOut(ctx, "dexpreopt", dexJarStem, "product_packages.txt")
Jiakai Zhang51b2a8b2023-06-26 16:47:38 +0100472 appProductPackagesStaging := appProductPackages.ReplaceExtension(ctx, "txt.tmp")
473 clcNames, _ := dexpreopt.ComputeClassLoaderContextDependencies(dexpreoptConfig.ClassLoaderContexts)
474 sort.Strings(clcNames) // The order needs to be deterministic.
475 productPackagesRule := android.NewRuleBuilder(pctx, ctx)
476 if len(clcNames) > 0 {
477 productPackagesRule.Command().
478 Text("grep -F -x").
479 FlagForEachArg("-e ", clcNames).
480 Input(productPackages).
481 FlagWithOutput("> ", appProductPackagesStaging).
482 Text("|| true")
483 } else {
484 productPackagesRule.Command().
485 Text("rm -f").Output(appProductPackagesStaging).
486 Text("&&").
487 Text("touch").Output(appProductPackagesStaging)
488 }
489 productPackagesRule.Command().
490 Text("rsync --checksum").
491 Input(appProductPackagesStaging).
492 Output(appProductPackages)
Spandan Das2069c3f2023-12-06 19:40:24 +0000493 productPackagesRule.Restat().Build("product_packages."+dexJarStem, "dexpreopt product_packages")
Jiakai Zhanga4496782023-05-17 16:57:30 +0100494
495 dexpreoptRule, err := dexpreopt.GenerateDexpreoptRule(
Colin Cross388c6612025-01-28 14:00:12 -0800496 ctx, globalSoong, global, dexpreoptConfig, appProductPackages)
Colin Cross43f08db2018-11-12 10:13:39 -0800497 if err != nil {
498 ctx.ModuleErrorf("error generating dexpreopt rule: %s", err.Error())
Jaewoong Jung4b97a562020-12-17 09:43:28 -0800499 return
Colin Cross43f08db2018-11-12 10:13:39 -0800500 }
501
Spandan Das2069c3f2023-12-06 19:40:24 +0000502 dexpreoptRule.Build("dexpreopt"+"."+dexJarStem, "dexpreopt")
Colin Cross43f08db2018-11-12 10:13:39 -0800503
Spandan Das2069c3f2023-12-06 19:40:24 +0000504 // The current ctx might be of a deapexer module created by a prebuilt apex
505 // Use the path of the dex file to determine the library name
506 isApexSystemServerJar := global.AllApexSystemServerJars(ctx).ContainsJar(dexJarStem)
Jiakai Zhang389a6472021-12-14 18:54:06 +0000507
Justin Yun613bdc52024-06-12 21:32:10 +0900508 dexpreoptPartition := d.installPath.Partition()
509 // dexpreoptPartition is set to empty for dexpreopts of system APEX and system_other.
510 // In case of system APEX, however, we can set it to "system" manually.
511 // TODO(b/346662300): Let dexpreopter generate the installPath for dexpreopt files instead of
512 // using the dex location to generate the installPath.
513 if isApexSystemServerJar {
Spandan Das906222c2024-10-17 18:29:54 +0000514 dexpreoptPartition = dexpreoptConfig.ApexPartition
Justin Yun613bdc52024-06-12 21:32:10 +0900515 }
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700516 for _, install := range dexpreoptRule.Installs() {
517 // Remove the "/" prefix because the path should be relative to $ANDROID_PRODUCT_OUT.
518 installDir := strings.TrimPrefix(filepath.Dir(install.To), "/")
Justin Yun613bdc52024-06-12 21:32:10 +0900519 partition := dexpreoptPartition
Justin Yun22c8aca2024-06-05 20:25:03 +0900520 if strings.HasPrefix(installDir, partition+"/") {
521 installDir = strings.TrimPrefix(installDir, partition+"/")
522 } else {
523 // If the partition for the installDir is different from the install partition, set the
524 // partition empty to install the dexpreopt files to the desired partition.
525 // TODO(b/346439786): Define and use the dexpreopt module type to avoid this mismatch.
526 partition = ""
527 }
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700528 installBase := filepath.Base(install.To)
Justin Yun22c8aca2024-06-05 20:25:03 +0900529 installPath := android.PathForModuleInPartitionInstall(ctx, partition, installDir)
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800530 isProfile := strings.HasSuffix(installBase, ".prof")
531
532 if isProfile {
Jiakai Zhang81e46812023-02-08 21:56:07 +0800533 d.outputProfilePathOnHost = install.From
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800534 }
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700535
Jiakai Zhang389a6472021-12-14 18:54:06 +0000536 if isApexSystemServerJar {
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800537 // Profiles are handled separately because they are installed into the APEX.
538 if !isProfile {
539 // APEX variants of java libraries are hidden from Make, so their dexpreopt
540 // outputs need special handling. Currently, for APEX variants of java
541 // libraries, only those in the system server classpath are handled here.
542 // Preopting of boot classpath jars in the ART APEX are handled in
543 // java/dexpreopt_bootjars.go, and other APEX jars are not preopted.
Colin Cross388c6612025-01-28 14:00:12 -0800544 // The installs will be handled the apex module that includes this library.
545 di := DexpreopterInstall{
546 OutputPathOnHost: install.From,
547 InstallDirOnDevice: installPath,
548 InstallFileOnDevice: installBase,
Spandan Das746161d2024-08-21 22:47:53 +0000549 }
Colin Cross388c6612025-01-28 14:00:12 -0800550 d.apexSystemServerDexpreoptInstalls = append(d.apexSystemServerDexpreoptInstalls, di)
Spandan Das746161d2024-08-21 22:47:53 +0000551
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800552 }
Colin Crossfa9bfcd2021-11-10 16:42:38 -0800553 } else if !d.preventInstall {
Colin Crossa6182ab2024-08-21 10:47:44 -0700554 // Install without adding to checkbuild to match behavior of previous Make-based checkbuild rules
555 ctx.InstallFileWithoutCheckbuild(installPath, installBase, install.From)
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000556 }
Colin Cross1d0eb7a2021-11-03 14:08:20 -0700557 }
558
Colin Cross388c6612025-01-28 14:00:12 -0800559 if isApexSystemServerJar {
560 // Store the dex jar location for system server jars in apexes, the apex will copy the file into
561 // a known location for dex2oat.
562 d.apexSystemServerDexJars = append(d.apexSystemServerDexJars, dexJarFile)
563 } else if isSystemServerJar && !d.preventInstall {
564 // Copy the dex jar into a known location for dex2oat for non-apex system server jars.
565 android.CopyFileRule(ctx, dexJarFile, android.PathForOutput(ctx, dexpreopt.SystemServerDexjarsDir, dexJarFile.Base()))
566 }
567
Jiakai Zhang389a6472021-12-14 18:54:06 +0000568 if !isApexSystemServerJar {
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000569 d.builtInstalled = dexpreoptRule.Installs().String()
570 }
571}
572
Justin Yun22c8aca2024-06-05 20:25:03 +0900573func getModuleInstallPathInfo(ctx android.ModuleContext, fullInstallPath string) (android.InstallPath, string, string) {
574 installPath := android.PathForModuleInstall(ctx)
575 installDir, installBase := filepath.Split(strings.TrimPrefix(fullInstallPath, "/"))
576
577 if !strings.HasPrefix(installDir, installPath.Partition()+"/") {
578 // Return empty filename if the install partition is not for the target image.
579 return installPath, "", ""
580 }
581 relDir, err := filepath.Rel(installPath.Partition(), installDir)
582 if err != nil {
583 panic(err)
584 }
585 return installPath, relDir, installBase
586}
587
Spandan Das29207b52024-07-30 23:28:17 +0000588// installFile will install the file if `install` path and the target install partition are the same.
589func installFile(ctx android.ModuleContext, install android.RuleBuilderInstall) {
Justin Yun22c8aca2024-06-05 20:25:03 +0900590 installPath, relDir, name := getModuleInstallPathInfo(ctx, install.To)
591 // Empty name means the install partition is not for the target image.
592 // For the system image, files for "apex" and "system_other" are skipped here.
593 // The skipped "apex" files are for testing only, for example,
594 // "/apex/art_boot_images/javalib/x86/boot.vdex".
595 // TODO(b/320196894): Files for "system_other" are skipped because soong creates the system
596 // image only for now.
597 if name != "" {
Spandan Das29207b52024-07-30 23:28:17 +0000598 ctx.InstallFile(installPath.Join(ctx, relDir), name, install.From)
Justin Yun22c8aca2024-06-05 20:25:03 +0900599 }
600}
601
Colin Cross388c6612025-01-28 14:00:12 -0800602func (d *dexpreopter) ApexSystemServerDexpreoptInstalls() []DexpreopterInstall {
603 return d.apexSystemServerDexpreoptInstalls
Jiakai Zhangca9bc982021-09-09 08:09:41 +0000604}
605
Colin Cross388c6612025-01-28 14:00:12 -0800606func (d *dexpreopter) ApexSystemServerDexJars() android.Paths {
607 return d.apexSystemServerDexJars
Jihoon Kangd4063812025-01-24 00:25:30 +0000608}
609
Jiakai Zhang81e46812023-02-08 21:56:07 +0800610func (d *dexpreopter) OutputProfilePathOnHost() android.Path {
611 return d.outputProfilePathOnHost
Jiakai Zhang3317ce72023-02-08 01:19:19 +0800612}
Spandan Das0727ba72024-02-13 16:37:43 +0000613
614func (d *dexpreopter) disableDexpreopt() {
615 d.shouldDisableDexpreopt = true
616}
Spandan Das3dbda182024-05-20 22:23:10 +0000617
Cole Fausteb032462024-09-19 11:12:54 -0700618func (d *dexpreopter) EnableProfileRewriting(ctx android.BaseModuleContext) bool {
619 return d.dexpreoptProperties.Dex_preopt.Enable_profile_rewriting.GetOrDefault(ctx, false)
Spandan Das3dbda182024-05-20 22:23:10 +0000620}
621
Cole Fausteb032462024-09-19 11:12:54 -0700622func (d *dexpreopter) GetProfile(ctx android.BaseModuleContext) string {
623 return d.dexpreoptProperties.Dex_preopt.Profile.GetOrDefault(ctx, "")
Spandan Das3dbda182024-05-20 22:23:10 +0000624}
625
Cole Fausteb032462024-09-19 11:12:54 -0700626func (d *dexpreopter) GetProfileGuided(ctx android.BaseModuleContext) bool {
627 return d.dexpreoptProperties.Dex_preopt.Profile_guided.GetOrDefault(ctx, false)
Spandan Das3dbda182024-05-20 22:23:10 +0000628}
629
630func (d *dexpreopter) GetRewrittenProfile() android.Path {
631 return d.rewrittenProfile
632}
633
634func (d *dexpreopter) SetRewrittenProfile(p android.Path) {
635 d.rewrittenProfile = p
636}