blob: a8b5c7db61076044b6f06ddab98ac2f45046ef51 [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 {
35 return collateRegisteredMutators(preArch, preDeps, postDeps, finalDeps)
36}
37
38// collateRegisteredMutators constructs a single list of mutators from the separate lists.
39func collateRegisteredMutators(preArch, preDeps, postDeps, 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
Martin Stjernholm710ec3a2020-01-16 15:12:04 +000056 mctx.finalPhase = true
57 register(finalDeps)
58
Paul Duffinc05b0342021-03-06 13:28:13 +000059 return mctx.mutators
Colin Cross795c3772017-03-16 16:50:10 -070060}
61
62type registerMutatorsContext struct {
Colin Crossb63d7b32023-12-07 16:54:51 -080063 mutators sortableComponents
64 finalPhase bool
Colin Cross795c3772017-03-16 16:50:10 -070065}
Colin Cross1e676be2016-10-12 14:38:15 -070066
67type RegisterMutatorsContext interface {
Colin Cross25de6c32019-06-06 14:29:25 -070068 TopDown(name string, m TopDownMutator) MutatorHandle
69 BottomUp(name string, m BottomUpMutator) MutatorHandle
Colin Cross617b88a2020-08-24 18:04:09 -070070 BottomUpBlueprint(name string, m blueprint.BottomUpMutator) MutatorHandle
Lukacs T. Berki6c716762022-06-13 20:50:39 +020071 Transition(name string, m TransitionMutator)
Colin Cross1e676be2016-10-12 14:38:15 -070072}
73
74type RegisterMutatorFunc func(RegisterMutatorsContext)
75
Colin Crosscec81712017-07-13 14:43:27 -070076var preArch = []RegisterMutatorFunc{
Dan Willemsen6e72ef72018-01-26 18:27:02 -080077 RegisterNamespaceMutator,
Paul Duffinaa4162e2020-05-05 11:35:43 +010078
Paul Duffinaa4162e2020-05-05 11:35:43 +010079 // Check the visibility rules are valid.
80 //
81 // This must run after the package renamer mutators so that any issues found during
82 // validation of the package's default_visibility property are reported using the
83 // correct package name and not the synthetic name.
84 //
85 // This must also be run before defaults mutators as the rules for validation are
86 // different before checking the rules than they are afterwards. e.g.
87 // visibility: ["//visibility:private", "//visibility:public"]
88 // would be invalid if specified in a module definition but is valid if it results
89 // from something like this:
90 //
91 // defaults {
92 // name: "defaults",
93 // // Be inaccessible outside a package by default.
94 // visibility: ["//visibility:private"]
95 // }
96 //
97 // defaultable_module {
98 // name: "defaultable_module",
99 // defaults: ["defaults"],
100 // // Override the default.
101 // visibility: ["//visibility:public"]
102 // }
103 //
Paul Duffin593b3c92019-12-05 14:31:48 +0000104 RegisterVisibilityRuleChecker,
Paul Duffinaa4162e2020-05-05 11:35:43 +0100105
Bob Badour37af0462021-01-07 03:34:31 +0000106 // Record the default_applicable_licenses for each package.
107 //
108 // This must run before the defaults so that defaults modules can pick up the package default.
109 RegisterLicensesPackageMapper,
110
Paul Duffinaa4162e2020-05-05 11:35:43 +0100111 // Apply properties from defaults modules to the referencing modules.
Paul Duffinafa9fa12020-04-29 16:47:28 +0100112 //
113 // Any mutators that are added before this will not see any modules created by
114 // a DefaultableHook.
Colin Cross89536d42017-07-07 14:35:50 -0700115 RegisterDefaultsPreArchMutators,
Paul Duffinaa4162e2020-05-05 11:35:43 +0100116
Paul Duffin44f1d842020-06-26 20:17:02 +0100117 // Add dependencies on any components so that any component references can be
118 // resolved within the deps mutator.
119 //
120 // Must be run after defaults so it can be used to create dependencies on the
121 // component modules that are creating in a DefaultableHook.
122 //
123 // Must be run before RegisterPrebuiltsPreArchMutators, i.e. before prebuilts are
124 // renamed. That is so that if a module creates components using a prebuilt module
125 // type that any dependencies (which must use prebuilt_ prefixes) are resolved to
126 // the prebuilt module and not the source module.
127 RegisterComponentsMutator,
128
Paul Duffinc988c8e2020-04-29 18:27:14 +0100129 // Create an association between prebuilt modules and their corresponding source
130 // modules (if any).
Paul Duffinafa9fa12020-04-29 16:47:28 +0100131 //
132 // Must be run after defaults mutators to ensure that any modules created by
133 // a DefaultableHook can be either a prebuilt or a source module with a matching
134 // prebuilt.
Paul Duffinc988c8e2020-04-29 18:27:14 +0100135 RegisterPrebuiltsPreArchMutators,
136
Bob Badour37af0462021-01-07 03:34:31 +0000137 // Gather the licenses properties for all modules for use during expansion and enforcement.
138 //
139 // This must come after the defaults mutators to ensure that any licenses supplied
140 // in a defaults module has been successfully applied before the rules are gathered.
141 RegisterLicensesPropertyGatherer,
142
Paul Duffinaa4162e2020-05-05 11:35:43 +0100143 // Gather the visibility rules for all modules for us during visibility enforcement.
144 //
145 // This must come after the defaults mutators to ensure that any visibility supplied
146 // in a defaults module has been successfully applied before the rules are gathered.
Paul Duffin593b3c92019-12-05 14:31:48 +0000147 RegisterVisibilityRuleGatherer,
Colin Crosscec81712017-07-13 14:43:27 -0700148}
149
Colin Crossae4c6182017-09-15 17:33:55 -0700150func registerArchMutator(ctx RegisterMutatorsContext) {
Colin Cross8bbc3d52024-09-11 15:33:54 -0700151 ctx.Transition("os", &osTransitionMutator{})
Jihoon Kang5402bbd2024-07-31 18:37:49 +0000152 ctx.Transition("image", &imageTransitionMutator{})
Colin Cross8bbc3d52024-09-11 15:33:54 -0700153 ctx.Transition("arch", &archTransitionMutator{})
Colin Crossae4c6182017-09-15 17:33:55 -0700154}
155
Colin Crosscec81712017-07-13 14:43:27 -0700156var preDeps = []RegisterMutatorFunc{
Colin Crossae4c6182017-09-15 17:33:55 -0700157 registerArchMutator,
Colin Crosscec81712017-07-13 14:43:27 -0700158}
159
160var postDeps = []RegisterMutatorFunc{
Colin Cross1b488422019-03-04 22:33:56 -0800161 registerPathDepsMutator,
Colin Cross5ea9bcc2017-07-27 15:41:32 -0700162 RegisterPrebuiltsPostDepsMutators,
Paul Duffin593b3c92019-12-05 14:31:48 +0000163 RegisterVisibilityRuleEnforcer,
Bob Badour37af0462021-01-07 03:34:31 +0000164 RegisterLicensesDependencyChecker,
Paul Duffin45338f02021-03-30 23:07:52 +0100165 registerNeverallowMutator,
Jaewoong Jungb639a6a2019-05-10 15:16:29 -0700166 RegisterOverridePostDepsMutators,
Colin Crosscec81712017-07-13 14:43:27 -0700167}
Colin Cross1e676be2016-10-12 14:38:15 -0700168
Martin Stjernholm710ec3a2020-01-16 15:12:04 +0000169var finalDeps = []RegisterMutatorFunc{}
170
Colin Cross1e676be2016-10-12 14:38:15 -0700171func PreArchMutators(f RegisterMutatorFunc) {
172 preArch = append(preArch, f)
173}
174
175func PreDepsMutators(f RegisterMutatorFunc) {
176 preDeps = append(preDeps, f)
177}
178
179func PostDepsMutators(f RegisterMutatorFunc) {
180 postDeps = append(postDeps, f)
181}
182
Martin Stjernholm710ec3a2020-01-16 15:12:04 +0000183func FinalDepsMutators(f RegisterMutatorFunc) {
184 finalDeps = append(finalDeps, f)
185}
186
Chris Parsons637458d2023-09-19 20:09:00 +0000187type BaseMutatorContext interface {
188 BaseModuleContext
189
190 // MutatorName returns the name that this mutator was registered with.
191 MutatorName() string
192
193 // Rename all variants of a module. The new name is not visible to calls to ModuleName,
194 // AddDependency or OtherModuleName until after this mutator pass is complete.
195 Rename(name string)
Colin Crossda279cf2024-09-17 14:25:45 -0700196
197 // CreateModule creates a new module by calling the factory method for the specified moduleType, and applies
198 // the specified property structs to it as if the properties were set in a blueprint file.
199 CreateModule(ModuleFactory, ...interface{}) Module
Chris Parsons637458d2023-09-19 20:09:00 +0000200}
201
202type TopDownMutator func(TopDownMutatorContext)
203
204type TopDownMutatorContext interface {
205 BaseMutatorContext
Chris Parsons637458d2023-09-19 20:09:00 +0000206}
207
Colin Cross25de6c32019-06-06 14:29:25 -0700208type topDownMutatorContext struct {
Colin Crossdc35e212019-06-06 16:13:11 -0700209 bp blueprint.TopDownMutatorContext
Colin Cross0ea8ba82019-06-06 14:33:29 -0700210 baseModuleContext
Colin Cross6362e272015-10-29 15:25:03 -0700211}
212
Colin Cross25de6c32019-06-06 14:29:25 -0700213type BottomUpMutator func(BottomUpMutatorContext)
Colin Cross6362e272015-10-29 15:25:03 -0700214
Colin Cross635c3b02016-05-18 15:37:25 -0700215type BottomUpMutatorContext interface {
Colin Cross9f35c3d2020-09-16 19:04:41 -0700216 BaseMutatorContext
Colin Crossb63d7b32023-12-07 16:54:51 -0800217
218 // AddDependency adds a dependency to the given module. It returns a slice of modules for each
219 // dependency (some entries may be nil).
220 //
221 // If the mutator is parallel (see MutatorHandle.Parallel), this method will pause until the
222 // new dependencies have had the current mutator called on them. If the mutator is not
223 // parallel this method does not affect the ordering of the current mutator pass, but will
224 // be ordered correctly for all future mutator passes.
225 AddDependency(module blueprint.Module, tag blueprint.DependencyTag, name ...string) []blueprint.Module
Colin Crossaabf6792017-11-29 00:27:14 -0800226
Colin Cross9f35c3d2020-09-16 19:04:41 -0700227 // AddReverseDependency adds a dependency from the destination to the given module.
228 // Does not affect the ordering of the current mutator pass, but will be ordered
229 // correctly for all future mutator passes. All reverse dependencies for a destination module are
230 // collected until the end of the mutator pass, sorted by name, and then appended to the destination
231 // module's dependency list.
Colin Crossaabf6792017-11-29 00:27:14 -0800232 AddReverseDependency(module blueprint.Module, tag blueprint.DependencyTag, name string)
Colin Cross9f35c3d2020-09-16 19:04:41 -0700233
Colin Cross9f35c3d2020-09-16 19:04:41 -0700234 // AddVariationDependencies adds deps as dependencies of the current module, but uses the variations
Colin Cross4f1dcb02020-09-16 18:45:04 -0700235 // argument to select which variant of the dependency to use. It returns a slice of modules for
236 // each dependency (some entries may be nil). A variant of the dependency must exist that matches
Usta Shresthac725f472022-01-11 02:44:21 -0500237 // all the non-local variations of the current module, plus the variations argument.
Colin Cross4f1dcb02020-09-16 18:45:04 -0700238 //
239 // If the mutator is parallel (see MutatorHandle.Parallel), this method will pause until the
240 // new dependencies have had the current mutator called on them. If the mutator is not
241 // parallel this method does not affect the ordering of the current mutator pass, but will
242 // be ordered correctly for all future mutator passes.
Usta Shresthac725f472022-01-11 02:44:21 -0500243 AddVariationDependencies(variations []blueprint.Variation, tag blueprint.DependencyTag, names ...string) []blueprint.Module
Colin Cross9f35c3d2020-09-16 19:04:41 -0700244
Cole Faustf4992e82024-09-30 15:16:18 -0700245 // AddReverseVariationDependencies adds a dependency from the named module to the current
246 // module. The given variations will be added to the current module's varations, and then the
247 // result will be used to find the correct variation of the depending module, which must exist.
248 //
249 // Does not affect the ordering of the current mutator pass, but will be ordered
250 // correctly for all future mutator passes. All reverse dependencies for a destination module are
251 // collected until the end of the mutator pass, sorted by name, and then appended to the destination
252 // module's dependency list.
253 AddReverseVariationDependency([]blueprint.Variation, blueprint.DependencyTag, string)
254
Colin Cross9f35c3d2020-09-16 19:04:41 -0700255 // AddFarVariationDependencies adds deps as dependencies of the current module, but uses the
Colin Cross4f1dcb02020-09-16 18:45:04 -0700256 // variations argument to select which variant of the dependency to use. It returns a slice of
257 // modules for each dependency (some entries may be nil). A variant of the dependency must
258 // exist that matches the variations argument, but may also have other variations.
Colin Cross9f35c3d2020-09-16 19:04:41 -0700259 // For any unspecified variation the first variant will be used.
260 //
261 // Unlike AddVariationDependencies, the variations of the current module are ignored - the
262 // dependency only needs to match the supplied variations.
Colin Cross4f1dcb02020-09-16 18:45:04 -0700263 //
264 // If the mutator is parallel (see MutatorHandle.Parallel), this method will pause until the
265 // new dependencies have had the current mutator called on them. If the mutator is not
266 // parallel this method does not affect the ordering of the current mutator pass, but will
267 // be ordered correctly for all future mutator passes.
268 AddFarVariationDependencies([]blueprint.Variation, blueprint.DependencyTag, ...string) []blueprint.Module
Colin Cross9f35c3d2020-09-16 19:04:41 -0700269
Colin Cross86771322024-05-01 14:19:51 -0700270 // ReplaceDependencies finds all the variants of the module with the specified name, then
271 // replaces all dependencies onto those variants with the current variant of this module.
272 // Replacements don't take effect until after the mutator pass is finished.
Colin Crossaabf6792017-11-29 00:27:14 -0800273 ReplaceDependencies(string)
Colin Cross9f35c3d2020-09-16 19:04:41 -0700274
Colin Cross86771322024-05-01 14:19:51 -0700275 // ReplaceDependenciesIf finds all the variants of the module with the specified name, then
276 // replaces all dependencies onto those variants with the current variant of this module
277 // as long as the supplied predicate returns true.
Colin Cross9f35c3d2020-09-16 19:04:41 -0700278 // Replacements don't take effect until after the mutator pass is finished.
Paul Duffin80342d72020-06-26 22:08:43 +0100279 ReplaceDependenciesIf(string, blueprint.ReplaceDependencyPredicate)
Colin Cross6362e272015-10-29 15:25:03 -0700280}
281
Colin Cross984223f2024-02-01 17:10:23 -0800282// An outgoingTransitionContextImpl and incomingTransitionContextImpl is created for every dependency of every module
283// for each transition mutator. bottomUpMutatorContext and topDownMutatorContext are created once for every module
284// for every BottomUp or TopDown mutator. Use a global pool for each to avoid reallocating every time.
285var (
286 outgoingTransitionContextPool = sync.Pool{
287 New: func() any { return &outgoingTransitionContextImpl{} },
288 }
289 incomingTransitionContextPool = sync.Pool{
290 New: func() any { return &incomingTransitionContextImpl{} },
291 }
292 bottomUpMutatorContextPool = sync.Pool{
293 New: func() any { return &bottomUpMutatorContext{} },
294 }
295
296 topDownMutatorContextPool = sync.Pool{
297 New: func() any { return &topDownMutatorContext{} },
298 }
299)
300
Colin Cross25de6c32019-06-06 14:29:25 -0700301type bottomUpMutatorContext struct {
Colin Crossdc35e212019-06-06 16:13:11 -0700302 bp blueprint.BottomUpMutatorContext
Colin Cross0ea8ba82019-06-06 14:33:29 -0700303 baseModuleContext
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400304 finalPhase bool
Colin Cross6362e272015-10-29 15:25:03 -0700305}
306
Colin Cross984223f2024-02-01 17:10:23 -0800307// callers must immediately follow the call to this function with defer bottomUpMutatorContextPool.Put(mctx).
Colin Cross617b88a2020-08-24 18:04:09 -0700308func bottomUpMutatorContextFactory(ctx blueprint.BottomUpMutatorContext, a Module,
Colin Crossb63d7b32023-12-07 16:54:51 -0800309 finalPhase bool) BottomUpMutatorContext {
Colin Cross617b88a2020-08-24 18:04:09 -0700310
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400311 moduleContext := a.base().baseModuleContextFactory(ctx)
Colin Cross984223f2024-02-01 17:10:23 -0800312 mctx := bottomUpMutatorContextPool.Get().(*bottomUpMutatorContext)
313 *mctx = bottomUpMutatorContext{
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400314 bp: ctx,
Cole Faust0abd4b42023-01-10 10:49:18 -0800315 baseModuleContext: moduleContext,
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400316 finalPhase: finalPhase,
Colin Cross617b88a2020-08-24 18:04:09 -0700317 }
Colin Cross984223f2024-02-01 17:10:23 -0800318 return mctx
Colin Cross617b88a2020-08-24 18:04:09 -0700319}
320
Colin Cross25de6c32019-06-06 14:29:25 -0700321func (x *registerMutatorsContext) BottomUp(name string, m BottomUpMutator) MutatorHandle {
Martin Stjernholm710ec3a2020-01-16 15:12:04 +0000322 finalPhase := x.finalPhase
Colin Cross798bfce2016-10-12 14:28:16 -0700323 f := func(ctx blueprint.BottomUpMutatorContext) {
Colin Cross635c3b02016-05-18 15:37:25 -0700324 if a, ok := ctx.Module().(Module); ok {
Colin Cross984223f2024-02-01 17:10:23 -0800325 mctx := bottomUpMutatorContextFactory(ctx, a, finalPhase)
326 defer bottomUpMutatorContextPool.Put(mctx)
327 m(mctx)
Colin Cross6362e272015-10-29 15:25:03 -0700328 }
Colin Cross798bfce2016-10-12 14:28:16 -0700329 }
Liz Kammer356f7d42021-01-26 09:18:53 -0500330 mutator := &mutator{name: x.mutatorName(name), bottomUpMutator: f}
Colin Cross795c3772017-03-16 16:50:10 -0700331 x.mutators = append(x.mutators, mutator)
Colin Cross798bfce2016-10-12 14:28:16 -0700332 return mutator
Colin Cross6362e272015-10-29 15:25:03 -0700333}
334
Colin Cross617b88a2020-08-24 18:04:09 -0700335func (x *registerMutatorsContext) BottomUpBlueprint(name string, m blueprint.BottomUpMutator) MutatorHandle {
336 mutator := &mutator{name: name, bottomUpMutator: m}
337 x.mutators = append(x.mutators, mutator)
338 return mutator
339}
340
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200341type IncomingTransitionContext interface {
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800342 ArchModuleContext
Colin Crossaf333f52024-04-15 15:20:23 -0700343 ModuleProviderContext
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800344
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200345 // Module returns the target of the dependency edge for which the transition
346 // is being computed
347 Module() Module
348
349 // Config returns the configuration for the build.
350 Config() Config
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800351
352 DeviceConfig() DeviceConfig
Colin Crosse1a85552024-06-14 12:17:37 -0700353
354 // IsAddingDependency returns true if the transition is being called while adding a dependency
355 // after the transition mutator has already run, or false if it is being called when the transition
356 // mutator is running. This should be used sparingly, all uses will have to be removed in order
357 // to support creating variants on demand.
358 IsAddingDependency() bool
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200359}
360
361type OutgoingTransitionContext interface {
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800362 ArchModuleContext
Colin Crossaf333f52024-04-15 15:20:23 -0700363 ModuleProviderContext
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800364
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200365 // Module returns the target of the dependency edge for which the transition
366 // is being computed
367 Module() Module
368
369 // DepTag() Returns the dependency tag through which this dependency is
370 // reached
371 DepTag() blueprint.DependencyTag
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800372
373 // Config returns the configuration for the build.
374 Config() Config
375
376 DeviceConfig() DeviceConfig
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200377}
Lukacs T. Berki0e691c12022-06-24 10:15:55 +0200378
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800379// TransitionMutator implements a top-down mechanism where a module tells its
Lukacs T. Berki0e691c12022-06-24 10:15:55 +0200380// direct dependencies what variation they should be built in but the dependency
381// has the final say.
382//
383// When implementing a transition mutator, one needs to implement four methods:
384// - Split() that tells what variations a module has by itself
385// - OutgoingTransition() where a module tells what it wants from its
386// dependency
387// - IncomingTransition() where a module has the final say about its own
388// variation
389// - Mutate() that changes the state of a module depending on its variation
390//
391// That the effective variation of module B when depended on by module A is the
392// composition the outgoing transition of module A and the incoming transition
393// of module B.
394//
395// the outgoing transition should not take the properties of the dependency into
396// account, only those of the module that depends on it. For this reason, the
397// dependency is not even passed into it as an argument. Likewise, the incoming
398// transition should not take the properties of the depending module into
399// account and is thus not informed about it. This makes for a nice
400// decomposition of the decision logic.
401//
402// A given transition mutator only affects its own variation; other variations
403// stay unchanged along the dependency edges.
404//
405// Soong makes sure that all modules are created in the desired variations and
406// that dependency edges are set up correctly. This ensures that "missing
407// variation" errors do not happen and allows for more flexible changes in the
408// value of the variation among dependency edges (as oppposed to bottom-up
409// mutators where if module A in variation X depends on module B and module B
410// has that variation X, A must depend on variation X of B)
411//
412// The limited power of the context objects passed to individual mutators
413// methods also makes it more difficult to shoot oneself in the foot. Complete
414// safety is not guaranteed because no one prevents individual transition
415// mutators from mutating modules in illegal ways and for e.g. Split() or
416// Mutate() to run their own visitations of the transitive dependency of the
417// module and both of these are bad ideas, but it's better than no guardrails at
418// all.
419//
420// This model is pretty close to Bazel's configuration transitions. The mapping
421// between concepts in Soong and Bazel is as follows:
422// - Module == configured target
423// - Variant == configuration
424// - Variation name == configuration flag
425// - Variation == configuration flag value
426// - Outgoing transition == attribute transition
427// - Incoming transition == rule transition
428//
429// The Split() method does not have a Bazel equivalent and Bazel split
430// transitions do not have a Soong equivalent.
431//
432// Mutate() does not make sense in Bazel due to the different models of the
433// two systems: when creating new variations, Soong clones the old module and
434// thus some way is needed to change it state whereas Bazel creates each
435// configuration of a given configured target anew.
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200436type TransitionMutator interface {
437 // Split returns the set of variations that should be created for a module no
438 // matter who depends on it. Used when Make depends on a particular variation
439 // or when the module knows its variations just based on information given to
440 // it in the Blueprint file. This method should not mutate the module it is
441 // called on.
442 Split(ctx BaseModuleContext) []string
443
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800444 // OutgoingTransition is called on a module to determine which variation it wants
445 // from its direct dependencies. The dependency itself can override this decision.
446 // This method should not mutate the module itself.
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200447 OutgoingTransition(ctx OutgoingTransitionContext, sourceVariation string) string
448
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800449 // IncomingTransition is called on a module to determine which variation it should
450 // be in based on the variation modules that depend on it want. This gives the module
451 // a final say about its own variations. This method should not mutate the module
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200452 // itself.
453 IncomingTransition(ctx IncomingTransitionContext, incomingVariation string) string
454
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800455 // Mutate is called after a module was split into multiple variations on each variation.
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200456 // It should not split the module any further but adding new dependencies is
457 // fine. Unlike all the other methods on TransitionMutator, this method is
458 // allowed to mutate the module.
459 Mutate(ctx BottomUpMutatorContext, variation string)
460}
461
462type androidTransitionMutator struct {
Colin Crossb63d7b32023-12-07 16:54:51 -0800463 finalPhase bool
464 mutator TransitionMutator
Colin Crossd67425d2024-04-15 15:17:05 -0700465 name string
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200466}
467
468func (a *androidTransitionMutator) Split(ctx blueprint.BaseModuleContext) []string {
Colin Crossd27205e2024-09-12 22:41:37 -0700469 if a.finalPhase {
470 panic("TransitionMutator not allowed in FinalDepsMutators")
471 }
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200472 if m, ok := ctx.Module().(Module); ok {
473 moduleContext := m.base().baseModuleContextFactory(ctx)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200474 return a.mutator.Split(&moduleContext)
475 } else {
476 return []string{""}
477 }
478}
479
480type outgoingTransitionContextImpl struct {
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800481 archModuleContext
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200482 bp blueprint.OutgoingTransitionContext
483}
484
485func (c *outgoingTransitionContextImpl) Module() Module {
486 return c.bp.Module().(Module)
487}
488
489func (c *outgoingTransitionContextImpl) DepTag() blueprint.DependencyTag {
490 return c.bp.DepTag()
491}
492
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800493func (c *outgoingTransitionContextImpl) Config() Config {
494 return c.bp.Config().(Config)
495}
496
497func (c *outgoingTransitionContextImpl) DeviceConfig() DeviceConfig {
498 return DeviceConfig{c.bp.Config().(Config).deviceConfig}
499}
500
Colin Crossaf333f52024-04-15 15:20:23 -0700501func (c *outgoingTransitionContextImpl) provider(provider blueprint.AnyProviderKey) (any, bool) {
502 return c.bp.Provider(provider)
503}
504
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800505func (a *androidTransitionMutator) OutgoingTransition(bpctx blueprint.OutgoingTransitionContext, sourceVariation string) string {
506 if m, ok := bpctx.Module().(Module); ok {
Colin Cross984223f2024-02-01 17:10:23 -0800507 ctx := outgoingTransitionContextPool.Get().(*outgoingTransitionContextImpl)
508 defer outgoingTransitionContextPool.Put(ctx)
509 *ctx = outgoingTransitionContextImpl{
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800510 archModuleContext: m.base().archModuleContextFactory(bpctx),
511 bp: bpctx,
512 }
513 return a.mutator.OutgoingTransition(ctx, sourceVariation)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200514 } else {
515 return ""
516 }
517}
518
519type incomingTransitionContextImpl struct {
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800520 archModuleContext
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200521 bp blueprint.IncomingTransitionContext
522}
523
524func (c *incomingTransitionContextImpl) Module() Module {
525 return c.bp.Module().(Module)
526}
527
528func (c *incomingTransitionContextImpl) Config() Config {
529 return c.bp.Config().(Config)
530}
531
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800532func (c *incomingTransitionContextImpl) DeviceConfig() DeviceConfig {
533 return DeviceConfig{c.bp.Config().(Config).deviceConfig}
534}
535
Colin Crosse1a85552024-06-14 12:17:37 -0700536func (c *incomingTransitionContextImpl) IsAddingDependency() bool {
537 return c.bp.IsAddingDependency()
538}
539
Colin Crossaf333f52024-04-15 15:20:23 -0700540func (c *incomingTransitionContextImpl) provider(provider blueprint.AnyProviderKey) (any, bool) {
541 return c.bp.Provider(provider)
542}
543
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800544func (a *androidTransitionMutator) IncomingTransition(bpctx blueprint.IncomingTransitionContext, incomingVariation string) string {
545 if m, ok := bpctx.Module().(Module); ok {
Colin Cross984223f2024-02-01 17:10:23 -0800546 ctx := incomingTransitionContextPool.Get().(*incomingTransitionContextImpl)
547 defer incomingTransitionContextPool.Put(ctx)
548 *ctx = incomingTransitionContextImpl{
Colin Cross4aa3e0a2024-01-18 17:22:58 -0800549 archModuleContext: m.base().archModuleContextFactory(bpctx),
550 bp: bpctx,
551 }
552 return a.mutator.IncomingTransition(ctx, incomingVariation)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200553 } else {
554 return ""
555 }
556}
557
558func (a *androidTransitionMutator) Mutate(ctx blueprint.BottomUpMutatorContext, variation string) {
559 if am, ok := ctx.Module().(Module); ok {
Colin Cross888046f2024-05-01 14:20:31 -0700560 if variation != "" {
561 // TODO: this should really be checking whether the TransitionMutator affected this module, not
562 // the empty variant, but TransitionMutator has no concept of skipping a module.
563 base := am.base()
564 base.commonProperties.DebugMutators = append(base.commonProperties.DebugMutators, a.name)
565 base.commonProperties.DebugVariations = append(base.commonProperties.DebugVariations, variation)
566 }
567
Colin Cross984223f2024-02-01 17:10:23 -0800568 mctx := bottomUpMutatorContextFactory(ctx, am, a.finalPhase)
569 defer bottomUpMutatorContextPool.Put(mctx)
570 a.mutator.Mutate(mctx, variation)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200571 }
572}
573
574func (x *registerMutatorsContext) Transition(name string, m TransitionMutator) {
575 atm := &androidTransitionMutator{
Colin Crossb63d7b32023-12-07 16:54:51 -0800576 finalPhase: x.finalPhase,
577 mutator: m,
Colin Crossd67425d2024-04-15 15:17:05 -0700578 name: name,
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200579 }
580 mutator := &mutator{
581 name: name,
582 transitionMutator: atm}
583 x.mutators = append(x.mutators, mutator)
584}
585
Liz Kammer356f7d42021-01-26 09:18:53 -0500586func (x *registerMutatorsContext) mutatorName(name string) string {
Liz Kammer356f7d42021-01-26 09:18:53 -0500587 return name
588}
589
Colin Cross25de6c32019-06-06 14:29:25 -0700590func (x *registerMutatorsContext) TopDown(name string, m TopDownMutator) MutatorHandle {
Colin Cross798bfce2016-10-12 14:28:16 -0700591 f := func(ctx blueprint.TopDownMutatorContext) {
Colin Cross635c3b02016-05-18 15:37:25 -0700592 if a, ok := ctx.Module().(Module); ok {
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400593 moduleContext := a.base().baseModuleContextFactory(ctx)
Colin Cross984223f2024-02-01 17:10:23 -0800594 actx := topDownMutatorContextPool.Get().(*topDownMutatorContext)
595 defer topDownMutatorContextPool.Put(actx)
596 *actx = topDownMutatorContext{
Colin Crossdc35e212019-06-06 16:13:11 -0700597 bp: ctx,
Chris Parsons5a34ffb2021-07-21 14:34:58 -0400598 baseModuleContext: moduleContext,
Colin Cross6362e272015-10-29 15:25:03 -0700599 }
Colin Cross798bfce2016-10-12 14:28:16 -0700600 m(actx)
Colin Cross6362e272015-10-29 15:25:03 -0700601 }
Colin Cross798bfce2016-10-12 14:28:16 -0700602 }
Liz Kammer356f7d42021-01-26 09:18:53 -0500603 mutator := &mutator{name: x.mutatorName(name), topDownMutator: f}
Colin Cross795c3772017-03-16 16:50:10 -0700604 x.mutators = append(x.mutators, mutator)
Colin Cross798bfce2016-10-12 14:28:16 -0700605 return mutator
606}
607
Paul Duffin1d2d42f2021-03-06 20:08:12 +0000608func (mutator *mutator) componentName() string {
609 return mutator.name
610}
611
612func (mutator *mutator) register(ctx *Context) {
613 blueprintCtx := ctx.Context
614 var handle blueprint.MutatorHandle
615 if mutator.bottomUpMutator != nil {
616 handle = blueprintCtx.RegisterBottomUpMutator(mutator.name, mutator.bottomUpMutator)
617 } else if mutator.topDownMutator != nil {
618 handle = blueprintCtx.RegisterTopDownMutator(mutator.name, mutator.topDownMutator)
Lukacs T. Berki6c716762022-06-13 20:50:39 +0200619 } else if mutator.transitionMutator != nil {
620 blueprintCtx.RegisterTransitionMutator(mutator.name, mutator.transitionMutator)
Paul Duffin1d2d42f2021-03-06 20:08:12 +0000621 }
622 if mutator.parallel {
623 handle.Parallel()
624 }
625}
626
Colin Cross798bfce2016-10-12 14:28:16 -0700627type MutatorHandle interface {
628 Parallel() MutatorHandle
629}
630
631func (mutator *mutator) Parallel() MutatorHandle {
632 mutator.parallel = true
633 return mutator
Colin Cross6362e272015-10-29 15:25:03 -0700634}
Colin Cross1e676be2016-10-12 14:38:15 -0700635
Paul Duffin44f1d842020-06-26 20:17:02 +0100636func RegisterComponentsMutator(ctx RegisterMutatorsContext) {
637 ctx.BottomUp("component-deps", componentDepsMutator).Parallel()
638}
639
640// A special mutator that runs just prior to the deps mutator to allow the dependencies
641// on component modules to be added so that they can depend directly on a prebuilt
642// module.
643func componentDepsMutator(ctx BottomUpMutatorContext) {
Cole Fausta963b942024-04-11 17:43:00 -0700644 ctx.Module().ComponentDepsMutator(ctx)
Paul Duffin44f1d842020-06-26 20:17:02 +0100645}
646
Colin Cross1e676be2016-10-12 14:38:15 -0700647func depsMutator(ctx BottomUpMutatorContext) {
Cole Fausta963b942024-04-11 17:43:00 -0700648 if m := ctx.Module(); m.Enabled(ctx) {
Ronald Braunstein73b08ff2023-12-19 10:24:47 -0800649 m.base().baseDepsMutator(ctx)
Colin Cross1e676be2016-10-12 14:38:15 -0700650 m.DepsMutator(ctx)
651 }
652}
Colin Crossd11fcda2017-10-23 17:59:01 -0700653
Liz Kammer356f7d42021-01-26 09:18:53 -0500654func registerDepsMutator(ctx RegisterMutatorsContext) {
655 ctx.BottomUp("deps", depsMutator).Parallel()
656}
657
Colin Crossdc35e212019-06-06 16:13:11 -0700658// android.topDownMutatorContext either has to embed blueprint.TopDownMutatorContext, in which case every method that
659// has an overridden version in android.BaseModuleContext has to be manually forwarded to BaseModuleContext to avoid
660// ambiguous method errors, or it has to store a blueprint.TopDownMutatorContext non-embedded, in which case every
661// non-overridden method has to be forwarded. There are fewer non-overridden methods, so use the latter. The following
662// methods forward to the identical blueprint versions for topDownMutatorContext and bottomUpMutatorContext.
663
Colin Crosscb55e082019-07-01 15:32:31 -0700664func (t *topDownMutatorContext) MutatorName() string {
665 return t.bp.MutatorName()
666}
667
Colin Crossdc35e212019-06-06 16:13:11 -0700668func (t *topDownMutatorContext) Rename(name string) {
669 t.bp.Rename(name)
Colin Cross9a362232019-07-01 15:32:45 -0700670 t.Module().base().commonProperties.DebugName = name
Colin Crossdc35e212019-06-06 16:13:11 -0700671}
672
Liz Kammerf31c9002022-04-26 09:08:55 -0400673func (t *topDownMutatorContext) createModule(factory blueprint.ModuleFactory, name string, props ...interface{}) blueprint.Module {
674 return t.bp.CreateModule(factory, name, props...)
675}
676
Colin Crosse003c4a2019-09-25 12:58:36 -0700677func (t *topDownMutatorContext) CreateModule(factory ModuleFactory, props ...interface{}) Module {
Liz Kammerf31c9002022-04-26 09:08:55 -0400678 return createModule(t, factory, "_topDownMutatorModule", props...)
Colin Crossdc35e212019-06-06 16:13:11 -0700679}
680
Liz Kammera060c452021-03-24 10:14:47 -0400681func (t *topDownMutatorContext) createModuleWithoutInheritance(factory ModuleFactory, props ...interface{}) Module {
Liz Kammerf31c9002022-04-26 09:08:55 -0400682 module := t.bp.CreateModule(ModuleFactoryAdaptor(factory), "", props...).(Module)
Liz Kammera060c452021-03-24 10:14:47 -0400683 return module
684}
685
Colin Crosscb55e082019-07-01 15:32:31 -0700686func (b *bottomUpMutatorContext) MutatorName() string {
687 return b.bp.MutatorName()
688}
689
Colin Crossdc35e212019-06-06 16:13:11 -0700690func (b *bottomUpMutatorContext) Rename(name string) {
691 b.bp.Rename(name)
Colin Cross9a362232019-07-01 15:32:45 -0700692 b.Module().base().commonProperties.DebugName = name
Colin Crossdc35e212019-06-06 16:13:11 -0700693}
694
Colin Crossda279cf2024-09-17 14:25:45 -0700695func (b *bottomUpMutatorContext) createModule(factory blueprint.ModuleFactory, name string, props ...interface{}) blueprint.Module {
696 return b.bp.CreateModule(factory, name, props...)
697}
698
699func (b *bottomUpMutatorContext) CreateModule(factory ModuleFactory, props ...interface{}) Module {
700 return createModule(b, factory, "_bottomUpMutatorModule", props...)
701}
702
Colin Cross4f1dcb02020-09-16 18:45:04 -0700703func (b *bottomUpMutatorContext) AddDependency(module blueprint.Module, tag blueprint.DependencyTag, name ...string) []blueprint.Module {
Liz Kammerc13f7852023-05-17 13:01:48 -0400704 if b.baseModuleContext.checkedMissingDeps() {
705 panic("Adding deps not allowed after checking for missing deps")
706 }
Colin Cross4f1dcb02020-09-16 18:45:04 -0700707 return b.bp.AddDependency(module, tag, name...)
Colin Crossdc35e212019-06-06 16:13:11 -0700708}
709
710func (b *bottomUpMutatorContext) AddReverseDependency(module blueprint.Module, tag blueprint.DependencyTag, name string) {
Liz Kammerc13f7852023-05-17 13:01:48 -0400711 if b.baseModuleContext.checkedMissingDeps() {
712 panic("Adding deps not allowed after checking for missing deps")
713 }
Colin Crossdc35e212019-06-06 16:13:11 -0700714 b.bp.AddReverseDependency(module, tag, name)
715}
Cole Faustf4992e82024-09-30 15:16:18 -0700716
717func (b *bottomUpMutatorContext) AddReverseVariationDependency(variations []blueprint.Variation, tag blueprint.DependencyTag, name string) {
718 if b.baseModuleContext.checkedMissingDeps() {
719 panic("Adding deps not allowed after checking for missing deps")
720 }
721 b.bp.AddReverseVariationDependency(variations, tag, name)
722}
723
Colin Crossdc35e212019-06-06 16:13:11 -0700724func (b *bottomUpMutatorContext) AddVariationDependencies(variations []blueprint.Variation, tag blueprint.DependencyTag,
Colin Cross4f1dcb02020-09-16 18:45:04 -0700725 names ...string) []blueprint.Module {
Liz Kammerc13f7852023-05-17 13:01:48 -0400726 if b.baseModuleContext.checkedMissingDeps() {
727 panic("Adding deps not allowed after checking for missing deps")
728 }
Colin Cross4f1dcb02020-09-16 18:45:04 -0700729 return b.bp.AddVariationDependencies(variations, tag, names...)
Colin Crossdc35e212019-06-06 16:13:11 -0700730}
731
732func (b *bottomUpMutatorContext) AddFarVariationDependencies(variations []blueprint.Variation,
Colin Cross4f1dcb02020-09-16 18:45:04 -0700733 tag blueprint.DependencyTag, 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 Crossdc35e212019-06-06 16:13:11 -0700737
Colin Cross4f1dcb02020-09-16 18:45:04 -0700738 return b.bp.AddFarVariationDependencies(variations, tag, names...)
Colin Crossdc35e212019-06-06 16:13:11 -0700739}
740
Colin Crossdc35e212019-06-06 16:13:11 -0700741func (b *bottomUpMutatorContext) ReplaceDependencies(name string) {
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 b.bp.ReplaceDependencies(name)
746}
Jaewoong Jung9f88ce22019-11-15 10:57:34 -0800747
Paul Duffin80342d72020-06-26 22:08:43 +0100748func (b *bottomUpMutatorContext) ReplaceDependenciesIf(name string, predicate blueprint.ReplaceDependencyPredicate) {
Liz Kammerc13f7852023-05-17 13:01:48 -0400749 if b.baseModuleContext.checkedMissingDeps() {
750 panic("Adding deps not allowed after checking for missing deps")
751 }
Paul Duffin80342d72020-06-26 22:08:43 +0100752 b.bp.ReplaceDependenciesIf(name, predicate)
753}