X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=rpc%2Fdial.go;h=945a75ff86c365fd1e251e205a0b68efbd2b3aef;hb=825f928d824f728088606bcbf112d30d7a76f627;hp=36092793e005e06991a64fe440b97ca74fac6a88;hpb=f2f2bc47e8729548f3c10117f7f008b547c4afc5;p=safcm%2Fsafcm.git

diff --git a/rpc/dial.go b/rpc/dial.go
index 3609279..945a75f 100644
--- a/rpc/dial.go
+++ b/rpc/dial.go
@@ -32,11 +32,15 @@ import (
 	"ruderich.org/simon/safcm/remote"
 )
 
-func (c *Conn) DialSSH(remote string) error {
+func (c *Conn) DialSSH(user, host, sshConfig string) error {
 	if c.events == nil {
 		return fmt.Errorf("cannot reuse Conn")
 	}
 
+	remote := host
+	if user != "" {
+		remote = user + "@" + host
+	}
 	c.debugf("DialSSH: connecting to %q", remote)
 
 	opts := "-eu"
@@ -44,8 +48,14 @@ func (c *Conn) DialSSH(remote string) error {
 		// Help debugging by showing executed shell commands
 		opts += "x"
 	}
-	c.cmd = exec.Command("ssh", remote, "/bin/sh", opts)
-	c.remote = remote
+
+	c.sshRemote = remote
+	if sshConfig != "" {
+		c.sshOpts = append(c.sshOpts, "-F", sshConfig)
+	}
+	c.cmd = exec.Command("ssh",
+		append(append([]string{}, c.sshOpts...),
+			c.sshRemote, "/bin/sh", opts)...)
 
 	stdin, err := c.cmd.StdinPipe()
 	if err != nil {
@@ -94,6 +104,36 @@ func (c *Conn) dialSSH(stdin io.Writer, stdout_ io.Reader) error {
 	path := fmt.Sprintf("/tmp/safcm-remote-%d", uid)
 
 	c.debugf("DialSSH: probing remote at %q", path)
+
+	// Compatibility for different operating systems
+	var compat string
+	switch goos {
+	case "linux":
+		compat = `
+dir_stat='drwxrwxrwt 0 0'
+file_stat="-rwx------ $(id -u) $(id -g)"
+compat_stat() {
+	stat -c '%A %u %g' "$1"
+}
+compat_sha512sum() {
+	sha512sum "$1"
+}
+`
+	case "freebsd", "openbsd":
+		compat = `
+dir_stat='41777 0 0'
+file_stat="100700 $(id -u) $(id -g)"
+compat_stat() {
+	stat -f '%p %u %g' "$1"
+}
+compat_sha512sum() {
+	sha512 -q "$1"
+}
+`
+	default:
+		return fmt.Errorf("internal error: no support for %q", goos)
+	}
+
 	// Use a function so the shell cannot execute the input line-wise.
 	// This is important because we're also using stdin to send data to
 	// the script. If the shell executes the input line-wise then our
@@ -112,18 +152,19 @@ func (c *Conn) dialSSH(stdin io.Writer, stdout_ io.Reader) error {
 	// `test -e` is only used to prevent error messages if the file
 	// doesn't exist. It does not guard against any races.
 	_, err = fmt.Fprintf(stdin, `
+%s
 f() {
 	x=%q
 
 	dir="$(dirname "$x")"
-	if ! test "$(stat -c '%%A %%u %%g' "$dir")" = 'drwxrwxrwt 0 0'; then
+	if ! test "$(compat_stat "$dir")" = "$dir_stat"; then
 		echo "unsafe permissions on $dir, aborting" >&2
 		exit 1
 	fi
 
-	if test -e "$x" && test "$(stat -c '%%A %%u' "$x")" = "-rwx------ $(id -u)"; then
+	if test -e "$x" && test "$(compat_stat "$x")" = "$file_stat"; then
 		# Report checksum
-		sha512sum "$x"
+		compat_sha512sum "$x"
 	else
 		# Empty checksum to request upload
 		echo
@@ -146,12 +187,14 @@ f() {
 		rm "$tmp"
 		# Make file executable
 		chmod 0700 "$x"
+		# Some BSD create files with group wheel in /tmp
+		chgrp "$(id -g)" "$x"
 	fi
 
-	exec "$x"
+	exec "$x" sync
 }
 f
-`, path)
+`, compat, path)
 	if err != nil {
 		return err
 	}
@@ -214,10 +257,12 @@ f
 		path = strings.TrimSuffix(path, "\n")
 
 		c.debugf("DialSSH: uploading new remote to %q at %q",
-			c.remote, path)
+			c.sshRemote, path)
 
-		cmd := exec.Command("ssh", c.remote,
-			fmt.Sprintf("cat > %q", path))
+		cmd := exec.Command("ssh",
+			append(append([]string{}, c.sshOpts...),
+				c.sshRemote,
+				fmt.Sprintf("cat > %q", path))...)
 		cmd.Stdin = bytes.NewReader(helper)
 		err = c.handleStderrAsEvents(cmd)
 		if err != nil {
@@ -239,7 +284,7 @@ f
 }
 
 func connGetGoos(stdin io.Writer, stdout *bufio.Reader) (string, error) {
-	_, err := fmt.Fprintln(stdin, "uname -o")
+	_, err := fmt.Fprintln(stdin, "uname")
 	if err != nil {
 		return "", err
 	}
@@ -252,10 +297,14 @@ func connGetGoos(stdin io.Writer, stdout *bufio.Reader) (string, error) {
 	// NOTE: Adapt helper uploading in dialSSH() when adding new systems
 	var goos string
 	switch x {
-	case "GNU/Linux":
+	case "Linux":
 		goos = "linux"
+	case "FreeBSD":
+		goos = "freebsd"
+	case "OpenBSD":
+		goos = "openbsd"
 	default:
-		return "", fmt.Errorf("unsupported OS %q (`uname -o`)", x)
+		return "", fmt.Errorf("unsupported OS %q (`uname`)", x)
 	}
 	return goos, nil
 }
@@ -274,8 +323,10 @@ func connGetGoarch(stdin io.Writer, stdout *bufio.Reader) (string, error) {
 	// NOTE: Adapt cmd/safcm-remote/build.sh when adding new architectures
 	var goarch string
 	switch x {
-	case "x86_64":
+	case "x86_64", "amd64":
 		goarch = "amd64"
+	case "armv7l":
+		goarch = "armv7l"
 	default:
 		return "", fmt.Errorf("unsupported arch %q (`uname -m`)", x)
 	}