This allows non-privileged users to use containerd. This is part of a larger track of work integrating containerd into Cloudfoundry's garden with support for rootless. [#156343575] Signed-off-by: Claudia Beresford <cberesford@pivotal.io>
		
			
				
	
	
		
			118 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			118 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
/*
 | 
						|
   Copyright The containerd Authors.
 | 
						|
 | 
						|
   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 rootfs
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"fmt"
 | 
						|
	"io/ioutil"
 | 
						|
	"os"
 | 
						|
 | 
						|
	"github.com/containerd/containerd/log"
 | 
						|
	"github.com/containerd/containerd/mount"
 | 
						|
	"github.com/containerd/containerd/snapshots"
 | 
						|
	digest "github.com/opencontainers/go-digest"
 | 
						|
	"github.com/pkg/errors"
 | 
						|
)
 | 
						|
 | 
						|
var (
 | 
						|
	initializers = map[string]initializerFunc{}
 | 
						|
)
 | 
						|
 | 
						|
type initializerFunc func(string) error
 | 
						|
 | 
						|
// Mounter handles mount and unmount
 | 
						|
type Mounter interface {
 | 
						|
	Mount(target string, mounts ...mount.Mount) error
 | 
						|
	Unmount(target string) error
 | 
						|
}
 | 
						|
 | 
						|
// InitRootFS initializes the snapshot for use as a rootfs
 | 
						|
func InitRootFS(ctx context.Context, name string, parent digest.Digest, readonly bool, snapshotter snapshots.Snapshotter, mounter Mounter) ([]mount.Mount, error) {
 | 
						|
	_, err := snapshotter.Stat(ctx, name)
 | 
						|
	if err == nil {
 | 
						|
		return nil, errors.Errorf("rootfs already exists")
 | 
						|
	}
 | 
						|
	// TODO: ensure not exist error once added to snapshot package
 | 
						|
 | 
						|
	parentS := parent.String()
 | 
						|
 | 
						|
	initName := defaultInitializer
 | 
						|
	initFn := initializers[initName]
 | 
						|
	if initFn != nil {
 | 
						|
		parentS, err = createInitLayer(ctx, parentS, initName, initFn, snapshotter, mounter)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if readonly {
 | 
						|
		return snapshotter.View(ctx, name, parentS)
 | 
						|
	}
 | 
						|
 | 
						|
	return snapshotter.Prepare(ctx, name, parentS)
 | 
						|
}
 | 
						|
 | 
						|
func createInitLayer(ctx context.Context, parent, initName string, initFn func(string) error, snapshotter snapshots.Snapshotter, mounter Mounter) (string, error) {
 | 
						|
	initS := fmt.Sprintf("%s %s", parent, initName)
 | 
						|
	if _, err := snapshotter.Stat(ctx, initS); err == nil {
 | 
						|
		return initS, nil
 | 
						|
	}
 | 
						|
	// TODO: ensure not exist error once added to snapshot package
 | 
						|
 | 
						|
	// Create tempdir
 | 
						|
	td, err := ioutil.TempDir(os.Getenv("XDG_RUNTIME_DIR"), "create-init-")
 | 
						|
	if err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	defer os.RemoveAll(td)
 | 
						|
 | 
						|
	mounts, err := snapshotter.Prepare(ctx, td, parent)
 | 
						|
	if err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
 | 
						|
	defer func() {
 | 
						|
		if err != nil {
 | 
						|
			if rerr := snapshotter.Remove(ctx, td); rerr != nil {
 | 
						|
				log.G(ctx).Errorf("Failed to remove snapshot %s: %v", td, rerr)
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}()
 | 
						|
 | 
						|
	if err = mounter.Mount(td, mounts...); err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
 | 
						|
	if err = initFn(td); err != nil {
 | 
						|
		if merr := mounter.Unmount(td); merr != nil {
 | 
						|
			log.G(ctx).Errorf("Failed to unmount %s: %v", td, merr)
 | 
						|
		}
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
 | 
						|
	if err = mounter.Unmount(td); err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
 | 
						|
	if err := snapshotter.Commit(ctx, initS, td); err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
 | 
						|
	return initS, nil
 | 
						|
}
 |