fix(agent/usershell): check shell on darwin via dscl

mafredri · mafredri · commit 48e1cb47cf22 · 2023-07-07T15:54:53.000+03:00
diff --git a/agent/usershell/usershell_darwin.go b/agent/usershell/usershell_darwin.go
@@ -1,8 +1,25 @@
 package usershell
 
-import "os"
+import (
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"golang.org/x/xerrors"
+)
 
 // Get returns the $SHELL environment variable.
-func Get(_ string) (string, error) {
-	return os.Getenv("SHELL"), nil
+func Get(username string) (string, error) {
+	// This command will output "UserShell: /bin/zsh" if successful, we
+	// can ignore the error since we have fallback behavior.
+	out, _ := exec.Command("dscl", ".", "-read", filepath.Join("/Users", username), "UserShell").Output()
+	s, ok := strings.CutPrefix(string(out), "UserShell: ")
+	if ok {
+		return strings.TrimSpace(s), nil
+	}
+	if s = os.Getenv("SHELL"); s != "" {
+		return s, nil
+	}
+	return "", xerrors.Errorf("shell for user %q not found via dscl or in $SHELL", username)
 }
diff --git a/agent/usershell/usershell_other.go b/agent/usershell/usershell_other.go
@@ -27,5 +27,8 @@ func Get(username string) (string, error) {
 		}
 		return parts[6], nil
 	}
-	return "", xerrors.Errorf("user %q not found in /etc/passwd", username)
+	if s := os.Getenv("SHELL"); s != "" {
+		return s, nil
+	}
+	return "", xerrors.Errorf("shell for user %q not found in /etc/passwd or $SHELL", username)
 }
diff --git a/agent/usershell/usershell_other_test.go b/agent/usershell/usershell_other_test.go
diff --git a/agent/usershell/usershell_test.go b/agent/usershell/usershell_test.go
@@ -0,0 +1,47 @@
+package usershell_test
+
+import (
+	"os/user"
+	"runtime"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/agent/usershell"
+)
+
+func TestGet(t *testing.T) {
+	t.Parallel()
+	t.Run("Has", func(t *testing.T) {
+		t.Parallel()
+		if runtime.GOOS == "windows" {
+			t.SkipNow()
+		}
+		shell, err := usershell.Get("root")
+		require.NoError(t, err)
+		require.NotEmpty(t, shell)
+	})
+}
+
+//nolint:paralleltest,tparallel // This test sets an environment variable.
+func TestGet_NoFallback(t *testing.T) {
+	if runtime.GOOS == "windows" {
+		t.SkipNow()
+	}
+
+	// Disable env fallback for these tests.
+	t.Setenv("SHELL", "")
+
+	t.Run("NotFound", func(t *testing.T) {
+		_, err := usershell.Get("notauser")
+		require.Error(t, err)
+	})
+
+	t.Run("User", func(t *testing.T) {
+		u, err := user.Current()
+		require.NoError(t, err)
+		shell, err := usershell.Get(u.Username)
+		require.NoError(t, err)
+		require.NotEmpty(t, shell)
+	})
+}

Original file line number	Diff line number	Diff line change
`@@ -27,5 +27,8 @@ func Get(username string) (string, error) {`
`27`	`27`	`}`
`28`	`28`	`return parts[6], nil`
`29`	`29`	`}`
`30`		`- return "", xerrors.Errorf("user %q not found in /etc/passwd", username)`
	`30`	`+ if s := os.Getenv("SHELL"); s != "" {`
	`31`	`+ return s, nil`
	`32`	`+ }`
	`33`	`+ return "", xerrors.Errorf("shell for user %q not found in /etc/passwd or $SHELL", username)`
`31`	`34`	`}`