Add test for tarfile exclusion

tools/cmd/ssh/main_test.go

@@ -6,10 +6,17 @@ import (
 	"encoding/binary"
 	"encoding/json"
 	"fmt"
+	"io/fs"
 	"kitty/tools/utils/shm"
+	"os"
+	"os/exec"
+	"path"
+	"path/filepath"
+	"strings"
 	"testing"
 
 	"github.com/google/go-cmp/cmp"
+	"golang.org/x/sys/unix"
 )
 
 var _ = fmt.Print
@@ -38,11 +45,17 @@ func TestCloneEnv(t *testing.T) {
 	}
 }
 
-func basic_connection_data() *connection_data {
-	return &connection_data{
-		script_type: "sh", request_id: "123-123", remote_args: []string{}, host_opts: NewConfig(),
+func basic_connection_data(overrides ...string) *connection_data {
+	ans := &connection_data{
+		script_type: "sh", request_id: "123-123", remote_args: []string{},
 		username: "testuser", hostname_for_match: "host.test",
 	}
+	opts, err := load_config(ans.hostname_for_match, ans.username, overrides, "")
+	if err != nil {
+		panic(err)
+	}
+	ans.host_opts = opts
+	return ans
 }
 
 func TestSSHBootstrapScriptLimit(t *testing.T) {
@@ -59,3 +72,80 @@ func TestSSHBootstrapScriptLimit(t *testing.T) {
 		t.Fatalf("Bootstrap script too large: %d bytes", total)
 	}
 }
+
+func TestSSHTarfile(t *testing.T) {
+	tdir := t.TempDir()
+	cd := basic_connection_data()
+	data, err := make_tarfile(cd, func(key string) (val string, found bool) { return })
+	if err != nil {
+		t.Fatal(err)
+	}
+	cmd := exec.Command("tar", "xpzf", "-", "-C", tdir)
+	cmd.Stderr = os.Stderr
+	inp, err := cmd.StdinPipe()
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = cmd.Start()
+	if err != nil {
+		t.Fatal(err)
+	}
+	_, err = inp.Write(data)
+	if err != nil {
+		t.Fatal(err)
+	}
+	inp.Close()
+	err = cmd.Wait()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	seen := map[string]bool{}
+	err = filepath.WalkDir(tdir, func(name string, d fs.DirEntry, werr error) error {
+		if werr != nil {
+			return werr
+		}
+		rname, werr := filepath.Rel(tdir, name)
+		if werr != nil {
+			return werr
+		}
+		rname = strings.ReplaceAll(rname, "\\", "/")
+		if rname == "." {
+			return nil
+		}
+		fi, werr := d.Info()
+		if werr != nil {
+			return werr
+		}
+		if fi.Mode().Perm()&0o600 == 0 {
+			return fmt.Errorf("%s is not rw for its owner. Actual permissions: %s", rname, fi.Mode().String())
+		}
+		seen[rname] = true
+		return nil
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !seen["data.sh"] {
+		t.Fatalf("data.sh missing")
+	}
+	for _, x := range []string{".terminfo/kitty.terminfo", ".terminfo/x/xterm-kitty"} {
+		if !seen["home/"+x] {
+			t.Fatalf("%s missing", x)
+		}
+	}
+	for _, x := range []string{"shell-integration/bash/kitty.bash", "shell-integration/fish/vendor_completions.d/kitty.fish"} {
+		if !seen[path.Join("home", cd.host_opts.Remote_dir, x)] {
+			t.Fatalf("%s missing", x)
+		}
+	}
+	for _, x := range []string{"kitty", "kitten"} {
+		p := filepath.Join(tdir, "home", cd.host_opts.Remote_dir, "kitty", "bin", x)
+		if err = unix.Access(p, unix.X_OK); err != nil {
+			t.Fatalf("Cannot execute %s with error: %s", x, err)
+		}
+	}
+	if seen[path.Join("home", cd.host_opts.Remote_dir, "shell-integration", "ssh", "kitten")] {
+		t.Fatalf("Contents of shell-integration/ssh not excluded")
+	}
+}