mirror of
https://github.com/tailscale/tailscale.git
synced 2025-04-03 14:55:47 +00:00
5b6a90fb33
Can write "wasm" instead of js || wasi1p, since there's only two:
$ go tool dist list | grep wasm
js/wasm
wasip1/wasm
Plus, if GOOS=wasip2 is added later, we're already set.
Updates #5794
Change-Id: Ifcfb187c3775c17c9141bc721512dc4577ac4434
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
//go:build !wasm && !windows && !plan9
|
|
|
|
package cli
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strconv"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func init() {
|
|
getSSHClientEnvVar = func() string {
|
|
if os.Getenv("SUDO_USER") == "" {
|
|
// No sudo, just check the env.
|
|
return os.Getenv("SSH_CLIENT")
|
|
}
|
|
if runtime.GOOS != "linux" {
|
|
// TODO(maisem): implement this for other platforms. It's not clear
|
|
// if there is a way to get the environment for a given process on
|
|
// darwin and bsd.
|
|
return ""
|
|
}
|
|
// SID is the session ID of the user's login session.
|
|
// It is also the process ID of the original shell that the user logged in with.
|
|
// We only need to check the environment of that process.
|
|
sid, err := unix.Getsid(os.Getpid())
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
b, err := os.ReadFile(filepath.Join("/proc", strconv.Itoa(sid), "environ"))
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
prefix := []byte("SSH_CLIENT=")
|
|
for _, env := range bytes.Split(b, []byte{0}) {
|
|
if bytes.HasPrefix(env, prefix) {
|
|
return string(env[len(prefix):])
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
}
|