pkg/archive/archive_unix_test.go
28842d3f
 // +build !windows
 
 package archive
 
 import (
1a714e76
 	"bytes"
 	"fmt"
 	"io/ioutil"
1a22418f
 	"os"
d6b78191
 	"path/filepath"
934328d8
 	"runtime"
1a714e76
 	"syscall"
28842d3f
 	"testing"
d6b78191
 
 	"github.com/docker/docker/pkg/system"
069fdc8a
 	"golang.org/x/sys/unix"
28842d3f
 )
 
 func TestCanonicalTarNameForPath(t *testing.T) {
 	cases := []struct{ in, expected string }{
 		{"foo", "foo"},
 		{"foo/bar", "foo/bar"},
 		{"foo/dir/", "foo/dir/"},
 	}
 	for _, v := range cases {
5ecab9e8
 		if out, err := CanonicalTarNameForPath(v.in); err != nil {
28842d3f
 			t.Fatalf("cannot get canonical name for path: %s: %v", v.in, err)
 		} else if out != v.expected {
 			t.Fatalf("wrong canonical tar name. expected:%s got:%s", v.expected, out)
 		}
 	}
 }
 
 func TestCanonicalTarName(t *testing.T) {
 	cases := []struct {
 		in       string
 		isDir    bool
 		expected string
 	}{
 		{"foo", false, "foo"},
 		{"foo", true, "foo/"},
 		{"foo/bar", false, "foo/bar"},
 		{"foo/bar", true, "foo/bar/"},
 	}
 	for _, v := range cases {
 		if out, err := canonicalTarName(v.in, v.isDir); err != nil {
 			t.Fatalf("cannot get canonical name for path: %s: %v", v.in, err)
 		} else if out != v.expected {
 			t.Fatalf("wrong canonical tar name. expected:%s got:%s", v.expected, out)
 		}
 	}
 }
1a22418f
 
 func TestChmodTarEntry(t *testing.T) {
 	cases := []struct {
 		in, expected os.FileMode
 	}{
 		{0000, 0000},
 		{0777, 0777},
 		{0644, 0644},
 		{0755, 0755},
 		{0444, 0444},
 	}
 	for _, v := range cases {
 		if out := chmodTarEntry(v.in); out != v.expected {
 			t.Fatalf("wrong chmod. expected:%v got:%v", v.expected, out)
 		}
 	}
 }
1a714e76
 
 func TestTarWithHardLink(t *testing.T) {
 	origin, err := ioutil.TempDir("", "docker-test-tar-hardlink")
 	if err != nil {
 		t.Fatal(err)
 	}
 	defer os.RemoveAll(origin)
d6b78191
 	if err := ioutil.WriteFile(filepath.Join(origin, "1"), []byte("hello world"), 0700); err != nil {
1a714e76
 		t.Fatal(err)
 	}
d6b78191
 	if err := os.Link(filepath.Join(origin, "1"), filepath.Join(origin, "2")); err != nil {
1a714e76
 		t.Fatal(err)
 	}
 
 	var i1, i2 uint64
d6b78191
 	if i1, err = getNlink(filepath.Join(origin, "1")); err != nil {
1a714e76
 		t.Fatal(err)
 	}
 	// sanity check that we can hardlink
 	if i1 != 2 {
 		t.Skipf("skipping since hardlinks don't work here; expected 2 links, got %d", i1)
 	}
 
 	dest, err := ioutil.TempDir("", "docker-test-tar-hardlink-dest")
 	if err != nil {
 		t.Fatal(err)
 	}
 	defer os.RemoveAll(dest)
 
 	// we'll do this in two steps to separate failure
 	fh, err := Tar(origin, Uncompressed)
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	// ensure we can read the whole thing with no error, before writing back out
 	buf, err := ioutil.ReadAll(fh)
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	bRdr := bytes.NewReader(buf)
 	err = Untar(bRdr, dest, &TarOptions{Compression: Uncompressed})
 	if err != nil {
 		t.Fatal(err)
 	}
 
d6b78191
 	if i1, err = getInode(filepath.Join(dest, "1")); err != nil {
1a714e76
 		t.Fatal(err)
 	}
d6b78191
 	if i2, err = getInode(filepath.Join(dest, "2")); err != nil {
1a714e76
 		t.Fatal(err)
 	}
 
 	if i1 != i2 {
 		t.Errorf("expected matching inodes, but got %d and %d", i1, i2)
 	}
 }
 
 func getNlink(path string) (uint64, error) {
 	stat, err := os.Stat(path)
 	if err != nil {
 		return 0, err
 	}
 	statT, ok := stat.Sys().(*syscall.Stat_t)
 	if !ok {
 		return 0, fmt.Errorf("expected type *syscall.Stat_t, got %t", stat.Sys())
 	}
 	// We need this conversion on ARM64
 	return uint64(statT.Nlink), nil
 }
 
 func getInode(path string) (uint64, error) {
 	stat, err := os.Stat(path)
 	if err != nil {
 		return 0, err
 	}
 	statT, ok := stat.Sys().(*syscall.Stat_t)
 	if !ok {
 		return 0, fmt.Errorf("expected type *syscall.Stat_t, got %t", stat.Sys())
 	}
 	return statT.Ino, nil
 }
d6b78191
 
 func TestTarWithBlockCharFifo(t *testing.T) {
 	origin, err := ioutil.TempDir("", "docker-test-tar-hardlink")
 	if err != nil {
 		t.Fatal(err)
 	}
 	defer os.RemoveAll(origin)
 	if err := ioutil.WriteFile(filepath.Join(origin, "1"), []byte("hello world"), 0700); err != nil {
 		t.Fatal(err)
 	}
069fdc8a
 	if err := system.Mknod(filepath.Join(origin, "2"), unix.S_IFBLK, int(system.Mkdev(int64(12), int64(5)))); err != nil {
d6b78191
 		t.Fatal(err)
 	}
069fdc8a
 	if err := system.Mknod(filepath.Join(origin, "3"), unix.S_IFCHR, int(system.Mkdev(int64(12), int64(5)))); err != nil {
d6b78191
 		t.Fatal(err)
 	}
069fdc8a
 	if err := system.Mknod(filepath.Join(origin, "4"), unix.S_IFIFO, int(system.Mkdev(int64(12), int64(5)))); err != nil {
d6b78191
 		t.Fatal(err)
 	}
 
 	dest, err := ioutil.TempDir("", "docker-test-tar-hardlink-dest")
 	if err != nil {
 		t.Fatal(err)
 	}
 	defer os.RemoveAll(dest)
 
 	// we'll do this in two steps to separate failure
 	fh, err := Tar(origin, Uncompressed)
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	// ensure we can read the whole thing with no error, before writing back out
 	buf, err := ioutil.ReadAll(fh)
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	bRdr := bytes.NewReader(buf)
 	err = Untar(bRdr, dest, &TarOptions{Compression: Uncompressed})
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	changes, err := ChangesDirs(origin, dest)
 	if err != nil {
 		t.Fatal(err)
 	}
 	if len(changes) > 0 {
 		t.Fatalf("Tar with special device (block, char, fifo) should keep them (recreate them when untar) : %v", changes)
 	}
 }
 
 // TestTarUntarWithXattr is Unix as Lsetxattr is not supported on Windows
 func TestTarUntarWithXattr(t *testing.T) {
934328d8
 	if runtime.GOOS == "solaris" {
 		t.Skip()
 	}
d6b78191
 	origin, err := ioutil.TempDir("", "docker-test-untar-origin")
 	if err != nil {
 		t.Fatal(err)
 	}
 	defer os.RemoveAll(origin)
 	if err := ioutil.WriteFile(filepath.Join(origin, "1"), []byte("hello world"), 0700); err != nil {
 		t.Fatal(err)
 	}
 	if err := ioutil.WriteFile(filepath.Join(origin, "2"), []byte("welcome!"), 0700); err != nil {
 		t.Fatal(err)
 	}
 	if err := ioutil.WriteFile(filepath.Join(origin, "3"), []byte("will be ignored"), 0700); err != nil {
 		t.Fatal(err)
 	}
 	if err := system.Lsetxattr(filepath.Join(origin, "2"), "security.capability", []byte{0x00}, 0); err != nil {
 		t.Fatal(err)
 	}
 
 	for _, c := range []Compression{
 		Uncompressed,
 		Gzip,
 	} {
 		changes, err := tarUntar(t, origin, &TarOptions{
 			Compression:     c,
 			ExcludePatterns: []string{"3"},
 		})
 
 		if err != nil {
 			t.Fatalf("Error tar/untar for compression %s: %s", c.Extension(), err)
 		}
 
 		if len(changes) != 1 || changes[0].Path != "/3" {
 			t.Fatalf("Unexpected differences after tarUntar: %v", changes)
 		}
 		capability, _ := system.Lgetxattr(filepath.Join(origin, "2"), "security.capability")
 		if capability == nil && capability[0] != 0x00 {
 			t.Fatalf("Untar should have kept the 'security.capability' xattr.")
 		}
 	}
 }