blob: 919564399910ff9d553fa82ee9bfdb40be1f0d0a [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 }
323 if specs := t.TransitivePackagingSpecs(); specs != nil {
324 // If the HostToolProvider has PackgingSpecs, which are definitions of the
325 // required relative locations of the tool and its dependencies, use those
326 // instead. They will be copied to those relative locations in the sbox
327 // sandbox.
Jiyong Park8fb0e972024-03-18 18:29:37 +0900328 // Care must be taken since TransitivePackagingSpec may return device-side
329 // paths via the required property. Filter them out.
330 for i, ps := range specs {
331 if ps.Partition() != "" {
332 if i == 0 {
333 panic("first PackagingSpec is assumed to be the host-side tool")
334 }
335 continue
336 }
337 packagedTools = append(packagedTools, ps)
338 }
Colin Crossba9e4032020-11-24 16:32:22 -0800339 // Assume that the first PackagingSpec of the module is the tool.
Colin Crossd11cf622021-03-23 22:30:35 -0700340 addLocationLabel(tag.label, packagedToolLocation{specs[0]})
Colin Crossba9e4032020-11-24 16:32:22 -0800341 } else {
342 tools = append(tools, path.Path())
Colin Crossd11cf622021-03-23 22:30:35 -0700343 addLocationLabel(tag.label, toolLocation{android.Paths{path.Path()}})
Colin Crossba9e4032020-11-24 16:32:22 -0800344 }
345 case bootstrap.GoBinaryTool:
346 // A GoBinaryTool provides the install path to a tool, which will be copied.
Colin Crossa44551f2021-10-25 15:36:21 -0700347 p := android.PathForGoBinary(ctx, t)
348 tools = append(tools, p)
349 addLocationLabel(tag.label, toolLocation{android.Paths{p}})
Colin Crossba9e4032020-11-24 16:32:22 -0800350 default:
Dan Willemsend6ba0d52017-09-13 15:46:47 -0700351 ctx.ModuleErrorf("%q is not a host tool provider", tool)
Colin Crossba9e4032020-11-24 16:32:22 -0800352 return
Dan Willemsen8eded0a2017-09-13 16:07:44 -0700353 }
354
Colin Crossba9e4032020-11-24 16:32:22 -0800355 seenTools[tag.label] = true
Colin Crossd350ecd2015-04-28 13:25:36 -0700356 }
Dan Willemsenf7f3d692016-04-20 14:54:32 -0700357 })
Colin Crossba71a3f2019-03-18 12:12:48 -0700358
359 // If AllowMissingDependencies is enabled, the build will not have stopped when
360 // AddFarVariationDependencies was called on a missing tool, which will result in nonsensical
Liz Kammer20ebfb42020-07-28 11:32:07 -0700361 // "cmd: unknown location label ..." errors later. Add a placeholder file to the local label.
362 // The command that uses this placeholder file will never be executed because the rule will be
363 // replaced with an android.Error rule reporting the missing dependencies.
Colin Crossba71a3f2019-03-18 12:12:48 -0700364 if ctx.Config().AllowMissingDependencies() {
365 for _, tool := range g.properties.Tools {
366 if !seenTools[tool] {
Colin Crossd11cf622021-03-23 22:30:35 -0700367 addLocationLabel(tool, errorLocation{"***missing tool " + tool + "***"})
Colin Crossba71a3f2019-03-18 12:12:48 -0700368 }
369 }
370 }
Dan Willemsenf7f3d692016-04-20 14:54:32 -0700371 }
Colin Crossd350ecd2015-04-28 13:25:36 -0700372
Dan Willemsend6ba0d52017-09-13 15:46:47 -0700373 if ctx.Failed() {
374 return
375 }
376
Colin Cross08f15ab2018-10-04 23:29:14 -0700377 for _, toolFile := range g.properties.Tool_files {
Colin Cross8a497952019-03-05 22:25:09 -0800378 paths := android.PathsForModuleSrc(ctx, []string{toolFile})
Colin Crossba9e4032020-11-24 16:32:22 -0800379 tools = append(tools, paths...)
Colin Crossd11cf622021-03-23 22:30:35 -0700380 addLocationLabel(toolFile, toolLocation{paths})
Colin Cross08f15ab2018-10-04 23:29:14 -0700381 }
382
Liz Kammer81fec182023-06-09 13:33:45 -0400383 addLabelsForInputs := func(propName string, include, exclude []string) android.Paths {
Liz Kammer81fec182023-06-09 13:33:45 -0400384 includeDirInPaths := ctx.DeviceConfig().BuildBrokenInputDir(g.Name())
385 var srcFiles android.Paths
386 for _, in := range include {
387 paths, missingDeps := android.PathsAndMissingDepsRelativeToModuleSourceDir(android.SourceInput{
388 Context: ctx, Paths: []string{in}, ExcludePaths: exclude, IncludeDirs: includeDirInPaths,
389 })
390 if len(missingDeps) > 0 {
391 if !ctx.Config().AllowMissingDependencies() {
392 panic(fmt.Errorf("should never get here, the missing dependencies %q should have been reported in DepsMutator",
393 missingDeps))
394 }
395
396 // If AllowMissingDependencies is enabled, the build will not have stopped when
397 // the dependency was added on a missing SourceFileProducer module, which will result in nonsensical
398 // "cmd: label ":..." has no files" errors later. Add a placeholder file to the local label.
399 // The command that uses this placeholder file will never be executed because the rule will be
400 // replaced with an android.Error rule reporting the missing dependencies.
401 ctx.AddMissingDependencies(missingDeps)
402 addLocationLabel(in, errorLocation{"***missing " + propName + " " + in + "***"})
403 } else {
404 srcFiles = append(srcFiles, paths...)
405 addLocationLabel(in, inputLocation{paths})
406 }
Colin Crossba71a3f2019-03-18 12:12:48 -0700407 }
Liz Kammer81fec182023-06-09 13:33:45 -0400408 return srcFiles
Colin Cross08f15ab2018-10-04 23:29:14 -0700409 }
Cole Faustbf1d92a2024-07-29 12:24:25 -0700410 g.properties.ResolvedSrcs = g.properties.Srcs.GetOrDefault(ctx, nil)
Inseob Kim2f730622024-07-23 14:03:40 +0900411 srcFiles := addLabelsForInputs("srcs", g.properties.ResolvedSrcs, g.properties.Exclude_srcs)
Colin Cross40213022023-12-13 15:19:49 -0800412 android.SetProvider(ctx, blueprint.SrcsFileProviderKey, blueprint.SrcsFileProviderData{SrcPaths: srcFiles.Strings()})
Colin Cross08f15ab2018-10-04 23:29:14 -0700413
Colin Cross1a527682019-09-23 15:55:30 -0700414 var copyFrom android.Paths
415 var outputFiles android.WritablePaths
416 var zipArgs strings.Builder
Colin Cross08f15ab2018-10-04 23:29:14 -0700417
Aleks Todorov1eb06c42024-06-03 15:23:56 +0100418 cmd := g.properties.Cmd.GetOrDefault(ctx, "")
Colin Crossf3bfd022021-09-27 15:15:06 -0700419 if g.CmdModifier != nil {
420 cmd = g.CmdModifier(ctx, cmd)
421 }
422
Liz Kammer796921d2023-07-11 08:21:41 -0400423 var extraInputs android.Paths
Alex Humesky29e3bbe2020-11-20 21:30:13 -0500424 // Generate tasks, either from genrule or gensrcs.
Liz Kammer81fec182023-06-09 13:33:45 -0400425 for i, task := range g.taskGenerator(ctx, cmd, srcFiles) {
Colin Cross3d680512020-11-13 16:23:53 -0800426 if len(task.out) == 0 {
427 ctx.ModuleErrorf("must have at least one output file")
428 return
Colin Cross85a2e892018-07-09 09:45:06 -0700429 }
430
Liz Kammer81fec182023-06-09 13:33:45 -0400431 // Only handle extra inputs once as these currently are the same across all tasks
432 if i == 0 {
433 for name, values := range task.extraInputs {
434 extraInputs = append(extraInputs, addLabelsForInputs(name, values, []string{})...)
435 }
436 }
437
Colin Crossf1a035e2020-11-16 17:32:30 -0800438 // Pick a unique path outside the task.genDir for the sbox manifest textproto,
439 // a unique rule name, and the user-visible description.
440 manifestName := "genrule.sbox.textproto"
441 desc := "generate"
442 name := "generator"
443 if task.shards > 0 {
444 manifestName = "genrule_" + strconv.Itoa(task.shard) + ".sbox.textproto"
445 desc += " " + strconv.Itoa(task.shard)
446 name += strconv.Itoa(task.shard)
447 } else if len(task.out) == 1 {
448 desc += " " + task.out[0].Base()
449 }
450
451 manifestPath := android.PathForModuleOut(ctx, manifestName)
452
453 // Use a RuleBuilder to create a rule that runs the command inside an sbox sandbox.
Yu Liu6a7940c2023-05-09 17:12:22 -0700454 rule := getSandboxedRuleBuilder(ctx, android.NewRuleBuilder(pctx, ctx).Sbox(task.genDir, manifestPath))
Justin Yun4da4ccc2023-07-06 10:56:29 +0900455 if Bool(g.properties.Write_if_changed) {
456 rule.Restat()
457 }
Colin Crossf1a035e2020-11-16 17:32:30 -0800458 cmd := rule.Command()
459
Colin Cross3d680512020-11-13 16:23:53 -0800460 for _, out := range task.out {
Colin Crossd11cf622021-03-23 22:30:35 -0700461 addLocationLabel(out.Rel(), outputLocation{out})
Colin Cross3d680512020-11-13 16:23:53 -0800462 }
463
Colin Cross3d680512020-11-13 16:23:53 -0800464 rawCommand, err := android.Expand(task.cmd, func(name string) (string, error) {
Colin Cross1a527682019-09-23 15:55:30 -0700465 // report the error directly without returning an error to android.Expand to catch multiple errors in a
466 // single run
Colin Cross3d680512020-11-13 16:23:53 -0800467 reportError := func(fmt string, args ...interface{}) (string, error) {
Colin Cross1a527682019-09-23 15:55:30 -0700468 ctx.PropertyErrorf("cmd", fmt, args...)
Colin Cross3d680512020-11-13 16:23:53 -0800469 return "SOONG_ERROR", nil
Colin Cross6f080df2016-11-04 15:32:58 -0700470 }
Colin Cross1a527682019-09-23 15:55:30 -0700471
Jihoon Kangc170af42022-08-20 05:26:38 +0000472 // Apply shell escape to each cases to prevent source file paths containing $ from being evaluated in shell
Colin Cross1a527682019-09-23 15:55:30 -0700473 switch name {
474 case "location":
475 if len(g.properties.Tools) == 0 && len(g.properties.Tool_files) == 0 {
476 return reportError("at least one `tools` or `tool_files` is required if $(location) is used")
Colin Cross6f080df2016-11-04 15:32:58 -0700477 }
Colin Crossd11cf622021-03-23 22:30:35 -0700478 loc := locationLabels[firstLabel]
479 paths := loc.Paths(cmd)
Colin Cross1a527682019-09-23 15:55:30 -0700480 if len(paths) == 0 {
481 return reportError("default label %q has no files", firstLabel)
482 } else if len(paths) > 1 {
483 return reportError("default label %q has multiple files, use $(locations %s) to reference it",
484 firstLabel, firstLabel)
Colin Cross08f15ab2018-10-04 23:29:14 -0700485 }
Jihoon Kangc170af42022-08-20 05:26:38 +0000486 return proptools.ShellEscape(paths[0]), nil
Colin Cross1a527682019-09-23 15:55:30 -0700487 case "in":
Jihoon Kangc170af42022-08-20 05:26:38 +0000488 return strings.Join(proptools.ShellEscapeList(cmd.PathsForInputs(srcFiles)), " "), nil
Colin Cross1a527682019-09-23 15:55:30 -0700489 case "out":
Colin Cross3d680512020-11-13 16:23:53 -0800490 var sandboxOuts []string
491 for _, out := range task.out {
Colin Crossf1a035e2020-11-16 17:32:30 -0800492 sandboxOuts = append(sandboxOuts, cmd.PathForOutput(out))
Colin Cross3d680512020-11-13 16:23:53 -0800493 }
Jihoon Kangc170af42022-08-20 05:26:38 +0000494 return strings.Join(proptools.ShellEscapeList(sandboxOuts), " "), nil
Colin Cross1a527682019-09-23 15:55:30 -0700495 case "genDir":
Jihoon Kangc170af42022-08-20 05:26:38 +0000496 return proptools.ShellEscape(cmd.PathForOutput(task.genDir)), nil
Cole Faust78f3c3a2024-08-15 17:19:34 -0700497 case "build_number_file":
498 if !proptools.Bool(g.properties.Uses_order_only_build_number_file) {
499 return reportError("to use the $(build_number_file) label, you must set uses_order_only_build_number_file: true")
500 }
501 return proptools.ShellEscape(cmd.PathForInput(ctx.Config().BuildNumberFile(ctx))), nil
Colin Cross1a527682019-09-23 15:55:30 -0700502 default:
503 if strings.HasPrefix(name, "location ") {
504 label := strings.TrimSpace(strings.TrimPrefix(name, "location "))
Colin Crossd11cf622021-03-23 22:30:35 -0700505 if loc, ok := locationLabels[label]; ok {
506 paths := loc.Paths(cmd)
Colin Cross1a527682019-09-23 15:55:30 -0700507 if len(paths) == 0 {
508 return reportError("label %q has no files", label)
509 } else if len(paths) > 1 {
510 return reportError("label %q has multiple files, use $(locations %s) to reference it",
511 label, label)
512 }
Jihoon Kangc170af42022-08-20 05:26:38 +0000513 return proptools.ShellEscape(paths[0]), nil
Colin Cross1a527682019-09-23 15:55:30 -0700514 } else {
Anton Hanssonbebf5262022-02-23 11:42:38 +0000515 return reportError("unknown location label %q is not in srcs, out, tools or tool_files.", label)
Colin Cross1a527682019-09-23 15:55:30 -0700516 }
517 } else if strings.HasPrefix(name, "locations ") {
518 label := strings.TrimSpace(strings.TrimPrefix(name, "locations "))
Colin Crossd11cf622021-03-23 22:30:35 -0700519 if loc, ok := locationLabels[label]; ok {
520 paths := loc.Paths(cmd)
Colin Cross1a527682019-09-23 15:55:30 -0700521 if len(paths) == 0 {
522 return reportError("label %q has no files", label)
523 }
Cole Faustce74a592023-12-07 14:58:45 -0800524 return strings.Join(proptools.ShellEscapeList(paths), " "), nil
Colin Cross1a527682019-09-23 15:55:30 -0700525 } else {
Anton Hanssonbebf5262022-02-23 11:42:38 +0000526 return reportError("unknown locations label %q is not in srcs, out, tools or tool_files.", label)
Colin Cross1a527682019-09-23 15:55:30 -0700527 }
528 } else {
529 return reportError("unknown variable '$(%s)'", name)
530 }
Colin Cross6f080df2016-11-04 15:32:58 -0700531 }
Colin Cross1a527682019-09-23 15:55:30 -0700532 })
533
534 if err != nil {
535 ctx.PropertyErrorf("cmd", "%s", err.Error())
536 return
Colin Cross6f080df2016-11-04 15:32:58 -0700537 }
Colin Cross6f080df2016-11-04 15:32:58 -0700538
Colin Cross1a527682019-09-23 15:55:30 -0700539 g.rawCommands = append(g.rawCommands, rawCommand)
Bill Peckhamc087be12020-02-13 15:55:10 -0800540
Colin Cross3d680512020-11-13 16:23:53 -0800541 cmd.Text(rawCommand)
Liz Kammer81fec182023-06-09 13:33:45 -0400542 cmd.Implicits(srcFiles) // need to be able to reference other srcs
543 cmd.Implicits(extraInputs)
Colin Cross3d680512020-11-13 16:23:53 -0800544 cmd.ImplicitOutputs(task.out)
545 cmd.Implicits(task.in)
Colin Crossba9e4032020-11-24 16:32:22 -0800546 cmd.ImplicitTools(tools)
Colin Crossba9e4032020-11-24 16:32:22 -0800547 cmd.ImplicitPackagedTools(packagedTools)
Cole Faust78f3c3a2024-08-15 17:19:34 -0700548 if proptools.Bool(g.properties.Uses_order_only_build_number_file) {
549 if _, ok := genrule_build_number_allowlist[ctx.ModuleDir()+":"+ctx.ModuleName()]; !ok {
550 ctx.ModuleErrorf("Only allowlisted modules may use uses_order_only_build_number_file: true")
551 }
552 cmd.OrderOnly(ctx.Config().BuildNumberFile(ctx))
553 }
Colin Cross3d680512020-11-13 16:23:53 -0800554
555 // Create the rule to run the genrule command inside sbox.
Colin Crossf1a035e2020-11-16 17:32:30 -0800556 rule.Build(name, desc)
Colin Cross1a527682019-09-23 15:55:30 -0700557
558 if len(task.copyTo) > 0 {
Colin Cross3d680512020-11-13 16:23:53 -0800559 // If copyTo is set, multiple shards need to be copied into a single directory.
560 // task.out contains the per-shard paths, and copyTo contains the corresponding
561 // final path. The files need to be copied into the final directory by a
562 // single rule so it can remove the directory before it starts to ensure no
563 // old files remain. zipsync already does this, so build up zipArgs that
564 // zip all the per-shard directories into a single zip.
Colin Cross1a527682019-09-23 15:55:30 -0700565 outputFiles = append(outputFiles, task.copyTo...)
566 copyFrom = append(copyFrom, task.out.Paths()...)
567 zipArgs.WriteString(" -C " + task.genDir.String())
568 zipArgs.WriteString(android.JoinWithPrefix(task.out.Strings(), " -f "))
569 } else {
570 outputFiles = append(outputFiles, task.out...)
571 }
Colin Cross6f080df2016-11-04 15:32:58 -0700572 }
573
Colin Cross1a527682019-09-23 15:55:30 -0700574 if len(copyFrom) > 0 {
Colin Cross3d680512020-11-13 16:23:53 -0800575 // Create a rule that zips all the per-shard directories into a single zip and then
576 // uses zipsync to unzip it into the final directory.
Colin Cross1a527682019-09-23 15:55:30 -0700577 ctx.Build(pctx, android.BuildParams{
Colin Crossf1885962020-11-20 15:28:30 -0800578 Rule: gensrcsMerge,
579 Implicits: copyFrom,
580 Outputs: outputFiles,
581 Description: "merge shards",
Colin Cross1a527682019-09-23 15:55:30 -0700582 Args: map[string]string{
583 "zipArgs": zipArgs.String(),
584 "tmpZip": android.PathForModuleGen(ctx, g.subDir+".zip").String(),
585 "genDir": android.PathForModuleGen(ctx, g.subDir).String(),
586 },
587 })
Colin Cross85a2e892018-07-09 09:45:06 -0700588 }
589
Colin Cross1a527682019-09-23 15:55:30 -0700590 g.outputFiles = outputFiles.Paths()
Chris Parsonsf874e462022-05-10 13:50:12 -0400591}
Jeff Gastonefc1b412017-03-29 17:29:06 -0700592
Chris Parsonsf874e462022-05-10 13:50:12 -0400593func (g *Module) GenerateAndroidBuildActions(ctx android.ModuleContext) {
594 g.generateCommonBuildActions(ctx)
595
596 // For <= 6 outputs, just embed those directly in the users. Right now, that covers >90% of
597 // the genrules on AOSP. That will make things simpler to look at the graph in the common
598 // case. For larger sets of outputs, inject a phony target in between to limit ninja file
599 // growth.
600 if len(g.outputFiles) <= 6 {
601 g.outputDeps = g.outputFiles
602 } else {
603 phonyFile := android.PathForModuleGen(ctx, "genrule-phony")
604 ctx.Build(pctx, android.BuildParams{
605 Rule: blueprint.Phony,
606 Output: phonyFile,
607 Inputs: g.outputFiles,
608 })
609 g.outputDeps = android.Paths{phonyFile}
Jeff Gaston02a684b2017-10-27 14:59:27 -0700610 }
mrziwang4514ef22024-06-07 13:31:48 -0700611
612 g.setOutputFiles(ctx)
613}
614
615func (g *Module) setOutputFiles(ctx android.ModuleContext) {
616 if len(g.outputFiles) == 0 {
617 return
618 }
619 ctx.SetOutputFiles(g.outputFiles, "")
620 // non-empty-string-tag should match one of the outputs
621 for _, files := range g.outputFiles {
622 ctx.SetOutputFiles(android.Paths{files}, files.Rel())
623 }
Chris Parsonsf874e462022-05-10 13:50:12 -0400624}
625
Brandon Lee5d45c6f2018-08-15 15:35:38 -0700626// Collect information for opening IDE project files in java/jdeps.go.
627func (g *Module) IDEInfo(dpInfo *android.IdeInfo) {
628 dpInfo.Srcs = append(dpInfo.Srcs, g.Srcs().Strings()...)
Inseob Kim2f730622024-07-23 14:03:40 +0900629 for _, src := range g.properties.ResolvedSrcs {
Brandon Lee5d45c6f2018-08-15 15:35:38 -0700630 if strings.HasPrefix(src, ":") {
631 src = strings.Trim(src, ":")
632 dpInfo.Deps = append(dpInfo.Deps, src)
633 }
634 }
635}
636
Colin Crossa4ad2b02019-03-18 22:15:32 -0700637func (g *Module) AndroidMk() android.AndroidMkData {
638 return android.AndroidMkData{
Anton Hansson72f18492020-10-30 16:34:45 +0000639 Class: "ETC",
Colin Crossa4ad2b02019-03-18 22:15:32 -0700640 OutputFile: android.OptionalPathForPath(g.outputFiles[0]),
641 SubName: g.subName,
642 Extra: []android.AndroidMkExtraFunc{
643 func(w io.Writer, outputFile android.Path) {
Anton Hansson72f18492020-10-30 16:34:45 +0000644 fmt.Fprintln(w, "LOCAL_UNINSTALLABLE_MODULE := true")
Colin Crossa4ad2b02019-03-18 22:15:32 -0700645 },
646 },
647 Custom: func(w io.Writer, name, prefix, moduleDir string, data android.AndroidMkData) {
648 android.WriteAndroidMkData(w, data)
649 if data.SubName != "" {
650 fmt.Fprintln(w, ".PHONY:", name)
651 fmt.Fprintln(w, name, ":", name+g.subName)
652 }
653 },
654 }
655}
656
Jiyong Park45bf82e2020-12-15 22:29:02 +0900657var _ android.ApexModule = (*Module)(nil)
658
659// Implements android.ApexModule
Dan Albertc8060532020-07-22 22:32:17 -0700660func (g *Module) ShouldSupportSdkVersion(ctx android.BaseModuleContext,
661 sdkVersion android.ApiLevel) error {
Jooyung Han749dc692020-04-15 11:03:39 +0900662 // Because generated outputs are checked by client modules(e.g. cc_library, ...)
663 // we can safely ignore the check here.
664 return nil
665}
666
Jeff Gaston437d23c2017-11-08 12:38:00 -0800667func generatorFactory(taskGenerator taskFunc, props ...interface{}) *Module {
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700668 module := &Module{
Jeff Gaston437d23c2017-11-08 12:38:00 -0800669 taskGenerator: taskGenerator,
Colin Crossd350ecd2015-04-28 13:25:36 -0700670 }
671
Colin Cross36242852017-06-23 15:06:31 -0700672 module.AddProperties(props...)
673 module.AddProperties(&module.properties)
Colin Crossd350ecd2015-04-28 13:25:36 -0700674
Colin Cross7228ecd2019-11-18 16:00:16 -0800675 module.ImageInterface = noopImageInterface{}
676
Colin Cross36242852017-06-23 15:06:31 -0700677 return module
Colin Crossd350ecd2015-04-28 13:25:36 -0700678}
679
Colin Cross7228ecd2019-11-18 16:00:16 -0800680type noopImageInterface struct{}
681
682func (x noopImageInterface) ImageMutatorBegin(android.BaseModuleContext) {}
Jihoon Kang47e91842024-06-19 00:51:16 +0000683func (x noopImageInterface) VendorVariantNeeded(android.BaseModuleContext) bool { return false }
684func (x noopImageInterface) ProductVariantNeeded(android.BaseModuleContext) bool { return false }
Colin Cross7228ecd2019-11-18 16:00:16 -0800685func (x noopImageInterface) CoreVariantNeeded(android.BaseModuleContext) bool { return false }
Yifan Hong1b3348d2020-01-21 15:53:22 -0800686func (x noopImageInterface) RamdiskVariantNeeded(android.BaseModuleContext) bool { return false }
Yifan Hong60e0cfb2020-10-21 15:17:56 -0700687func (x noopImageInterface) VendorRamdiskVariantNeeded(android.BaseModuleContext) bool { return false }
Inseob Kim08758f02021-04-08 21:13:22 +0900688func (x noopImageInterface) DebugRamdiskVariantNeeded(android.BaseModuleContext) bool { return false }
Colin Cross7228ecd2019-11-18 16:00:16 -0800689func (x noopImageInterface) RecoveryVariantNeeded(android.BaseModuleContext) bool { return false }
690func (x noopImageInterface) ExtraImageVariations(ctx android.BaseModuleContext) []string { return nil }
Jihoon Kang7583e832024-06-13 21:25:45 +0000691func (x noopImageInterface) SetImageVariation(ctx android.BaseModuleContext, variation string) {
Colin Cross7228ecd2019-11-18 16:00:16 -0800692}
693
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700694func NewGenSrcs() *Module {
Colin Crossd350ecd2015-04-28 13:25:36 -0700695 properties := &genSrcsProperties{}
696
Colin Crossf1885962020-11-20 15:28:30 -0800697 // finalSubDir is the name of the subdirectory that output files will be generated into.
698 // It is used so that per-shard directories can be placed alongside it an then finally
699 // merged into it.
700 const finalSubDir = "gensrcs"
701
Colin Cross1a527682019-09-23 15:55:30 -0700702 taskGenerator := func(ctx android.ModuleContext, rawCommand string, srcFiles android.Paths) []generateTask {
Colin Cross1a527682019-09-23 15:55:30 -0700703 shardSize := defaultShardSize
704 if s := properties.Shard_size; s != nil {
705 shardSize = int(*s)
706 }
Jeff Gaston437d23c2017-11-08 12:38:00 -0800707
Colin Crossf1885962020-11-20 15:28:30 -0800708 // gensrcs rules can easily hit command line limits by repeating the command for
709 // every input file. Shard the input files into groups.
Colin Cross1a527682019-09-23 15:55:30 -0700710 shards := android.ShardPaths(srcFiles, shardSize)
711 var generateTasks []generateTask
Colin Crossbaccf5b2018-02-21 14:07:48 -0800712
Colin Cross1a527682019-09-23 15:55:30 -0700713 for i, shard := range shards {
714 var commands []string
715 var outFiles android.WritablePaths
716 var copyTo android.WritablePaths
Colin Cross1a527682019-09-23 15:55:30 -0700717
Colin Crossf1885962020-11-20 15:28:30 -0800718 // When sharding is enabled (i.e. len(shards) > 1), the sbox rules for each
719 // shard will be write to their own directories and then be merged together
720 // into finalSubDir. If sharding is not enabled (i.e. len(shards) == 1),
721 // the sbox rule will write directly to finalSubDir.
722 genSubDir := finalSubDir
Colin Cross1a527682019-09-23 15:55:30 -0700723 if len(shards) > 1 {
Colin Crossf1885962020-11-20 15:28:30 -0800724 genSubDir = strconv.Itoa(i)
Jeff Gaston437d23c2017-11-08 12:38:00 -0800725 }
726
Colin Crossf1885962020-11-20 15:28:30 -0800727 genDir := android.PathForModuleGen(ctx, genSubDir)
Colin Crossf1a035e2020-11-16 17:32:30 -0800728 // TODO(ccross): this RuleBuilder is a hack to be able to call
729 // rule.Command().PathForOutput. Replace this with passing the rule into the
730 // generator.
Yu Liu6a7940c2023-05-09 17:12:22 -0700731 rule := getSandboxedRuleBuilder(ctx, android.NewRuleBuilder(pctx, ctx).Sbox(genDir, nil))
Jeff Gaston437d23c2017-11-08 12:38:00 -0800732
Colin Cross3ea4eb82020-11-24 13:07:27 -0800733 for _, in := range shard {
yangbill6d032dd2024-04-18 03:05:49 +0000734 outFile := android.GenPathWithExtAndTrimExt(ctx, finalSubDir, in, String(properties.Output_extension), String(properties.Trim_extension))
Colin Crossf1885962020-11-20 15:28:30 -0800735
736 // If sharding is enabled, then outFile is the path to the output file in
737 // the shard directory, and copyTo is the path to the output file in the
738 // final directory.
Colin Cross1a527682019-09-23 15:55:30 -0700739 if len(shards) > 1 {
yangbill6d032dd2024-04-18 03:05:49 +0000740 shardFile := android.GenPathWithExtAndTrimExt(ctx, genSubDir, in, String(properties.Output_extension), String(properties.Trim_extension))
Colin Cross1a527682019-09-23 15:55:30 -0700741 copyTo = append(copyTo, outFile)
742 outFile = shardFile
743 }
744
745 outFiles = append(outFiles, outFile)
Colin Cross1a527682019-09-23 15:55:30 -0700746
Colin Crossf1885962020-11-20 15:28:30 -0800747 // pre-expand the command line to replace $in and $out with references to
748 // a single input and output file.
Colin Cross1a527682019-09-23 15:55:30 -0700749 command, err := android.Expand(rawCommand, func(name string) (string, error) {
750 switch name {
751 case "in":
752 return in.String(), nil
753 case "out":
Colin Crossf1a035e2020-11-16 17:32:30 -0800754 return rule.Command().PathForOutput(outFile), nil
Colin Cross1a527682019-09-23 15:55:30 -0700755 default:
756 return "$(" + name + ")", nil
757 }
758 })
759 if err != nil {
760 ctx.PropertyErrorf("cmd", err.Error())
761 }
762
763 // escape the command in case for example it contains '#', an odd number of '"', etc
764 command = fmt.Sprintf("bash -c %v", proptools.ShellEscape(command))
765 commands = append(commands, command)
766 }
767 fullCommand := strings.Join(commands, " && ")
768
769 generateTasks = append(generateTasks, generateTask{
Cole Faust55492572024-01-25 18:00:33 -0800770 in: shard,
771 out: outFiles,
772 copyTo: copyTo,
773 genDir: genDir,
774 cmd: fullCommand,
775 shard: i,
776 shards: len(shards),
Liz Kammer81fec182023-06-09 13:33:45 -0400777 extraInputs: map[string][]string{
778 "data": properties.Data,
779 },
Colin Cross1a527682019-09-23 15:55:30 -0700780 })
Jeff Gaston437d23c2017-11-08 12:38:00 -0800781 }
Colin Cross1a527682019-09-23 15:55:30 -0700782
783 return generateTasks
Colin Crossd350ecd2015-04-28 13:25:36 -0700784 }
785
Colin Cross1a527682019-09-23 15:55:30 -0700786 g := generatorFactory(taskGenerator, properties)
Colin Crossf1885962020-11-20 15:28:30 -0800787 g.subDir = finalSubDir
Colin Cross1a527682019-09-23 15:55:30 -0700788 return g
Colin Crossd350ecd2015-04-28 13:25:36 -0700789}
790
Colin Cross54190b32017-10-09 15:34:10 -0700791func GenSrcsFactory() android.Module {
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700792 m := NewGenSrcs()
793 android.InitAndroidModule(m)
Colin Cross483b4c42024-05-09 13:08:02 -0700794 android.InitDefaultableModule(m)
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700795 return m
796}
797
Colin Crossd350ecd2015-04-28 13:25:36 -0700798type genSrcsProperties struct {
Colin Cross7d5136f2015-05-11 13:39:40 -0700799 // extension that will be substituted for each output file
Nan Zhanga5e7cb42017-11-09 22:42:32 -0800800 Output_extension *string
Colin Cross1a527682019-09-23 15:55:30 -0700801
802 // maximum number of files that will be passed on a single command line.
803 Shard_size *int64
Liz Kammer81fec182023-06-09 13:33:45 -0400804
805 // Additional files needed for build that are not tooling related.
806 Data []string `android:"path"`
yangbill6d032dd2024-04-18 03:05:49 +0000807
808 // Trim the matched extension for each input file, and it should start with ".".
809 Trim_extension *string
Colin Cross5049f022015-03-18 13:28:46 -0700810}
811
Evgenii Stepanovf47c90d2020-12-02 18:55:09 -0800812const defaultShardSize = 50
Colin Cross1a527682019-09-23 15:55:30 -0700813
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700814func NewGenRule() *Module {
Colin Crossd350ecd2015-04-28 13:25:36 -0700815 properties := &genRuleProperties{}
Colin Cross5049f022015-03-18 13:28:46 -0700816
Colin Cross1a527682019-09-23 15:55:30 -0700817 taskGenerator := func(ctx android.ModuleContext, rawCommand string, srcFiles android.Paths) []generateTask {
Dan Willemsen9c8681f2016-09-28 16:21:00 -0700818 outs := make(android.WritablePaths, len(properties.Out))
819 for i, out := range properties.Out {
Cole Faust55492572024-01-25 18:00:33 -0800820 outs[i] = android.PathForModuleGen(ctx, out)
Dan Willemsen9c8681f2016-09-28 16:21:00 -0700821 }
Colin Cross1a527682019-09-23 15:55:30 -0700822 return []generateTask{{
Cole Faust55492572024-01-25 18:00:33 -0800823 in: srcFiles,
824 out: outs,
825 genDir: android.PathForModuleGen(ctx),
826 cmd: rawCommand,
Colin Cross1a527682019-09-23 15:55:30 -0700827 }}
Colin Cross5049f022015-03-18 13:28:46 -0700828 }
Colin Crossd350ecd2015-04-28 13:25:36 -0700829
Jeff Gaston437d23c2017-11-08 12:38:00 -0800830 return generatorFactory(taskGenerator, properties)
Colin Cross5049f022015-03-18 13:28:46 -0700831}
832
Colin Cross54190b32017-10-09 15:34:10 -0700833func GenRuleFactory() android.Module {
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700834 m := NewGenRule()
835 android.InitAndroidModule(m)
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800836 android.InitDefaultableModule(m)
Dan Willemsen3e5bdf22017-09-13 18:37:08 -0700837 return m
838}
839
Colin Crossd350ecd2015-04-28 13:25:36 -0700840type genRuleProperties struct {
Dan Willemsen9c8681f2016-09-28 16:21:00 -0700841 // names of the output files that will be generated
kellyhung750334a2024-03-14 01:03:49 +0800842 Out []string `android:"arch_variant"`
Colin Cross5049f022015-03-18 13:28:46 -0700843}
Nan Zhangea568a42017-11-08 21:20:04 -0800844
845var Bool = proptools.Bool
846var String = proptools.String
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800847
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800848// Defaults
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800849type Defaults struct {
850 android.ModuleBase
851 android.DefaultsModuleBase
852}
853
Jaewoong Jung98716bd2018-12-10 08:13:18 -0800854func defaultsFactory() android.Module {
855 return DefaultsFactory()
856}
857
858func DefaultsFactory(props ...interface{}) android.Module {
859 module := &Defaults{}
860
861 module.AddProperties(props...)
862 module.AddProperties(
863 &generatorProperties{},
864 &genRuleProperties{},
865 )
866
867 android.InitDefaultsModule(module)
868
869 return module
870}
Yu Liu6a7940c2023-05-09 17:12:22 -0700871
Yu Liue7f7cbf2023-06-13 18:50:03 +0000872var sandboxingAllowlistKey = android.NewOnceKey("genruleSandboxingAllowlistKey")
873
874type sandboxingAllowlistSets struct {
875 sandboxingDenyModuleSet map[string]bool
Yu Liue7f7cbf2023-06-13 18:50:03 +0000876}
877
878func getSandboxingAllowlistSets(ctx android.PathContext) *sandboxingAllowlistSets {
879 return ctx.Config().Once(sandboxingAllowlistKey, func() interface{} {
880 sandboxingDenyModuleSet := map[string]bool{}
Yu Liue7f7cbf2023-06-13 18:50:03 +0000881
Cole Faust55492572024-01-25 18:00:33 -0800882 android.AddToStringSet(sandboxingDenyModuleSet, SandboxingDenyModuleList)
Yu Liue7f7cbf2023-06-13 18:50:03 +0000883 return &sandboxingAllowlistSets{
884 sandboxingDenyModuleSet: sandboxingDenyModuleSet,
Yu Liue7f7cbf2023-06-13 18:50:03 +0000885 }
886 }).(*sandboxingAllowlistSets)
887}
Liz Kammer0db0e342023-07-18 11:39:30 -0400888
Yu Liu6a7940c2023-05-09 17:12:22 -0700889func getSandboxedRuleBuilder(ctx android.ModuleContext, r *android.RuleBuilder) *android.RuleBuilder {
Yu Liu45d6af52023-05-24 23:10:18 +0000890 if !ctx.DeviceConfig().GenruleSandboxing() {
891 return r.SandboxTools()
892 }
Yu Liue7f7cbf2023-06-13 18:50:03 +0000893 sandboxingAllowlistSets := getSandboxingAllowlistSets(ctx)
Cole Fauste762b942024-03-15 12:46:14 -0700894 if sandboxingAllowlistSets.sandboxingDenyModuleSet[ctx.ModuleName()] {
Yu Liu6a7940c2023-05-09 17:12:22 -0700895 return r.SandboxTools()
896 }
897 return r.SandboxInputs()
898}