777 lines
25 KiB
Go
777 lines
25 KiB
Go
// Copyright 2015 Google Inc. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// A genrule module takes a list of source files ("srcs" property), an optional
|
|
// list of tools ("tools" property), and a command line ("cmd" property), to
|
|
// generate output files ("out" property).
|
|
|
|
package genrule
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/google/blueprint"
|
|
"github.com/google/blueprint/bootstrap"
|
|
"github.com/google/blueprint/proptools"
|
|
|
|
"android/soong/android"
|
|
"android/soong/bazel"
|
|
)
|
|
|
|
func init() {
|
|
RegisterGenruleBuildComponents(android.InitRegistrationContext)
|
|
}
|
|
|
|
func RegisterGenruleBuildComponents(ctx android.RegistrationContext) {
|
|
ctx.RegisterModuleType("genrule_defaults", defaultsFactory)
|
|
|
|
ctx.RegisterModuleType("gensrcs", GenSrcsFactory)
|
|
ctx.RegisterModuleType("genrule", GenRuleFactory)
|
|
|
|
ctx.FinalDepsMutators(func(ctx android.RegisterMutatorsContext) {
|
|
ctx.BottomUp("genrule_tool_deps", toolDepsMutator).Parallel()
|
|
})
|
|
}
|
|
|
|
var (
|
|
pctx = android.NewPackageContext("android/soong/genrule")
|
|
|
|
// Used by gensrcs when there is more than 1 shard to merge the outputs
|
|
// of each shard into a zip file.
|
|
gensrcsMerge = pctx.AndroidStaticRule("gensrcsMerge", blueprint.RuleParams{
|
|
Command: "${soongZip} -o ${tmpZip} @${tmpZip}.rsp && ${zipSync} -d ${genDir} ${tmpZip}",
|
|
CommandDeps: []string{"${soongZip}", "${zipSync}"},
|
|
Rspfile: "${tmpZip}.rsp",
|
|
RspfileContent: "${zipArgs}",
|
|
}, "tmpZip", "genDir", "zipArgs")
|
|
)
|
|
|
|
func init() {
|
|
pctx.Import("android/soong/android")
|
|
pctx.HostBinToolVariable("sboxCmd", "sbox")
|
|
|
|
pctx.HostBinToolVariable("soongZip", "soong_zip")
|
|
pctx.HostBinToolVariable("zipSync", "zipsync")
|
|
}
|
|
|
|
type SourceFileGenerator interface {
|
|
GeneratedSourceFiles() android.Paths
|
|
GeneratedHeaderDirs() android.Paths
|
|
GeneratedDeps() android.Paths
|
|
}
|
|
|
|
// Alias for android.HostToolProvider
|
|
// Deprecated: use android.HostToolProvider instead.
|
|
type HostToolProvider interface {
|
|
android.HostToolProvider
|
|
}
|
|
|
|
type hostToolDependencyTag struct {
|
|
blueprint.BaseDependencyTag
|
|
label string
|
|
}
|
|
type generatorProperties struct {
|
|
// The command to run on one or more input files. Cmd supports substitution of a few variables
|
|
//
|
|
// Available variables for substitution:
|
|
//
|
|
// $(location): the path to the first entry in tools or tool_files
|
|
// $(location <label>): the path to the tool, tool_file, input or output with name <label>
|
|
// $(in): one or more input files
|
|
// $(out): a single output file
|
|
// $(depfile): a file to which dependencies will be written, if the depfile property is set to true
|
|
// $(genDir): the sandbox directory for this tool; contains $(out)
|
|
// $$: a literal $
|
|
Cmd *string
|
|
|
|
// Enable reading a file containing dependencies in gcc format after the command completes
|
|
Depfile *bool
|
|
|
|
// name of the modules (if any) that produces the host executable. Leave empty for
|
|
// prebuilts or scripts that do not need a module to build them.
|
|
Tools []string
|
|
|
|
// Local file that is used as the tool
|
|
Tool_files []string `android:"path"`
|
|
|
|
// List of directories to export generated headers from
|
|
Export_include_dirs []string
|
|
|
|
// list of input files
|
|
Srcs []string `android:"path,arch_variant"`
|
|
|
|
// input files to exclude
|
|
Exclude_srcs []string `android:"path,arch_variant"`
|
|
|
|
// Properties for Bazel migration purposes.
|
|
bazel.Properties
|
|
}
|
|
|
|
type Module struct {
|
|
android.ModuleBase
|
|
android.DefaultableModuleBase
|
|
android.ApexModuleBase
|
|
|
|
// For other packages to make their own genrules with extra
|
|
// properties
|
|
Extra interface{}
|
|
android.ImageInterface
|
|
|
|
properties generatorProperties
|
|
|
|
// For the different tasks that genrule and gensrc generate. genrule will
|
|
// generate 1 task, and gensrc will generate 1 or more tasks based on the
|
|
// number of shards the input files are sharded into.
|
|
taskGenerator taskFunc
|
|
|
|
deps android.Paths
|
|
rule blueprint.Rule
|
|
rawCommands []string
|
|
|
|
exportedIncludeDirs android.Paths
|
|
|
|
outputFiles android.Paths
|
|
outputDeps android.Paths
|
|
|
|
subName string
|
|
subDir string
|
|
|
|
// Collect the module directory for IDE info in java/jdeps.go.
|
|
modulePaths []string
|
|
}
|
|
|
|
type taskFunc func(ctx android.ModuleContext, rawCommand string, srcFiles android.Paths) []generateTask
|
|
|
|
type generateTask struct {
|
|
in android.Paths
|
|
out android.WritablePaths
|
|
depFile android.WritablePath
|
|
copyTo android.WritablePaths // For gensrcs to set on gensrcsMerge rule.
|
|
genDir android.WritablePath
|
|
extraTools android.Paths // dependencies on tools used by the generator
|
|
|
|
cmd string
|
|
// For gensrsc sharding.
|
|
shard int
|
|
shards int
|
|
}
|
|
|
|
func (g *Module) GeneratedSourceFiles() android.Paths {
|
|
return g.outputFiles
|
|
}
|
|
|
|
func (g *Module) Srcs() android.Paths {
|
|
return append(android.Paths{}, g.outputFiles...)
|
|
}
|
|
|
|
func (g *Module) GeneratedHeaderDirs() android.Paths {
|
|
return g.exportedIncludeDirs
|
|
}
|
|
|
|
func (g *Module) GeneratedDeps() android.Paths {
|
|
return g.outputDeps
|
|
}
|
|
|
|
func toolDepsMutator(ctx android.BottomUpMutatorContext) {
|
|
if g, ok := ctx.Module().(*Module); ok {
|
|
for _, tool := range g.properties.Tools {
|
|
tag := hostToolDependencyTag{label: tool}
|
|
if m := android.SrcIsModule(tool); m != "" {
|
|
tool = m
|
|
}
|
|
ctx.AddFarVariationDependencies(ctx.Config().BuildOSTarget.Variations(), tag, tool)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Returns true if information was available from Bazel, false if bazel invocation still needs to occur.
|
|
func (c *Module) generateBazelBuildActions(ctx android.ModuleContext, label string) bool {
|
|
bazelCtx := ctx.Config().BazelContext
|
|
filePaths, ok := bazelCtx.GetAllFiles(label)
|
|
if ok {
|
|
var bazelOutputFiles android.Paths
|
|
for _, bazelOutputFile := range filePaths {
|
|
bazelOutputFiles = append(bazelOutputFiles, android.PathForSource(ctx, bazelOutputFile))
|
|
}
|
|
c.outputFiles = bazelOutputFiles
|
|
c.outputDeps = bazelOutputFiles
|
|
}
|
|
return ok
|
|
}
|
|
|
|
func (g *Module) GenerateAndroidBuildActions(ctx android.ModuleContext) {
|
|
g.subName = ctx.ModuleSubDir()
|
|
|
|
// Collect the module directory for IDE info in java/jdeps.go.
|
|
g.modulePaths = append(g.modulePaths, ctx.ModuleDir())
|
|
|
|
if len(g.properties.Export_include_dirs) > 0 {
|
|
for _, dir := range g.properties.Export_include_dirs {
|
|
g.exportedIncludeDirs = append(g.exportedIncludeDirs,
|
|
android.PathForModuleGen(ctx, g.subDir, ctx.ModuleDir(), dir))
|
|
}
|
|
} else {
|
|
g.exportedIncludeDirs = append(g.exportedIncludeDirs, android.PathForModuleGen(ctx, g.subDir))
|
|
}
|
|
|
|
locationLabels := map[string][]string{}
|
|
firstLabel := ""
|
|
|
|
addLocationLabel := func(label string, paths []string) {
|
|
if firstLabel == "" {
|
|
firstLabel = label
|
|
}
|
|
if _, exists := locationLabels[label]; !exists {
|
|
locationLabels[label] = paths
|
|
} else {
|
|
ctx.ModuleErrorf("multiple labels for %q, %q and %q",
|
|
label, strings.Join(locationLabels[label], " "), strings.Join(paths, " "))
|
|
}
|
|
}
|
|
|
|
if len(g.properties.Tools) > 0 {
|
|
seenTools := make(map[string]bool)
|
|
|
|
ctx.VisitDirectDepsBlueprint(func(module blueprint.Module) {
|
|
switch tag := ctx.OtherModuleDependencyTag(module).(type) {
|
|
case hostToolDependencyTag:
|
|
tool := ctx.OtherModuleName(module)
|
|
var path android.OptionalPath
|
|
|
|
if t, ok := module.(android.HostToolProvider); ok {
|
|
if !t.(android.Module).Enabled() {
|
|
if ctx.Config().AllowMissingDependencies() {
|
|
ctx.AddMissingDependencies([]string{tool})
|
|
} else {
|
|
ctx.ModuleErrorf("depends on disabled module %q", tool)
|
|
}
|
|
break
|
|
}
|
|
path = t.HostToolPath()
|
|
} else if t, ok := module.(bootstrap.GoBinaryTool); ok {
|
|
if s, err := filepath.Rel(android.PathForOutput(ctx).String(), t.InstallPath()); err == nil {
|
|
path = android.OptionalPathForPath(android.PathForOutput(ctx, s))
|
|
} else {
|
|
ctx.ModuleErrorf("cannot find path for %q: %v", tool, err)
|
|
break
|
|
}
|
|
} else {
|
|
ctx.ModuleErrorf("%q is not a host tool provider", tool)
|
|
break
|
|
}
|
|
|
|
if path.Valid() {
|
|
g.deps = append(g.deps, path.Path())
|
|
addLocationLabel(tag.label, []string{path.Path().String()})
|
|
seenTools[tag.label] = true
|
|
} else {
|
|
ctx.ModuleErrorf("host tool %q missing output file", tool)
|
|
}
|
|
}
|
|
})
|
|
|
|
// If AllowMissingDependencies is enabled, the build will not have stopped when
|
|
// AddFarVariationDependencies was called on a missing tool, which will result in nonsensical
|
|
// "cmd: unknown location label ..." errors later. Add a placeholder file to the local label.
|
|
// The command that uses this placeholder file will never be executed because the rule will be
|
|
// replaced with an android.Error rule reporting the missing dependencies.
|
|
if ctx.Config().AllowMissingDependencies() {
|
|
for _, tool := range g.properties.Tools {
|
|
if !seenTools[tool] {
|
|
addLocationLabel(tool, []string{"***missing tool " + tool + "***"})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if ctx.Failed() {
|
|
return
|
|
}
|
|
|
|
for _, toolFile := range g.properties.Tool_files {
|
|
paths := android.PathsForModuleSrc(ctx, []string{toolFile})
|
|
g.deps = append(g.deps, paths...)
|
|
addLocationLabel(toolFile, paths.Strings())
|
|
}
|
|
|
|
var srcFiles android.Paths
|
|
for _, in := range g.properties.Srcs {
|
|
paths, missingDeps := android.PathsAndMissingDepsForModuleSrcExcludes(ctx, []string{in}, g.properties.Exclude_srcs)
|
|
if len(missingDeps) > 0 {
|
|
if !ctx.Config().AllowMissingDependencies() {
|
|
panic(fmt.Errorf("should never get here, the missing dependencies %q should have been reported in DepsMutator",
|
|
missingDeps))
|
|
}
|
|
|
|
// If AllowMissingDependencies is enabled, the build will not have stopped when
|
|
// the dependency was added on a missing SourceFileProducer module, which will result in nonsensical
|
|
// "cmd: label ":..." has no files" errors later. Add a placeholder file to the local label.
|
|
// The command that uses this placeholder file will never be executed because the rule will be
|
|
// replaced with an android.Error rule reporting the missing dependencies.
|
|
ctx.AddMissingDependencies(missingDeps)
|
|
addLocationLabel(in, []string{"***missing srcs " + in + "***"})
|
|
} else {
|
|
srcFiles = append(srcFiles, paths...)
|
|
addLocationLabel(in, paths.Strings())
|
|
}
|
|
}
|
|
|
|
var copyFrom android.Paths
|
|
var outputFiles android.WritablePaths
|
|
var zipArgs strings.Builder
|
|
|
|
// Generate tasks, either from genrule or gensrcs.
|
|
for _, task := range g.taskGenerator(ctx, String(g.properties.Cmd), srcFiles) {
|
|
if len(task.out) == 0 {
|
|
ctx.ModuleErrorf("must have at least one output file")
|
|
return
|
|
}
|
|
|
|
// Pick a unique path outside the task.genDir for the sbox manifest textproto,
|
|
// a unique rule name, and the user-visible description.
|
|
manifestName := "genrule.sbox.textproto"
|
|
desc := "generate"
|
|
name := "generator"
|
|
if task.shards > 0 {
|
|
manifestName = "genrule_" + strconv.Itoa(task.shard) + ".sbox.textproto"
|
|
desc += " " + strconv.Itoa(task.shard)
|
|
name += strconv.Itoa(task.shard)
|
|
} else if len(task.out) == 1 {
|
|
desc += " " + task.out[0].Base()
|
|
}
|
|
|
|
manifestPath := android.PathForModuleOut(ctx, manifestName)
|
|
|
|
// Use a RuleBuilder to create a rule that runs the command inside an sbox sandbox.
|
|
rule := android.NewRuleBuilder(pctx, ctx).Sbox(task.genDir, manifestPath)
|
|
cmd := rule.Command()
|
|
|
|
for _, out := range task.out {
|
|
addLocationLabel(out.Rel(), []string{cmd.PathForOutput(out)})
|
|
}
|
|
|
|
referencedDepfile := false
|
|
|
|
rawCommand, err := android.Expand(task.cmd, func(name string) (string, error) {
|
|
// report the error directly without returning an error to android.Expand to catch multiple errors in a
|
|
// single run
|
|
reportError := func(fmt string, args ...interface{}) (string, error) {
|
|
ctx.PropertyErrorf("cmd", fmt, args...)
|
|
return "SOONG_ERROR", nil
|
|
}
|
|
|
|
switch name {
|
|
case "location":
|
|
if len(g.properties.Tools) == 0 && len(g.properties.Tool_files) == 0 {
|
|
return reportError("at least one `tools` or `tool_files` is required if $(location) is used")
|
|
}
|
|
paths := locationLabels[firstLabel]
|
|
if len(paths) == 0 {
|
|
return reportError("default label %q has no files", firstLabel)
|
|
} else if len(paths) > 1 {
|
|
return reportError("default label %q has multiple files, use $(locations %s) to reference it",
|
|
firstLabel, firstLabel)
|
|
}
|
|
return locationLabels[firstLabel][0], nil
|
|
case "in":
|
|
return strings.Join(srcFiles.Strings(), " "), nil
|
|
case "out":
|
|
var sandboxOuts []string
|
|
for _, out := range task.out {
|
|
sandboxOuts = append(sandboxOuts, cmd.PathForOutput(out))
|
|
}
|
|
return strings.Join(sandboxOuts, " "), nil
|
|
case "depfile":
|
|
referencedDepfile = true
|
|
if !Bool(g.properties.Depfile) {
|
|
return reportError("$(depfile) used without depfile property")
|
|
}
|
|
return "__SBOX_DEPFILE__", nil
|
|
case "genDir":
|
|
return cmd.PathForOutput(task.genDir), nil
|
|
default:
|
|
if strings.HasPrefix(name, "location ") {
|
|
label := strings.TrimSpace(strings.TrimPrefix(name, "location "))
|
|
if paths, ok := locationLabels[label]; ok {
|
|
if len(paths) == 0 {
|
|
return reportError("label %q has no files", label)
|
|
} else if len(paths) > 1 {
|
|
return reportError("label %q has multiple files, use $(locations %s) to reference it",
|
|
label, label)
|
|
}
|
|
return paths[0], nil
|
|
} else {
|
|
return reportError("unknown location label %q", label)
|
|
}
|
|
} else if strings.HasPrefix(name, "locations ") {
|
|
label := strings.TrimSpace(strings.TrimPrefix(name, "locations "))
|
|
if paths, ok := locationLabels[label]; ok {
|
|
if len(paths) == 0 {
|
|
return reportError("label %q has no files", label)
|
|
}
|
|
return strings.Join(paths, " "), nil
|
|
} else {
|
|
return reportError("unknown locations label %q", label)
|
|
}
|
|
} else {
|
|
return reportError("unknown variable '$(%s)'", name)
|
|
}
|
|
}
|
|
})
|
|
|
|
if err != nil {
|
|
ctx.PropertyErrorf("cmd", "%s", err.Error())
|
|
return
|
|
}
|
|
|
|
if Bool(g.properties.Depfile) && !referencedDepfile {
|
|
ctx.PropertyErrorf("cmd", "specified depfile=true but did not include a reference to '${depfile}' in cmd")
|
|
return
|
|
}
|
|
g.rawCommands = append(g.rawCommands, rawCommand)
|
|
|
|
cmd.Text(rawCommand)
|
|
cmd.ImplicitOutputs(task.out)
|
|
cmd.Implicits(task.in)
|
|
cmd.Implicits(g.deps)
|
|
cmd.Implicits(task.extraTools)
|
|
if Bool(g.properties.Depfile) {
|
|
cmd.ImplicitDepFile(task.depFile)
|
|
}
|
|
|
|
// Create the rule to run the genrule command inside sbox.
|
|
rule.Build(name, desc)
|
|
|
|
if len(task.copyTo) > 0 {
|
|
// If copyTo is set, multiple shards need to be copied into a single directory.
|
|
// task.out contains the per-shard paths, and copyTo contains the corresponding
|
|
// final path. The files need to be copied into the final directory by a
|
|
// single rule so it can remove the directory before it starts to ensure no
|
|
// old files remain. zipsync already does this, so build up zipArgs that
|
|
// zip all the per-shard directories into a single zip.
|
|
outputFiles = append(outputFiles, task.copyTo...)
|
|
copyFrom = append(copyFrom, task.out.Paths()...)
|
|
zipArgs.WriteString(" -C " + task.genDir.String())
|
|
zipArgs.WriteString(android.JoinWithPrefix(task.out.Strings(), " -f "))
|
|
} else {
|
|
outputFiles = append(outputFiles, task.out...)
|
|
}
|
|
}
|
|
|
|
if len(copyFrom) > 0 {
|
|
// Create a rule that zips all the per-shard directories into a single zip and then
|
|
// uses zipsync to unzip it into the final directory.
|
|
ctx.Build(pctx, android.BuildParams{
|
|
Rule: gensrcsMerge,
|
|
Implicits: copyFrom,
|
|
Outputs: outputFiles,
|
|
Description: "merge shards",
|
|
Args: map[string]string{
|
|
"zipArgs": zipArgs.String(),
|
|
"tmpZip": android.PathForModuleGen(ctx, g.subDir+".zip").String(),
|
|
"genDir": android.PathForModuleGen(ctx, g.subDir).String(),
|
|
},
|
|
})
|
|
}
|
|
|
|
g.outputFiles = outputFiles.Paths()
|
|
|
|
bazelModuleLabel := g.properties.Bazel_module.Label
|
|
bazelActionsUsed := false
|
|
if ctx.Config().BazelContext.BazelEnabled() && len(bazelModuleLabel) > 0 {
|
|
bazelActionsUsed = g.generateBazelBuildActions(ctx, bazelModuleLabel)
|
|
}
|
|
if !bazelActionsUsed {
|
|
// For <= 6 outputs, just embed those directly in the users. Right now, that covers >90% of
|
|
// the genrules on AOSP. That will make things simpler to look at the graph in the common
|
|
// case. For larger sets of outputs, inject a phony target in between to limit ninja file
|
|
// growth.
|
|
if len(g.outputFiles) <= 6 {
|
|
g.outputDeps = g.outputFiles
|
|
} else {
|
|
phonyFile := android.PathForModuleGen(ctx, "genrule-phony")
|
|
ctx.Build(pctx, android.BuildParams{
|
|
Rule: blueprint.Phony,
|
|
Output: phonyFile,
|
|
Inputs: g.outputFiles,
|
|
})
|
|
g.outputDeps = android.Paths{phonyFile}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Collect information for opening IDE project files in java/jdeps.go.
|
|
func (g *Module) IDEInfo(dpInfo *android.IdeInfo) {
|
|
dpInfo.Srcs = append(dpInfo.Srcs, g.Srcs().Strings()...)
|
|
for _, src := range g.properties.Srcs {
|
|
if strings.HasPrefix(src, ":") {
|
|
src = strings.Trim(src, ":")
|
|
dpInfo.Deps = append(dpInfo.Deps, src)
|
|
}
|
|
}
|
|
dpInfo.Paths = append(dpInfo.Paths, g.modulePaths...)
|
|
}
|
|
|
|
func (g *Module) AndroidMk() android.AndroidMkData {
|
|
return android.AndroidMkData{
|
|
Class: "ETC",
|
|
OutputFile: android.OptionalPathForPath(g.outputFiles[0]),
|
|
SubName: g.subName,
|
|
Extra: []android.AndroidMkExtraFunc{
|
|
func(w io.Writer, outputFile android.Path) {
|
|
fmt.Fprintln(w, "LOCAL_UNINSTALLABLE_MODULE := true")
|
|
},
|
|
},
|
|
Custom: func(w io.Writer, name, prefix, moduleDir string, data android.AndroidMkData) {
|
|
android.WriteAndroidMkData(w, data)
|
|
if data.SubName != "" {
|
|
fmt.Fprintln(w, ".PHONY:", name)
|
|
fmt.Fprintln(w, name, ":", name+g.subName)
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
func (g *Module) ShouldSupportSdkVersion(ctx android.BaseModuleContext,
|
|
sdkVersion android.ApiLevel) error {
|
|
// Because generated outputs are checked by client modules(e.g. cc_library, ...)
|
|
// we can safely ignore the check here.
|
|
return nil
|
|
}
|
|
|
|
func generatorFactory(taskGenerator taskFunc, props ...interface{}) *Module {
|
|
module := &Module{
|
|
taskGenerator: taskGenerator,
|
|
}
|
|
|
|
module.AddProperties(props...)
|
|
module.AddProperties(&module.properties)
|
|
|
|
module.ImageInterface = noopImageInterface{}
|
|
|
|
return module
|
|
}
|
|
|
|
type noopImageInterface struct{}
|
|
|
|
func (x noopImageInterface) ImageMutatorBegin(android.BaseModuleContext) {}
|
|
func (x noopImageInterface) CoreVariantNeeded(android.BaseModuleContext) bool { return false }
|
|
func (x noopImageInterface) RamdiskVariantNeeded(android.BaseModuleContext) bool { return false }
|
|
func (x noopImageInterface) VendorRamdiskVariantNeeded(android.BaseModuleContext) bool { return false }
|
|
func (x noopImageInterface) RecoveryVariantNeeded(android.BaseModuleContext) bool { return false }
|
|
func (x noopImageInterface) ExtraImageVariations(ctx android.BaseModuleContext) []string { return nil }
|
|
func (x noopImageInterface) SetImageVariation(ctx android.BaseModuleContext, variation string, module android.Module) {
|
|
}
|
|
|
|
func NewGenSrcs() *Module {
|
|
properties := &genSrcsProperties{}
|
|
|
|
// finalSubDir is the name of the subdirectory that output files will be generated into.
|
|
// It is used so that per-shard directories can be placed alongside it an then finally
|
|
// merged into it.
|
|
const finalSubDir = "gensrcs"
|
|
|
|
taskGenerator := func(ctx android.ModuleContext, rawCommand string, srcFiles android.Paths) []generateTask {
|
|
shardSize := defaultShardSize
|
|
if s := properties.Shard_size; s != nil {
|
|
shardSize = int(*s)
|
|
}
|
|
|
|
// gensrcs rules can easily hit command line limits by repeating the command for
|
|
// every input file. Shard the input files into groups.
|
|
shards := android.ShardPaths(srcFiles, shardSize)
|
|
var generateTasks []generateTask
|
|
|
|
for i, shard := range shards {
|
|
var commands []string
|
|
var outFiles android.WritablePaths
|
|
var commandDepFiles []string
|
|
var copyTo android.WritablePaths
|
|
|
|
// When sharding is enabled (i.e. len(shards) > 1), the sbox rules for each
|
|
// shard will be write to their own directories and then be merged together
|
|
// into finalSubDir. If sharding is not enabled (i.e. len(shards) == 1),
|
|
// the sbox rule will write directly to finalSubDir.
|
|
genSubDir := finalSubDir
|
|
if len(shards) > 1 {
|
|
genSubDir = strconv.Itoa(i)
|
|
}
|
|
|
|
genDir := android.PathForModuleGen(ctx, genSubDir)
|
|
// TODO(ccross): this RuleBuilder is a hack to be able to call
|
|
// rule.Command().PathForOutput. Replace this with passing the rule into the
|
|
// generator.
|
|
rule := android.NewRuleBuilder(pctx, ctx).Sbox(genDir, nil)
|
|
|
|
for _, in := range shard {
|
|
outFile := android.GenPathWithExt(ctx, finalSubDir, in, String(properties.Output_extension))
|
|
|
|
// If sharding is enabled, then outFile is the path to the output file in
|
|
// the shard directory, and copyTo is the path to the output file in the
|
|
// final directory.
|
|
if len(shards) > 1 {
|
|
shardFile := android.GenPathWithExt(ctx, genSubDir, in, String(properties.Output_extension))
|
|
copyTo = append(copyTo, outFile)
|
|
outFile = shardFile
|
|
}
|
|
|
|
outFiles = append(outFiles, outFile)
|
|
|
|
// pre-expand the command line to replace $in and $out with references to
|
|
// a single input and output file.
|
|
command, err := android.Expand(rawCommand, func(name string) (string, error) {
|
|
switch name {
|
|
case "in":
|
|
return in.String(), nil
|
|
case "out":
|
|
return rule.Command().PathForOutput(outFile), nil
|
|
case "depfile":
|
|
// Generate a depfile for each output file. Store the list for
|
|
// later in order to combine them all into a single depfile.
|
|
depFile := rule.Command().PathForOutput(outFile.ReplaceExtension(ctx, "d"))
|
|
commandDepFiles = append(commandDepFiles, depFile)
|
|
return depFile, nil
|
|
default:
|
|
return "$(" + name + ")", nil
|
|
}
|
|
})
|
|
if err != nil {
|
|
ctx.PropertyErrorf("cmd", err.Error())
|
|
}
|
|
|
|
// escape the command in case for example it contains '#', an odd number of '"', etc
|
|
command = fmt.Sprintf("bash -c %v", proptools.ShellEscape(command))
|
|
commands = append(commands, command)
|
|
}
|
|
fullCommand := strings.Join(commands, " && ")
|
|
|
|
var outputDepfile android.WritablePath
|
|
var extraTools android.Paths
|
|
if len(commandDepFiles) > 0 {
|
|
// Each command wrote to a depfile, but ninja can only handle one
|
|
// depfile per rule. Use the dep_fixer tool at the end of the
|
|
// command to combine all the depfiles into a single output depfile.
|
|
outputDepfile = android.PathForModuleGen(ctx, genSubDir, "gensrcs.d")
|
|
depFixerTool := ctx.Config().HostToolPath(ctx, "dep_fixer")
|
|
fullCommand += fmt.Sprintf(" && %s -o $(depfile) %s",
|
|
depFixerTool.String(), strings.Join(commandDepFiles, " "))
|
|
extraTools = append(extraTools, depFixerTool)
|
|
}
|
|
|
|
generateTasks = append(generateTasks, generateTask{
|
|
in: shard,
|
|
out: outFiles,
|
|
depFile: outputDepfile,
|
|
copyTo: copyTo,
|
|
genDir: genDir,
|
|
cmd: fullCommand,
|
|
shard: i,
|
|
shards: len(shards),
|
|
extraTools: extraTools,
|
|
})
|
|
}
|
|
|
|
return generateTasks
|
|
}
|
|
|
|
g := generatorFactory(taskGenerator, properties)
|
|
g.subDir = finalSubDir
|
|
return g
|
|
}
|
|
|
|
func GenSrcsFactory() android.Module {
|
|
m := NewGenSrcs()
|
|
android.InitAndroidModule(m)
|
|
return m
|
|
}
|
|
|
|
type genSrcsProperties struct {
|
|
// extension that will be substituted for each output file
|
|
Output_extension *string
|
|
|
|
// maximum number of files that will be passed on a single command line.
|
|
Shard_size *int64
|
|
}
|
|
|
|
const defaultShardSize = 50
|
|
|
|
func NewGenRule() *Module {
|
|
properties := &genRuleProperties{}
|
|
|
|
taskGenerator := func(ctx android.ModuleContext, rawCommand string, srcFiles android.Paths) []generateTask {
|
|
outs := make(android.WritablePaths, len(properties.Out))
|
|
var depFile android.WritablePath
|
|
for i, out := range properties.Out {
|
|
outPath := android.PathForModuleGen(ctx, out)
|
|
if i == 0 {
|
|
depFile = outPath.ReplaceExtension(ctx, "d")
|
|
}
|
|
outs[i] = outPath
|
|
}
|
|
return []generateTask{{
|
|
in: srcFiles,
|
|
out: outs,
|
|
depFile: depFile,
|
|
genDir: android.PathForModuleGen(ctx),
|
|
cmd: rawCommand,
|
|
}}
|
|
}
|
|
|
|
return generatorFactory(taskGenerator, properties)
|
|
}
|
|
|
|
func GenRuleFactory() android.Module {
|
|
m := NewGenRule()
|
|
android.InitAndroidModule(m)
|
|
android.InitDefaultableModule(m)
|
|
return m
|
|
}
|
|
|
|
type genRuleProperties struct {
|
|
// names of the output files that will be generated
|
|
Out []string `android:"arch_variant"`
|
|
}
|
|
|
|
var Bool = proptools.Bool
|
|
var String = proptools.String
|
|
|
|
//
|
|
// Defaults
|
|
//
|
|
type Defaults struct {
|
|
android.ModuleBase
|
|
android.DefaultsModuleBase
|
|
}
|
|
|
|
func defaultsFactory() android.Module {
|
|
return DefaultsFactory()
|
|
}
|
|
|
|
func DefaultsFactory(props ...interface{}) android.Module {
|
|
module := &Defaults{}
|
|
|
|
module.AddProperties(props...)
|
|
module.AddProperties(
|
|
&generatorProperties{},
|
|
&genRuleProperties{},
|
|
)
|
|
|
|
android.InitDefaultsModule(module)
|
|
|
|
return module
|
|
}
|