mirror of
https://github.com/gravitational/teleport.git
synced 2026-09-24 16:17:11 +08:00
Zac Bergquist
0bae0d255f
- Don't assume an explicit $GOPATH is set - Remove golint from linters - it's been deprecated for over a year and golangci-lint prints a warning instead of running it.
90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
/*
|
|
Copyright 2017 Gravitational, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package shell
|
|
|
|
/*
|
|
#cgo solaris CFLAGS: -D_POSIX_PTHREAD_SEMANTICS
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <pwd.h>
|
|
#include <stdlib.h>
|
|
|
|
static int mygetpwnam_r(const char *name, struct passwd *pwd,
|
|
char *buf, size_t buflen, struct passwd **result) {
|
|
return getpwnam_r(name, pwd, buf, buflen, result);
|
|
}
|
|
*/
|
|
import "C"
|
|
|
|
import (
|
|
"os/user"
|
|
"strings"
|
|
"syscall"
|
|
"unsafe"
|
|
|
|
"github.com/gravitational/trace"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// getLoginShell determines the login shell for a given username
|
|
func getLoginShell(username string) (string, error) {
|
|
// See if the username is valid.
|
|
_, err := user.Lookup(username)
|
|
if err != nil {
|
|
return "", trace.Wrap(err)
|
|
}
|
|
|
|
// Based on stdlib user/lookup_unix.go packages which does not return
|
|
// user shell: https://golang.org/src/os/user/lookup_unix.go
|
|
var pwd C.struct_passwd
|
|
var result *C.struct_passwd
|
|
|
|
bufSize := C.sysconf(C._SC_GETPW_R_SIZE_MAX)
|
|
if bufSize == -1 {
|
|
bufSize = 1024
|
|
}
|
|
if bufSize <= 0 || bufSize > 1<<20 {
|
|
return "", trace.BadParameter("lookupPosixShell: unreasonable _SC_GETPW_R_SIZE_MAX of %d", bufSize)
|
|
}
|
|
buf := C.malloc(C.size_t(bufSize))
|
|
defer C.free(buf)
|
|
var rv C.int
|
|
nameC := C.CString(username)
|
|
defer C.free(unsafe.Pointer(nameC))
|
|
rv = C.mygetpwnam_r(nameC,
|
|
&pwd,
|
|
(*C.char)(buf),
|
|
C.size_t(bufSize),
|
|
&result)
|
|
if rv != 0 || result == nil {
|
|
log.Errorf("lookupPosixShell: lookup username %s: %s", username, syscall.Errno(rv))
|
|
return "", trace.BadParameter("cannot determine shell for %s", username)
|
|
}
|
|
|
|
// If no shell was found, return trace.NotFound to allow the caller to set
|
|
// the default shell.
|
|
shellCmd := strings.TrimSpace(C.GoString(pwd.pw_shell))
|
|
if len(shellCmd) == 0 {
|
|
return "", trace.NotFound("no shell specified for %v", username)
|
|
}
|
|
|
|
return shellCmd, nil
|
|
}
|