X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/bcb6f55e2edfddd9edfa08c6463013b89ed0b78d..d640a546f2cd55cfbd0a959bcfb84be1eecdd6a6:/services/crunch-run/copier.go diff --git a/services/crunch-run/copier.go b/services/crunch-run/copier.go index b6093ce092..4c45f6acb9 100644 --- a/services/crunch-run/copier.go +++ b/services/crunch-run/copier.go @@ -54,6 +54,7 @@ type copier struct { keepClient IKeepClient hostOutputDir string ctrOutputDir string + binds []string mounts map[string]arvados.Mount secretMounts map[string]arvados.Mount logger printfer @@ -157,8 +158,12 @@ func (cp *copier) walkMount(dest, src string, maxSymlinks int, walkMountsBelow b return err } cp.manifest += mft.Extract(srcRelPath, dest).Text - case srcRoot == cp.ctrOutputDir: - f, err := os.Open(filepath.Join(cp.hostOutputDir, ".arvados#collection")) + default: + hostRoot, err := cp.hostRoot(srcRoot) + if err != nil { + return err + } + f, err := os.Open(filepath.Join(hostRoot, ".arvados#collection")) if err != nil { return err } @@ -170,8 +175,6 @@ func (cp *copier) walkMount(dest, src string, maxSymlinks int, walkMountsBelow b } mft := manifest.Manifest{Text: coll.ManifestText} cp.manifest += mft.Extract(srcRelPath, dest).Text - default: - return fmt.Errorf("cannot output %q as %q: writable collection mounted at %q", src, dest, srcRoot) } if walkMountsBelow { return cp.walkMountsBelow(dest, src) @@ -185,7 +188,7 @@ func (cp *copier) walkMountsBelow(dest, src string) error { if !strings.HasPrefix(mnt, src+"/") { continue } - if mntinfo.Kind == "text" || mntinfo.Kind == "json" { + if cp.copyRegularFiles(mntinfo) { // These got copied into the nearest parent // mount as regular files during setup, so // they get copied as regular files when we @@ -284,15 +287,15 @@ func (cp *copier) walkHostFS(dest, src string, maxSymlinks int, includeMounts bo if _, isSecret := cp.secretMounts[src]; isSecret { continue } - if mntinfo, isMount := cp.mounts[src]; isMount && mntinfo.Kind != "text" && mntinfo.Kind != "json" { + if mntinfo, isMount := cp.mounts[src]; isMount && !cp.copyRegularFiles(mntinfo) { // If a regular file/dir somehow // exists at a path that's also a // mount target, ignore the file -- // the mount has already been included // with walkMountsBelow(). // - // (...except json/text mounts, which - // are handled as regular files.) + // (...except mount types that are + // handled as regular files.) continue } err = cp.walkHostFS(dest, src, maxSymlinks, false) @@ -316,6 +319,25 @@ func (cp *copier) walkHostFS(dest, src string, maxSymlinks int, includeMounts bo return fmt.Errorf("Unsupported file type (mode %o) in output dir: %q", fi.Mode(), src) } +// Return the host path that was mounted at the given path in the +// container. +func (cp *copier) hostRoot(ctrRoot string) (string, error) { + if ctrRoot == cp.ctrOutputDir { + return cp.hostOutputDir, nil + } + for _, bind := range cp.binds { + tokens := strings.Split(bind, ":") + if len(tokens) >= 2 && tokens[1] == ctrRoot { + return tokens[0], nil + } + } + return "", fmt.Errorf("not bind-mounted: %q", ctrRoot) +} + +func (cp *copier) copyRegularFiles(m arvados.Mount) bool { + return m.Kind == "text" || m.Kind == "json" || (m.Kind == "collection" && m.Writable) +} + func (cp *copier) getManifest(pdh string) (*manifest.Manifest, error) { if mft, ok := cp.manifestCache[pdh]; ok { return mft, nil