Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 1 | // Copyright 2017 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 | |
| 15 | package android |
| 16 | |
| 17 | import ( |
| 18 | "path/filepath" |
| 19 | "reflect" |
Anton Hansson | 4537640 | 2020-04-09 14:18:21 +0100 | [diff] [blame] | 20 | "regexp" |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 21 | "strconv" |
| 22 | "strings" |
| 23 | |
| 24 | "github.com/google/blueprint/proptools" |
| 25 | ) |
| 26 | |
| 27 | // "neverallow" rules for the build system. |
| 28 | // |
| 29 | // This allows things which aren't related to the build system and are enforced |
Joe Onorato | b4638c1 | 2021-10-27 15:47:06 -0700 | [diff] [blame] | 30 | // against assumptions, in progress code refactors, or policy to be expressed in a |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 31 | // straightforward away disjoint from implementations and tests which should |
| 32 | // work regardless of these restrictions. |
| 33 | // |
| 34 | // A module is disallowed if all of the following are true: |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 35 | // - it is in one of the "In" paths |
| 36 | // - it is not in one of the "NotIn" paths |
| 37 | // - it has all "With" properties matched |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 38 | // - - values are matched in their entirety |
| 39 | // - - nil is interpreted as an empty string |
| 40 | // - - nested properties are separated with a '.' |
| 41 | // - - if the property is a list, any of the values in the list being matches |
| 42 | // counts as a match |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 43 | // - it has none of the "Without" properties matched (same rules as above) |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 44 | |
Paul Duffin | 45338f0 | 2021-03-30 23:07:52 +0100 | [diff] [blame] | 45 | func registerNeverallowMutator(ctx RegisterMutatorsContext) { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 46 | ctx.BottomUp("neverallow", neverallowMutator).Parallel() |
| 47 | } |
| 48 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 49 | var neverallows = []Rule{} |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 50 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 51 | func init() { |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 52 | AddNeverAllowRules(createIncludeDirsRules()...) |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 53 | AddNeverAllowRules(createTrebleRules()...) |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 54 | AddNeverAllowRules(createJavaDeviceForHostRules()...) |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 55 | AddNeverAllowRules(createCcSdkVariantRules()...) |
David Srbecky | e033cba | 2020-05-20 22:20:28 +0100 | [diff] [blame] | 56 | AddNeverAllowRules(createUncompressDexRules()...) |
Yifan Hong | 696ed4d | 2020-07-27 12:59:58 -0700 | [diff] [blame] | 57 | AddNeverAllowRules(createMakefileGoalRules()...) |
Inseob Kim | 800d114 | 2021-06-14 12:03:51 +0900 | [diff] [blame] | 58 | AddNeverAllowRules(createInitFirstStageRules()...) |
Neil Fuller | df5f356 | 2018-10-21 17:19:10 +0100 | [diff] [blame] | 59 | } |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 60 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 61 | // Add a NeverAllow rule to the set of rules to apply. |
| 62 | func AddNeverAllowRules(rules ...Rule) { |
| 63 | neverallows = append(neverallows, rules...) |
| 64 | } |
| 65 | |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 66 | func createIncludeDirsRules() []Rule { |
Steven Moreland | 8fc8dbf | 2021-04-27 02:31:07 +0000 | [diff] [blame] | 67 | notInIncludeDir := []string{ |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 68 | "art", |
Orion Hodson | 6341f01 | 2019-11-06 13:39:46 +0000 | [diff] [blame] | 69 | "art/libnativebridge", |
| 70 | "art/libnativeloader", |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 71 | "libcore", |
| 72 | "libnativehelper", |
| 73 | "external/apache-harmony", |
| 74 | "external/apache-xml", |
| 75 | "external/boringssl", |
| 76 | "external/bouncycastle", |
| 77 | "external/conscrypt", |
| 78 | "external/icu", |
| 79 | "external/okhttp", |
| 80 | "external/vixl", |
| 81 | "external/wycheproof", |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 82 | } |
Steven Moreland | 8fc8dbf | 2021-04-27 02:31:07 +0000 | [diff] [blame] | 83 | noUseIncludeDir := []string{ |
Steven Moreland | f36a3ac | 2021-04-27 18:03:14 +0000 | [diff] [blame] | 84 | "frameworks/av/apex", |
| 85 | "frameworks/av/tools", |
| 86 | "frameworks/native/cmds", |
| 87 | "system/apex", |
| 88 | "system/bpf", |
| 89 | "system/gatekeeper", |
| 90 | "system/hwservicemanager", |
| 91 | "system/libbase", |
Steven Moreland | 8fc8dbf | 2021-04-27 02:31:07 +0000 | [diff] [blame] | 92 | "system/libfmq", |
Steven Moreland | f36a3ac | 2021-04-27 18:03:14 +0000 | [diff] [blame] | 93 | "system/libvintf", |
Steven Moreland | 8fc8dbf | 2021-04-27 02:31:07 +0000 | [diff] [blame] | 94 | } |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 95 | |
Steven Moreland | 8fc8dbf | 2021-04-27 02:31:07 +0000 | [diff] [blame] | 96 | rules := make([]Rule, 0, len(notInIncludeDir)+len(noUseIncludeDir)) |
| 97 | |
| 98 | for _, path := range notInIncludeDir { |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 99 | rule := |
| 100 | NeverAllow(). |
| 101 | WithMatcher("include_dirs", StartsWith(path+"/")). |
| 102 | Because("include_dirs is deprecated, all usages of '" + path + "' have been migrated" + |
| 103 | " to use alternate mechanisms and so can no longer be used.") |
| 104 | |
| 105 | rules = append(rules, rule) |
| 106 | } |
| 107 | |
Steven Moreland | 8fc8dbf | 2021-04-27 02:31:07 +0000 | [diff] [blame] | 108 | for _, path := range noUseIncludeDir { |
| 109 | rule := NeverAllow().In(path+"/").WithMatcher("include_dirs", isSetMatcherInstance). |
| 110 | Because("include_dirs is deprecated, all usages of them in '" + path + "' have been migrated" + |
| 111 | " to use alternate mechanisms and so can no longer be used.") |
| 112 | rules = append(rules, rule) |
| 113 | } |
| 114 | |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 115 | return rules |
| 116 | } |
| 117 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 118 | func createTrebleRules() []Rule { |
| 119 | return []Rule{ |
| 120 | NeverAllow(). |
| 121 | In("vendor", "device"). |
| 122 | With("vndk.enabled", "true"). |
| 123 | Without("vendor", "true"). |
Justin Yun | 0ecf0b2 | 2020-02-28 15:07:59 +0900 | [diff] [blame] | 124 | Without("product_specific", "true"). |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 125 | Because("the VNDK can never contain a library that is device dependent."), |
| 126 | NeverAllow(). |
| 127 | With("vndk.enabled", "true"). |
| 128 | Without("vendor", "true"). |
| 129 | Without("owner", ""). |
| 130 | Because("a VNDK module can never have an owner."), |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 131 | |
Neil Fuller | df5f356 | 2018-10-21 17:19:10 +0100 | [diff] [blame] | 132 | // TODO(b/67974785): always enforce the manifest |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 133 | NeverAllow(). |
Steven Moreland | 51ce4f6 | 2020-02-10 17:21:32 -0800 | [diff] [blame] | 134 | Without("name", "libhidlbase-combined-impl"). |
| 135 | Without("name", "libhidlbase"). |
| 136 | Without("name", "libhidlbase_pgo"). |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 137 | With("product_variables.enforce_vintf_manifest.cflags", "*"). |
| 138 | Because("manifest enforcement should be independent of ."), |
Neil Fuller | df5f356 | 2018-10-21 17:19:10 +0100 | [diff] [blame] | 139 | |
| 140 | // TODO(b/67975799): vendor code should always use /vendor/bin/sh |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 141 | NeverAllow(). |
| 142 | Without("name", "libc_bionic_ndk"). |
| 143 | With("product_variables.treble_linker_namespaces.cflags", "*"). |
| 144 | Because("nothing should care if linker namespaces are enabled or not"), |
Neil Fuller | df5f356 | 2018-10-21 17:19:10 +0100 | [diff] [blame] | 145 | |
| 146 | // Example: |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 147 | // *NeverAllow().with("Srcs", "main.cpp")) |
Neil Fuller | df5f356 | 2018-10-21 17:19:10 +0100 | [diff] [blame] | 148 | } |
| 149 | } |
| 150 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 151 | func createJavaDeviceForHostRules() []Rule { |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 152 | javaDeviceForHostProjectsAllowedList := []string{ |
Dan Willemsen | 9fe1410 | 2021-07-13 21:52:04 -0700 | [diff] [blame] | 153 | "development/build", |
Colin Cross | b5191a5 | 2019-04-11 14:07:38 -0700 | [diff] [blame] | 154 | "external/guava", |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 155 | "external/robolectric-shadows", |
Jerome Gaillard | 655ee02 | 2021-09-23 11:38:08 +0000 | [diff] [blame] | 156 | "frameworks/layoutlib", |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 157 | } |
| 158 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 159 | return []Rule{ |
| 160 | NeverAllow(). |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 161 | NotIn(javaDeviceForHostProjectsAllowedList...). |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 162 | ModuleType("java_device_for_host", "java_host_for_device"). |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 163 | Because("java_device_for_host can only be used in allowed projects"), |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 164 | } |
| 165 | } |
| 166 | |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 167 | func createCcSdkVariantRules() []Rule { |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 168 | sdkVersionOnlyAllowedList := []string{ |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 169 | // derive_sdk_prefer32 has stem: "derive_sdk" which conflicts with the derive_sdk. |
| 170 | // This sometimes works because the APEX modules that contain derive_sdk and |
| 171 | // derive_sdk_prefer32 suppress the platform installation rules, but fails when |
| 172 | // the APEX modules contain the SDK variant and the platform variant still exists. |
Anton Hansson | 4b8e64b | 2020-05-27 18:25:23 +0100 | [diff] [blame] | 173 | "packages/modules/SdkExtensions/derive_sdk", |
Dan Albert | e2054a9 | 2020-04-20 14:46:47 -0700 | [diff] [blame] | 174 | // These are for apps and shouldn't be used by non-SDK variant modules. |
| 175 | "prebuilts/ndk", |
| 176 | "tools/test/graphicsbenchmark/apps/sample_app", |
| 177 | "tools/test/graphicsbenchmark/functional_tests/java", |
Dan Albert | 5557605 | 2020-04-20 14:46:47 -0700 | [diff] [blame] | 178 | "vendor/xts/gts-tests/hostsidetests/gamedevicecert/apps/javatests", |
Chang Li | 66d3cb7 | 2021-06-18 14:04:50 +0000 | [diff] [blame] | 179 | "external/libtextclassifier/native", |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 180 | } |
| 181 | |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 182 | platformVariantPropertiesAllowedList := []string{ |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 183 | // android_native_app_glue and libRSSupport use native_window.h but target old |
| 184 | // sdk versions (minimum and 9 respectively) where libnativewindow didn't exist, |
| 185 | // so they can't add libnativewindow to shared_libs to get the header directory |
| 186 | // for the platform variant. Allow them to use the platform variant |
| 187 | // property to set shared_libs. |
| 188 | "prebuilts/ndk", |
| 189 | "frameworks/rs", |
| 190 | } |
| 191 | |
| 192 | return []Rule{ |
| 193 | NeverAllow(). |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 194 | NotIn(sdkVersionOnlyAllowedList...). |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 195 | WithMatcher("sdk_variant_only", isSetMatcherInstance). |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 196 | Because("sdk_variant_only can only be used in allowed projects"), |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 197 | NeverAllow(). |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 198 | NotIn(platformVariantPropertiesAllowedList...). |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 199 | WithMatcher("platform.shared_libs", isSetMatcherInstance). |
Colin Cross | 440e0d0 | 2020-06-11 11:32:11 -0700 | [diff] [blame] | 200 | Because("platform variant properties can only be used in allowed projects"), |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 201 | } |
| 202 | } |
| 203 | |
David Srbecky | e033cba | 2020-05-20 22:20:28 +0100 | [diff] [blame] | 204 | func createUncompressDexRules() []Rule { |
| 205 | return []Rule{ |
| 206 | NeverAllow(). |
| 207 | NotIn("art"). |
| 208 | WithMatcher("uncompress_dex", isSetMatcherInstance). |
| 209 | Because("uncompress_dex is only allowed for certain jars for test in art."), |
| 210 | } |
| 211 | } |
| 212 | |
Yifan Hong | 696ed4d | 2020-07-27 12:59:58 -0700 | [diff] [blame] | 213 | func createMakefileGoalRules() []Rule { |
| 214 | return []Rule{ |
| 215 | NeverAllow(). |
| 216 | ModuleType("makefile_goal"). |
Yuntao Xu | feb0756 | 2021-11-18 22:33:02 +0000 | [diff] [blame] | 217 | // TODO(b/33691272): remove this after migrating seapp to Soong |
| 218 | Without("product_out_path", "obj/ETC/plat_seapp_contexts_intermediates/plat_seapp_contexts"). |
| 219 | Without("product_out_path", "obj/ETC/plat_seapp_neverallows_intermediates/plat_seapp_neverallows"). |
Yifan Hong | 696ed4d | 2020-07-27 12:59:58 -0700 | [diff] [blame] | 220 | WithoutMatcher("product_out_path", Regexp("^boot[0-9a-zA-Z.-]*[.]img$")). |
Yuntao Xu | feb0756 | 2021-11-18 22:33:02 +0000 | [diff] [blame] | 221 | Because("Only boot images and seapp contexts may be imported as a makefile goal."), |
Yifan Hong | 696ed4d | 2020-07-27 12:59:58 -0700 | [diff] [blame] | 222 | } |
| 223 | } |
| 224 | |
Inseob Kim | 800d114 | 2021-06-14 12:03:51 +0900 | [diff] [blame] | 225 | func createInitFirstStageRules() []Rule { |
| 226 | return []Rule{ |
| 227 | NeverAllow(). |
| 228 | Without("name", "init_first_stage"). |
| 229 | With("install_in_root", "true"). |
| 230 | Because("install_in_root is only for init_first_stage."), |
| 231 | } |
| 232 | } |
| 233 | |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 234 | func neverallowMutator(ctx BottomUpMutatorContext) { |
| 235 | m, ok := ctx.Module().(Module) |
| 236 | if !ok { |
| 237 | return |
| 238 | } |
| 239 | |
| 240 | dir := ctx.ModuleDir() + "/" |
| 241 | properties := m.GetProperties() |
| 242 | |
Paul Duffin | f1c9bbe | 2019-07-26 10:48:06 +0100 | [diff] [blame] | 243 | osClass := ctx.Module().Target().Os.Class |
| 244 | |
Paul Duffin | 115445b | 2019-08-07 15:31:07 +0100 | [diff] [blame] | 245 | for _, r := range neverallowRules(ctx.Config()) { |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 246 | n := r.(*rule) |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 247 | if !n.appliesToPath(dir) { |
| 248 | continue |
| 249 | } |
| 250 | |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 251 | if !n.appliesToModuleType(ctx.ModuleType()) { |
| 252 | continue |
| 253 | } |
| 254 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 255 | if !n.appliesToProperties(ctx, properties) { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 256 | continue |
| 257 | } |
| 258 | |
Paul Duffin | f1c9bbe | 2019-07-26 10:48:06 +0100 | [diff] [blame] | 259 | if !n.appliesToOsClass(osClass) { |
| 260 | continue |
| 261 | } |
| 262 | |
Paul Duffin | 3578188 | 2019-07-25 15:41:09 +0100 | [diff] [blame] | 263 | if !n.appliesToDirectDeps(ctx) { |
| 264 | continue |
| 265 | } |
| 266 | |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 267 | if !n.appliesToBootclasspathJar(ctx) { |
| 268 | continue |
| 269 | } |
| 270 | |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 271 | ctx.ModuleErrorf("violates " + n.String()) |
| 272 | } |
| 273 | } |
| 274 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 275 | type ValueMatcherContext interface { |
| 276 | Config() Config |
| 277 | } |
| 278 | |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 279 | type ValueMatcher interface { |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 280 | Test(ValueMatcherContext, string) bool |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 281 | String() string |
| 282 | } |
| 283 | |
| 284 | type equalMatcher struct { |
| 285 | expected string |
| 286 | } |
| 287 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 288 | func (m *equalMatcher) Test(ctx ValueMatcherContext, value string) bool { |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 289 | return m.expected == value |
| 290 | } |
| 291 | |
| 292 | func (m *equalMatcher) String() string { |
| 293 | return "=" + m.expected |
| 294 | } |
| 295 | |
| 296 | type anyMatcher struct { |
| 297 | } |
| 298 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 299 | func (m *anyMatcher) Test(ctx ValueMatcherContext, value string) bool { |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 300 | return true |
| 301 | } |
| 302 | |
| 303 | func (m *anyMatcher) String() string { |
| 304 | return "=*" |
| 305 | } |
| 306 | |
| 307 | var anyMatcherInstance = &anyMatcher{} |
| 308 | |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 309 | type startsWithMatcher struct { |
| 310 | prefix string |
| 311 | } |
| 312 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 313 | func (m *startsWithMatcher) Test(ctx ValueMatcherContext, value string) bool { |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 314 | return strings.HasPrefix(value, m.prefix) |
| 315 | } |
| 316 | |
| 317 | func (m *startsWithMatcher) String() string { |
| 318 | return ".starts-with(" + m.prefix + ")" |
| 319 | } |
| 320 | |
Anton Hansson | 4537640 | 2020-04-09 14:18:21 +0100 | [diff] [blame] | 321 | type regexMatcher struct { |
| 322 | re *regexp.Regexp |
| 323 | } |
| 324 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 325 | func (m *regexMatcher) Test(ctx ValueMatcherContext, value string) bool { |
Anton Hansson | 4537640 | 2020-04-09 14:18:21 +0100 | [diff] [blame] | 326 | return m.re.MatchString(value) |
| 327 | } |
| 328 | |
| 329 | func (m *regexMatcher) String() string { |
| 330 | return ".regexp(" + m.re.String() + ")" |
| 331 | } |
| 332 | |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 333 | type notInListMatcher struct { |
| 334 | allowed []string |
| 335 | } |
| 336 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 337 | func (m *notInListMatcher) Test(ctx ValueMatcherContext, value string) bool { |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 338 | return !InList(value, m.allowed) |
| 339 | } |
| 340 | |
| 341 | func (m *notInListMatcher) String() string { |
| 342 | return ".not-in-list(" + strings.Join(m.allowed, ",") + ")" |
| 343 | } |
| 344 | |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 345 | type isSetMatcher struct{} |
| 346 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 347 | func (m *isSetMatcher) Test(ctx ValueMatcherContext, value string) bool { |
Colin Cross | c511bc5 | 2020-04-07 16:50:32 +0000 | [diff] [blame] | 348 | return value != "" |
| 349 | } |
| 350 | |
| 351 | func (m *isSetMatcher) String() string { |
| 352 | return ".is-set" |
| 353 | } |
| 354 | |
| 355 | var isSetMatcherInstance = &isSetMatcher{} |
| 356 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 357 | type sdkVersionMatcher struct { |
| 358 | condition func(ctx ValueMatcherContext, spec SdkSpec) bool |
| 359 | description string |
| 360 | } |
| 361 | |
| 362 | func (m *sdkVersionMatcher) Test(ctx ValueMatcherContext, value string) bool { |
| 363 | return m.condition(ctx, SdkSpecFromWithConfig(ctx.Config(), value)) |
| 364 | } |
| 365 | |
| 366 | func (m *sdkVersionMatcher) String() string { |
| 367 | return ".sdk-version(" + m.description + ")" |
| 368 | } |
| 369 | |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 370 | type ruleProperty struct { |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 371 | fields []string // e.x.: Vndk.Enabled |
| 372 | matcher ValueMatcher |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 373 | } |
| 374 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 375 | // A NeverAllow rule. |
| 376 | type Rule interface { |
| 377 | In(path ...string) Rule |
| 378 | |
| 379 | NotIn(path ...string) Rule |
| 380 | |
Paul Duffin | 3578188 | 2019-07-25 15:41:09 +0100 | [diff] [blame] | 381 | InDirectDeps(deps ...string) Rule |
| 382 | |
Paul Duffin | f1c9bbe | 2019-07-26 10:48:06 +0100 | [diff] [blame] | 383 | WithOsClass(osClasses ...OsClass) Rule |
| 384 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 385 | ModuleType(types ...string) Rule |
| 386 | |
| 387 | NotModuleType(types ...string) Rule |
| 388 | |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 389 | BootclasspathJar() Rule |
| 390 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 391 | With(properties, value string) Rule |
| 392 | |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 393 | WithMatcher(properties string, matcher ValueMatcher) Rule |
| 394 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 395 | Without(properties, value string) Rule |
| 396 | |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 397 | WithoutMatcher(properties string, matcher ValueMatcher) Rule |
| 398 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 399 | Because(reason string) Rule |
| 400 | } |
| 401 | |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 402 | type rule struct { |
| 403 | // User string for why this is a thing. |
| 404 | reason string |
| 405 | |
| 406 | paths []string |
| 407 | unlessPaths []string |
| 408 | |
Paul Duffin | 3578188 | 2019-07-25 15:41:09 +0100 | [diff] [blame] | 409 | directDeps map[string]bool |
| 410 | |
Paul Duffin | f1c9bbe | 2019-07-26 10:48:06 +0100 | [diff] [blame] | 411 | osClasses []OsClass |
| 412 | |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 413 | moduleTypes []string |
| 414 | unlessModuleTypes []string |
| 415 | |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 416 | props []ruleProperty |
| 417 | unlessProps []ruleProperty |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 418 | |
| 419 | onlyBootclasspathJar bool |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 420 | } |
| 421 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 422 | // Create a new NeverAllow rule. |
| 423 | func NeverAllow() Rule { |
Paul Duffin | 3578188 | 2019-07-25 15:41:09 +0100 | [diff] [blame] | 424 | return &rule{directDeps: make(map[string]bool)} |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 425 | } |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 426 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 427 | func (r *rule) In(path ...string) Rule { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 428 | r.paths = append(r.paths, cleanPaths(path)...) |
| 429 | return r |
| 430 | } |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 431 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 432 | func (r *rule) NotIn(path ...string) Rule { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 433 | r.unlessPaths = append(r.unlessPaths, cleanPaths(path)...) |
| 434 | return r |
| 435 | } |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 436 | |
Paul Duffin | 3578188 | 2019-07-25 15:41:09 +0100 | [diff] [blame] | 437 | func (r *rule) InDirectDeps(deps ...string) Rule { |
| 438 | for _, d := range deps { |
| 439 | r.directDeps[d] = true |
| 440 | } |
| 441 | return r |
| 442 | } |
| 443 | |
Paul Duffin | f1c9bbe | 2019-07-26 10:48:06 +0100 | [diff] [blame] | 444 | func (r *rule) WithOsClass(osClasses ...OsClass) Rule { |
| 445 | r.osClasses = append(r.osClasses, osClasses...) |
| 446 | return r |
| 447 | } |
| 448 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 449 | func (r *rule) ModuleType(types ...string) Rule { |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 450 | r.moduleTypes = append(r.moduleTypes, types...) |
| 451 | return r |
| 452 | } |
| 453 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 454 | func (r *rule) NotModuleType(types ...string) Rule { |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 455 | r.unlessModuleTypes = append(r.unlessModuleTypes, types...) |
| 456 | return r |
| 457 | } |
| 458 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 459 | func (r *rule) With(properties, value string) Rule { |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 460 | return r.WithMatcher(properties, selectMatcher(value)) |
| 461 | } |
| 462 | |
| 463 | func (r *rule) WithMatcher(properties string, matcher ValueMatcher) Rule { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 464 | r.props = append(r.props, ruleProperty{ |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 465 | fields: fieldNamesForProperties(properties), |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 466 | matcher: matcher, |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 467 | }) |
| 468 | return r |
| 469 | } |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 470 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 471 | func (r *rule) Without(properties, value string) Rule { |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 472 | return r.WithoutMatcher(properties, selectMatcher(value)) |
| 473 | } |
| 474 | |
| 475 | func (r *rule) WithoutMatcher(properties string, matcher ValueMatcher) Rule { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 476 | r.unlessProps = append(r.unlessProps, ruleProperty{ |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 477 | fields: fieldNamesForProperties(properties), |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 478 | matcher: matcher, |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 479 | }) |
| 480 | return r |
| 481 | } |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 482 | |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 483 | func selectMatcher(expected string) ValueMatcher { |
| 484 | if expected == "*" { |
| 485 | return anyMatcherInstance |
| 486 | } |
| 487 | return &equalMatcher{expected: expected} |
| 488 | } |
| 489 | |
Paul Duffin | 730f2a5 | 2019-06-27 14:08:51 +0100 | [diff] [blame] | 490 | func (r *rule) Because(reason string) Rule { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 491 | r.reason = reason |
| 492 | return r |
| 493 | } |
| 494 | |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 495 | func (r *rule) BootclasspathJar() Rule { |
| 496 | r.onlyBootclasspathJar = true |
| 497 | return r |
| 498 | } |
| 499 | |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 500 | func (r *rule) String() string { |
| 501 | s := "neverallow" |
| 502 | for _, v := range r.paths { |
| 503 | s += " dir:" + v + "*" |
| 504 | } |
| 505 | for _, v := range r.unlessPaths { |
| 506 | s += " -dir:" + v + "*" |
| 507 | } |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 508 | for _, v := range r.moduleTypes { |
| 509 | s += " type:" + v |
| 510 | } |
| 511 | for _, v := range r.unlessModuleTypes { |
| 512 | s += " -type:" + v |
| 513 | } |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 514 | for _, v := range r.props { |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 515 | s += " " + strings.Join(v.fields, ".") + v.matcher.String() |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 516 | } |
| 517 | for _, v := range r.unlessProps { |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 518 | s += " -" + strings.Join(v.fields, ".") + v.matcher.String() |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 519 | } |
Paul Duffin | 3578188 | 2019-07-25 15:41:09 +0100 | [diff] [blame] | 520 | for k := range r.directDeps { |
| 521 | s += " deps:" + k |
| 522 | } |
Paul Duffin | f1c9bbe | 2019-07-26 10:48:06 +0100 | [diff] [blame] | 523 | for _, v := range r.osClasses { |
| 524 | s += " os:" + v.String() |
| 525 | } |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 526 | if r.onlyBootclasspathJar { |
| 527 | s += " inBcp" |
| 528 | } |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 529 | if len(r.reason) != 0 { |
| 530 | s += " which is restricted because " + r.reason |
| 531 | } |
| 532 | return s |
| 533 | } |
| 534 | |
| 535 | func (r *rule) appliesToPath(dir string) bool { |
Jaewoong Jung | 3aff578 | 2020-02-11 07:54:35 -0800 | [diff] [blame] | 536 | includePath := len(r.paths) == 0 || HasAnyPrefix(dir, r.paths) |
| 537 | excludePath := HasAnyPrefix(dir, r.unlessPaths) |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 538 | return includePath && !excludePath |
| 539 | } |
| 540 | |
Paul Duffin | 3578188 | 2019-07-25 15:41:09 +0100 | [diff] [blame] | 541 | func (r *rule) appliesToDirectDeps(ctx BottomUpMutatorContext) bool { |
| 542 | if len(r.directDeps) == 0 { |
| 543 | return true |
| 544 | } |
| 545 | |
| 546 | matches := false |
| 547 | ctx.VisitDirectDeps(func(m Module) { |
| 548 | if !matches { |
| 549 | name := ctx.OtherModuleName(m) |
| 550 | matches = r.directDeps[name] |
| 551 | } |
| 552 | }) |
| 553 | |
| 554 | return matches |
| 555 | } |
| 556 | |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 557 | func (r *rule) appliesToBootclasspathJar(ctx BottomUpMutatorContext) bool { |
| 558 | if !r.onlyBootclasspathJar { |
| 559 | return true |
| 560 | } |
| 561 | |
| 562 | return InList(ctx.ModuleName(), ctx.Config().BootJars()) |
| 563 | } |
| 564 | |
Paul Duffin | f1c9bbe | 2019-07-26 10:48:06 +0100 | [diff] [blame] | 565 | func (r *rule) appliesToOsClass(osClass OsClass) bool { |
| 566 | if len(r.osClasses) == 0 { |
| 567 | return true |
| 568 | } |
| 569 | |
| 570 | for _, c := range r.osClasses { |
| 571 | if c == osClass { |
| 572 | return true |
| 573 | } |
| 574 | } |
| 575 | |
| 576 | return false |
| 577 | } |
| 578 | |
Colin Cross | fd4f743 | 2019-03-05 15:06:16 -0800 | [diff] [blame] | 579 | func (r *rule) appliesToModuleType(moduleType string) bool { |
| 580 | return (len(r.moduleTypes) == 0 || InList(moduleType, r.moduleTypes)) && !InList(moduleType, r.unlessModuleTypes) |
| 581 | } |
| 582 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 583 | func (r *rule) appliesToProperties(ctx ValueMatcherContext, |
| 584 | properties []interface{}) bool { |
| 585 | includeProps := hasAllProperties(ctx, properties, r.props) |
| 586 | excludeProps := hasAnyProperty(ctx, properties, r.unlessProps) |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 587 | return includeProps && !excludeProps |
| 588 | } |
| 589 | |
Paul Duffin | c811170 | 2019-07-22 12:13:55 +0100 | [diff] [blame] | 590 | func StartsWith(prefix string) ValueMatcher { |
| 591 | return &startsWithMatcher{prefix} |
| 592 | } |
| 593 | |
Anton Hansson | 4537640 | 2020-04-09 14:18:21 +0100 | [diff] [blame] | 594 | func Regexp(re string) ValueMatcher { |
| 595 | r, err := regexp.Compile(re) |
| 596 | if err != nil { |
| 597 | panic(err) |
| 598 | } |
| 599 | return ®exMatcher{r} |
| 600 | } |
| 601 | |
Andrei Onea | 115e7e7 | 2020-06-05 21:14:03 +0100 | [diff] [blame] | 602 | func NotInList(allowed []string) ValueMatcher { |
| 603 | return ¬InListMatcher{allowed} |
| 604 | } |
| 605 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 606 | func LessThanSdkVersion(sdk string) ValueMatcher { |
| 607 | return &sdkVersionMatcher{ |
| 608 | condition: func(ctx ValueMatcherContext, spec SdkSpec) bool { |
| 609 | return spec.ApiLevel.LessThan( |
| 610 | SdkSpecFromWithConfig(ctx.Config(), sdk).ApiLevel) |
| 611 | }, |
| 612 | description: "lessThan=" + sdk, |
| 613 | } |
| 614 | } |
| 615 | |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 616 | // assorted utils |
| 617 | |
| 618 | func cleanPaths(paths []string) []string { |
| 619 | res := make([]string, len(paths)) |
| 620 | for i, v := range paths { |
| 621 | res[i] = filepath.Clean(v) + "/" |
| 622 | } |
| 623 | return res |
| 624 | } |
| 625 | |
| 626 | func fieldNamesForProperties(propertyNames string) []string { |
| 627 | names := strings.Split(propertyNames, ".") |
| 628 | for i, v := range names { |
| 629 | names[i] = proptools.FieldNameForProperty(v) |
| 630 | } |
| 631 | return names |
| 632 | } |
| 633 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 634 | func hasAnyProperty(ctx ValueMatcherContext, properties []interface{}, |
| 635 | props []ruleProperty) bool { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 636 | for _, v := range props { |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 637 | if hasProperty(ctx, properties, v) { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 638 | return true |
| 639 | } |
| 640 | } |
| 641 | return false |
| 642 | } |
| 643 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 644 | func hasAllProperties(ctx ValueMatcherContext, properties []interface{}, |
| 645 | props []ruleProperty) bool { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 646 | for _, v := range props { |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 647 | if !hasProperty(ctx, properties, v) { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 648 | return false |
| 649 | } |
| 650 | } |
| 651 | return true |
| 652 | } |
| 653 | |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 654 | func hasProperty(ctx ValueMatcherContext, properties []interface{}, |
| 655 | prop ruleProperty) bool { |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 656 | for _, propertyStruct := range properties { |
| 657 | propertiesValue := reflect.ValueOf(propertyStruct).Elem() |
| 658 | for _, v := range prop.fields { |
| 659 | if !propertiesValue.IsValid() { |
| 660 | break |
| 661 | } |
| 662 | propertiesValue = propertiesValue.FieldByName(v) |
| 663 | } |
| 664 | if !propertiesValue.IsValid() { |
| 665 | continue |
| 666 | } |
| 667 | |
Paul Duffin | 73bf054 | 2019-07-12 14:12:49 +0100 | [diff] [blame] | 668 | check := func(value string) bool { |
Remi NGUYEN VAN | 1fdd6ca | 2021-12-02 19:39:35 +0900 | [diff] [blame] | 669 | return prop.matcher.Test(ctx, value) |
Steven Moreland | 65b3fd9 | 2017-12-06 14:18:35 -0800 | [diff] [blame] | 670 | } |
| 671 | |
| 672 | if matchValue(propertiesValue, check) { |
| 673 | return true |
| 674 | } |
| 675 | } |
| 676 | return false |
| 677 | } |
| 678 | |
| 679 | func matchValue(value reflect.Value, check func(string) bool) bool { |
| 680 | if !value.IsValid() { |
| 681 | return false |
| 682 | } |
| 683 | |
| 684 | if value.Kind() == reflect.Ptr { |
| 685 | if value.IsNil() { |
| 686 | return check("") |
| 687 | } |
| 688 | value = value.Elem() |
| 689 | } |
| 690 | |
| 691 | switch value.Kind() { |
| 692 | case reflect.String: |
| 693 | return check(value.String()) |
| 694 | case reflect.Bool: |
| 695 | return check(strconv.FormatBool(value.Bool())) |
| 696 | case reflect.Int: |
| 697 | return check(strconv.FormatInt(value.Int(), 10)) |
| 698 | case reflect.Slice: |
| 699 | slice, ok := value.Interface().([]string) |
| 700 | if !ok { |
| 701 | panic("Can only handle slice of string") |
| 702 | } |
| 703 | for _, v := range slice { |
| 704 | if check(v) { |
| 705 | return true |
| 706 | } |
| 707 | } |
| 708 | return false |
| 709 | } |
| 710 | |
| 711 | panic("Can't handle type: " + value.Kind().String()) |
| 712 | } |
Paul Duffin | 115445b | 2019-08-07 15:31:07 +0100 | [diff] [blame] | 713 | |
| 714 | var neverallowRulesKey = NewOnceKey("neverallowRules") |
| 715 | |
| 716 | func neverallowRules(config Config) []Rule { |
| 717 | return config.Once(neverallowRulesKey, func() interface{} { |
| 718 | // No test rules were set by setTestNeverallowRules, use the global rules |
| 719 | return neverallows |
| 720 | }).([]Rule) |
| 721 | } |
| 722 | |
| 723 | // Overrides the default neverallow rules for the supplied config. |
| 724 | // |
| 725 | // For testing only. |
Paul Duffin | 45338f0 | 2021-03-30 23:07:52 +0100 | [diff] [blame] | 726 | func setTestNeverallowRules(config Config, testRules []Rule) { |
Paul Duffin | 115445b | 2019-08-07 15:31:07 +0100 | [diff] [blame] | 727 | config.Once(neverallowRulesKey, func() interface{} { return testRules }) |
| 728 | } |
Paul Duffin | 45338f0 | 2021-03-30 23:07:52 +0100 | [diff] [blame] | 729 | |
| 730 | // Prepares for a test by setting neverallow rules and enabling the mutator. |
| 731 | // |
| 732 | // If the supplied rules are nil then the default rules are used. |
| 733 | func PrepareForTestWithNeverallowRules(testRules []Rule) FixturePreparer { |
| 734 | return GroupFixturePreparers( |
| 735 | FixtureModifyConfig(func(config Config) { |
| 736 | if testRules != nil { |
| 737 | setTestNeverallowRules(config, testRules) |
| 738 | } |
| 739 | }), |
| 740 | FixtureRegisterWithContext(func(ctx RegistrationContext) { |
| 741 | ctx.PostDepsMutators(registerNeverallowMutator) |
| 742 | }), |
| 743 | ) |
| 744 | } |