1// Copyright 2019 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
15package java
16
17import (
18	"github.com/google/blueprint"
19
20	"android/soong/android"
21)
22
23var (
24	hiddenAPIGenerateCSVRule = pctx.AndroidStaticRule("hiddenAPIGenerateCSV", blueprint.RuleParams{
25		Command:     "${config.Class2NonSdkList} --stub-api-flags ${stubAPIFlags} $in $outFlag $out",
26		CommandDeps: []string{"${config.Class2NonSdkList}"},
27	}, "outFlag", "stubAPIFlags")
28
29	hiddenAPIGenerateIndexRule = pctx.AndroidStaticRule("hiddenAPIGenerateIndex", blueprint.RuleParams{
30		Command:     "${config.MergeCsvCommand} --zip_input --key_field signature --output=$out $in",
31		CommandDeps: []string{"${config.MergeCsvCommand}"},
32	})
33)
34
35type hiddenAPI struct {
36	// True if the module containing this structure contributes to the hiddenapi information or has
37	// that information encoded within it.
38	active bool
39
40	// The path to the dex jar that is in the boot class path. If this is unset then the associated
41	// module is not a boot jar, but could be one of the <x>-hiddenapi modules that provide additional
42	// annotations for the <x> boot dex jar but which do not actually provide a boot dex jar
43	// themselves.
44	//
45	// This must be the path to the unencoded dex jar as the encoded dex jar indirectly depends on
46	// this file so using the encoded dex jar here would result in a cycle in the ninja rules.
47	bootDexJarPath    OptionalDexJarPath
48	bootDexJarPathErr error
49
50	// The paths to the classes jars that contain classes and class members annotated with
51	// the UnsupportedAppUsage annotation that need to be extracted as part of the hidden API
52	// processing.
53	classesJarPaths android.Paths
54
55	// The compressed state of the dex file being encoded. This is used to ensure that the encoded
56	// dex file has the same state.
57	uncompressDexState *bool
58}
59
60func (h *hiddenAPI) bootDexJar(ctx android.ModuleErrorfContext) OptionalDexJarPath {
61	if h.bootDexJarPathErr != nil {
62		ctx.ModuleErrorf(h.bootDexJarPathErr.Error())
63	}
64	return h.bootDexJarPath
65}
66
67func (h *hiddenAPI) classesJars() android.Paths {
68	return h.classesJarPaths
69}
70
71func (h *hiddenAPI) uncompressDex() *bool {
72	return h.uncompressDexState
73}
74
75// hiddenAPIModule is the interface a module that embeds the hiddenAPI structure must implement.
76type hiddenAPIModule interface {
77	android.Module
78	hiddenAPIIntf
79
80	MinSdkVersion(ctx android.EarlyModuleContext) android.ApiLevel
81}
82
83type hiddenAPIIntf interface {
84	bootDexJar(ctx android.ModuleErrorfContext) OptionalDexJarPath
85	classesJars() android.Paths
86	uncompressDex() *bool
87}
88
89var _ hiddenAPIIntf = (*hiddenAPI)(nil)
90
91// Initialize the hiddenapi structure
92//
93// uncompressedDexState should be nil when the module is a prebuilt and so does not require hidden
94// API encoding.
95func (h *hiddenAPI) initHiddenAPI(ctx android.ModuleContext, dexJar OptionalDexJarPath, classesJar android.Path, uncompressedDexState *bool) {
96
97	// Save the classes jars even if this is not active as they may be used by modular hidden API
98	// processing.
99	classesJars := android.Paths{classesJar}
100	ctx.VisitDirectDepsWithTag(hiddenApiAnnotationsTag, func(dep android.Module) {
101		javaInfo, _ := android.OtherModuleProvider(ctx, dep, JavaInfoProvider)
102		classesJars = append(classesJars, javaInfo.ImplementationJars...)
103	})
104	h.classesJarPaths = classesJars
105
106	// Save the unencoded dex jar so it can be used when generating the
107	// hiddenAPISingletonPathsStruct.stubFlags file.
108	h.bootDexJarPath = dexJar
109
110	h.uncompressDexState = uncompressedDexState
111
112	// If hiddenapi processing is disabled treat this as inactive.
113	if ctx.Config().DisableHiddenApiChecks() {
114		return
115	}
116
117	// The context module must implement hiddenAPIModule.
118	module := ctx.Module().(hiddenAPIModule)
119
120	// If the frameworks/base directories does not exist and no prebuilt hidden API flag files have
121	// been configured then it is not possible to do hidden API encoding.
122	if !ctx.Config().FrameworksBaseDirExists(ctx) && ctx.Config().PrebuiltHiddenApiDir(ctx) == "" {
123		return
124	}
125
126	// It is important that hiddenapi information is only gathered for/from modules that are actually
127	// on the boot jars list because the runtime only enforces access to the hidden API for the
128	// bootclassloader. If information is gathered for modules not on the list then that will cause
129	// failures in the CtsHiddenApiBlocklist... tests.
130	h.active = isModuleInBootClassPath(ctx, module)
131}
132
133// Store any error encountered during the initialization of hiddenapi structure (e.g. unflagged co-existing prebuilt apexes)
134func (h *hiddenAPI) initHiddenAPIError(err error) {
135	h.bootDexJarPathErr = err
136}
137
138func isModuleInBootClassPath(ctx android.BaseModuleContext, module android.Module) bool {
139	// Get the configured platform and apex boot jars.
140	nonApexBootJars := ctx.Config().NonApexBootJars()
141	apexBootJars := ctx.Config().ApexBootJars()
142	active := isModuleInConfiguredList(ctx, module, nonApexBootJars) ||
143		isModuleInConfiguredList(ctx, module, apexBootJars)
144	return active
145}
146
147// hiddenAPIEncodeDex is called by any module that needs to encode dex files.
148//
149// It ignores any module that has not had initHiddenApi() called on it and which is not in the boot
150// jar list. In that case it simply returns the supplied dex jar path.
151//
152// Otherwise, it creates a copy of the supplied dex file into which it has encoded the hiddenapi
153// flags and returns this instead of the supplied dex jar.
154func (h *hiddenAPI) hiddenAPIEncodeDex(ctx android.ModuleContext, dexJar android.OutputPath) android.OutputPath {
155
156	if !h.active {
157		return dexJar
158	}
159
160	// A nil uncompressDexState prevents the dex file from being encoded.
161	if h.uncompressDexState == nil {
162		ctx.ModuleErrorf("cannot encode dex file %s when uncompressDexState is nil", dexJar)
163	}
164	uncompressDex := *h.uncompressDexState
165
166	// Create a copy of the dex jar which has been encoded with hiddenapi flags.
167	flagsCSV := hiddenAPISingletonPaths(ctx).flags
168	outputDir := android.PathForModuleOut(ctx, "hiddenapi").OutputPath
169	encodedDex := hiddenAPIEncodeDex(ctx, dexJar, flagsCSV, uncompressDex, android.NoneApiLevel, outputDir)
170
171	// Use the encoded dex jar from here onwards.
172	return encodedDex
173}
174
175// buildRuleToGenerateAnnotationFlags builds a ninja rule to generate the annotation-flags.csv file
176// from the classes jars and stub-flags.csv files.
177//
178// The annotation-flags.csv file contains mappings from Java signature to various flags derived from
179// annotations in the source, e.g. whether it is public or the sdk version above which it can no
180// longer be used.
181//
182// It is created by the Class2NonSdkList tool which processes the .class files in the class
183// implementation jar looking for UnsupportedAppUsage and CovariantReturnType annotations. The
184// tool also consumes the hiddenAPISingletonPathsStruct.stubFlags file in order to perform
185// consistency checks on the information in the annotations and to filter out bridge methods
186// that are already part of the public API.
187func buildRuleToGenerateAnnotationFlags(ctx android.ModuleContext, desc string, classesJars android.Paths, stubFlagsCSV android.Path, outputPath android.WritablePath) {
188	ctx.Build(pctx, android.BuildParams{
189		Rule:        hiddenAPIGenerateCSVRule,
190		Description: desc,
191		Inputs:      classesJars,
192		Output:      outputPath,
193		Implicit:    stubFlagsCSV,
194		Args: map[string]string{
195			"outFlag":      "--write-flags-csv",
196			"stubAPIFlags": stubFlagsCSV.String(),
197		},
198	})
199}
200
201// buildRuleToGenerateMetadata builds a ninja rule to generate the metadata.csv file from
202// the classes jars and stub-flags.csv files.
203//
204// The metadata.csv file contains mappings from Java signature to the value of properties specified
205// on UnsupportedAppUsage annotations in the source.
206//
207// Like the annotation-flags.csv file this is also created by the Class2NonSdkList in the same way.
208// Although the two files could potentially be created in a single invocation of the
209// Class2NonSdkList at the moment they are created using their own invocation, with the behavior
210// being determined by the property that is used.
211func buildRuleToGenerateMetadata(ctx android.ModuleContext, desc string, classesJars android.Paths, stubFlagsCSV android.Path, metadataCSV android.WritablePath) {
212	ctx.Build(pctx, android.BuildParams{
213		Rule:        hiddenAPIGenerateCSVRule,
214		Description: desc,
215		Inputs:      classesJars,
216		Output:      metadataCSV,
217		Implicit:    stubFlagsCSV,
218		Args: map[string]string{
219			"outFlag":      "--write-metadata-csv",
220			"stubAPIFlags": stubFlagsCSV.String(),
221		},
222	})
223}
224
225// buildRuleToGenerateIndex builds a ninja rule to generate the index.csv file from the classes
226// jars.
227//
228// The index.csv file contains mappings from Java signature to source location information.
229//
230// It is created by the merge_csv tool which processes the class implementation jar, extracting
231// all the files ending in .uau (which are CSV files) and merges them together. The .uau files are
232// created by the unsupported app usage annotation processor during compilation of the class
233// implementation jar.
234func buildRuleToGenerateIndex(ctx android.ModuleContext, desc string, classesJars android.Paths, indexCSV android.WritablePath) {
235	ctx.Build(pctx, android.BuildParams{
236		Rule:        hiddenAPIGenerateIndexRule,
237		Description: desc,
238		Inputs:      classesJars,
239		Output:      indexCSV,
240	})
241}
242
243var hiddenAPIEncodeDexRule = pctx.AndroidStaticRule("hiddenAPIEncodeDex", blueprint.RuleParams{
244	Command: `rm -rf $tmpDir && mkdir -p $tmpDir && mkdir $tmpDir/dex-input && mkdir $tmpDir/dex-output &&
245		unzip -qoDD $in 'classes*.dex' -d $tmpDir/dex-input &&
246		for INPUT_DEX in $$(find $tmpDir/dex-input -maxdepth 1 -name 'classes*.dex' | sort); do
247		  echo "--input-dex=$${INPUT_DEX}";
248		  echo "--output-dex=$tmpDir/dex-output/$$(basename $${INPUT_DEX})";
249		done | xargs ${config.HiddenAPI} encode --api-flags=$flagsCsv $hiddenapiFlags &&
250		${config.SoongZipCmd} $soongZipFlags -o $tmpDir/dex.jar -C $tmpDir/dex-output -f "$tmpDir/dex-output/classes*.dex" &&
251		${config.MergeZipsCmd} -j -D -zipToNotStrip $tmpDir/dex.jar -stripFile "classes*.dex" -stripFile "**/*.uau" $out $tmpDir/dex.jar $in`,
252	CommandDeps: []string{
253		"${config.HiddenAPI}",
254		"${config.SoongZipCmd}",
255		"${config.MergeZipsCmd}",
256	},
257}, "flagsCsv", "hiddenapiFlags", "tmpDir", "soongZipFlags")
258
259// hiddenAPIEncodeDex generates the build rule that will encode the supplied dex jar and place the
260// encoded dex jar in a file of the same name in the output directory.
261//
262// The encode dex rule requires unzipping, encoding and rezipping the classes.dex files along with
263// all the resources from the input jar. It also ensures that if it was uncompressed in the input
264// it stays uncompressed in the output.
265func hiddenAPIEncodeDex(ctx android.ModuleContext, dexInput, flagsCSV android.Path, uncompressDex bool, minSdkVersion android.ApiLevel, outputDir android.OutputPath) android.OutputPath {
266
267	// The output file has the same name as the input file and is in the output directory.
268	output := outputDir.Join(ctx, dexInput.Base())
269
270	// Create a jar specific temporary directory in which to do the work just in case this is called
271	// with the same output directory for multiple modules.
272	tmpDir := outputDir.Join(ctx, dexInput.Base()+"-tmp")
273
274	// If the input is uncompressed then generate the output of the encode rule to an intermediate
275	// file as the final output will need further processing after encoding.
276	soongZipFlags := ""
277	encodeRuleOutput := output
278	if uncompressDex {
279		soongZipFlags = "-L 0"
280		encodeRuleOutput = outputDir.Join(ctx, "unaligned", dexInput.Base())
281	}
282
283	// b/149353192: when a module is instrumented, jacoco adds synthetic members
284	// $jacocoData and $jacocoInit. Since they don't exist when building the hidden API flags,
285	// don't complain when we don't find hidden API flags for the synthetic members.
286	hiddenapiFlags := ""
287	if j, ok := ctx.Module().(interface {
288		shouldInstrument(android.BaseModuleContext) bool
289	}); ok && j.shouldInstrument(ctx) {
290		hiddenapiFlags = "--no-force-assign-all"
291	}
292
293	// If the library is targeted for Q and/or R then make sure that they do not
294	// have any S+ flags encoded as that will break the runtime.
295	minApiLevel := minSdkVersion
296	if !minApiLevel.IsNone() {
297		if minApiLevel.LessThanOrEqualTo(android.ApiLevelOrPanic(ctx, "R")) {
298			hiddenapiFlags = hiddenapiFlags + " --max-hiddenapi-level=max-target-r"
299		}
300	}
301
302	ctx.Build(pctx, android.BuildParams{
303		Rule:        hiddenAPIEncodeDexRule,
304		Description: "hiddenapi encode dex",
305		Input:       dexInput,
306		Output:      encodeRuleOutput,
307		Implicit:    flagsCSV,
308		Args: map[string]string{
309			"flagsCsv":       flagsCSV.String(),
310			"tmpDir":         tmpDir.String(),
311			"soongZipFlags":  soongZipFlags,
312			"hiddenapiFlags": hiddenapiFlags,
313		},
314	})
315
316	if uncompressDex {
317		TransformZipAlign(ctx, output, encodeRuleOutput, nil)
318	}
319
320	return output
321}
322
323type hiddenApiAnnotationsDependencyTag struct {
324	blueprint.BaseDependencyTag
325	android.LicenseAnnotationSharedDependencyTag
326}
327
328// Tag used to mark dependencies on java_library instances that contains Java source files whose
329// sole purpose is to provide additional hiddenapi annotations.
330var hiddenApiAnnotationsTag hiddenApiAnnotationsDependencyTag
331
332// Mark this tag so dependencies that use it are excluded from APEX contents.
333func (t hiddenApiAnnotationsDependencyTag) ExcludeFromApexContents() {}
334
335var _ android.ExcludeFromApexContentsTag = hiddenApiAnnotationsTag
336