|
| 1 | +/* |
| 2 | + * Copyright (c) 2021, 2021, Oracle and/or its affiliates. All rights reserved. |
| 3 | + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| 4 | + * |
| 5 | + * This code is free software; you can redistribute it and/or modify it |
| 6 | + * under the terms of the GNU General Public License version 2 only, as |
| 7 | + * published by the Free Software Foundation. Oracle designates this |
| 8 | + * particular file as subject to the "Classpath" exception as provided |
| 9 | + * by Oracle in the LICENSE file that accompanied this code. |
| 10 | + * |
| 11 | + * This code is distributed in the hope that it will be useful, but WITHOUT |
| 12 | + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| 13 | + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| 14 | + * version 2 for more details (a copy is included in the LICENSE file that |
| 15 | + * accompanied this code). |
| 16 | + * |
| 17 | + * You should have received a copy of the GNU General Public License version |
| 18 | + * 2 along with this work; if not, write to the Free Software Foundation, |
| 19 | + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| 20 | + * |
| 21 | + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| 22 | + * or visit www.oracle.com if you need additional information or have any |
| 23 | + * questions. |
| 24 | + */ |
| 25 | +package com.oracle.svm.core.jdk; |
| 26 | + |
| 27 | +// Checkstyle: allow reflection |
| 28 | + |
| 29 | +import java.io.File; |
| 30 | +import java.net.MalformedURLException; |
| 31 | +import java.net.URL; |
| 32 | +import java.security.CodeSource; |
| 33 | +import java.security.ProtectionDomain; |
| 34 | +import java.security.cert.Certificate; |
| 35 | +import java.util.function.Supplier; |
| 36 | + |
| 37 | +import org.graalvm.compiler.options.Option; |
| 38 | +import org.graalvm.nativeimage.ImageSingletons; |
| 39 | +import org.graalvm.nativeimage.Platform; |
| 40 | +import org.graalvm.nativeimage.Platforms; |
| 41 | +import org.graalvm.nativeimage.ProcessProperties; |
| 42 | +import org.graalvm.nativeimage.hosted.Feature; |
| 43 | + |
| 44 | +import com.oracle.svm.core.annotate.AutomaticFeature; |
| 45 | +import com.oracle.svm.core.option.HostedOptionKey; |
| 46 | +import com.oracle.svm.core.util.LazyFinalReference; |
| 47 | +import com.oracle.svm.util.ReflectionUtil; |
| 48 | + |
| 49 | +import sun.security.util.SecurityConstants; |
| 50 | + |
| 51 | +/** |
| 52 | + * All classes that do not set a ProtectionDomain explicitly have a "default" ProtectionDomain. And |
| 53 | + * every ProtectionDomain has a CodeSource with a URL as its location. |
| 54 | + * |
| 55 | + * If an application invokes {@link CodeSource#getLocation()}, it can reasonably expect a non-null |
| 56 | + * URL. We use the executable image's name as the location, because that is the best value we can |
| 57 | + * provide. |
| 58 | + * |
| 59 | + * But computing the URL for the location pulls in a lot of JDK dependencies. For a simple |
| 60 | + * application like "Hello World", that would significantly increase the image size. So we only add |
| 61 | + * code to compute the URL if the application explicitly invokes {@link CodeSource#getLocation()}. |
| 62 | + * This is done using a reachability handler registered in |
| 63 | + * {@link ProtectionDomainFeature#beforeAnalysis}. |
| 64 | + * |
| 65 | + * Note that this still leads to observable differences in places where the location is used |
| 66 | + * implicitly, like {@link CodeSource#toString} and {@link CodeSource#implies}. We accept that |
| 67 | + * difference in behavior. |
| 68 | + */ |
| 69 | +public final class ProtectionDomainSupport { |
| 70 | + |
| 71 | + public static class Options { |
| 72 | + @Option(help = "Return a the application path as the Class.getProtectionDomain().getCodeSource().getLocation() for all classes that have no explicit ProtectionDomain")// |
| 73 | + public static final HostedOptionKey<Boolean> UseApplicationCodeSourceLocation = new HostedOptionKey<>(null); |
| 74 | + } |
| 75 | + |
| 76 | + private final LazyFinalReference<ProtectionDomain> allPermDomain = new LazyFinalReference<>(this::createAllPermDomain); |
| 77 | + |
| 78 | + /** Remains null as long as the reachability handler has not triggered. */ |
| 79 | + Supplier<URL> executableURLSupplier; |
| 80 | + |
| 81 | + public static ProtectionDomain allPermDomain() { |
| 82 | + return ImageSingletons.lookup(ProtectionDomainSupport.class).allPermDomain.get(); |
| 83 | + } |
| 84 | + |
| 85 | + private ProtectionDomain createAllPermDomain() { |
| 86 | + java.security.Permissions perms = new java.security.Permissions(); |
| 87 | + perms.add(SecurityConstants.ALL_PERMISSION); |
| 88 | + URL executableURL = executableURLSupplier != null ? executableURLSupplier.get() : null; |
| 89 | + CodeSource cs = new CodeSource(executableURL, (Certificate[]) null); |
| 90 | + return new java.security.ProtectionDomain(cs, perms); |
| 91 | + } |
| 92 | + |
| 93 | + private static URL createExecutableURL() { |
| 94 | + /* |
| 95 | + * Try to use executable image's name as code source for the class. The file location can be |
| 96 | + * used by Java code to determine its location on disk, similar to argv[0]. |
| 97 | + */ |
| 98 | + String executableName = ProcessProperties.getExecutableName(); |
| 99 | + if (executableName != null) { |
| 100 | + try { |
| 101 | + return new File(executableName).toURI().toURL(); |
| 102 | + } catch (MalformedURLException ex) { |
| 103 | + /* |
| 104 | + * This should not really happen; the file is canonicalized, absolute, so it should |
| 105 | + * always have a file:// URL. But if it happens, it is OK to just ignore. |
| 106 | + */ |
| 107 | + } |
| 108 | + } |
| 109 | + return null; |
| 110 | + } |
| 111 | + |
| 112 | + @Platforms(Platform.HOSTED_ONLY.class) |
| 113 | + static void enableCodeSource(Feature.DuringAnalysisAccess access) { |
| 114 | + ImageSingletons.lookup(ProtectionDomainSupport.class).executableURLSupplier = ProtectionDomainSupport::createExecutableURL; |
| 115 | + if (access != null) { |
| 116 | + access.requireAnalysisIteration(); |
| 117 | + } |
| 118 | + } |
| 119 | +} |
| 120 | + |
| 121 | +@AutomaticFeature |
| 122 | +final class ProtectionDomainFeature implements Feature { |
| 123 | + |
| 124 | + @Override |
| 125 | + public void afterRegistration(AfterRegistrationAccess access) { |
| 126 | + ImageSingletons.add(ProtectionDomainSupport.class, new ProtectionDomainSupport()); |
| 127 | + } |
| 128 | + |
| 129 | + @Override |
| 130 | + public void beforeAnalysis(BeforeAnalysisAccess access) { |
| 131 | + Boolean useApplicationCodeSourceLocation = ProtectionDomainSupport.Options.UseApplicationCodeSourceLocation.getValue(); |
| 132 | + if (useApplicationCodeSourceLocation == null) { |
| 133 | + /* Option not set explicitly, so use automatic behavior based on reachability. */ |
| 134 | + access.registerReachabilityHandler(ProtectionDomainSupport::enableCodeSource, |
| 135 | + ReflectionUtil.lookupMethod(CodeSource.class, "getLocation")); |
| 136 | + } else if (useApplicationCodeSourceLocation) { |
| 137 | + /* Always enabled. */ |
| 138 | + ProtectionDomainSupport.enableCodeSource(null); |
| 139 | + } else { |
| 140 | + /* Always disabled - nothing to do. */ |
| 141 | + } |
| 142 | + } |
| 143 | +} |
0 commit comments