diff options
author | Philipp Wollermann <philwo@google.com> | 2016-09-28 12:27:23 +0000 |
---|---|---|
committer | Yun Peng <pcloudy@google.com> | 2016-09-28 14:01:18 +0000 |
commit | a1cf35928c20b9a5bde490934f95df7813fb2e5e (patch) | |
tree | 85879cc7b6dc49c495574daa3736a3b15dfdb067 /src/main/java/com/google/devtools/build/lib/worker/SandboxedWorker.java | |
parent | f009901a502fd7d45ee79c53612cad54ed3ecb9b (diff) |
Add support for sandboxing persistent workers.
--
MOS_MIGRATED_REVID=134523222
Diffstat (limited to 'src/main/java/com/google/devtools/build/lib/worker/SandboxedWorker.java')
-rw-r--r-- | src/main/java/com/google/devtools/build/lib/worker/SandboxedWorker.java | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/src/main/java/com/google/devtools/build/lib/worker/SandboxedWorker.java b/src/main/java/com/google/devtools/build/lib/worker/SandboxedWorker.java new file mode 100644 index 0000000000..7192c3e029 --- /dev/null +++ b/src/main/java/com/google/devtools/build/lib/worker/SandboxedWorker.java @@ -0,0 +1,52 @@ +// Copyright 2016 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.worker; + +import com.google.common.collect.ImmutableSet; +import com.google.devtools.build.lib.sandbox.SymlinkedExecRoot; +import com.google.devtools.build.lib.vfs.FileSystemUtils; +import com.google.devtools.build.lib.vfs.Path; +import java.io.IOException; + +/** A {@link Worker} that runs inside a sandboxed execution root. */ +final class SandboxedWorker extends Worker { + private final Path workDir; + private final SymlinkedExecRoot symlinkedExecRoot; + + SandboxedWorker(WorkerKey workerKey, int workerId, Path workDir, Path logFile) { + super(workerKey, workerId, workDir, logFile); + this.workDir = workDir; + this.symlinkedExecRoot = new SymlinkedExecRoot(workDir); + } + + @Override + void destroy() throws IOException { + super.destroy(); + if (symlinkedExecRoot != null) { + FileSystemUtils.deleteTree(workDir); + } + } + + @Override + public void prepareExecution(WorkerKey key) throws IOException { + symlinkedExecRoot.createFileSystem( + key.getInputFiles(), key.getOutputFiles(), ImmutableSet.<Path>of()); + } + + @Override + public void finishExecution(WorkerKey key) throws IOException { + symlinkedExecRoot.copyOutputs(key.getExecRoot(), key.getOutputFiles()); + } +} |