pkg/osutil: support glob patterns in CopyFiles

Follow up to #2053

1 files changed, 97 insertions, 0 deletions

diff --git a/pkg/osutil/osutil_test.go b/pkg/osutil/osutil_test.go
index ee4cb6705..3d367a0ca 100644
--- a/pkg/osutil/osutil_test.go
+++ b/pkg/osutil/osutil_test.go
@@ -4,7 +4,11 @@
 package osutil
 
 import (
+	"fmt"
+	"io/ioutil"
 	"os"
+	"path/filepath"
+	"strings"
 	"testing"
 )
 
@@ -16,3 +20,96 @@ func TestIsExist(t *testing.T) {
 		t.Fatalf("file %v exists", f)
 	}
 }
+
+func TestCopyFiles(t *testing.T) {
+	type Test struct {
+		files    []string
+		patterns map[string]bool
+		err      string
+	}
+	tests := []Test{
+		{
+			files: []string{
+				"foo",
+				"bar",
+				"baz/foo",
+				"baz/bar",
+			},
+			patterns: map[string]bool{
+				"foo":     true,
+				"bar":     false,
+				"qux":     false,
+				"baz/foo": true,
+				"baz/bar": false,
+			},
+		},
+		{
+			files: []string{
+				"foo",
+			},
+			patterns: map[string]bool{
+				"bar": true,
+			},
+			err: "file bar does not exist",
+		},
+		{
+			files: []string{
+				"baz/foo",
+				"baz/bar",
+			},
+			patterns: map[string]bool{
+				"baz/*": true,
+			},
+		},
+		{
+			files: []string{
+				"qux/foo/foo",
+				"qux/foo/bar",
+				"qux/bar/foo",
+				"qux/bar/bar",
+			},
+			patterns: map[string]bool{
+				"qux/*/*": false,
+			},
+		},
+	}
+	for i, test := range tests {
+		t.Run(fmt.Sprint(i), func(t *testing.T) {
+			dir, err := ioutil.TempDir("", "syz-osutil-test")
+			if err != nil {
+				t.Fatal(err)
+			}
+			defer os.RemoveAll(dir)
+			src := filepath.Join(dir, "src")
+			dst := filepath.Join(dir, "dst")
+			for _, file := range test.files {
+				file = filepath.Join(src, filepath.FromSlash(file))
+				if err := MkdirAll(filepath.Dir(file)); err != nil {
+					t.Fatal(err)
+				}
+				if err := WriteFile(file, []byte{'a'}); err != nil {
+					t.Fatal(err)
+				}
+			}
+			if err := CopyFiles(src, dst, test.patterns); err != nil {
+				if test.err != "" {
+					if strings.Contains(err.Error(), test.err) {
+						return
+					}
+					t.Fatalf("got err %q, want %q", err, test.err)
+				}
+				t.Fatal(err)
+			} else if test.err != "" {
+				t.Fatalf("got no err, want %q", test.err)
+			}
+			if err := os.RemoveAll(src); err != nil {
+				t.Fatal(err)
+			}
+			for _, file := range test.files {
+				if !IsExist(filepath.Join(dst, filepath.FromSlash(file))) {
+					t.Fatalf("%v does not exist in dst", file)
+				}
+			}
+		})
+	}
+}