blob: b8b99682011359ed28efd5bff98db4651660d64c [file] [log] [blame]
Colin Cross5049f022015-03-18 13:28:46 -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
Alex Humesky29e3bbe2020-11-20 21:30:13 -050015// A genrule module takes a list of source files ("srcs" property), an optional
16// list of tools ("tools" property), and a command line ("cmd" property), to
17// generate output files ("out" property).
18
Colin Cross5049f022015-03-18 13:28:46 -070019package genrule
20
21import (
Colin Cross6f080df2016-11-04 15:32:58 -070022 "fmt"
Colin Crossa4ad2b02019-03-18 22:15:32 -070023 "io"
Colin Cross1a527682019-09-23 15:55:30 -070024 "strconv"
Colin Cross6f080df2016-11-04 15:32:58 -070025 "strings"
Dan Willemsen3f4539b2016-09-28 16:19:10 -070026
Colin Cross70b40592015-03-23 12:57:34 -070027 "github.com/google/blueprint"
Dan Willemsen8eded0a2017-09-13 16:07:44 -070028 "github.com/google/blueprint/bootstrap"
Nan Zhangea568a42017-11-08 21:20:04 -080029 "github.com/google/blueprint/proptools"
Colin Cross5049f022015-03-18 13:28:46 -070030
Colin Cross635c3b02016-05-18 15:37:25 -070031 "android/soong/android"
Colin Cross5049f022015-03-18 13:28:46 -070032)
33
Colin Cross463a90e2015-06-17 14:20:06 -070034func init() {
Colin Crosse9fe2942020-11-10 18:12:15 -080035 RegisterGenruleBuildComponents(android.InitRegistrationContext)
Martin Stjernholm710ec3a2020-01-16 15:12:04 +000036}
Jaewoong Jung98716bd2018-12-10 08:13:18 -080037
Paul Duffin672cb9f2021-03-03 02:30:37 +000038// Test fixture preparer that will register most genrule build components.
39//
40// Singletons and mutators should only be added here if they are needed for a majority of genrule
41// module types, otherwise they should be added under a separate preparer to allow them to be
42// selected only when needed to reduce test execution time.
43//
44// Module types do not have much of an overhead unless they are used so this should include as many
45// module types as possible. The exceptions are those module types that require mutators and/or
46// singletons in order to function in which case they should be kept together in a separate
47// preparer.
48var PrepareForTestWithGenRuleBuildComponents = android.GroupFixturePreparers(
49 android.FixtureRegisterWithContext(RegisterGenruleBuildComponents),
50)
51
52// Prepare a fixture to use all genrule module types, mutators and singletons fully.
53//
54// This should only be used by tests that want to run with as much of the build enabled as possible.
55var PrepareForIntegrationTestWithGenrule = android.GroupFixturePreparers(
56 PrepareForTestWithGenRuleBuildComponents,
57)
58
Colin Crosse9fe2942020-11-10 18:12:15 -080059func RegisterGenruleBuildComponents(ctx android.RegistrationContext) {
Martin Stjernholm710ec3a2020-01-16 15:12:04 +000060 ctx.RegisterModuleType("genrule_defaults", defaultsFactory)
61
62 ctx.RegisterModuleType("gensrcs", GenSrcsFactory)
63 ctx.RegisterModuleType("genrule", GenRuleFactory)
64
65 ctx.FinalDepsMutators(func(ctx android.RegisterMutatorsContext) {
66 ctx.BottomUp("genrule_tool_deps", toolDepsMutator).Parallel()
67 })
Liz Kammer356f7d42021-01-26 09:18:53 -050068}
69
Colin Cross5049f022015-03-18 13:28:46 -070070var (
Colin Cross635c3b02016-05-18 15:37:25 -070071 pctx = android.NewPackageContext("android/soong/genrule")
Colin Cross1a527682019-09-23 15:55:30 -070072
Alex Humesky29e3bbe2020-11-20 21:30:13 -050073 // Used by gensrcs when there is more than 1 shard to merge the outputs
74 // of each shard into a zip file.
Colin Cross1a527682019-09-23 15:55:30 -070075 gensrcsMerge = pctx.AndroidStaticRule("gensrcsMerge", blueprint.RuleParams{
76 Command: "${soongZip} -o ${tmpZip} @${tmpZip}.rsp && ${zipSync} -d ${genDir} ${tmpZip}",
77 CommandDeps: []string{"${soongZip}", "${zipSync}"},
78 Rspfile: "${tmpZip}.rsp",
79 RspfileContent: "${zipArgs}",
80 }, "tmpZip", "genDir", "zipArgs")
Colin Cross5049f022015-03-18 13:28:46 -070081)
82
Jeff Gastonefc1b412017-03-29 17:29:06 -070083func init() {
Dan Willemsenddf504c2019-08-09 16:21:29 -070084 pctx.Import("android/soong/android")
Colin Cross1a527682019-09-23 15:55:30 -070085
86 pctx.HostBinToolVariable("soongZip", "soong_zip")
87 pctx.HostBinToolVariable("zipSync", "zipsync")
Jeff Gastonefc1b412017-03-29 17:29:06 -070088}
89
Colin Cross5049f022015-03-18 13:28:46 -070090type SourceFileGenerator interface {
Colin Cross635c3b02016-05-18 15:37:25 -070091 GeneratedSourceFiles() android.Paths
Colin Cross5ed99c62016-11-22 12:55:55 -080092 GeneratedHeaderDirs() android.Paths
Dan Willemsen9da9d492018-02-21 18:28:18 -080093 GeneratedDeps() android.Paths
Colin Cross5049f022015-03-18 13:28:46 -070094}
95
Colin Crossfe17f6f2019-03-28 19:30:56 -070096// Alias for android.HostToolProvider
97// Deprecated: use android.HostToolProvider instead.
Colin Crossd350ecd2015-04-28 13:25:36 -070098type HostToolProvider interface {
Colin Crossfe17f6f2019-03-28 19:30:56 -070099 android.HostToolProvider
Colin Crossd350ecd2015-04-28 13:25:36 -0700100}
Colin Cross5049f022015-03-18 13:28:46 -0700101
Dan Willemsend6ba0d52017-09-13 15:46:47 -0700102type hostToolDependencyTag struct {
103 blueprint.BaseDependencyTag
Colin Cross65cb3142021-12-10 23:05:02 +0000104 android.LicenseAnnotationToolchainDependencyTag
Colin Cross08f15ab2018-10-04 23:29:14 -0700105 label string
Dan Willemsend6ba0d52017-09-13 15:46:47 -0700106}
Martin Stjernholmdbd814d2022-01-12 23:18:30 +0000107
108func (t hostToolDependencyTag) AllowDisabledModuleDependency(target android.Module) bool {
109 // Allow depending on a disabled module if it's replaced by a prebuilt
110 // counterpart. We get the prebuilt through android.PrebuiltGetPreferred in
111 // GenerateAndroidBuildActions.
112 return target.IsReplacedByPrebuilt()
113}
114
115var _ android.AllowDisabledModuleDependency = (*hostToolDependencyTag)(nil)
116
Colin Cross7d5136f2015-05-11 13:39:40 -0700117type generatorProperties struct {
Spandan Das93e95992021-07-29 18:26:39 +0000118 // The command to run on one or more input files. Cmd supports substitution of a few variables.
Jeff Gastonefc1b412017-03-29 17:29:06 -0700119 //
120 // Available variables for substitution:
121 //
Spandan Das93e95992021-07-29 18:26:39 +0000122 // $(location): the path to the first entry in tools or tool_files.
123 // $(location <label>): the path to the tool, tool_file, input or output with name <label>. Use $(location) if <label> refers to a rule that outputs exactly one file.
124 // $(locations <label>): the paths to the tools, tool_files, inputs or outputs with name <label>. Use $(locations) if <label> refers to a rule that outputs two or more files.
125 // $(in): one or more input files.
126 // $(out): a single output file.
Spandan Das93e95992021-07-29 18:26:39 +0000127 // $(genDir): the sandbox directory for this tool; contains $(out).
Colin Cross2296f5b2017-10-17 21:38:14 -0700128 // $$: a literal $
Aleks Todorov1eb06c42024-06-03 15:23:56 +0100129 Cmd proptools.Configurable[string] `android:"replace_instead_of_append"`
Colin Cross7d5136f2015-05-11 13:39:40 -0700130
Colin Cross6f080df2016-11-04 15:32:58 -0700131 // name of the modules (if any) that produces the host executable. Leave empty for
Colin Cross7d5136f2015-05-11 13:39:40 -0700132 // prebuilts or scripts that do not need a module to build them.
Colin Cross6f080df2016-11-04 15:32:58 -0700133 Tools []string
Dan Willemsenf7f3d692016-04-20 14:54:32 -0700134
Sam Delmericof8775632023-08-14 23:45:41 +0000135 // Local files that are used by the tool
Colin Cross27b922f2019-03-04 22:35:41 -0800136 Tool_files []string `android:"path"`
Colin Cross5ed99c62016-11-22 12:55:55 -0800137
138 // List of directories to export generated headers from
139 Export_include_dirs []string
Colin Cross708c4242017-01-13 18:05:49 -0800140
141 // list of input files
Inseob Kim2f730622024-07-23 14:03:40 +0900142 Srcs proptools.Configurable[[]string] `android:"path,arch_variant"`
143 ResolvedSrcs []string `blueprint:"mutated"`
Dan Willemseneefa0262018-11-17 14:01:18 -0800144
145 // input files to exclude
Colin Cross27b922f2019-03-04 22:35:41 -0800146 Exclude_srcs []string `android:"path,arch_variant"`
Justin Yun4da4ccc2023-07-06 10:56:29 +0900147
148 // Enable restat to update the output only if the output is changed
149 Write_if_changed *bool
Cole Faust78f3c3a2024-08-15 17:19:34 -0700150
151 // When set to true, an additional $(build_number_file) label will be available
152 // to use in the cmd. This will be the location of a text file containing the
153 // build number. The dependency on this file will be "order-only", meaning that
154 // the genrule will not rerun when only this file changes, to avoid rerunning
155 // the genrule every build, because the build number changes every build.
156 // This also means that you should not attempt to consume the build number from
157 // the result of this genrule in another build rule. If you do, the build number
158 // in the second build rule will be stale when the second build rule rebuilds
159 // but this genrule does not. Only certain allowlisted modules are allowed to
160 // use this property, usages of the build number should be kept to the absolute
161 // minimum. Particularly no modules on the system image may include the build
162 // number. Prefer using libbuildversion via the use_version_lib property on
163 // cc modules.
164 Uses_order_only_build_number_file *bool
Chris Parsonsf3c96ef2020-09-29 02:23:17 -0400165}
Alex Humesky29e3bbe2020-11-20 21:30:13 -0500166
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700167type Module struct {
Colin Cross635c3b02016-05-18 15:37:25 -0700168 android.ModuleBase
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800169 android.DefaultableModuleBase
Jiyong Parkfc752ca2019-06-12 13:27:29 +0900170 android.ApexModuleBase
Colin Crossd350ecd2015-04-28 13:25:36 -0700171
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700172 // For other packages to make their own genrules with extra
173 // properties
174 Extra interface{}
Colin Crossf3bfd022021-09-27 15:15:06 -0700175
176 // CmdModifier can be set by wrappers around genrule to modify the command, for example to
177 // prefix environment variables to it.
178 CmdModifier func(ctx android.ModuleContext, cmd string) string
179
Colin Cross7228ecd2019-11-18 16:00:16 -0800180 android.ImageInterface
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700181
Colin Cross7d5136f2015-05-11 13:39:40 -0700182 properties generatorProperties
Colin Crossd350ecd2015-04-28 13:25:36 -0700183
Alex Humesky29e3bbe2020-11-20 21:30:13 -0500184 // For the different tasks that genrule and gensrc generate. genrule will
185 // generate 1 task, and gensrc will generate 1 or more tasks based on the
186 // number of shards the input files are sharded into.
Jeff Gaston437d23c2017-11-08 12:38:00 -0800187 taskGenerator taskFunc
Colin Crossd350ecd2015-04-28 13:25:36 -0700188
Colin Cross1a527682019-09-23 15:55:30 -0700189 rule blueprint.Rule
190 rawCommands []string
Colin Crossd350ecd2015-04-28 13:25:36 -0700191
Colin Cross5ed99c62016-11-22 12:55:55 -0800192 exportedIncludeDirs android.Paths
Dan Willemsenb40aab62016-04-20 14:21:14 -0700193
Colin Cross635c3b02016-05-18 15:37:25 -0700194 outputFiles android.Paths
Dan Willemsen9da9d492018-02-21 18:28:18 -0800195 outputDeps android.Paths
Colin Crossa4ad2b02019-03-18 22:15:32 -0700196
197 subName string
Colin Cross1a527682019-09-23 15:55:30 -0700198 subDir string
Colin Crossd350ecd2015-04-28 13:25:36 -0700199}
200
Colin Cross1a527682019-09-23 15:55:30 -0700201type taskFunc func(ctx android.ModuleContext, rawCommand string, srcFiles android.Paths) []generateTask
Colin Crossd350ecd2015-04-28 13:25:36 -0700202
203type generateTask struct {
Liz Kammer81fec182023-06-09 13:33:45 -0400204 in android.Paths
205 out android.WritablePaths
Liz Kammer81fec182023-06-09 13:33:45 -0400206 copyTo android.WritablePaths // For gensrcs to set on gensrcsMerge rule.
207 genDir android.WritablePath
Liz Kammer81fec182023-06-09 13:33:45 -0400208 extraInputs map[string][]string
Colin Cross3ea4eb82020-11-24 13:07:27 -0800209
Alex Humesky29e3bbe2020-11-20 21:30:13 -0500210 cmd string
211 // For gensrsc sharding.
Colin Cross3ea4eb82020-11-24 13:07:27 -0800212 shard int
213 shards int
Colin Crossd350ecd2015-04-28 13:25:36 -0700214}
215
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700216func (g *Module) GeneratedSourceFiles() android.Paths {
Colin Crossd350ecd2015-04-28 13:25:36 -0700217 return g.outputFiles
218}
219
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700220func (g *Module) Srcs() android.Paths {
Nan Zhange42777a2018-03-27 16:19:42 -0700221 return append(android.Paths{}, g.outputFiles...)
Colin Cross068e0fe2016-12-13 15:23:47 -0800222}
223
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700224func (g *Module) GeneratedHeaderDirs() android.Paths {
Colin Cross5ed99c62016-11-22 12:55:55 -0800225 return g.exportedIncludeDirs
Dan Willemsenb40aab62016-04-20 14:21:14 -0700226}
227
Dan Willemsen9da9d492018-02-21 18:28:18 -0800228func (g *Module) GeneratedDeps() android.Paths {
229 return g.outputDeps
230}
231
Jooyung Han8c7e3ed2021-06-28 17:35:58 +0900232var _ android.SourceFileProducer = (*Module)(nil)
Jooyung Han8c7e3ed2021-06-28 17:35:58 +0900233
Martin Stjernholm710ec3a2020-01-16 15:12:04 +0000234func toolDepsMutator(ctx android.BottomUpMutatorContext) {
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700235 if g, ok := ctx.Module().(*Module); ok {
Colin Cross08f15ab2018-10-04 23:29:14 -0700236 for _, tool := range g.properties.Tools {
237 tag := hostToolDependencyTag{label: tool}
238 if m := android.SrcIsModule(tool); m != "" {
239 tool = m
240 }
Colin Cross0f7d2ef2019-10-16 11:03:10 -0700241 ctx.AddFarVariationDependencies(ctx.Config().BuildOSTarget.Variations(), tag, tool)
Colin Cross6362e272015-10-29 15:25:03 -0700242 }
Colin Crossd350ecd2015-04-28 13:25:36 -0700243 }
Colin Crossd350ecd2015-04-28 13:25:36 -0700244}
245
Cole Faust78f3c3a2024-08-15 17:19:34 -0700246// This allowlist should be kept to the bare minimum, it's
247// intended for things that existed before the build number
248// was tightly controlled. Prefer using libbuildversion
249// via the use_version_lib property of cc modules.
250var genrule_build_number_allowlist = map[string]bool{
251 "build/soong/tests:gen": true,
252 "tools/tradefederation/core:tradefed_zip": true,
253}
254
Chris Parsonsf874e462022-05-10 13:50:12 -0400255// generateCommonBuildActions contains build action generation logic
256// common to both the mixed build case and the legacy case of genrule processing.
257// To fully support genrule in mixed builds, the contents of this function should
258// approach zero; there should be no genrule action registration done directly
259// by Soong logic in the mixed-build case.
260func (g *Module) generateCommonBuildActions(ctx android.ModuleContext) {
Colin Crossa4ad2b02019-03-18 22:15:32 -0700261 g.subName = ctx.ModuleSubDir()
262
Colin Cross5ed99c62016-11-22 12:55:55 -0800263 if len(g.properties.Export_include_dirs) > 0 {
264 for _, dir := range g.properties.Export_include_dirs {
265 g.exportedIncludeDirs = append(g.exportedIncludeDirs,
Colin Cross1a527682019-09-23 15:55:30 -0700266 android.PathForModuleGen(ctx, g.subDir, ctx.ModuleDir(), dir))
Liz Kammerd38c87c2023-07-17 09:58:50 -0400267 // Also export without ModuleDir for consistency with Export_include_dirs not being set
268 g.exportedIncludeDirs = append(g.exportedIncludeDirs,
269 android.PathForModuleGen(ctx, g.subDir, dir))
Colin Cross5ed99c62016-11-22 12:55:55 -0800270 }
271 } else {
Colin Cross1a527682019-09-23 15:55:30 -0700272 g.exportedIncludeDirs = append(g.exportedIncludeDirs, android.PathForModuleGen(ctx, g.subDir))
Colin Cross5ed99c62016-11-22 12:55:55 -0800273 }
Dan Willemsen3f4539b2016-09-28 16:19:10 -0700274
Colin Crossd11cf622021-03-23 22:30:35 -0700275 locationLabels := map[string]location{}
Colin Cross08f15ab2018-10-04 23:29:14 -0700276 firstLabel := ""
277
Colin Crossd11cf622021-03-23 22:30:35 -0700278 addLocationLabel := func(label string, loc location) {
Colin Cross08f15ab2018-10-04 23:29:14 -0700279 if firstLabel == "" {
280 firstLabel = label
281 }
282 if _, exists := locationLabels[label]; !exists {
Colin Crossd11cf622021-03-23 22:30:35 -0700283 locationLabels[label] = loc
Colin Cross08f15ab2018-10-04 23:29:14 -0700284 } else {
Anton Hansson7cd41e52021-10-08 16:13:10 +0100285 ctx.ModuleErrorf("multiple locations for label %q: %q and %q (do you have duplicate srcs entries?)",
Colin Crossd11cf622021-03-23 22:30:35 -0700286 label, locationLabels[label], loc)
Colin Cross08f15ab2018-10-04 23:29:14 -0700287 }
288 }
Dan Willemsen3f4539b2016-09-28 16:19:10 -0700289
Colin Crossba9e4032020-11-24 16:32:22 -0800290 var tools android.Paths
291 var packagedTools []android.PackagingSpec
Colin Cross6f080df2016-11-04 15:32:58 -0700292 if len(g.properties.Tools) > 0 {
Colin Crossba71a3f2019-03-18 12:12:48 -0700293 seenTools := make(map[string]bool)
294
Colin Cross35143d02017-11-16 00:11:20 -0800295 ctx.VisitDirectDepsBlueprint(func(module blueprint.Module) {
Colin Cross08f15ab2018-10-04 23:29:14 -0700296 switch tag := ctx.OtherModuleDependencyTag(module).(type) {
297 case hostToolDependencyTag:
Dan Willemsend6ba0d52017-09-13 15:46:47 -0700298 tool := ctx.OtherModuleName(module)
Martin Stjernholmdbd814d2022-01-12 23:18:30 +0000299 if m, ok := module.(android.Module); ok {
300 // Necessary to retrieve any prebuilt replacement for the tool, since
301 // toolDepsMutator runs too late for the prebuilt mutators to have
302 // replaced the dependency.
303 module = android.PrebuiltGetPreferred(ctx, m)
304 }
Dan Willemsend6ba0d52017-09-13 15:46:47 -0700305
Colin Crossba9e4032020-11-24 16:32:22 -0800306 switch t := module.(type) {
307 case android.HostToolProvider:
308 // A HostToolProvider provides the path to a tool, which will be copied
309 // into the sandbox.
Cole Fausta963b942024-04-11 17:43:00 -0700310 if !t.(android.Module).Enabled(ctx) {
Colin Cross6510f912017-11-29 00:27:14 -0800311 if ctx.Config().AllowMissingDependencies() {
Colin Cross35143d02017-11-16 00:11:20 -0800312 ctx.AddMissingDependencies([]string{tool})
313 } else {
314 ctx.ModuleErrorf("depends on disabled module %q", tool)
315 }
Colin Crossba9e4032020-11-24 16:32:22 -0800316 return
Colin Cross35143d02017-11-16 00:11:20 -0800317 }
Colin Crossba9e4032020-11-24 16:32:22 -0800318 path := t.HostToolPath()
319 if !path.Valid() {
320 ctx.ModuleErrorf("host tool %q missing output file", tool)
321 return
322 }
Yu Liubad1eef2024-08-21 22:37:35 +0000323 if specs := android.OtherModuleProviderOrDefault(
324 ctx, t, android.InstallFilesProvider).TransitivePackagingSpecs.ToList(); specs != nil {
Colin Crossba9e4032020-11-24 16:32:22 -0800325 // If the HostToolProvider has PackgingSpecs, which are definitions of the
326 // required relative locations of the tool and its dependencies, use those
327 // instead. They will be copied to those relative locations in the sbox
328 // sandbox.
Jiyong Park8fb0e972024-03-18 18:29:37 +0900329 // Care must be taken since TransitivePackagingSpec may return device-side
330 // paths via the required property. Filter them out.
331 for i, ps := range specs {
332 if ps.Partition() != "" {
333 if i == 0 {
334 panic("first PackagingSpec is assumed to be the host-side tool")
335 }
336 continue
337 }
338 packagedTools = append(packagedTools, ps)
339 }
Colin Crossba9e4032020-11-24 16:32:22 -0800340 // Assume that the first PackagingSpec of the module is the tool.
Colin Crossd11cf622021-03-23 22:30:35 -0700341 addLocationLabel(tag.label, packagedToolLocation{specs[0]})
Colin Crossba9e4032020-11-24 16:32:22 -0800342 } else {
343 tools = append(tools, path.Path())
Colin Crossd11cf622021-03-23 22:30:35 -0700344 addLocationLabel(tag.label, toolLocation{android.Paths{path.Path()}})
Colin Crossba9e4032020-11-24 16:32:22 -0800345 }
346 case bootstrap.GoBinaryTool:
347 // A GoBinaryTool provides the install path to a tool, which will be copied.
Colin Crossa44551f2021-10-25 15:36:21 -0700348 p := android.PathForGoBinary(ctx, t)
349 tools = append(tools, p)
350 addLocationLabel(tag.label, toolLocation{android.Paths{p}})
Colin Crossba9e4032020-11-24 16:32:22 -0800351 default:
Dan Willemsend6ba0d52017-09-13 15:46:47 -0700352 ctx.ModuleErrorf("%q is not a host tool provider", tool)
Colin Crossba9e4032020-11-24 16:32:22 -0800353 return
Dan Willemsen8eded0a2017-09-13 16:07:44 -0700354 }
355
Colin Crossba9e4032020-11-24 16:32:22 -0800356 seenTools[tag.label] = true
Colin Crossd350ecd2015-04-28 13:25:36 -0700357 }
Dan Willemsenf7f3d692016-04-20 14:54:32 -0700358 })
Colin Crossba71a3f2019-03-18 12:12:48 -0700359
360 // If AllowMissingDependencies is enabled, the build will not have stopped when
361 // AddFarVariationDependencies was called on a missing tool, which will result in nonsensical
Liz Kammer20ebfb42020-07-28 11:32:07 -0700362 // "cmd: unknown location label ..." errors later. Add a placeholder file to the local label.
363 // The command that uses this placeholder file will never be executed because the rule will be
364 // replaced with an android.Error rule reporting the missing dependencies.
Colin Crossba71a3f2019-03-18 12:12:48 -0700365 if ctx.Config().AllowMissingDependencies() {
366 for _, tool := range g.properties.Tools {
367 if !seenTools[tool] {
Colin Crossd11cf622021-03-23 22:30:35 -0700368 addLocationLabel(tool, errorLocation{"***missing tool " + tool + "***"})
Colin Crossba71a3f2019-03-18 12:12:48 -0700369 }
370 }
371 }
Dan Willemsenf7f3d692016-04-20 14:54:32 -0700372 }
Colin Crossd350ecd2015-04-28 13:25:36 -0700373
Dan Willemsend6ba0d52017-09-13 15:46:47 -0700374 if ctx.Failed() {
375 return
376 }
377
Colin Cross08f15ab2018-10-04 23:29:14 -0700378 for _, toolFile := range g.properties.Tool_files {
Colin Cross8a497952019-03-05 22:25:09 -0800379 paths := android.PathsForModuleSrc(ctx, []string{toolFile})
Colin Crossba9e4032020-11-24 16:32:22 -0800380 tools = append(tools, paths...)
Colin Crossd11cf622021-03-23 22:30:35 -0700381 addLocationLabel(toolFile, toolLocation{paths})
Colin Cross08f15ab2018-10-04 23:29:14 -0700382 }
383
Liz Kammer81fec182023-06-09 13:33:45 -0400384 addLabelsForInputs := func(propName string, include, exclude []string) android.Paths {
Liz Kammer81fec182023-06-09 13:33:45 -0400385 includeDirInPaths := ctx.DeviceConfig().BuildBrokenInputDir(g.Name())
386 var srcFiles android.Paths
387 for _, in := range include {
388 paths, missingDeps := android.PathsAndMissingDepsRelativeToModuleSourceDir(android.SourceInput{
389 Context: ctx, Paths: []string{in}, ExcludePaths: exclude, IncludeDirs: includeDirInPaths,
390 })
391 if len(missingDeps) > 0 {
392 if !ctx.Config().AllowMissingDependencies() {
393 panic(fmt.Errorf("should never get here, the missing dependencies %q should have been reported in DepsMutator",
394 missingDeps))
395 }
396
397 // If AllowMissingDependencies is enabled, the build will not have stopped when
398 // the dependency was added on a missing SourceFileProducer module, which will result in nonsensical
399 // "cmd: label ":..." has no files" errors later. Add a placeholder file to the local label.
400 // The command that uses this placeholder file will never be executed because the rule will be
401 // replaced with an android.Error rule reporting the missing dependencies.
402 ctx.AddMissingDependencies(missingDeps)
403 addLocationLabel(in, errorLocation{"***missing " + propName + " " + in + "***"})
404 } else {
405 srcFiles = append(srcFiles, paths...)
406 addLocationLabel(in, inputLocation{paths})
407 }
Colin Crossba71a3f2019-03-18 12:12:48 -0700408 }
Liz Kammer81fec182023-06-09 13:33:45 -0400409 return srcFiles
Colin Cross08f15ab2018-10-04 23:29:14 -0700410 }
Cole Faustbf1d92a2024-07-29 12:24:25 -0700411 g.properties.ResolvedSrcs = g.properties.Srcs.GetOrDefault(ctx, nil)
Inseob Kim2f730622024-07-23 14:03:40 +0900412 srcFiles := addLabelsForInputs("srcs", g.properties.ResolvedSrcs, g.properties.Exclude_srcs)
Colin Cross40213022023-12-13 15:19:49 -0800413 android.SetProvider(ctx, blueprint.SrcsFileProviderKey, blueprint.SrcsFileProviderData{SrcPaths: srcFiles.Strings()})
Colin Cross08f15ab2018-10-04 23:29:14 -0700414
Colin Cross1a527682019-09-23 15:55:30 -0700415 var copyFrom android.Paths
416 var outputFiles android.WritablePaths
417 var zipArgs strings.Builder
Colin Cross08f15ab2018-10-04 23:29:14 -0700418
Aleks Todorov1eb06c42024-06-03 15:23:56 +0100419 cmd := g.properties.Cmd.GetOrDefault(ctx, "")
Colin Crossf3bfd022021-09-27 15:15:06 -0700420 if g.CmdModifier != nil {
421 cmd = g.CmdModifier(ctx, cmd)
422 }
423
Liz Kammer796921d2023-07-11 08:21:41 -0400424 var extraInputs android.Paths
Alex Humesky29e3bbe2020-11-20 21:30:13 -0500425 // Generate tasks, either from genrule or gensrcs.
Liz Kammer81fec182023-06-09 13:33:45 -0400426 for i, task := range g.taskGenerator(ctx, cmd, srcFiles) {
Colin Cross3d680512020-11-13 16:23:53 -0800427 if len(task.out) == 0 {
428 ctx.ModuleErrorf("must have at least one output file")
429 return
Colin Cross85a2e892018-07-09 09:45:06 -0700430 }
431
Liz Kammer81fec182023-06-09 13:33:45 -0400432 // Only handle extra inputs once as these currently are the same across all tasks
433 if i == 0 {
434 for name, values := range task.extraInputs {
435 extraInputs = append(extraInputs, addLabelsForInputs(name, values, []string{})...)
436 }
437 }
438
Colin Crossf1a035e2020-11-16 17:32:30 -0800439 // Pick a unique path outside the task.genDir for the sbox manifest textproto,
440 // a unique rule name, and the user-visible description.
441 manifestName := "genrule.sbox.textproto"
442 desc := "generate"
443 name := "generator"
444 if task.shards > 0 {
445 manifestName = "genrule_" + strconv.Itoa(task.shard) + ".sbox.textproto"
446 desc += " " + strconv.Itoa(task.shard)
447 name += strconv.Itoa(task.shard)
448 } else if len(task.out) == 1 {
449 desc += " " + task.out[0].Base()
450 }
451
452 manifestPath := android.PathForModuleOut(ctx, manifestName)
453
454 // Use a RuleBuilder to create a rule that runs the command inside an sbox sandbox.
Yu Liu6a7940c2023-05-09 17:12:22 -0700455 rule := getSandboxedRuleBuilder(ctx, android.NewRuleBuilder(pctx, ctx).Sbox(task.genDir, manifestPath))
Justin Yun4da4ccc2023-07-06 10:56:29 +0900456 if Bool(g.properties.Write_if_changed) {
457 rule.Restat()
458 }
Colin Crossf1a035e2020-11-16 17:32:30 -0800459 cmd := rule.Command()
460
Colin Cross3d680512020-11-13 16:23:53 -0800461 for _, out := range task.out {
Colin Crossd11cf622021-03-23 22:30:35 -0700462 addLocationLabel(out.Rel(), outputLocation{out})
Colin Cross3d680512020-11-13 16:23:53 -0800463 }
464
Colin Cross3d680512020-11-13 16:23:53 -0800465 rawCommand, err := android.Expand(task.cmd, func(name string) (string, error) {
Colin Cross1a527682019-09-23 15:55:30 -0700466 // report the error directly without returning an error to android.Expand to catch multiple errors in a
467 // single run
Colin Cross3d680512020-11-13 16:23:53 -0800468 reportError := func(fmt string, args ...interface{}) (string, error) {
Colin Cross1a527682019-09-23 15:55:30 -0700469 ctx.PropertyErrorf("cmd", fmt, args...)
Colin Cross3d680512020-11-13 16:23:53 -0800470 return "SOONG_ERROR", nil
Colin Cross6f080df2016-11-04 15:32:58 -0700471 }
Colin Cross1a527682019-09-23 15:55:30 -0700472
Jihoon Kangc170af42022-08-20 05:26:38 +0000473 // Apply shell escape to each cases to prevent source file paths containing $ from being evaluated in shell
Colin Cross1a527682019-09-23 15:55:30 -0700474 switch name {
475 case "location":
476 if len(g.properties.Tools) == 0 && len(g.properties.Tool_files) == 0 {
477 return reportError("at least one `tools` or `tool_files` is required if $(location) is used")
Colin Cross6f080df2016-11-04 15:32:58 -0700478 }
Colin Crossd11cf622021-03-23 22:30:35 -0700479 loc := locationLabels[firstLabel]
480 paths := loc.Paths(cmd)
Colin Cross1a527682019-09-23 15:55:30 -0700481 if len(paths) == 0 {
482 return reportError("default label %q has no files", firstLabel)
483 } else if len(paths) > 1 {
484 return reportError("default label %q has multiple files, use $(locations %s) to reference it",
485 firstLabel, firstLabel)
Colin Cross08f15ab2018-10-04 23:29:14 -0700486 }
Jihoon Kangc170af42022-08-20 05:26:38 +0000487 return proptools.ShellEscape(paths[0]), nil
Colin Cross1a527682019-09-23 15:55:30 -0700488 case "in":
Jihoon Kangc170af42022-08-20 05:26:38 +0000489 return strings.Join(proptools.ShellEscapeList(cmd.PathsForInputs(srcFiles)), " "), nil
Colin Cross1a527682019-09-23 15:55:30 -0700490 case "out":
Colin Cross3d680512020-11-13 16:23:53 -0800491 var sandboxOuts []string
492 for _, out := range task.out {
Colin Crossf1a035e2020-11-16 17:32:30 -0800493 sandboxOuts = append(sandboxOuts, cmd.PathForOutput(out))
Colin Cross3d680512020-11-13 16:23:53 -0800494 }
Jihoon Kangc170af42022-08-20 05:26:38 +0000495 return strings.Join(proptools.ShellEscapeList(sandboxOuts), " "), nil
Colin Cross1a527682019-09-23 15:55:30 -0700496 case "genDir":
Jihoon Kangc170af42022-08-20 05:26:38 +0000497 return proptools.ShellEscape(cmd.PathForOutput(task.genDir)), nil
Cole Faust78f3c3a2024-08-15 17:19:34 -0700498 case "build_number_file":
499 if !proptools.Bool(g.properties.Uses_order_only_build_number_file) {
500 return reportError("to use the $(build_number_file) label, you must set uses_order_only_build_number_file: true")
501 }
502 return proptools.ShellEscape(cmd.PathForInput(ctx.Config().BuildNumberFile(ctx))), nil
Colin Cross1a527682019-09-23 15:55:30 -0700503 default:
504 if strings.HasPrefix(name, "location ") {
505 label := strings.TrimSpace(strings.TrimPrefix(name, "location "))
Colin Crossd11cf622021-03-23 22:30:35 -0700506 if loc, ok := locationLabels[label]; ok {
507 paths := loc.Paths(cmd)
Colin Cross1a527682019-09-23 15:55:30 -0700508 if len(paths) == 0 {
509 return reportError("label %q has no files", label)
510 } else if len(paths) > 1 {
511 return reportError("label %q has multiple files, use $(locations %s) to reference it",
512 label, label)
513 }
Jihoon Kangc170af42022-08-20 05:26:38 +0000514 return proptools.ShellEscape(paths[0]), nil
Colin Cross1a527682019-09-23 15:55:30 -0700515 } else {
Anton Hanssonbebf5262022-02-23 11:42:38 +0000516 return reportError("unknown location label %q is not in srcs, out, tools or tool_files.", label)
Colin Cross1a527682019-09-23 15:55:30 -0700517 }
518 } else if strings.HasPrefix(name, "locations ") {
519 label := strings.TrimSpace(strings.TrimPrefix(name, "locations "))
Colin Crossd11cf622021-03-23 22:30:35 -0700520 if loc, ok := locationLabels[label]; ok {
521 paths := loc.Paths(cmd)
Colin Cross1a527682019-09-23 15:55:30 -0700522 if len(paths) == 0 {
523 return reportError("label %q has no files", label)
524 }
Cole Faustce74a592023-12-07 14:58:45 -0800525 return strings.Join(proptools.ShellEscapeList(paths), " "), nil
Colin Cross1a527682019-09-23 15:55:30 -0700526 } else {
Anton Hanssonbebf5262022-02-23 11:42:38 +0000527 return reportError("unknown locations label %q is not in srcs, out, tools or tool_files.", label)
Colin Cross1a527682019-09-23 15:55:30 -0700528 }
529 } else {
530 return reportError("unknown variable '$(%s)'", name)
531 }
Colin Cross6f080df2016-11-04 15:32:58 -0700532 }
Colin Cross1a527682019-09-23 15:55:30 -0700533 })
534
535 if err != nil {
536 ctx.PropertyErrorf("cmd", "%s", err.Error())
537 return
Colin Cross6f080df2016-11-04 15:32:58 -0700538 }
Colin Cross6f080df2016-11-04 15:32:58 -0700539
Colin Cross1a527682019-09-23 15:55:30 -0700540 g.rawCommands = append(g.rawCommands, rawCommand)
Bill Peckhamc087be12020-02-13 15:55:10 -0800541
Colin Cross3d680512020-11-13 16:23:53 -0800542 cmd.Text(rawCommand)
Liz Kammer81fec182023-06-09 13:33:45 -0400543 cmd.Implicits(srcFiles) // need to be able to reference other srcs
544 cmd.Implicits(extraInputs)
Colin Cross3d680512020-11-13 16:23:53 -0800545 cmd.ImplicitOutputs(task.out)
546 cmd.Implicits(task.in)
Colin Crossba9e4032020-11-24 16:32:22 -0800547 cmd.ImplicitTools(tools)
Colin Crossba9e4032020-11-24 16:32:22 -0800548 cmd.ImplicitPackagedTools(packagedTools)
Cole Faust78f3c3a2024-08-15 17:19:34 -0700549 if proptools.Bool(g.properties.Uses_order_only_build_number_file) {
550 if _, ok := genrule_build_number_allowlist[ctx.ModuleDir()+":"+ctx.ModuleName()]; !ok {
551 ctx.ModuleErrorf("Only allowlisted modules may use uses_order_only_build_number_file: true")
552 }
553 cmd.OrderOnly(ctx.Config().BuildNumberFile(ctx))
554 }
Colin Cross3d680512020-11-13 16:23:53 -0800555
556 // Create the rule to run the genrule command inside sbox.
Colin Crossf1a035e2020-11-16 17:32:30 -0800557 rule.Build(name, desc)
Colin Cross1a527682019-09-23 15:55:30 -0700558
559 if len(task.copyTo) > 0 {
Colin Cross3d680512020-11-13 16:23:53 -0800560 // If copyTo is set, multiple shards need to be copied into a single directory.
561 // task.out contains the per-shard paths, and copyTo contains the corresponding
562 // final path. The files need to be copied into the final directory by a
563 // single rule so it can remove the directory before it starts to ensure no
564 // old files remain. zipsync already does this, so build up zipArgs that
565 // zip all the per-shard directories into a single zip.
Colin Cross1a527682019-09-23 15:55:30 -0700566 outputFiles = append(outputFiles, task.copyTo...)
567 copyFrom = append(copyFrom, task.out.Paths()...)
568 zipArgs.WriteString(" -C " + task.genDir.String())
569 zipArgs.WriteString(android.JoinWithPrefix(task.out.Strings(), " -f "))
570 } else {
571 outputFiles = append(outputFiles, task.out...)
572 }
Colin Cross6f080df2016-11-04 15:32:58 -0700573 }
574
Colin Cross1a527682019-09-23 15:55:30 -0700575 if len(copyFrom) > 0 {
Colin Cross3d680512020-11-13 16:23:53 -0800576 // Create a rule that zips all the per-shard directories into a single zip and then
577 // uses zipsync to unzip it into the final directory.
Colin Cross1a527682019-09-23 15:55:30 -0700578 ctx.Build(pctx, android.BuildParams{
Colin Crossf1885962020-11-20 15:28:30 -0800579 Rule: gensrcsMerge,
580 Implicits: copyFrom,
581 Outputs: outputFiles,
582 Description: "merge shards",
Colin Cross1a527682019-09-23 15:55:30 -0700583 Args: map[string]string{
584 "zipArgs": zipArgs.String(),
585 "tmpZip": android.PathForModuleGen(ctx, g.subDir+".zip").String(),
586 "genDir": android.PathForModuleGen(ctx, g.subDir).String(),
587 },
588 })
Colin Cross85a2e892018-07-09 09:45:06 -0700589 }
590
Colin Cross1a527682019-09-23 15:55:30 -0700591 g.outputFiles = outputFiles.Paths()
Chris Parsonsf874e462022-05-10 13:50:12 -0400592}
Jeff Gastonefc1b412017-03-29 17:29:06 -0700593
Chris Parsonsf874e462022-05-10 13:50:12 -0400594func (g *Module) GenerateAndroidBuildActions(ctx android.ModuleContext) {
595 g.generateCommonBuildActions(ctx)
596
597 // For <= 6 outputs, just embed those directly in the users. Right now, that covers >90% of
598 // the genrules on AOSP. That will make things simpler to look at the graph in the common
599 // case. For larger sets of outputs, inject a phony target in between to limit ninja file
600 // growth.
601 if len(g.outputFiles) <= 6 {
602 g.outputDeps = g.outputFiles
603 } else {
604 phonyFile := android.PathForModuleGen(ctx, "genrule-phony")
605 ctx.Build(pctx, android.BuildParams{
606 Rule: blueprint.Phony,
607 Output: phonyFile,
608 Inputs: g.outputFiles,
609 })
610 g.outputDeps = android.Paths{phonyFile}
Jeff Gaston02a684b2017-10-27 14:59:27 -0700611 }
mrziwang4514ef22024-06-07 13:31:48 -0700612
613 g.setOutputFiles(ctx)
614}
615
616func (g *Module) setOutputFiles(ctx android.ModuleContext) {
617 if len(g.outputFiles) == 0 {
618 return
619 }
620 ctx.SetOutputFiles(g.outputFiles, "")
621 // non-empty-string-tag should match one of the outputs
622 for _, files := range g.outputFiles {
623 ctx.SetOutputFiles(android.Paths{files}, files.Rel())
624 }
Chris Parsonsf874e462022-05-10 13:50:12 -0400625}
626
Brandon Lee5d45c6f2018-08-15 15:35:38 -0700627// Collect information for opening IDE project files in java/jdeps.go.
628func (g *Module) IDEInfo(dpInfo *android.IdeInfo) {
629 dpInfo.Srcs = append(dpInfo.Srcs, g.Srcs().Strings()...)
Inseob Kim2f730622024-07-23 14:03:40 +0900630 for _, src := range g.properties.ResolvedSrcs {
Brandon Lee5d45c6f2018-08-15 15:35:38 -0700631 if strings.HasPrefix(src, ":") {
632 src = strings.Trim(src, ":")
633 dpInfo.Deps = append(dpInfo.Deps, src)
634 }
635 }
636}
637
Colin Crossa4ad2b02019-03-18 22:15:32 -0700638func (g *Module) AndroidMk() android.AndroidMkData {
639 return android.AndroidMkData{
Anton Hansson72f18492020-10-30 16:34:45 +0000640 Class: "ETC",
Colin Crossa4ad2b02019-03-18 22:15:32 -0700641 OutputFile: android.OptionalPathForPath(g.outputFiles[0]),
642 SubName: g.subName,
643 Extra: []android.AndroidMkExtraFunc{
644 func(w io.Writer, outputFile android.Path) {
Anton Hansson72f18492020-10-30 16:34:45 +0000645 fmt.Fprintln(w, "LOCAL_UNINSTALLABLE_MODULE := true")
Colin Crossa4ad2b02019-03-18 22:15:32 -0700646 },
647 },
648 Custom: func(w io.Writer, name, prefix, moduleDir string, data android.AndroidMkData) {
649 android.WriteAndroidMkData(w, data)
650 if data.SubName != "" {
651 fmt.Fprintln(w, ".PHONY:", name)
652 fmt.Fprintln(w, name, ":", name+g.subName)
653 }
654 },
655 }
656}
657
Jiyong Park45bf82e2020-12-15 22:29:02 +0900658var _ android.ApexModule = (*Module)(nil)
659
660// Implements android.ApexModule
Dan Albertc8060532020-07-22 22:32:17 -0700661func (g *Module) ShouldSupportSdkVersion(ctx android.BaseModuleContext,
662 sdkVersion android.ApiLevel) error {
Jooyung Han749dc692020-04-15 11:03:39 +0900663 // Because generated outputs are checked by client modules(e.g. cc_library, ...)
664 // we can safely ignore the check here.
665 return nil
666}
667
Jeff Gaston437d23c2017-11-08 12:38:00 -0800668func generatorFactory(taskGenerator taskFunc, props ...interface{}) *Module {
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700669 module := &Module{
Jeff Gaston437d23c2017-11-08 12:38:00 -0800670 taskGenerator: taskGenerator,
Colin Crossd350ecd2015-04-28 13:25:36 -0700671 }
672
Colin Cross36242852017-06-23 15:06:31 -0700673 module.AddProperties(props...)
674 module.AddProperties(&module.properties)
Colin Crossd350ecd2015-04-28 13:25:36 -0700675
Colin Cross7228ecd2019-11-18 16:00:16 -0800676 module.ImageInterface = noopImageInterface{}
677
Colin Cross36242852017-06-23 15:06:31 -0700678 return module
Colin Crossd350ecd2015-04-28 13:25:36 -0700679}
680
Colin Cross7228ecd2019-11-18 16:00:16 -0800681type noopImageInterface struct{}
682
683func (x noopImageInterface) ImageMutatorBegin(android.BaseModuleContext) {}
Jihoon Kang47e91842024-06-19 00:51:16 +0000684func (x noopImageInterface) VendorVariantNeeded(android.BaseModuleContext) bool { return false }
685func (x noopImageInterface) ProductVariantNeeded(android.BaseModuleContext) bool { return false }
Colin Cross7228ecd2019-11-18 16:00:16 -0800686func (x noopImageInterface) CoreVariantNeeded(android.BaseModuleContext) bool { return false }
Yifan Hong1b3348d2020-01-21 15:53:22 -0800687func (x noopImageInterface) RamdiskVariantNeeded(android.BaseModuleContext) bool { return false }
Yifan Hong60e0cfb2020-10-21 15:17:56 -0700688func (x noopImageInterface) VendorRamdiskVariantNeeded(android.BaseModuleContext) bool { return false }
Inseob Kim08758f02021-04-08 21:13:22 +0900689func (x noopImageInterface) DebugRamdiskVariantNeeded(android.BaseModuleContext) bool { return false }
Colin Cross7228ecd2019-11-18 16:00:16 -0800690func (x noopImageInterface) RecoveryVariantNeeded(android.BaseModuleContext) bool { return false }
691func (x noopImageInterface) ExtraImageVariations(ctx android.BaseModuleContext) []string { return nil }
Jihoon Kang7583e832024-06-13 21:25:45 +0000692func (x noopImageInterface) SetImageVariation(ctx android.BaseModuleContext, variation string) {
Colin Cross7228ecd2019-11-18 16:00:16 -0800693}
694
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700695func NewGenSrcs() *Module {
Colin Crossd350ecd2015-04-28 13:25:36 -0700696 properties := &genSrcsProperties{}
697
Colin Crossf1885962020-11-20 15:28:30 -0800698 // finalSubDir is the name of the subdirectory that output files will be generated into.
699 // It is used so that per-shard directories can be placed alongside it an then finally
700 // merged into it.
701 const finalSubDir = "gensrcs"
702
Colin Cross1a527682019-09-23 15:55:30 -0700703 taskGenerator := func(ctx android.ModuleContext, rawCommand string, srcFiles android.Paths) []generateTask {
Colin Cross1a527682019-09-23 15:55:30 -0700704 shardSize := defaultShardSize
705 if s := properties.Shard_size; s != nil {
706 shardSize = int(*s)
707 }
Jeff Gaston437d23c2017-11-08 12:38:00 -0800708
Colin Crossf1885962020-11-20 15:28:30 -0800709 // gensrcs rules can easily hit command line limits by repeating the command for
710 // every input file. Shard the input files into groups.
Colin Cross1a527682019-09-23 15:55:30 -0700711 shards := android.ShardPaths(srcFiles, shardSize)
712 var generateTasks []generateTask
Colin Crossbaccf5b2018-02-21 14:07:48 -0800713
Colin Cross1a527682019-09-23 15:55:30 -0700714 for i, shard := range shards {
715 var commands []string
716 var outFiles android.WritablePaths
717 var copyTo android.WritablePaths
Colin Cross1a527682019-09-23 15:55:30 -0700718
Colin Crossf1885962020-11-20 15:28:30 -0800719 // When sharding is enabled (i.e. len(shards) > 1), the sbox rules for each
720 // shard will be write to their own directories and then be merged together
721 // into finalSubDir. If sharding is not enabled (i.e. len(shards) == 1),
722 // the sbox rule will write directly to finalSubDir.
723 genSubDir := finalSubDir
Colin Cross1a527682019-09-23 15:55:30 -0700724 if len(shards) > 1 {
Colin Crossf1885962020-11-20 15:28:30 -0800725 genSubDir = strconv.Itoa(i)
Jeff Gaston437d23c2017-11-08 12:38:00 -0800726 }
727
Colin Crossf1885962020-11-20 15:28:30 -0800728 genDir := android.PathForModuleGen(ctx, genSubDir)
Colin Crossf1a035e2020-11-16 17:32:30 -0800729 // TODO(ccross): this RuleBuilder is a hack to be able to call
730 // rule.Command().PathForOutput. Replace this with passing the rule into the
731 // generator.
Yu Liu6a7940c2023-05-09 17:12:22 -0700732 rule := getSandboxedRuleBuilder(ctx, android.NewRuleBuilder(pctx, ctx).Sbox(genDir, nil))
Jeff Gaston437d23c2017-11-08 12:38:00 -0800733
Colin Cross3ea4eb82020-11-24 13:07:27 -0800734 for _, in := range shard {
yangbill6d032dd2024-04-18 03:05:49 +0000735 outFile := android.GenPathWithExtAndTrimExt(ctx, finalSubDir, in, String(properties.Output_extension), String(properties.Trim_extension))
Colin Crossf1885962020-11-20 15:28:30 -0800736
737 // If sharding is enabled, then outFile is the path to the output file in
738 // the shard directory, and copyTo is the path to the output file in the
739 // final directory.
Colin Cross1a527682019-09-23 15:55:30 -0700740 if len(shards) > 1 {
yangbill6d032dd2024-04-18 03:05:49 +0000741 shardFile := android.GenPathWithExtAndTrimExt(ctx, genSubDir, in, String(properties.Output_extension), String(properties.Trim_extension))
Colin Cross1a527682019-09-23 15:55:30 -0700742 copyTo = append(copyTo, outFile)
743 outFile = shardFile
744 }
745
746 outFiles = append(outFiles, outFile)
Colin Cross1a527682019-09-23 15:55:30 -0700747
Colin Crossf1885962020-11-20 15:28:30 -0800748 // pre-expand the command line to replace $in and $out with references to
749 // a single input and output file.
Colin Cross1a527682019-09-23 15:55:30 -0700750 command, err := android.Expand(rawCommand, func(name string) (string, error) {
751 switch name {
752 case "in":
753 return in.String(), nil
754 case "out":
Colin Crossf1a035e2020-11-16 17:32:30 -0800755 return rule.Command().PathForOutput(outFile), nil
Colin Cross1a527682019-09-23 15:55:30 -0700756 default:
757 return "$(" + name + ")", nil
758 }
759 })
760 if err != nil {
761 ctx.PropertyErrorf("cmd", err.Error())
762 }
763
764 // escape the command in case for example it contains '#', an odd number of '"', etc
765 command = fmt.Sprintf("bash -c %v", proptools.ShellEscape(command))
766 commands = append(commands, command)
767 }
768 fullCommand := strings.Join(commands, " && ")
769
770 generateTasks = append(generateTasks, generateTask{
Cole Faust55492572024-01-25 18:00:33 -0800771 in: shard,
772 out: outFiles,
773 copyTo: copyTo,
774 genDir: genDir,
775 cmd: fullCommand,
776 shard: i,
777 shards: len(shards),
Liz Kammer81fec182023-06-09 13:33:45 -0400778 extraInputs: map[string][]string{
779 "data": properties.Data,
780 },
Colin Cross1a527682019-09-23 15:55:30 -0700781 })
Jeff Gaston437d23c2017-11-08 12:38:00 -0800782 }
Colin Cross1a527682019-09-23 15:55:30 -0700783
784 return generateTasks
Colin Crossd350ecd2015-04-28 13:25:36 -0700785 }
786
Colin Cross1a527682019-09-23 15:55:30 -0700787 g := generatorFactory(taskGenerator, properties)
Colin Crossf1885962020-11-20 15:28:30 -0800788 g.subDir = finalSubDir
Colin Cross1a527682019-09-23 15:55:30 -0700789 return g
Colin Crossd350ecd2015-04-28 13:25:36 -0700790}
791
Colin Cross54190b32017-10-09 15:34:10 -0700792func GenSrcsFactory() android.Module {
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700793 m := NewGenSrcs()
794 android.InitAndroidModule(m)
Colin Cross483b4c42024-05-09 13:08:02 -0700795 android.InitDefaultableModule(m)
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700796 return m
797}
798
Colin Crossd350ecd2015-04-28 13:25:36 -0700799type genSrcsProperties struct {
Colin Cross7d5136f2015-05-11 13:39:40 -0700800 // extension that will be substituted for each output file
Nan Zhanga5e7cb42017-11-09 22:42:32 -0800801 Output_extension *string
Colin Cross1a527682019-09-23 15:55:30 -0700802
803 // maximum number of files that will be passed on a single command line.
804 Shard_size *int64
Liz Kammer81fec182023-06-09 13:33:45 -0400805
806 // Additional files needed for build that are not tooling related.
807 Data []string `android:"path"`
yangbill6d032dd2024-04-18 03:05:49 +0000808
809 // Trim the matched extension for each input file, and it should start with ".".
810 Trim_extension *string
Colin Cross5049f022015-03-18 13:28:46 -0700811}
812
Evgenii Stepanovf47c90d2020-12-02 18:55:09 -0800813const defaultShardSize = 50
Colin Cross1a527682019-09-23 15:55:30 -0700814
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700815func NewGenRule() *Module {
Colin Crossd350ecd2015-04-28 13:25:36 -0700816 properties := &genRuleProperties{}
Colin Cross5049f022015-03-18 13:28:46 -0700817
Colin Cross1a527682019-09-23 15:55:30 -0700818 taskGenerator := func(ctx android.ModuleContext, rawCommand string, srcFiles android.Paths) []generateTask {
Dan Willemsen9c8681f2016-09-28 16:21:00 -0700819 outs := make(android.WritablePaths, len(properties.Out))
820 for i, out := range properties.Out {
Cole Faust55492572024-01-25 18:00:33 -0800821 outs[i] = android.PathForModuleGen(ctx, out)
Dan Willemsen9c8681f2016-09-28 16:21:00 -0700822 }
Colin Cross1a527682019-09-23 15:55:30 -0700823 return []generateTask{{
Cole Faust55492572024-01-25 18:00:33 -0800824 in: srcFiles,
825 out: outs,
826 genDir: android.PathForModuleGen(ctx),
827 cmd: rawCommand,
Colin Cross1a527682019-09-23 15:55:30 -0700828 }}
Colin Cross5049f022015-03-18 13:28:46 -0700829 }
Colin Crossd350ecd2015-04-28 13:25:36 -0700830
Jeff Gaston437d23c2017-11-08 12:38:00 -0800831 return generatorFactory(taskGenerator, properties)
Colin Cross5049f022015-03-18 13:28:46 -0700832}
833
Colin Cross54190b32017-10-09 15:34:10 -0700834func GenRuleFactory() android.Module {
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700835 m := NewGenRule()
836 android.InitAndroidModule(m)
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800837 android.InitDefaultableModule(m)
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700838 return m
839}
840
Colin Crossd350ecd2015-04-28 13:25:36 -0700841type genRuleProperties struct {
Dan Willemsen9c8681f2016-09-28 16:21:00 -0700842 // names of the output files that will be generated
kellyhung750334a2024-03-14 01:03:49 +0800843 Out []string `android:"arch_variant"`
Colin Cross5049f022015-03-18 13:28:46 -0700844}
Nan Zhangea568a42017-11-08 21:20:04 -0800845
846var Bool = proptools.Bool
847var String = proptools.String
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800848
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800849// Defaults
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800850type Defaults struct {
851 android.ModuleBase
852 android.DefaultsModuleBase
853}
854
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800855func defaultsFactory() android.Module {
856 return DefaultsFactory()
857}
858
859func DefaultsFactory(props ...interface{}) android.Module {
860 module := &Defaults{}
861
862 module.AddProperties(props...)
863 module.AddProperties(
864 &generatorProperties{},
865 &genRuleProperties{},
866 )
867
868 android.InitDefaultsModule(module)
869
870 return module
871}
Yu Liu6a7940c2023-05-09 17:12:22 -0700872
Yu Liue7f7cbf2023-06-13 18:50:03 +0000873var sandboxingAllowlistKey = android.NewOnceKey("genruleSandboxingAllowlistKey")
874
875type sandboxingAllowlistSets struct {
876 sandboxingDenyModuleSet map[string]bool
Yu Liue7f7cbf2023-06-13 18:50:03 +0000877}
878
879func getSandboxingAllowlistSets(ctx android.PathContext) *sandboxingAllowlistSets {
880 return ctx.Config().Once(sandboxingAllowlistKey, func() interface{} {
881 sandboxingDenyModuleSet := map[string]bool{}
Yu Liue7f7cbf2023-06-13 18:50:03 +0000882
Cole Faust55492572024-01-25 18:00:33 -0800883 android.AddToStringSet(sandboxingDenyModuleSet, SandboxingDenyModuleList)
Yu Liue7f7cbf2023-06-13 18:50:03 +0000884 return &sandboxingAllowlistSets{
885 sandboxingDenyModuleSet: sandboxingDenyModuleSet,
Yu Liue7f7cbf2023-06-13 18:50:03 +0000886 }
887 }).(*sandboxingAllowlistSets)
888}
Liz Kammer0db0e342023-07-18 11:39:30 -0400889
Yu Liu6a7940c2023-05-09 17:12:22 -0700890func getSandboxedRuleBuilder(ctx android.ModuleContext, r *android.RuleBuilder) *android.RuleBuilder {
Yu Liu45d6af52023-05-24 23:10:18 +0000891 if !ctx.DeviceConfig().GenruleSandboxing() {
892 return r.SandboxTools()
893 }
Yu Liue7f7cbf2023-06-13 18:50:03 +0000894 sandboxingAllowlistSets := getSandboxingAllowlistSets(ctx)
Cole Fauste762b942024-03-15 12:46:14 -0700895 if sandboxingAllowlistSets.sandboxingDenyModuleSet[ctx.ModuleName()] {
Yu Liu6a7940c2023-05-09 17:12:22 -0700896 return r.SandboxTools()
897 }
898 return r.SandboxInputs()
899}