blob: 8265458ba092ef085bf1e52a014badc480dbc5a9 [file] [log] [blame]
Colin Cross6362e272015-10-29 15:25:03 -07001// Copyright 2015 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
Colin Cross635c3b02016-05-18 15:37:25 -070015package android
Colin Cross6362e272015-10-29 15:25:03 -070016
Colin Cross795c3772017-03-16 16:50:10 -070017import (
Colin Cross984223f2024-02-01 17:10:23 -080018 "sync"
19
Colin Cross795c3772017-03-16 16:50:10 -070020 "github.com/google/blueprint"
21)
Colin Cross6362e272015-10-29 15:25:03 -070022
Jeff Gastonaf3cc2d2017-09-27 17:01:44 -070023// Phases:
24// run Pre-arch mutators
25// run archMutator
26// run Pre-deps mutators
27// run depsMutator
28// run PostDeps mutators
Colin Cross1e954b62024-09-13 13:50:00 -070029// run FinalDeps mutators (TransitionMutators disallowed in this phase)
Jeff Gastonaf3cc2d2017-09-27 17:01:44 -070030// continue on to GenerateAndroidBuildActions
Colin Cross1e676be2016-10-12 14:38:15 -070031
Paul Duffinc05b0342021-03-06 13:28:13 +000032// collateGloballyRegisteredMutators constructs the list of mutators that have been registered
33// with the InitRegistrationContext and will be used at runtime.
34func collateGloballyRegisteredMutators() sortableComponents {
Colin Crossf22fe412024-10-01 14:02:12 -070035 return collateRegisteredMutators(preArch, preDeps, postDeps, postApex, finalDeps)
Paul Duffinc05b0342021-03-06 13:28:13 +000036}
37
38// collateRegisteredMutators constructs a single list of mutators from the separate lists.
Colin Crossf22fe412024-10-01 14:02:12 -070039func collateRegisteredMutators(preArch, preDeps, postDeps, postApex, finalDeps []RegisterMutatorFunc) sortableComponents {
Colin Crosscec81712017-07-13 14:43:27 -070040 mctx := &registerMutatorsContext{}
Nan Zhangdb0b9a32017-02-27 10:12:13 -080041
42 register := func(funcs []RegisterMutatorFunc) {
43 for _, f := range funcs {
Colin Crosscec81712017-07-13 14:43:27 -070044 f(mctx)
Nan Zhangdb0b9a32017-02-27 10:12:13 -080045 }
46 }
47
Colin Crosscec81712017-07-13 14:43:27 -070048 register(preArch)
Nan Zhangdb0b9a32017-02-27 10:12:13 -080049
Colin Crosscec81712017-07-13 14:43:27 -070050 register(preDeps)
51
Liz Kammer356f7d42021-01-26 09:18:53 -050052 register([]RegisterMutatorFunc{registerDepsMutator})
Colin Crosscec81712017-07-13 14:43:27 -070053
54 register(postDeps)
55
Colin Crossf22fe412024-10-01 14:02:12 -070056 register(postApex)
57
Martin Stjernholm710ec3a2020-01-16 15:12:04 +000058 mctx.finalPhase = true
59 register(finalDeps)
60
Paul Duffinc05b0342021-03-06 13:28:13 +000061 return mctx.mutators
Colin Cross795c3772017-03-16 16:50:10 -070062}
63
64type registerMutatorsContext struct {
Colin Crossb63d7b32023-12-07 16:54:51 -080065 mutators sortableComponents
66 finalPhase bool
Colin Cross795c3772017-03-16 16:50:10 -070067}
Colin Cross1e676be2016-10-12 14:38:15 -070068
69type RegisterMutatorsContext interface {
Colin Cross25de6c32019-06-06 14:29:25 -070070 TopDown(name string, m TopDownMutator) MutatorHandle
71 BottomUp(name string, m BottomUpMutator) MutatorHandle
Colin Cross617b88a2020-08-24 18:04:09 -070072 BottomUpBlueprint(name string, m blueprint.BottomUpMutator) MutatorHandle
Lukacs T. Berki6c716762022-06-13 20:50:39 +020073 Transition(name string, m TransitionMutator)
Colin Cross1e676be2016-10-12 14:38:15 -070074}
75
76type RegisterMutatorFunc func(RegisterMutatorsContext)
77
Colin Crosscec81712017-07-13 14:43:27 -070078var preArch = []RegisterMutatorFunc{
Dan Willemsen6e72ef72018-01-26 18:27:02 -080079 RegisterNamespaceMutator,
Paul Duffinaa4162e2020-05-05 11:35:43 +010080
Paul Duffinaa4162e2020-05-05 11:35:43 +010081 // Check the visibility rules are valid.
82 //
83 // This must run after the package renamer mutators so that any issues found during
84 // validation of the package's default_visibility property are reported using the
85 // correct package name and not the synthetic name.
86 //
87 // This must also be run before defaults mutators as the rules for validation are
88 // different before checking the rules than they are afterwards. e.g.
89 // visibility: ["//visibility:private", "//visibility:public"]
90 // would be invalid if specified in a module definition but is valid if it results
91 // from something like this:
92 //
93 // defaults {
94 // name: "defaults",
95 // // Be inaccessible outside a package by default.
96 // visibility: ["//visibility:private"]
97 // }
98 //
99 // defaultable_module {
100 // name: "defaultable_module",
101 // defaults: ["defaults"],
102 // // Override the default.
103 // visibility: ["//visibility:public"]
104 // }
105 //
Paul Duffin593b3c92019-12-05 14:31:48 +0000106 RegisterVisibilityRuleChecker,
Paul Duffinaa4162e2020-05-05 11:35:43 +0100107
Bob Badour37af0462021-01-07 03:34:31 +0000108 // Record the default_applicable_licenses for each package.
109 //
110 // This must run before the defaults so that defaults modules can pick up the package default.
111 RegisterLicensesPackageMapper,
112
Paul Duffinaa4162e2020-05-05 11:35:43 +0100113 // Apply properties from defaults modules to the referencing modules.
Paul Duffinafa9fa12020-04-29 16:47:28 +0100114 //
115 // Any mutators that are added before this will not see any modules created by
116 // a DefaultableHook.
Colin Cross89536d42017-07-07 14:35:50 -0700117 RegisterDefaultsPreArchMutators,
Paul Duffinaa4162e2020-05-05 11:35:43 +0100118
Paul Duffin44f1d842020-06-26 20:17:02 +0100119 // Add dependencies on any components so that any component references can be
120 // resolved within the deps mutator.
121 //
122 // Must be run after defaults so it can be used to create dependencies on the
123 // component modules that are creating in a DefaultableHook.
124 //
125 // Must be run before RegisterPrebuiltsPreArchMutators, i.e. before prebuilts are
126 // renamed. That is so that if a module creates components using a prebuilt module
127 // type that any dependencies (which must use prebuilt_ prefixes) are resolved to
128 // the prebuilt module and not the source module.
129 RegisterComponentsMutator,
130
Paul Duffinc988c8e2020-04-29 18:27:14 +0100131 // Create an association between prebuilt modules and their corresponding source
132 // modules (if any).
Paul Duffinafa9fa12020-04-29 16:47:28 +0100133 //
134 // Must be run after defaults mutators to ensure that any modules created by
135 // a DefaultableHook can be either a prebuilt or a source module with a matching
136 // prebuilt.
Paul Duffinc988c8e2020-04-29 18:27:14 +0100137 RegisterPrebuiltsPreArchMutators,
138
Bob Badour37af0462021-01-07 03:34:31 +0000139 // Gather the licenses properties for all modules for use during expansion and enforcement.
140 //
141 // This must come after the defaults mutators to ensure that any licenses supplied
142 // in a defaults module has been successfully applied before the rules are gathered.
143 RegisterLicensesPropertyGatherer,
144
Paul Duffinaa4162e2020-05-05 11:35:43 +0100145 // Gather the visibility rules for all modules for us during visibility enforcement.
146 //
147 // This must come after the defaults mutators to ensure that any visibility supplied
148 // in a defaults module has been successfully applied before the rules are gathered.
Paul Duffin593b3c92019-12-05 14:31:48 +0000149 RegisterVisibilityRuleGatherer,
Colin Crosscec81712017-07-13 14:43:27 -0700150}
151
Colin Crossae4c6182017-09-15 17:33:55 -0700152func registerArchMutator(ctx RegisterMutatorsContext) {
Colin Cross8bbc3d52024-09-11 15:33:54 -0700153 ctx.Transition("os", &osTransitionMutator{})
Jihoon Kang5402bbd2024-07-31 18:37:49 +0000154 ctx.Transition("image", &imageTransitionMutator{})
Colin Cross8bbc3d52024-09-11 15:33:54 -0700155 ctx.Transition("arch", &archTransitionMutator{})
Colin Crossae4c6182017-09-15 17:33:55 -0700156}
157
Colin Crosscec81712017-07-13 14:43:27 -0700158var preDeps = []RegisterMutatorFunc{
Colin Crossae4c6182017-09-15 17:33:55 -0700159 registerArchMutator,
Colin Crosscec81712017-07-13 14:43:27 -0700160}
161
162var postDeps = []RegisterMutatorFunc{
Colin Cross1b488422019-03-04 22:33:56 -0800163 registerPathDepsMutator,
Colin Cross5ea9bcc2017-07-27 15:41:32 -0700164 RegisterPrebuiltsPostDepsMutators,
Paul Duffin593b3c92019-12-05 14:31:48 +0000165 RegisterVisibilityRuleEnforcer,
Bob Badour37af0462021-01-07 03:34:31 +0000166 RegisterLicensesDependencyChecker,
Paul Duffin45338f02021-03-30 23:07:52 +0100167 registerNeverallowMutator,
Jaewoong Jungb639a6a2019-05-10 15:16:29 -0700168 RegisterOverridePostDepsMutators,
Colin Crosscec81712017-07-13 14:43:27 -0700169}
Colin Cross1e676be2016-10-12 14:38:15 -0700170
Colin Crossf22fe412024-10-01 14:02:12 -0700171var postApex = []RegisterMutatorFunc{}
172
Martin Stjernholm710ec3a2020-01-16 15:12:04 +0000173var finalDeps = []RegisterMutatorFunc{}
174
Colin Cross1e676be2016-10-12 14:38:15 -0700175func PreArchMutators(f RegisterMutatorFunc) {
176 preArch = append(preArch, f)
177}
178
179func PreDepsMutators(f RegisterMutatorFunc) {
180 preDeps = append(preDeps, f)
181}
182
183func PostDepsMutators(f RegisterMutatorFunc) {
184 postDeps = append(postDeps, f)
185}
186
Colin Crossf22fe412024-10-01 14:02:12 -0700187func PostApexMutators(f RegisterMutatorFunc) {
188 postApex = append(postApex, f)
189}
190
Martin Stjernholm710ec3a2020-01-16 15:12:04 +0000191func FinalDepsMutators(f RegisterMutatorFunc) {
192 finalDeps = append(finalDeps, f)
193}
194
Chris Parsons637458d2023-09-19 20:09:00 +0000195type BaseMutatorContext interface {
196 BaseModuleContext
197
198 // MutatorName returns the name that this mutator was registered with.
199 MutatorName() string
200
201 // Rename all variants of a module. The new name is not visible to calls to ModuleName,
202 // AddDependency or OtherModuleName until after this mutator pass is complete.
203 Rename(name string)
Colin Crossda279cf2024-09-17 14:25:45 -0700204
205 // CreateModule creates a new module by calling the factory method for the specified moduleType, and applies
206 // the specified property structs to it as if the properties were set in a blueprint file.
207 CreateModule(ModuleFactory, ...interface{}) Module
Chris Parsons637458d2023-09-19 20:09:00 +0000208}
209
210type TopDownMutator func(TopDownMutatorContext)
211
212type TopDownMutatorContext interface {
213 BaseMutatorContext
Chris Parsons637458d2023-09-19 20:09:00 +0000214}
215
Colin Cross25de6c32019-06-06 14:29:25 -0700216type topDownMutatorContext struct {
Colin Crossdc35e212019-06-06 16:13:11 -0700217 bp blueprint.TopDownMutatorContext
Colin Cross0ea8ba82019-06-06 14:33:29 -0700218 baseModuleContext
Colin Cross6362e272015-10-29 15:25:03 -0700219}
220
Colin Cross25de6c32019-06-06 14:29:25 -0700221type BottomUpMutator func(BottomUpMutatorContext)
Colin Cross6362e272015-10-29 15:25:03 -0700222
Colin Cross635c3b02016-05-18 15:37:25 -0700223type BottomUpMutatorContext interface {
Colin Cross9f35c3d2020-09-16 19:04:41 -0700224 BaseMutatorContext
Colin Crossb63d7b32023-12-07 16:54:51 -0800225
226 // AddDependency adds a dependency to the given module. It returns a slice of modules for each
227 // dependency (some entries may be nil).
228 //
229 // If the mutator is parallel (see MutatorHandle.Parallel), this method will pause until the
230 // new dependencies have had the current mutator called on them. If the mutator is not
231 // parallel this method does not affect the ordering of the current mutator pass, but will
232 // be ordered correctly for all future mutator passes.
233 AddDependency(module blueprint.Module, tag blueprint.DependencyTag, name ...string) []blueprint.Module
Colin Crossaabf6792017-11-29 00:27:14 -0800234
Colin Cross9f35c3d2020-09-16 19:04:41 -0700235 // AddReverseDependency adds a dependency from the destination to the given module.
236 // Does not affect the ordering of the current mutator pass, but will be ordered
237 // correctly for all future mutator passes. All reverse dependencies for a destination module are
238 // collected until the end of the mutator pass, sorted by name, and then appended to the destination
239 // module's dependency list.
Colin Crossaabf6792017-11-29 00:27:14 -0800240 AddReverseDependency(module blueprint.Module, tag blueprint.DependencyTag, name string)
Colin Cross9f35c3d2020-09-16 19:04:41 -0700241
Colin Cross9f35c3d2020-09-16 19:04:41 -0700242 // AddVariationDependencies adds deps as dependencies of the current module, but uses the variations
Colin Cross4f1dcb02020-09-16 18:45:04 -0700243 // argument to select which variant of the dependency to use. It returns a slice of modules for
244 // each dependency (some entries may be nil). A variant of the dependency must exist that matches
Usta Shresthac725f472022-01-11 02:44:21 -0500245 // all the non-local variations of the current module, plus the variations argument.
Colin Cross4f1dcb02020-09-16 18:45:04 -0700246 //
247 // If the mutator is parallel (see MutatorHandle.Parallel), this method will pause until the
248 // new dependencies have had the current mutator called on them. If the mutator is not
249 // parallel this method does not affect the ordering of the current mutator pass, but will
250 // be ordered correctly for all future mutator passes.
Usta Shresthac725f472022-01-11 02:44:21 -0500251 AddVariationDependencies(variations []blueprint.Variation, tag blueprint.DependencyTag, names ...string) []blueprint.Module
Colin Cross9f35c3d2020-09-16 19:04:41 -0700252
Cole Faustf4992e82024-09-30 15:16:18 -0700253 // AddReverseVariationDependencies adds a dependency from the named module to the current
254 // module. The given variations will be added to the current module's varations, and then the
255 // result will be used to find the correct variation of the depending module, which must exist.
256 //
257 // Does not affect the ordering of the current mutator pass, but will be ordered
258 // correctly for all future mutator passes. All reverse dependencies for a destination module are
259 // collected until the end of the mutator pass, sorted by name, and then appended to the destination
260 // module's dependency list.
261 AddReverseVariationDependency([]blueprint.Variation, blueprint.DependencyTag, string)
262
Colin Cross9f35c3d2020-09-16 19:04:41 -0700263 // AddFarVariationDependencies adds deps as dependencies of the current module, but uses the
Colin Cross4f1dcb02020-09-16 18:45:04 -0700264 // variations argument to select which variant of the dependency to use. It returns a slice of
265 // modules for each dependency (some entries may be nil). A variant of the dependency must
266 // exist that matches the variations argument, but may also have other variations.
Colin Cross9f35c3d2020-09-16 19:04:41 -0700267 // For any unspecified variation the first variant will be used.
268 //
269 // Unlike AddVariationDependencies, the variations of the current module are ignored - the
270 // dependency only needs to match the supplied variations.
Colin Cross4f1dcb02020-09-16 18:45:04 -0700271 //
272 // If the mutator is parallel (see MutatorHandle.Parallel), this method will pause until the
273 // new dependencies have had the current mutator called on them. If the mutator is not
274 // parallel this method does not affect the ordering of the current mutator pass, but will
275 // be ordered correctly for all future mutator passes.
276 AddFarVariationDependencies([]blueprint.Variation, blueprint.DependencyTag, ...string) []blueprint.Module
Colin Cross9f35c3d2020-09-16 19:04:41 -0700277
Colin Cross86771322024-05-01 14:19:51 -0700278 // ReplaceDependencies finds all the variants of the module with the specified name, then
279 // replaces all dependencies onto those variants with the current variant of this module.
280 // Replacements don't take effect until after the mutator pass is finished.
Colin Crossaabf6792017-11-29 00:27:14 -0800281 ReplaceDependencies(string)
Colin Cross9f35c3d2020-09-16 19:04:41 -0700282
Colin Cross86771322024-05-01 14:19:51 -0700283 // ReplaceDependenciesIf finds all the variants of the module with the specified name, then
284 // replaces all dependencies onto those variants with the current variant of this module
285 // as long as the supplied predicate returns true.
Colin Cross9f35c3d2020-09-16 19:04:41 -0700286 // Replacements don't take effect until after the mutator pass is finished.
Paul Duffin80342d72020-06-26 22:08:43 +0100287 ReplaceDependenciesIf(string, blueprint.ReplaceDependencyPredicate)
Colin Cross6362e272015-10-29 15:25:03 -0700288}
289
Colin Cross984223f2024-02-01 17:10:23 -0800290// An outgoingTransitionContextImpl and incomingTransitionContextImpl is created for every dependency of every module
291// for each transition mutator. bottomUpMutatorContext and topDownMutatorContext are created once for every module
292// for every BottomUp or TopDown mutator. Use a global pool for each to avoid reallocating every time.
293var (
294 outgoingTransitionContextPool = sync.Pool{
295 New: func() any { return &outgoingTransitionContextImpl{} },
296 }
297 incomingTransitionContextPool = sync.Pool{
298 New: func() any { return &incomingTransitionContextImpl{} },
299 }
300 bottomUpMutatorContextPool = sync.Pool{
301 New: func() any { return &bottomUpMutatorContext{} },
302 }
303
304 topDownMutatorContextPool = sync.Pool{
305 New: func() any { return &topDownMutatorContext{} },
306 }
307)
308
Colin Cross25de6c32019-06-06 14:29:25 -0700309type bottomUpMutatorContext struct {
Colin Crossdc35e212019-06-06 16:13:11 -0700310 bp blueprint.BottomUpMutatorContext
Colin Cross0ea8ba82019-06-06 14:33:29 -0700311 baseModuleContext
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400312 finalPhase bool
Colin Cross6362e272015-10-29 15:25:03 -0700313}
314
Colin Cross984223f2024-02-01 17:10:23 -0800315// callers must immediately follow the call to this function with defer bottomUpMutatorContextPool.Put(mctx).
Colin Cross617b88a2020-08-24 18:04:09 -0700316func bottomUpMutatorContextFactory(ctx blueprint.BottomUpMutatorContext, a Module,
Colin Crossb63d7b32023-12-07 16:54:51 -0800317 finalPhase bool) BottomUpMutatorContext {
Colin Cross617b88a2020-08-24 18:04:09 -0700318
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400319 moduleContext := a.base().baseModuleContextFactory(ctx)
Colin Cross984223f2024-02-01 17:10:23 -0800320 mctx := bottomUpMutatorContextPool.Get().(*bottomUpMutatorContext)
321 *mctx = bottomUpMutatorContext{
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400322 bp: ctx,
Cole Faust0abd4b42023-01-10 10:49:18 -0800323 baseModuleContext: moduleContext,
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400324 finalPhase: finalPhase,
Colin Cross617b88a2020-08-24 18:04:09 -0700325 }
Colin Cross984223f2024-02-01 17:10:23 -0800326 return mctx
Colin Cross617b88a2020-08-24 18:04:09 -0700327}
328
Colin Cross25de6c32019-06-06 14:29:25 -0700329func (x *registerMutatorsContext) BottomUp(name string, m BottomUpMutator) MutatorHandle {
Martin Stjernholm710ec3a2020-01-16 15:12:04 +0000330 finalPhase := x.finalPhase
Colin Cross798bfce2016-10-12 14:28:16 -0700331 f := func(ctx blueprint.BottomUpMutatorContext) {
Colin Cross635c3b02016-05-18 15:37:25 -0700332 if a, ok := ctx.Module().(Module); ok {
Colin Cross984223f2024-02-01 17:10:23 -0800333 mctx := bottomUpMutatorContextFactory(ctx, a, finalPhase)
334 defer bottomUpMutatorContextPool.Put(mctx)
335 m(mctx)
Colin Cross6362e272015-10-29 15:25:03 -0700336 }
Colin Cross798bfce2016-10-12 14:28:16 -0700337 }
Liz Kammer356f7d42021-01-26 09:18:53 -0500338 mutator := &mutator{name: x.mutatorName(name), bottomUpMutator: f}
Colin Cross795c3772017-03-16 16:50:10 -0700339 x.mutators = append(x.mutators, mutator)
Colin Cross798bfce2016-10-12 14:28:16 -0700340 return mutator
Colin Cross6362e272015-10-29 15:25:03 -0700341}
342
Colin Cross617b88a2020-08-24 18:04:09 -0700343func (x *registerMutatorsContext) BottomUpBlueprint(name string, m blueprint.BottomUpMutator) MutatorHandle {
344 mutator := &mutator{name: name, bottomUpMutator: m}
345 x.mutators = append(x.mutators, mutator)
346 return mutator
347}
348
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200349type IncomingTransitionContext interface {
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800350 ArchModuleContext
Colin Crossaf333f52024-04-15 15:20:23 -0700351 ModuleProviderContext
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800352
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200353 // Module returns the target of the dependency edge for which the transition
354 // is being computed
355 Module() Module
356
357 // Config returns the configuration for the build.
358 Config() Config
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800359
360 DeviceConfig() DeviceConfig
Colin Crosse1a85552024-06-14 12:17:37 -0700361
362 // IsAddingDependency returns true if the transition is being called while adding a dependency
363 // after the transition mutator has already run, or false if it is being called when the transition
364 // mutator is running. This should be used sparingly, all uses will have to be removed in order
365 // to support creating variants on demand.
366 IsAddingDependency() bool
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200367}
368
369type OutgoingTransitionContext interface {
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800370 ArchModuleContext
Colin Crossaf333f52024-04-15 15:20:23 -0700371 ModuleProviderContext
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800372
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200373 // Module returns the target of the dependency edge for which the transition
374 // is being computed
375 Module() Module
376
377 // DepTag() Returns the dependency tag through which this dependency is
378 // reached
379 DepTag() blueprint.DependencyTag
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800380
381 // Config returns the configuration for the build.
382 Config() Config
383
384 DeviceConfig() DeviceConfig
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200385}
Lukacs T. Berki0e691c12022-06-24 10:15:55 +0200386
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800387// TransitionMutator implements a top-down mechanism where a module tells its
Lukacs T. Berki0e691c12022-06-24 10:15:55 +0200388// direct dependencies what variation they should be built in but the dependency
389// has the final say.
390//
391// When implementing a transition mutator, one needs to implement four methods:
392// - Split() that tells what variations a module has by itself
393// - OutgoingTransition() where a module tells what it wants from its
394// dependency
395// - IncomingTransition() where a module has the final say about its own
396// variation
397// - Mutate() that changes the state of a module depending on its variation
398//
399// That the effective variation of module B when depended on by module A is the
400// composition the outgoing transition of module A and the incoming transition
401// of module B.
402//
403// the outgoing transition should not take the properties of the dependency into
404// account, only those of the module that depends on it. For this reason, the
405// dependency is not even passed into it as an argument. Likewise, the incoming
406// transition should not take the properties of the depending module into
407// account and is thus not informed about it. This makes for a nice
408// decomposition of the decision logic.
409//
410// A given transition mutator only affects its own variation; other variations
411// stay unchanged along the dependency edges.
412//
413// Soong makes sure that all modules are created in the desired variations and
414// that dependency edges are set up correctly. This ensures that "missing
415// variation" errors do not happen and allows for more flexible changes in the
416// value of the variation among dependency edges (as oppposed to bottom-up
417// mutators where if module A in variation X depends on module B and module B
418// has that variation X, A must depend on variation X of B)
419//
420// The limited power of the context objects passed to individual mutators
421// methods also makes it more difficult to shoot oneself in the foot. Complete
422// safety is not guaranteed because no one prevents individual transition
423// mutators from mutating modules in illegal ways and for e.g. Split() or
424// Mutate() to run their own visitations of the transitive dependency of the
425// module and both of these are bad ideas, but it's better than no guardrails at
426// all.
427//
428// This model is pretty close to Bazel's configuration transitions. The mapping
429// between concepts in Soong and Bazel is as follows:
430// - Module == configured target
431// - Variant == configuration
432// - Variation name == configuration flag
433// - Variation == configuration flag value
434// - Outgoing transition == attribute transition
435// - Incoming transition == rule transition
436//
437// The Split() method does not have a Bazel equivalent and Bazel split
438// transitions do not have a Soong equivalent.
439//
440// Mutate() does not make sense in Bazel due to the different models of the
441// two systems: when creating new variations, Soong clones the old module and
442// thus some way is needed to change it state whereas Bazel creates each
443// configuration of a given configured target anew.
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200444type TransitionMutator interface {
445 // Split returns the set of variations that should be created for a module no
446 // matter who depends on it. Used when Make depends on a particular variation
447 // or when the module knows its variations just based on information given to
448 // it in the Blueprint file. This method should not mutate the module it is
449 // called on.
450 Split(ctx BaseModuleContext) []string
451
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800452 // OutgoingTransition is called on a module to determine which variation it wants
453 // from its direct dependencies. The dependency itself can override this decision.
454 // This method should not mutate the module itself.
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200455 OutgoingTransition(ctx OutgoingTransitionContext, sourceVariation string) string
456
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800457 // IncomingTransition is called on a module to determine which variation it should
458 // be in based on the variation modules that depend on it want. This gives the module
459 // a final say about its own variations. This method should not mutate the module
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200460 // itself.
461 IncomingTransition(ctx IncomingTransitionContext, incomingVariation string) string
462
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800463 // Mutate is called after a module was split into multiple variations on each variation.
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200464 // It should not split the module any further but adding new dependencies is
465 // fine. Unlike all the other methods on TransitionMutator, this method is
466 // allowed to mutate the module.
467 Mutate(ctx BottomUpMutatorContext, variation string)
468}
469
470type androidTransitionMutator struct {
Colin Crossb63d7b32023-12-07 16:54:51 -0800471 finalPhase bool
472 mutator TransitionMutator
Colin Crossd67425d2024-04-15 15:17:05 -0700473 name string
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200474}
475
476func (a *androidTransitionMutator) Split(ctx blueprint.BaseModuleContext) []string {
Colin Crossd27205e2024-09-12 22:41:37 -0700477 if a.finalPhase {
478 panic("TransitionMutator not allowed in FinalDepsMutators")
479 }
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200480 if m, ok := ctx.Module().(Module); ok {
481 moduleContext := m.base().baseModuleContextFactory(ctx)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200482 return a.mutator.Split(&moduleContext)
483 } else {
484 return []string{""}
485 }
486}
487
488type outgoingTransitionContextImpl struct {
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800489 archModuleContext
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200490 bp blueprint.OutgoingTransitionContext
491}
492
493func (c *outgoingTransitionContextImpl) Module() Module {
494 return c.bp.Module().(Module)
495}
496
497func (c *outgoingTransitionContextImpl) DepTag() blueprint.DependencyTag {
498 return c.bp.DepTag()
499}
500
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800501func (c *outgoingTransitionContextImpl) Config() Config {
502 return c.bp.Config().(Config)
503}
504
505func (c *outgoingTransitionContextImpl) DeviceConfig() DeviceConfig {
506 return DeviceConfig{c.bp.Config().(Config).deviceConfig}
507}
508
Colin Crossaf333f52024-04-15 15:20:23 -0700509func (c *outgoingTransitionContextImpl) provider(provider blueprint.AnyProviderKey) (any, bool) {
510 return c.bp.Provider(provider)
511}
512
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800513func (a *androidTransitionMutator) OutgoingTransition(bpctx blueprint.OutgoingTransitionContext, sourceVariation string) string {
514 if m, ok := bpctx.Module().(Module); ok {
Colin Cross984223f2024-02-01 17:10:23 -0800515 ctx := outgoingTransitionContextPool.Get().(*outgoingTransitionContextImpl)
516 defer outgoingTransitionContextPool.Put(ctx)
517 *ctx = outgoingTransitionContextImpl{
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800518 archModuleContext: m.base().archModuleContextFactory(bpctx),
519 bp: bpctx,
520 }
521 return a.mutator.OutgoingTransition(ctx, sourceVariation)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200522 } else {
523 return ""
524 }
525}
526
527type incomingTransitionContextImpl struct {
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800528 archModuleContext
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200529 bp blueprint.IncomingTransitionContext
530}
531
532func (c *incomingTransitionContextImpl) Module() Module {
533 return c.bp.Module().(Module)
534}
535
536func (c *incomingTransitionContextImpl) Config() Config {
537 return c.bp.Config().(Config)
538}
539
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800540func (c *incomingTransitionContextImpl) DeviceConfig() DeviceConfig {
541 return DeviceConfig{c.bp.Config().(Config).deviceConfig}
542}
543
Colin Crosse1a85552024-06-14 12:17:37 -0700544func (c *incomingTransitionContextImpl) IsAddingDependency() bool {
545 return c.bp.IsAddingDependency()
546}
547
Colin Crossaf333f52024-04-15 15:20:23 -0700548func (c *incomingTransitionContextImpl) provider(provider blueprint.AnyProviderKey) (any, bool) {
549 return c.bp.Provider(provider)
550}
551
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800552func (a *androidTransitionMutator) IncomingTransition(bpctx blueprint.IncomingTransitionContext, incomingVariation string) string {
553 if m, ok := bpctx.Module().(Module); ok {
Colin Cross984223f2024-02-01 17:10:23 -0800554 ctx := incomingTransitionContextPool.Get().(*incomingTransitionContextImpl)
555 defer incomingTransitionContextPool.Put(ctx)
556 *ctx = incomingTransitionContextImpl{
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800557 archModuleContext: m.base().archModuleContextFactory(bpctx),
558 bp: bpctx,
559 }
560 return a.mutator.IncomingTransition(ctx, incomingVariation)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200561 } else {
562 return ""
563 }
564}
565
566func (a *androidTransitionMutator) Mutate(ctx blueprint.BottomUpMutatorContext, variation string) {
567 if am, ok := ctx.Module().(Module); ok {
Colin Cross888046f2024-05-01 14:20:31 -0700568 if variation != "" {
569 // TODO: this should really be checking whether the TransitionMutator affected this module, not
570 // the empty variant, but TransitionMutator has no concept of skipping a module.
571 base := am.base()
572 base.commonProperties.DebugMutators = append(base.commonProperties.DebugMutators, a.name)
573 base.commonProperties.DebugVariations = append(base.commonProperties.DebugVariations, variation)
574 }
575
Colin Cross984223f2024-02-01 17:10:23 -0800576 mctx := bottomUpMutatorContextFactory(ctx, am, a.finalPhase)
577 defer bottomUpMutatorContextPool.Put(mctx)
578 a.mutator.Mutate(mctx, variation)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200579 }
580}
581
582func (x *registerMutatorsContext) Transition(name string, m TransitionMutator) {
583 atm := &androidTransitionMutator{
Colin Crossb63d7b32023-12-07 16:54:51 -0800584 finalPhase: x.finalPhase,
585 mutator: m,
Colin Crossd67425d2024-04-15 15:17:05 -0700586 name: name,
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200587 }
588 mutator := &mutator{
589 name: name,
590 transitionMutator: atm}
591 x.mutators = append(x.mutators, mutator)
592}
593
Liz Kammer356f7d42021-01-26 09:18:53 -0500594func (x *registerMutatorsContext) mutatorName(name string) string {
Liz Kammer356f7d42021-01-26 09:18:53 -0500595 return name
596}
597
Colin Cross25de6c32019-06-06 14:29:25 -0700598func (x *registerMutatorsContext) TopDown(name string, m TopDownMutator) MutatorHandle {
Colin Cross798bfce2016-10-12 14:28:16 -0700599 f := func(ctx blueprint.TopDownMutatorContext) {
Colin Cross635c3b02016-05-18 15:37:25 -0700600 if a, ok := ctx.Module().(Module); ok {
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400601 moduleContext := a.base().baseModuleContextFactory(ctx)
Colin Cross984223f2024-02-01 17:10:23 -0800602 actx := topDownMutatorContextPool.Get().(*topDownMutatorContext)
603 defer topDownMutatorContextPool.Put(actx)
604 *actx = topDownMutatorContext{
Colin Crossdc35e212019-06-06 16:13:11 -0700605 bp: ctx,
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400606 baseModuleContext: moduleContext,
Colin Cross6362e272015-10-29 15:25:03 -0700607 }
Colin Cross798bfce2016-10-12 14:28:16 -0700608 m(actx)
Colin Cross6362e272015-10-29 15:25:03 -0700609 }
Colin Cross798bfce2016-10-12 14:28:16 -0700610 }
Liz Kammer356f7d42021-01-26 09:18:53 -0500611 mutator := &mutator{name: x.mutatorName(name), topDownMutator: f}
Colin Cross795c3772017-03-16 16:50:10 -0700612 x.mutators = append(x.mutators, mutator)
Colin Cross798bfce2016-10-12 14:28:16 -0700613 return mutator
614}
615
Paul Duffin1d2d42f2021-03-06 20:08:12 +0000616func (mutator *mutator) componentName() string {
617 return mutator.name
618}
619
620func (mutator *mutator) register(ctx *Context) {
621 blueprintCtx := ctx.Context
622 var handle blueprint.MutatorHandle
623 if mutator.bottomUpMutator != nil {
624 handle = blueprintCtx.RegisterBottomUpMutator(mutator.name, mutator.bottomUpMutator)
625 } else if mutator.topDownMutator != nil {
626 handle = blueprintCtx.RegisterTopDownMutator(mutator.name, mutator.topDownMutator)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200627 } else if mutator.transitionMutator != nil {
628 blueprintCtx.RegisterTransitionMutator(mutator.name, mutator.transitionMutator)
Paul Duffin1d2d42f2021-03-06 20:08:12 +0000629 }
630 if mutator.parallel {
631 handle.Parallel()
632 }
633}
634
Colin Cross798bfce2016-10-12 14:28:16 -0700635type MutatorHandle interface {
636 Parallel() MutatorHandle
637}
638
639func (mutator *mutator) Parallel() MutatorHandle {
640 mutator.parallel = true
641 return mutator
Colin Cross6362e272015-10-29 15:25:03 -0700642}
Colin Cross1e676be2016-10-12 14:38:15 -0700643
Paul Duffin44f1d842020-06-26 20:17:02 +0100644func RegisterComponentsMutator(ctx RegisterMutatorsContext) {
645 ctx.BottomUp("component-deps", componentDepsMutator).Parallel()
646}
647
648// A special mutator that runs just prior to the deps mutator to allow the dependencies
649// on component modules to be added so that they can depend directly on a prebuilt
650// module.
651func componentDepsMutator(ctx BottomUpMutatorContext) {
Cole Fausta963b942024-04-11 17:43:00 -0700652 ctx.Module().ComponentDepsMutator(ctx)
Paul Duffin44f1d842020-06-26 20:17:02 +0100653}
654
Colin Cross1e676be2016-10-12 14:38:15 -0700655func depsMutator(ctx BottomUpMutatorContext) {
Cole Fausta963b942024-04-11 17:43:00 -0700656 if m := ctx.Module(); m.Enabled(ctx) {
Ronald Braunstein73b08ff2023-12-19 10:24:47 -0800657 m.base().baseDepsMutator(ctx)
Colin Cross1e676be2016-10-12 14:38:15 -0700658 m.DepsMutator(ctx)
659 }
660}
Colin Crossd11fcda2017-10-23 17:59:01 -0700661
Liz Kammer356f7d42021-01-26 09:18:53 -0500662func registerDepsMutator(ctx RegisterMutatorsContext) {
663 ctx.BottomUp("deps", depsMutator).Parallel()
664}
665
Colin Crossdc35e212019-06-06 16:13:11 -0700666// android.topDownMutatorContext either has to embed blueprint.TopDownMutatorContext, in which case every method that
667// has an overridden version in android.BaseModuleContext has to be manually forwarded to BaseModuleContext to avoid
668// ambiguous method errors, or it has to store a blueprint.TopDownMutatorContext non-embedded, in which case every
669// non-overridden method has to be forwarded. There are fewer non-overridden methods, so use the latter. The following
670// methods forward to the identical blueprint versions for topDownMutatorContext and bottomUpMutatorContext.
671
Colin Crosscb55e082019-07-01 15:32:31 -0700672func (t *topDownMutatorContext) MutatorName() string {
673 return t.bp.MutatorName()
674}
675
Colin Crossdc35e212019-06-06 16:13:11 -0700676func (t *topDownMutatorContext) Rename(name string) {
677 t.bp.Rename(name)
Colin Cross9a362232019-07-01 15:32:45 -0700678 t.Module().base().commonProperties.DebugName = name
Colin Crossdc35e212019-06-06 16:13:11 -0700679}
680
Liz Kammerf31c9002022-04-26 09:08:55 -0400681func (t *topDownMutatorContext) createModule(factory blueprint.ModuleFactory, name string, props ...interface{}) blueprint.Module {
682 return t.bp.CreateModule(factory, name, props...)
683}
684
Colin Crosse003c4a2019-09-25 12:58:36 -0700685func (t *topDownMutatorContext) CreateModule(factory ModuleFactory, props ...interface{}) Module {
Liz Kammerf31c9002022-04-26 09:08:55 -0400686 return createModule(t, factory, "_topDownMutatorModule", props...)
Colin Crossdc35e212019-06-06 16:13:11 -0700687}
688
Liz Kammera060c452021-03-24 10:14:47 -0400689func (t *topDownMutatorContext) createModuleWithoutInheritance(factory ModuleFactory, props ...interface{}) Module {
Liz Kammerf31c9002022-04-26 09:08:55 -0400690 module := t.bp.CreateModule(ModuleFactoryAdaptor(factory), "", props...).(Module)
Liz Kammera060c452021-03-24 10:14:47 -0400691 return module
692}
693
Colin Crosscb55e082019-07-01 15:32:31 -0700694func (b *bottomUpMutatorContext) MutatorName() string {
695 return b.bp.MutatorName()
696}
697
Colin Crossdc35e212019-06-06 16:13:11 -0700698func (b *bottomUpMutatorContext) Rename(name string) {
699 b.bp.Rename(name)
Colin Cross9a362232019-07-01 15:32:45 -0700700 b.Module().base().commonProperties.DebugName = name
Colin Crossdc35e212019-06-06 16:13:11 -0700701}
702
Colin Crossda279cf2024-09-17 14:25:45 -0700703func (b *bottomUpMutatorContext) createModule(factory blueprint.ModuleFactory, name string, props ...interface{}) blueprint.Module {
704 return b.bp.CreateModule(factory, name, props...)
705}
706
707func (b *bottomUpMutatorContext) CreateModule(factory ModuleFactory, props ...interface{}) Module {
708 return createModule(b, factory, "_bottomUpMutatorModule", props...)
709}
710
Colin Cross4f1dcb02020-09-16 18:45:04 -0700711func (b *bottomUpMutatorContext) AddDependency(module blueprint.Module, tag blueprint.DependencyTag, name ...string) []blueprint.Module {
Liz Kammerc13f7852023-05-17 13:01:48 -0400712 if b.baseModuleContext.checkedMissingDeps() {
713 panic("Adding deps not allowed after checking for missing deps")
714 }
Colin Cross4f1dcb02020-09-16 18:45:04 -0700715 return b.bp.AddDependency(module, tag, name...)
Colin Crossdc35e212019-06-06 16:13:11 -0700716}
717
718func (b *bottomUpMutatorContext) AddReverseDependency(module blueprint.Module, tag blueprint.DependencyTag, name string) {
Liz Kammerc13f7852023-05-17 13:01:48 -0400719 if b.baseModuleContext.checkedMissingDeps() {
720 panic("Adding deps not allowed after checking for missing deps")
721 }
Colin Crossdc35e212019-06-06 16:13:11 -0700722 b.bp.AddReverseDependency(module, tag, name)
723}
Cole Faustf4992e82024-09-30 15:16:18 -0700724
725func (b *bottomUpMutatorContext) AddReverseVariationDependency(variations []blueprint.Variation, tag blueprint.DependencyTag, name string) {
726 if b.baseModuleContext.checkedMissingDeps() {
727 panic("Adding deps not allowed after checking for missing deps")
728 }
729 b.bp.AddReverseVariationDependency(variations, tag, name)
730}
731
Colin Crossdc35e212019-06-06 16:13:11 -0700732func (b *bottomUpMutatorContext) AddVariationDependencies(variations []blueprint.Variation, tag blueprint.DependencyTag,
Colin Cross4f1dcb02020-09-16 18:45:04 -0700733 names ...string) []blueprint.Module {
Liz Kammerc13f7852023-05-17 13:01:48 -0400734 if b.baseModuleContext.checkedMissingDeps() {
735 panic("Adding deps not allowed after checking for missing deps")
736 }
Colin Cross4f1dcb02020-09-16 18:45:04 -0700737 return b.bp.AddVariationDependencies(variations, tag, names...)
Colin Crossdc35e212019-06-06 16:13:11 -0700738}
739
740func (b *bottomUpMutatorContext) AddFarVariationDependencies(variations []blueprint.Variation,
Colin Cross4f1dcb02020-09-16 18:45:04 -0700741 tag blueprint.DependencyTag, names ...string) []blueprint.Module {
Liz Kammerc13f7852023-05-17 13:01:48 -0400742 if b.baseModuleContext.checkedMissingDeps() {
743 panic("Adding deps not allowed after checking for missing deps")
744 }
Colin Crossdc35e212019-06-06 16:13:11 -0700745
Colin Cross4f1dcb02020-09-16 18:45:04 -0700746 return b.bp.AddFarVariationDependencies(variations, tag, names...)
Colin Crossdc35e212019-06-06 16:13:11 -0700747}
748
Colin Crossdc35e212019-06-06 16:13:11 -0700749func (b *bottomUpMutatorContext) ReplaceDependencies(name string) {
Liz Kammerc13f7852023-05-17 13:01:48 -0400750 if b.baseModuleContext.checkedMissingDeps() {
751 panic("Adding deps not allowed after checking for missing deps")
752 }
Colin Crossdc35e212019-06-06 16:13:11 -0700753 b.bp.ReplaceDependencies(name)
754}
Jaewoong Jung9f88ce22019-11-15 10:57:34 -0800755
Paul Duffin80342d72020-06-26 22:08:43 +0100756func (b *bottomUpMutatorContext) ReplaceDependenciesIf(name string, predicate blueprint.ReplaceDependencyPredicate) {
Liz Kammerc13f7852023-05-17 13:01:48 -0400757 if b.baseModuleContext.checkedMissingDeps() {
758 panic("Adding deps not allowed after checking for missing deps")
759 }
Paul Duffin80342d72020-06-26 22:08:43 +0100760 b.bp.ReplaceDependenciesIf(name, predicate)
761}