1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright 2021 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.
// Package bloaty implements a singleton that measures binary (e.g. ELF
// executable, shared library or Rust rlib) section sizes at build time.
package bloaty
import (
"android/soong/android"
"github.com/google/blueprint"
)
const bloatyDescriptorExt = ".bloaty.csv"
const protoFilename = "binary_sizes.pb.gz"
var (
fileSizeMeasurerKey blueprint.ProviderKey
pctx = android.NewPackageContext("android/soong/bloaty")
// bloaty is used to measure a binary section sizes.
bloaty = pctx.AndroidStaticRule("bloaty",
blueprint.RuleParams{
Command: "${bloaty} -n 0 --csv ${in} > ${out}",
CommandDeps: []string{"${bloaty}"},
})
// The bloaty merger script is used to combine the outputs from bloaty
// into a single protobuf.
bloatyMerger = pctx.AndroidStaticRule("bloatyMerger",
blueprint.RuleParams{
Command: "${bloatyMerger} ${out}.lst ${out}",
CommandDeps: []string{"${bloatyMerger}"},
Rspfile: "${out}.lst",
RspfileContent: "${in}",
})
)
func init() {
pctx.VariableConfigMethod("hostPrebuiltTag", android.Config.PrebuiltOS)
pctx.SourcePathVariable("bloaty", "prebuilts/build-tools/${hostPrebuiltTag}/bin/bloaty")
pctx.HostBinToolVariable("bloatyMerger", "bloaty_merger")
android.RegisterSingletonType("file_metrics", fileSizesSingleton)
fileSizeMeasurerKey = blueprint.NewProvider(measuredFiles{})
}
// measuredFiles contains the paths of the files measured by a module.
type measuredFiles struct {
paths []android.WritablePath
}
// MeasureSizeForPaths should be called by binary producers to measure the
// sizes of artifacts. It must only be called once per module; it will panic
// otherwise.
func MeasureSizeForPaths(ctx android.ModuleContext, paths ...android.OptionalPath) {
mf := measuredFiles{}
for _, p := range paths {
if !p.Valid() {
continue
}
if p, ok := p.Path().(android.WritablePath); ok {
mf.paths = append(mf.paths, p)
}
}
ctx.SetProvider(fileSizeMeasurerKey, mf)
}
type sizesSingleton struct{}
func fileSizesSingleton() android.Singleton {
return &sizesSingleton{}
}
func (singleton *sizesSingleton) GenerateBuildActions(ctx android.SingletonContext) {
var deps android.Paths
ctx.VisitAllModules(func(m android.Module) {
if !ctx.ModuleHasProvider(m, fileSizeMeasurerKey) {
return
}
filePaths := ctx.ModuleProvider(m, fileSizeMeasurerKey).(measuredFiles)
for _, path := range filePaths.paths {
filePath := path.(android.ModuleOutPath)
sizeFile := filePath.InSameDir(ctx, filePath.Base()+bloatyDescriptorExt)
ctx.Build(pctx, android.BuildParams{
Rule: bloaty,
Description: "bloaty " + filePath.Rel(),
Input: filePath,
Output: sizeFile,
})
deps = append(deps, sizeFile)
}
})
ctx.Build(pctx, android.BuildParams{
Rule: bloatyMerger,
Inputs: android.SortedUniquePaths(deps),
Output: android.PathForOutput(ctx, protoFilename),
})
}
func (singleton *sizesSingleton) MakeVars(ctx android.MakeVarsContext) {
ctx.DistForGoalWithFilename("checkbuild", android.PathForOutput(ctx, protoFilename), protoFilename)
}
|
