| // Copyright 2014 The Bazel Authors. 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 com.google.devtools.build.lib.skyframe; | |
| import static com.google.devtools.build.lib.vfs.UnixGlob.DEFAULT_SYSCALLS; | |
| import static java.nio.charset.StandardCharsets.UTF_8; | |
| import com.google.common.annotations.VisibleForTesting; | |
| import com.google.common.base.Preconditions; | |
| import com.google.common.base.Verify; | |
| import com.google.common.collect.Collections2; | |
| import com.google.common.collect.ImmutableList; | |
| import com.google.devtools.build.lib.actions.Artifact; | |
| import com.google.devtools.build.lib.actions.Artifact.TreeFileArtifact; | |
| import com.google.devtools.build.lib.actions.FileArtifactValue; | |
| import com.google.devtools.build.lib.actions.FileStateType; | |
| import com.google.devtools.build.lib.actions.FileStateValue; | |
| import com.google.devtools.build.lib.actions.FileStateValue.RegularFileStateValue; | |
| import com.google.devtools.build.lib.actions.FileValue; | |
| import com.google.devtools.build.lib.actions.HasDigest; | |
| import com.google.devtools.build.lib.collect.nestedset.NestedSet; | |
| import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder; | |
| import com.google.devtools.build.lib.events.Event; | |
| import com.google.devtools.build.lib.io.FileSymlinkException; | |
| import com.google.devtools.build.lib.io.FileSymlinkInfiniteExpansionException; | |
| import com.google.devtools.build.lib.io.FileSymlinkInfiniteExpansionUniquenessFunction; | |
| import com.google.devtools.build.lib.packages.BuildFileNotFoundException; | |
| import com.google.devtools.build.lib.profiler.Profiler; | |
| import com.google.devtools.build.lib.profiler.ProfilerTask; | |
| import com.google.devtools.build.lib.profiler.SilentCloseable; | |
| import com.google.devtools.build.lib.server.FailureDetails; | |
| import com.google.devtools.build.lib.skyframe.RecursiveFilesystemTraversalValue.FileType; | |
| import com.google.devtools.build.lib.skyframe.RecursiveFilesystemTraversalValue.ResolvedFile; | |
| import com.google.devtools.build.lib.skyframe.RecursiveFilesystemTraversalValue.ResolvedFileFactory; | |
| import com.google.devtools.build.lib.skyframe.RecursiveFilesystemTraversalValue.TraversalRequest; | |
| import com.google.devtools.build.lib.skyframe.serialization.autocodec.AutoCodec; | |
| import com.google.devtools.build.lib.util.Fingerprint; | |
| import com.google.devtools.build.lib.vfs.Dirent; | |
| import com.google.devtools.build.lib.vfs.FileStatus; | |
| import com.google.devtools.build.lib.vfs.Path; | |
| import com.google.devtools.build.lib.vfs.PathFragment; | |
| import com.google.devtools.build.lib.vfs.Root; | |
| import com.google.devtools.build.lib.vfs.RootedPath; | |
| import com.google.devtools.build.lib.vfs.Symlinks; | |
| import com.google.devtools.build.skyframe.SkyFunction; | |
| import com.google.devtools.build.skyframe.SkyFunctionException; | |
| import com.google.devtools.build.skyframe.SkyKey; | |
| import com.google.devtools.build.skyframe.SkyValue; | |
| import java.io.IOException; | |
| import java.math.BigInteger; | |
| import java.util.ArrayList; | |
| import java.util.Collection; | |
| import java.util.Collections; | |
| import java.util.HashMap; | |
| import java.util.List; | |
| import java.util.Map; | |
| import javax.annotation.Nullable; | |
| /** A {@link SkyFunction} to build {@link RecursiveFilesystemTraversalValue}s. */ | |
| public final class RecursiveFilesystemTraversalFunction implements SkyFunction { | |
| private static final byte[] MISSING_FINGERPRINT = | |
| new BigInteger(1, "NonexistentFileStateValue".getBytes(UTF_8)).toByteArray(); | |
| @AutoCodec @AutoCodec.VisibleForSerialization | |
| static final HasDigest NON_EXISTENT_HAS_DIGEST = () -> MISSING_FINGERPRINT; | |
| private static final FileInfo NON_EXISTENT_FILE_INFO = | |
| new FileInfo(FileType.NONEXISTENT, NON_EXISTENT_HAS_DIGEST, null, null); | |
| /** The exception that {@link RecursiveFilesystemTraversalFunctionException} wraps. */ | |
| public static class RecursiveFilesystemTraversalException extends Exception { | |
| /** | |
| * Categories of errors that prevent normal {@link RecursiveFilesystemTraversalFunction} | |
| * evaluation. | |
| */ | |
| public enum Type { | |
| /** | |
| * The traversal encountered a subdirectory with a BUILD file but is not allowed to recurse | |
| * into it. See {@code PackageBoundaryMode#REPORT_ERROR}. | |
| */ | |
| CANNOT_CROSS_PACKAGE_BOUNDARY, | |
| /** A dangling symlink was dereferenced. */ | |
| DANGLING_SYMLINK, | |
| /** A file operation failed. */ | |
| FILE_OPERATION_FAILURE, | |
| /** A generated directory's root-relative path conflicts with a package's path. */ | |
| GENERATED_PATH_CONFLICT, | |
| /** A file/directory visited was part of a symlink cycle or infinite expansion. */ | |
| SYMLINK_CYCLE_OR_INFINITE_EXPANSION, | |
| } | |
| private final Type type; | |
| RecursiveFilesystemTraversalException(String message, Type type) { | |
| super(message); | |
| this.type = type; | |
| } | |
| public Type getType() { | |
| return type; | |
| } | |
| } | |
| /** | |
| * Thrown when a dangling symlink is attempted to be dereferenced. | |
| * | |
| * <p>Note: this class is not identical to the one in com.google.devtools.build.lib.view.fileset | |
| * and it's not easy to merge the two because of the dependency structure. The other one will | |
| * probably be removed along with the rest of the legacy Fileset code. | |
| */ | |
| static final class DanglingSymlinkException extends RecursiveFilesystemTraversalException { | |
| DanglingSymlinkException(String path, String unresolvedLink) { | |
| super( | |
| String.format( | |
| "Found dangling symlink: %s, unresolved path: \"%s\"", path, unresolvedLink), | |
| Type.DANGLING_SYMLINK); | |
| Preconditions.checkArgument(path != null && !path.isEmpty()); | |
| Preconditions.checkArgument(unresolvedLink != null && !unresolvedLink.isEmpty()); | |
| } | |
| } | |
| /** Exception type thrown by {@link RecursiveFilesystemTraversalFunction#compute}. */ | |
| private static final class RecursiveFilesystemTraversalFunctionException extends | |
| SkyFunctionException { | |
| RecursiveFilesystemTraversalFunctionException(RecursiveFilesystemTraversalException e) { | |
| super(e, Transience.PERSISTENT); | |
| } | |
| } | |
| @Nullable | |
| @Override | |
| public SkyValue compute(SkyKey skyKey, Environment env) | |
| throws RecursiveFilesystemTraversalFunctionException, InterruptedException { | |
| TraversalRequest traversal = (TraversalRequest) skyKey.argument(); | |
| try (SilentCloseable c = | |
| Profiler.instance() | |
| .profile(ProfilerTask.FILESYSTEM_TRAVERSAL, traversal.getRoot().toString())) { | |
| // Stat the traversal root. | |
| FileInfo rootInfo = lookUpFileInfo(env, traversal); | |
| if (rootInfo == null) { | |
| return null; | |
| } | |
| if (!rootInfo.type.exists()) { | |
| // May be a dangling symlink or a non-existent file. Handle gracefully. | |
| if (rootInfo.type.isSymlink()) { | |
| return resultForDanglingSymlink(traversal.root.asRootedPath(), rootInfo); | |
| } else { | |
| return RecursiveFilesystemTraversalValue.EMPTY; | |
| } | |
| } | |
| if (rootInfo.type.isFile()) { | |
| return resultForFileRoot(traversal.root.asRootedPath(), rootInfo); | |
| } else if (rootInfo.type.isDirectory() && rootInfo.metadata instanceof TreeArtifactValue) { | |
| final TreeArtifactValue value = (TreeArtifactValue) rootInfo.metadata; | |
| ImmutableList.Builder<RecursiveFilesystemTraversalValue> list = ImmutableList.builder(); | |
| for (Map.Entry<TreeFileArtifact, FileArtifactValue> entry | |
| : value.getChildValues().entrySet()) { | |
| RootedPath path = | |
| RootedPath.toRootedPath( | |
| traversal.root.asRootedPath().getRoot(), entry.getKey().getPath()); | |
| list.add( | |
| resultForFileRoot( | |
| path, | |
| // TreeArtifact can't have symbolic inside. So the assumption for FileType.FILE | |
| // is always true. | |
| new FileInfo(FileType.FILE, entry.getValue(), path, null))); | |
| } | |
| return resultForDirectory(traversal, rootInfo, list.build()); | |
| } | |
| // Otherwise the root is a directory or a symlink to one. | |
| PkgLookupResult pkgLookupResult = checkIfPackage(env, traversal, rootInfo); | |
| if (pkgLookupResult == null) { | |
| return null; | |
| } | |
| traversal = pkgLookupResult.traversal; | |
| if (pkgLookupResult.isConflicting()) { | |
| // The traversal was requested for an output directory whose root-relative path conflicts | |
| // with a source package. We can't handle that, bail out. | |
| throw createGeneratedPathConflictException(traversal); | |
| } else if (pkgLookupResult.isPackage() && !traversal.skipTestingForSubpackage) { | |
| // The traversal was requested for a directory that defines a package. | |
| String msg = | |
| traversal.errorInfo | |
| + " crosses package boundary into package rooted at " | |
| + traversal.root.asRootedPath().getRootRelativePath().getPathString(); | |
| switch (traversal.crossPkgBoundaries) { | |
| case CROSS: | |
| // We are free to traverse the subpackage but we need to display a warning. | |
| env.getListener().handle(Event.warn(null, msg)); | |
| break; | |
| case DONT_CROSS: | |
| // We cannot traverse the subpackage and should skip it silently. Return empty results. | |
| return RecursiveFilesystemTraversalValue.EMPTY; | |
| case REPORT_ERROR: | |
| // We cannot traverse the subpackage and should complain loudly (display an error). | |
| throw new RecursiveFilesystemTraversalFunctionException( | |
| new RecursiveFilesystemTraversalException( | |
| msg, RecursiveFilesystemTraversalException.Type.CANNOT_CROSS_PACKAGE_BOUNDARY)); | |
| default: | |
| throw new IllegalStateException(traversal.toString()); | |
| } | |
| } | |
| // We are free to traverse this directory. | |
| Collection<SkyKey> dependentKeys = createRecursiveTraversalKeys(env, traversal); | |
| if (dependentKeys == null) { | |
| return null; | |
| } | |
| Collection<RecursiveFilesystemTraversalValue> subdirTraversals = | |
| traverseChildren(env, dependentKeys, /*inline=*/ traversal.isRootGenerated); | |
| if (subdirTraversals == null) { | |
| return null; | |
| } | |
| return resultForDirectory(traversal, rootInfo, subdirTraversals); | |
| } catch (IOException | BuildFileNotFoundException e) { | |
| String message = | |
| String.format( | |
| "Error while traversing directory %s: %s", | |
| traversal.root.getRelativePart(), e.getMessage()); | |
| // Trying to stat the starting point of this root may have failed with a symlink cycle or | |
| // trying to get a package lookup value may have failed due to a symlink cycle. | |
| RecursiveFilesystemTraversalException.Type exceptionType = | |
| RecursiveFilesystemTraversalException.Type.FILE_OPERATION_FAILURE; | |
| if (e instanceof FileSymlinkException) { | |
| exceptionType = | |
| RecursiveFilesystemTraversalException.Type.SYMLINK_CYCLE_OR_INFINITE_EXPANSION; | |
| } | |
| if (e instanceof DetailedException) { | |
| FailureDetails.PackageLoading.Code code = | |
| ((DetailedException) e) | |
| .getDetailedExitCode() | |
| .getFailureDetail() | |
| .getPackageLoading() | |
| .getCode(); | |
| if (code == FailureDetails.PackageLoading.Code.SYMLINK_CYCLE_OR_INFINITE_EXPANSION) { | |
| exceptionType = | |
| RecursiveFilesystemTraversalException.Type.SYMLINK_CYCLE_OR_INFINITE_EXPANSION; | |
| } | |
| } | |
| throw new RecursiveFilesystemTraversalFunctionException( | |
| new RecursiveFilesystemTraversalException(message, exceptionType)); | |
| } | |
| } | |
| private static RecursiveFilesystemTraversalFunctionException createGeneratedPathConflictException( | |
| TraversalRequest traversal) { | |
| String message = | |
| String.format( | |
| "Generated directory %s conflicts with package under the same path. " | |
| + "Additional info: %s", | |
| traversal.root.asRootedPath().getRootRelativePath().getPathString(), | |
| traversal.errorInfo != null ? traversal.errorInfo : traversal.toString()); | |
| return new RecursiveFilesystemTraversalFunctionException( | |
| new RecursiveFilesystemTraversalException( | |
| message, RecursiveFilesystemTraversalException.Type.GENERATED_PATH_CONFLICT)); | |
| } | |
| @Override | |
| public String extractTag(SkyKey skyKey) { | |
| return null; | |
| } | |
| private static final class FileInfo { | |
| final FileType type; | |
| final HasDigest metadata; | |
| @Nullable final RootedPath realPath; | |
| @Nullable final PathFragment unresolvedSymlinkTarget; | |
| FileInfo( | |
| FileType type, | |
| HasDigest metadata, | |
| @Nullable RootedPath realPath, | |
| @Nullable PathFragment unresolvedSymlinkTarget) { | |
| Preconditions.checkNotNull(metadata.getDigest(), metadata); | |
| this.type = Preconditions.checkNotNull(type); | |
| this.metadata = metadata; | |
| this.realPath = realPath; | |
| this.unresolvedSymlinkTarget = unresolvedSymlinkTarget; | |
| } | |
| @Override | |
| public String toString() { | |
| if (type.isSymlink()) { | |
| return String.format("(%s: link_value=%s, real_path=%s)", type, | |
| unresolvedSymlinkTarget.getPathString(), realPath); | |
| } else { | |
| return String.format("(%s: real_path=%s)", type, realPath); | |
| } | |
| } | |
| } | |
| @Nullable | |
| private static FileInfo lookUpFileInfo(Environment env, TraversalRequest traversal) | |
| throws IOException, InterruptedException { | |
| if (traversal.isRootGenerated) { | |
| HasDigest fsVal = null; | |
| if (traversal.root.getOutputArtifact() != null) { | |
| Artifact artifact = traversal.root.getOutputArtifact(); | |
| SkyKey artifactKey = Artifact.key(artifact); | |
| SkyValue value = env.getValue(artifactKey); | |
| if (env.valuesMissing()) { | |
| return null; | |
| } | |
| if (value instanceof FileArtifactValue || value instanceof TreeArtifactValue) { | |
| fsVal = (HasDigest) value; | |
| } else if (value instanceof ActionExecutionValue) { | |
| fsVal = ((ActionExecutionValue) value).getExistingFileArtifactValue(artifact); | |
| } else { | |
| return NON_EXISTENT_FILE_INFO; | |
| } | |
| } | |
| RootedPath realPath = traversal.root.asRootedPath(); | |
| if (traversal.strictOutputFiles) { | |
| Preconditions.checkNotNull(fsVal, "Strict Fileset output tree has null FileArtifactValue"); | |
| return new FileInfo( | |
| (fsVal instanceof TreeArtifactValue ? FileType.DIRECTORY : FileType.FILE), | |
| fsVal, | |
| realPath, | |
| null); | |
| } else { | |
| // FileArtifactValue does not currently track symlinks. If it did, we could potentially | |
| // remove some of the filesystem operations we're doing here. | |
| Path path = traversal.root.asRootedPath().asPath(); | |
| FileStateValue fileState = | |
| FileStateValue.create(traversal.root.asRootedPath(), DEFAULT_SYSCALLS, null); | |
| if (fileState.getType() == FileStateType.NONEXISTENT) { | |
| throw new IOException("Missing file: " + path); | |
| } | |
| FileStatus followStat = path.statIfFound(Symlinks.FOLLOW); | |
| FileType type; | |
| PathFragment unresolvedLinkTarget = null; | |
| if (followStat == null) { | |
| type = FileType.DANGLING_SYMLINK; | |
| if (fileState.getType() != FileStateType.SYMLINK) { | |
| throw new IOException("Expected symlink for " + path + ", but got: " + fileState); | |
| } | |
| unresolvedLinkTarget = path.readSymbolicLink(); | |
| } else if (fileState.getType() == FileStateType.REGULAR_FILE) { | |
| type = FileType.FILE; | |
| } else if (fileState.getType() == FileStateType.DIRECTORY) { | |
| type = FileType.DIRECTORY; | |
| } else { | |
| unresolvedLinkTarget = path.readSymbolicLink(); | |
| realPath = | |
| RootedPath.toRootedPath( | |
| Root.absoluteRoot(path.getFileSystem()), path.resolveSymbolicLinks()); | |
| type = followStat.isFile() ? FileType.SYMLINK_TO_FILE : FileType.SYMLINK_TO_DIRECTORY; | |
| } | |
| if (fsVal == null) { | |
| fsVal = fileState; | |
| } | |
| return new FileInfo(type, withDigest(fsVal, path), realPath, unresolvedLinkTarget); | |
| } | |
| } else { | |
| // Stat the file. | |
| FileValue fileValue = | |
| (FileValue) env.getValueOrThrow( | |
| FileValue.key(traversal.root.asRootedPath()), IOException.class); | |
| if (env.valuesMissing()) { | |
| return null; | |
| } | |
| if (fileValue.unboundedAncestorSymlinkExpansionChain() != null) { | |
| SkyKey uniquenessKey = | |
| FileSymlinkInfiniteExpansionUniquenessFunction.key( | |
| fileValue.unboundedAncestorSymlinkExpansionChain()); | |
| env.getValue(uniquenessKey); | |
| if (env.valuesMissing()) { | |
| return null; | |
| } | |
| throw new FileSymlinkInfiniteExpansionException( | |
| fileValue.pathToUnboundedAncestorSymlinkExpansionChain(), | |
| fileValue.unboundedAncestorSymlinkExpansionChain()); | |
| } | |
| if (fileValue.exists()) { | |
| // If it exists, it may either be a symlink or a file/directory. | |
| PathFragment unresolvedLinkTarget = null; | |
| FileType type; | |
| if (fileValue.isSymlink()) { | |
| unresolvedLinkTarget = fileValue.getUnresolvedLinkTarget(); | |
| type = fileValue.isDirectory() ? FileType.SYMLINK_TO_DIRECTORY : FileType.SYMLINK_TO_FILE; | |
| } else { | |
| type = fileValue.isDirectory() ? FileType.DIRECTORY : FileType.FILE; | |
| } | |
| Path path = traversal.root.asRootedPath().asPath(); | |
| return new FileInfo( | |
| type, | |
| withDigest(fileValue.realFileStateValue(), path), | |
| fileValue.realRootedPath(), | |
| unresolvedLinkTarget); | |
| } else { | |
| // If it doesn't exist, or it's a dangling symlink, we still want to handle that gracefully. | |
| return new FileInfo( | |
| fileValue.isSymlink() ? FileType.DANGLING_SYMLINK : FileType.NONEXISTENT, | |
| withDigest(fileValue.realFileStateValue(), null), | |
| null, | |
| fileValue.isSymlink() ? fileValue.getUnresolvedLinkTarget() : null); | |
| } | |
| } | |
| } | |
| /** | |
| * Transform the HasDigest to the appropriate type based on the current state of the digest. If | |
| * fsVal is type RegularFileStateValue or FileArtifactValue and has a valid digest value, then we | |
| * want to convert it to a new FileArtifactValue type. Otherwise if they are of the two | |
| * forementioned types but do not have a digest, then we will create a FileArtifactValue using its | |
| * {@link Path}. Otherwise we will fingerprint the digest and return it as a new {@link | |
| * HasDigest.ByteStringDigest} object. | |
| * | |
| * @param fsVal - the HasDigest value that was in the graph. | |
| * @param path - the Path of the digest. | |
| * @return transformed HasDigest value based on the digest field and object type. | |
| */ | |
| @VisibleForTesting | |
| static HasDigest withDigest(HasDigest fsVal, Path path) throws IOException { | |
| if (fsVal instanceof FileStateValue) { | |
| FileStateValue fsv = (FileStateValue) fsVal; | |
| if (fsv instanceof RegularFileStateValue) { | |
| RegularFileStateValue rfsv = (RegularFileStateValue) fsv; | |
| return rfsv.getDigest() != null | |
| // If we have the digest, then simply convert it with the digest value. | |
| ? FileArtifactValue.createForVirtualActionInput(rfsv.getDigest(), rfsv.getSize()) | |
| // Otherwise, create a file FileArtifactValue (RegularFileArtifactValue) based on the | |
| // path and size. | |
| : FileArtifactValue.createForNormalFileUsingPath(path, rfsv.getSize()); | |
| } | |
| return new HasDigest.ByteStringDigest(fsv.getValueFingerprint()); | |
| } else if (fsVal instanceof FileArtifactValue) { | |
| FileArtifactValue fav = ((FileArtifactValue) fsVal); | |
| if (fav.getDigest() != null) { | |
| return fav; | |
| } | |
| // In the case there is a directory, the HasDigest value should not be converted. Otherwise, | |
| // if the HasDigest value is a file, convert it using the Path and size values. | |
| return fav.getType().isFile() | |
| ? FileArtifactValue.createForNormalFileUsingPath(path, fav.getSize()) | |
| : new HasDigest.ByteStringDigest(fav.getValueFingerprint()); | |
| } | |
| return fsVal; | |
| } | |
| private static final class PkgLookupResult { | |
| private enum Type { | |
| CONFLICT, DIRECTORY, PKG | |
| } | |
| private final Type type; | |
| final TraversalRequest traversal; | |
| final FileInfo rootInfo; | |
| /** Result for a generated directory that conflicts with a source package. */ | |
| static PkgLookupResult conflict(TraversalRequest traversal, FileInfo rootInfo) { | |
| return new PkgLookupResult(Type.CONFLICT, traversal, rootInfo); | |
| } | |
| /** Result for a source or generated directory (not a package). */ | |
| static PkgLookupResult directory(TraversalRequest traversal, FileInfo rootInfo) { | |
| return new PkgLookupResult(Type.DIRECTORY, traversal, rootInfo); | |
| } | |
| /** Result for a package, i.e. a directory with a BUILD file. */ | |
| static PkgLookupResult pkg(TraversalRequest traversal, FileInfo rootInfo) { | |
| return new PkgLookupResult(Type.PKG, traversal, rootInfo); | |
| } | |
| private PkgLookupResult(Type type, TraversalRequest traversal, FileInfo rootInfo) { | |
| this.type = Preconditions.checkNotNull(type); | |
| this.traversal = Preconditions.checkNotNull(traversal); | |
| this.rootInfo = Preconditions.checkNotNull(rootInfo); | |
| } | |
| boolean isPackage() { | |
| return type == Type.PKG; | |
| } | |
| boolean isConflicting() { | |
| return type == Type.CONFLICT; | |
| } | |
| @Override | |
| public String toString() { | |
| return String.format("(%s: info=%s, traversal=%s)", type, rootInfo, traversal); | |
| } | |
| } | |
| /** | |
| * Checks whether the {@code traversal}'s path refers to a package directory. | |
| * | |
| * @return the result of the lookup; it contains potentially new {@link TraversalRequest} and | |
| * {@link FileInfo} so the caller should use these instead of the old ones (this happens when | |
| * a package is found, but under a different root than expected) | |
| */ | |
| private static PkgLookupResult checkIfPackage( | |
| Environment env, TraversalRequest traversal, FileInfo rootInfo) | |
| throws IOException, InterruptedException, BuildFileNotFoundException { | |
| Preconditions.checkArgument(rootInfo.type.exists() && !rootInfo.type.isFile(), | |
| "{%s} {%s}", traversal, rootInfo); | |
| // PackageLookupFunction/dependencies can only throw IOException, BuildFileNotFoundException, | |
| // and RepositoryFetchException, and RepositoryFetchException is not in play here. Note that | |
| // run-of-the-mill circular symlinks will *not* throw here, and will trigger later errors during | |
| // the recursive traversal. | |
| PackageLookupValue pkgLookup = | |
| (PackageLookupValue) | |
| env.getValueOrThrow( | |
| PackageLookupValue.key(traversal.root.asRootedPath().getRootRelativePath()), | |
| BuildFileNotFoundException.class, | |
| IOException.class); | |
| if (env.valuesMissing()) { | |
| return null; | |
| } | |
| if (pkgLookup.packageExists()) { | |
| if (traversal.isRootGenerated) { | |
| // The traversal's root was a generated directory, but its root-relative path conflicts with | |
| // an existing package. | |
| return PkgLookupResult.conflict(traversal, rootInfo); | |
| } else { | |
| // The traversal's root was a source directory and it defines a package. | |
| Root pkgRoot = pkgLookup.getRoot(); | |
| if (!pkgRoot.equals(traversal.root.asRootedPath().getRoot())) { | |
| // However the root of this package is different from what we expected. stat() the real | |
| // BUILD file of that package. | |
| traversal = traversal.forChangedRootPath(pkgRoot); | |
| rootInfo = lookUpFileInfo(env, traversal); | |
| Verify.verify(rootInfo.type.exists(), "{%s} {%s}", traversal, rootInfo); | |
| } | |
| return PkgLookupResult.pkg(traversal, rootInfo); | |
| } | |
| } else { | |
| // The traversal's root was a directory (source or generated one), no package exists under the | |
| // same root-relative path. | |
| return PkgLookupResult.directory(traversal, rootInfo); | |
| } | |
| } | |
| /** | |
| * List the directory and create {@code SkyKey}s to request contents of its children recursively. | |
| * | |
| * <p>The returned keys are of type {@link SkyFunctions#RECURSIVE_FILESYSTEM_TRAVERSAL}. | |
| */ | |
| @Nullable | |
| private static Collection<SkyKey> createRecursiveTraversalKeys( | |
| Environment env, TraversalRequest traversal) throws InterruptedException, IOException { | |
| // Use the traversal's path, even if it's a symlink. The contents of the directory, as listed | |
| // in the result, must be relative to it. | |
| Iterable<Dirent> dirents; | |
| if (traversal.isRootGenerated) { | |
| // If we're dealing with an output file, read the directory directly instead of creating | |
| // filesystem nodes under the output tree. | |
| List<Dirent> direntsCollection = | |
| new ArrayList<>( | |
| traversal.root.asRootedPath().asPath().readdir(Symlinks.FOLLOW)); | |
| Collections.sort(direntsCollection); | |
| dirents = direntsCollection; | |
| } else { | |
| DirectoryListingValue dirListingValue = | |
| (DirectoryListingValue) | |
| env.getValueOrThrow( | |
| DirectoryListingValue.key(traversal.root.asRootedPath()), IOException.class); | |
| if (dirListingValue == null) { | |
| return null; | |
| } | |
| dirents = dirListingValue.getDirents(); | |
| } | |
| List<SkyKey> result = new ArrayList<>(); | |
| for (Dirent dirent : dirents) { | |
| RootedPath childPath = | |
| RootedPath.toRootedPath( | |
| traversal.root.asRootedPath().getRoot(), | |
| traversal.root.asRootedPath().getRootRelativePath().getRelative(dirent.getName())); | |
| TraversalRequest childTraversal = traversal.forChildEntry(childPath); | |
| result.add(childTraversal); | |
| } | |
| return result; | |
| } | |
| /** | |
| * Creates result for a dangling symlink. | |
| * | |
| * @param linkName path to the symbolic link | |
| * @param info the {@link FileInfo} associated with the link file | |
| */ | |
| private static RecursiveFilesystemTraversalValue resultForDanglingSymlink(RootedPath linkName, | |
| FileInfo info) { | |
| Preconditions.checkState(info.type.isSymlink() && !info.type.exists(), "{%s} {%s}", linkName, | |
| info.type); | |
| return RecursiveFilesystemTraversalValue.of( | |
| ResolvedFileFactory.danglingSymlink(linkName, info.unresolvedSymlinkTarget, info.metadata)); | |
| } | |
| /** | |
| * Creates results for a file or for a symlink that points to one. | |
| * | |
| * <p>A symlink may be direct (points to a file) or transitive (points at a direct or transitive | |
| * symlink). | |
| */ | |
| private static RecursiveFilesystemTraversalValue resultForFileRoot(RootedPath path, | |
| FileInfo info) { | |
| Preconditions.checkState(info.type.isFile() && info.type.exists(), "{%s} {%s}", path, | |
| info.type); | |
| if (info.type.isSymlink()) { | |
| return RecursiveFilesystemTraversalValue.of( | |
| ResolvedFileFactory.symlinkToFile( | |
| info.realPath, path, info.unresolvedSymlinkTarget, info.metadata)); | |
| } else { | |
| return RecursiveFilesystemTraversalValue.of( | |
| ResolvedFileFactory.regularFile(path, info.metadata)); | |
| } | |
| } | |
| private static RecursiveFilesystemTraversalValue resultForDirectory(TraversalRequest traversal, | |
| FileInfo rootInfo, Collection<RecursiveFilesystemTraversalValue> subdirTraversals) { | |
| // Collect transitive closure of files in subdirectories. | |
| NestedSetBuilder<ResolvedFile> paths = NestedSetBuilder.stableOrder(); | |
| for (RecursiveFilesystemTraversalValue child : subdirTraversals) { | |
| paths.addTransitive(child.getTransitiveFiles()); | |
| } | |
| ResolvedFile root; | |
| if (rootInfo.type.isSymlink()) { | |
| NestedSet<ResolvedFile> children = paths.build(); | |
| root = | |
| ResolvedFileFactory.symlinkToDirectory( | |
| rootInfo.realPath, | |
| traversal.root.asRootedPath(), | |
| rootInfo.unresolvedSymlinkTarget, | |
| hashDirectorySymlink(children, rootInfo.metadata)); | |
| paths = NestedSetBuilder.<ResolvedFile>stableOrder().addTransitive(children).add(root); | |
| } else { | |
| root = ResolvedFileFactory.directory(rootInfo.realPath); | |
| } | |
| return RecursiveFilesystemTraversalValue.of(root, paths.build()); | |
| } | |
| private static HasDigest hashDirectorySymlink( | |
| NestedSet<ResolvedFile> children, HasDigest metadata) { | |
| // If the root is a directory symlink, the associated FileStateValue does not change when the | |
| // linked directory's contents change, so we can't use the FileStateValue as metadata like we | |
| // do with other ResolvedFile kinds. Instead we compute a metadata hash from the child | |
| // elements and return that as the ResolvedFile's metadata hash. | |
| Fingerprint fp = new Fingerprint(); | |
| fp.addBytes(metadata.getDigest()); | |
| for (ResolvedFile file : children.toList()) { | |
| fp.addPath(file.getNameInSymlinkTree()); | |
| fp.addBytes(file.getMetadata().getDigest()); | |
| } | |
| byte[] result = fp.digestAndReset(); | |
| return new HasDigest.ByteStringDigest(result); | |
| } | |
| /** | |
| * Requests Skyframe to compute the dependent values and returns them. | |
| * | |
| * <p>The keys must all be {@link SkyFunctions#RECURSIVE_FILESYSTEM_TRAVERSAL} keys. | |
| */ | |
| @Nullable | |
| private Collection<RecursiveFilesystemTraversalValue> traverseChildren( | |
| Environment env, Iterable<SkyKey> keys, boolean inline) | |
| throws InterruptedException, RecursiveFilesystemTraversalFunctionException { | |
| Map<SkyKey, SkyValue> values; | |
| if (inline) { | |
| // Don't create Skyframe nodes for a recursive traversal over the output tree. | |
| // Instead, inline the recursion in the top-level request. | |
| values = new HashMap<>(); | |
| for (SkyKey depKey : keys) { | |
| values.put(depKey, compute(depKey, env)); | |
| } | |
| } else { | |
| values = env.getValues(keys); | |
| } | |
| if (env.valuesMissing()) { | |
| return null; | |
| } | |
| return Collections2.transform(values.values(), RecursiveFilesystemTraversalValue.class::cast); | |
| } | |
| } |