Move archive to pkg/archive

Signed-off-by: Derek McGowan <derek@mcg.dev>

pkg/archive/tar_linux_test.go

@@ -0,0 +1,171 @@
+/*
+   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 archive
+
+import (
+	"bytes"
+	"context"
+	"fmt"
+	"io"
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/containerd/containerd/v2/core/mount"
+	"github.com/containerd/containerd/v2/pkg/testutil"
+	"github.com/containerd/containerd/v2/plugins/snapshots/overlay/overlayutils"
+	"github.com/containerd/continuity/fs"
+	"github.com/containerd/continuity/fs/fstest"
+	"github.com/containerd/log/logtest"
+)
+
+func TestOverlayApply(t *testing.T) {
+	testutil.RequiresRoot(t)
+
+	base := t.TempDir()
+
+	if err := overlayutils.Supported(base); err != nil {
+		t.Skipf("skipping because overlay is not supported %v", err)
+	}
+	fstest.FSSuite(t, overlayDiffApplier{
+		tmp:  base,
+		diff: WriteDiff,
+		t:    t,
+	})
+}
+
+func TestOverlayApplyNoParents(t *testing.T) {
+	testutil.RequiresRoot(t)
+
+	base := t.TempDir()
+
+	if err := overlayutils.Supported(base); err != nil {
+		t.Skipf("skipping because overlay is not supported %v", err)
+	}
+	fstest.FSSuite(t, overlayDiffApplier{
+		tmp: base,
+		diff: func(ctx context.Context, w io.Writer, a, b string, _ ...WriteDiffOpt) error {
+			cw := NewChangeWriter(w, b)
+			cw.addedDirs = nil
+			err := fs.Changes(ctx, a, b, cw.HandleChange)
+			if err != nil {
+				return fmt.Errorf("failed to create diff tar stream: %w", err)
+			}
+			return cw.Close()
+		},
+		t: t,
+	})
+}
+
+type overlayDiffApplier struct {
+	tmp  string
+	diff func(context.Context, io.Writer, string, string, ...WriteDiffOpt) error
+	t    *testing.T
+}
+
+type overlayContext struct {
+	merged  string
+	lowers  []string
+	mounted bool
+}
+
+type contextKey struct{}
+
+func (d overlayDiffApplier) TestContext(ctx context.Context) (context.Context, func(), error) {
+	merged, err := os.MkdirTemp(d.tmp, "merged")
+	if err != nil {
+		return ctx, nil, fmt.Errorf("failed to make merged dir: %w", err)
+	}
+
+	oc := &overlayContext{
+		merged: merged,
+	}
+
+	ctx = logtest.WithT(ctx, d.t)
+
+	return context.WithValue(ctx, contextKey{}, oc), func() {
+		if oc.mounted {
+			mount.Unmount(oc.merged, 0)
+		}
+	}, nil
+}
+
+func (d overlayDiffApplier) Apply(ctx context.Context, a fstest.Applier) (string, func(), error) {
+	oc := ctx.Value(contextKey{}).(*overlayContext)
+
+	applyCopy, err := os.MkdirTemp(d.tmp, "apply-copy-")
+	if err != nil {
+		return "", nil, fmt.Errorf("failed to create temp dir: %w", err)
+	}
+	defer os.RemoveAll(applyCopy)
+
+	base := oc.merged
+	if len(oc.lowers) == 1 {
+		base = oc.lowers[0]
+	}
+
+	if err = fs.CopyDir(applyCopy, base); err != nil {
+		return "", nil, fmt.Errorf("failed to copy base: %w", err)
+	}
+
+	if err := a.Apply(applyCopy); err != nil {
+		return "", nil, fmt.Errorf("failed to apply changes to copy of base: %w", err)
+	}
+
+	buf := bytes.NewBuffer(nil)
+
+	if err := d.diff(ctx, buf, base, applyCopy); err != nil {
+		return "", nil, fmt.Errorf("failed to create diff: %w", err)
+	}
+
+	if oc.mounted {
+		if err := mount.Unmount(oc.merged, 0); err != nil {
+			return "", nil, fmt.Errorf("failed to unmount: %w", err)
+		}
+		oc.mounted = false
+	}
+
+	next, err := os.MkdirTemp(d.tmp, "lower-")
+	if err != nil {
+		return "", nil, fmt.Errorf("failed to create temp dir: %w", err)
+	}
+
+	if _, err = Apply(ctx, next, buf, WithConvertWhiteout(OverlayConvertWhiteout), WithParents(oc.lowers)); err != nil {
+		return "", nil, fmt.Errorf("failed to apply tar stream: %w", err)
+	}
+
+	oc.lowers = append([]string{next}, oc.lowers...)
+
+	if len(oc.lowers) == 1 {
+		return oc.lowers[0], nil, nil
+	}
+
+	m := mount.Mount{
+		Type:   "overlay",
+		Source: "overlay",
+		Options: []string{
+			fmt.Sprintf("lowerdir=%s", strings.Join(oc.lowers, ":")),
+		},
+	}
+
+	if err := m.Mount(oc.merged); err != nil {
+		return "", nil, fmt.Errorf("failed to mount: %v: %w", m, err)
+	}
+	oc.mounted = true
+
+	return oc.merged, nil, nil
+}