09b184c15a
rootfs.CreateDiff might be canceled by context for some reason. Based on this case, the defer function should use the new ctx to do cleanup temporary snapshotter instead of the canceled one. Signed-off-by: Wei Fu <fuweid89@gmail.com>
71 lines
2.3 KiB
Go
71 lines
2.3 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"
|
|
|
|
"github.com/containerd/containerd/diff"
|
|
"github.com/containerd/containerd/mount"
|
|
"github.com/containerd/containerd/namespaces"
|
|
"github.com/containerd/containerd/snapshots"
|
|
ocispec "github.com/opencontainers/image-spec/specs-go/v1"
|
|
)
|
|
|
|
// CreateDiff creates a layer diff for the given snapshot identifier from the
|
|
// parent of the snapshot. A content ref is provided to track the progress of
|
|
// the content creation and the provided snapshotter and mount differ are used
|
|
// for calculating the diff. The descriptor for the layer diff is returned.
|
|
func CreateDiff(ctx context.Context, snapshotID string, sn snapshots.Snapshotter, d diff.Comparer, opts ...diff.Opt) (ocispec.Descriptor, error) {
|
|
// dctx is used to handle cleanup things just in case the param ctx
|
|
// has been canceled, which causes that the defer cleanup fails.
|
|
dctx := context.Background()
|
|
if ns, ok := namespaces.Namespace(ctx); ok {
|
|
dctx = namespaces.WithNamespace(dctx, ns)
|
|
}
|
|
|
|
info, err := sn.Stat(ctx, snapshotID)
|
|
if err != nil {
|
|
return ocispec.Descriptor{}, err
|
|
}
|
|
|
|
lowerKey := fmt.Sprintf("%s-parent-view", info.Parent)
|
|
lower, err := sn.View(ctx, lowerKey, info.Parent)
|
|
if err != nil {
|
|
return ocispec.Descriptor{}, err
|
|
}
|
|
defer sn.Remove(dctx, lowerKey)
|
|
|
|
var upper []mount.Mount
|
|
if info.Kind == snapshots.KindActive {
|
|
upper, err = sn.Mounts(ctx, snapshotID)
|
|
if err != nil {
|
|
return ocispec.Descriptor{}, err
|
|
}
|
|
} else {
|
|
upperKey := fmt.Sprintf("%s-view", snapshotID)
|
|
upper, err = sn.View(ctx, upperKey, snapshotID)
|
|
if err != nil {
|
|
return ocispec.Descriptor{}, err
|
|
}
|
|
defer sn.Remove(dctx, upperKey)
|
|
}
|
|
|
|
return d.Compare(ctx, lower, upper, opts...)
|
|
}
|