1// Copyright 2020 The Android Open Source Project 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 rust 16 17import ( 18 "github.com/google/blueprint" 19 20 "android/soong/cc" 21) 22 23var CovLibraryName = "libprofile-clang-extras" 24var ProfilerBuiltins = "libprofiler_builtins.rust_sysroot" 25 26// Add '%c' to default specifier after we resolve http://b/210012154 27const profileInstrFlag = "-fprofile-instr-generate=/data/misc/trace/clang-%p-%m.profraw" 28 29type coverage struct { 30 Properties cc.CoverageProperties 31 32 // Whether binaries containing this module need --coverage added to their ldflags 33 linkCoverage bool 34} 35 36func (cov *coverage) props() []interface{} { 37 return []interface{}{&cov.Properties} 38} 39 40func (cov *coverage) deps(ctx DepsContext, deps Deps) Deps { 41 if cov.Properties.NeedCoverageVariant { 42 if ctx.Device() { 43 ctx.AddVariationDependencies([]blueprint.Variation{ 44 {Mutator: "link", Variation: "static"}, 45 }, cc.CoverageDepTag, CovLibraryName) 46 } 47 48 // no_std modules are missing libprofiler_builtins which provides coverage, so we need to add it as a dependency. 49 if rustModule, ok := ctx.Module().(*Module); ok && rustModule.compiler.noStdlibs() { 50 ctx.AddVariationDependencies([]blueprint.Variation{{Mutator: "rust_libraries", Variation: "rlib"}}, rlibDepTag, ProfilerBuiltins) 51 } 52 } 53 54 return deps 55} 56 57func (cov *coverage) flags(ctx ModuleContext, flags Flags, deps PathDeps) (Flags, PathDeps) { 58 59 if !ctx.DeviceConfig().NativeCoverageEnabled() { 60 return flags, deps 61 } 62 63 if cov.Properties.CoverageEnabled { 64 flags.Coverage = true 65 flags.RustFlags = append(flags.RustFlags, 66 "-C instrument-coverage", "-g") 67 if ctx.Device() { 68 coverage := ctx.GetDirectDepWithTag(CovLibraryName, cc.CoverageDepTag).(cc.LinkableInterface) 69 flags.LinkFlags = append(flags.LinkFlags, 70 profileInstrFlag, "-g", coverage.OutputFile().Path().String(), "-Wl,--wrap,open") 71 deps.StaticLibs = append(deps.StaticLibs, coverage.OutputFile().Path()) 72 } 73 74 // no_std modules are missing libprofiler_builtins which provides coverage, so we need to add it as a dependency. 75 if rustModule, ok := ctx.Module().(*Module); ok && rustModule.compiler.noStdlibs() { 76 profiler_builtins := ctx.GetDirectDepWithTag(ProfilerBuiltins, rlibDepTag).(*Module) 77 deps.RLibs = append(deps.RLibs, RustLibrary{Path: profiler_builtins.OutputFile().Path(), CrateName: profiler_builtins.CrateName()}) 78 } 79 80 if cc.EnableContinuousCoverage(ctx) { 81 flags.RustFlags = append(flags.RustFlags, "-C llvm-args=--runtime-counter-relocation") 82 flags.LinkFlags = append(flags.LinkFlags, "-Wl,-mllvm,-runtime-counter-relocation") 83 } 84 } 85 86 return flags, deps 87} 88 89func (cov *coverage) begin(ctx BaseModuleContext) { 90 // Update useSdk and sdkVersion args if Rust modules become SDK aware. 91 cov.Properties = cc.SetCoverageProperties(ctx, cov.Properties, ctx.RustModule().nativeCoverage(), false, "") 92} 93